ABCMB 0.2.1__tar.gz → 0.2.3__tar.gz

{abcmb-0.2.1 → abcmb-0.2.3}/ABCMB.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ABCMB
-Version: 0.2.1
+Version: 0.2.3
 Summary: A fast, differentiable, and extensible CMB code
 Home-page: https://github.com/TonyZhou729/ABCMB
 Author: Zilu Zhou, Cara Giovanetti, Hongwan Liu
@@ -19,6 +19,7 @@ Classifier: Programming Language :: Python :: 3.13
 Classifier: Programming Language :: Python :: 3.14
 Requires-Python: >=3.8
 Description-Content-Type: text/markdown
+License-File: LICENSE
 Requires-Dist: numpy
 Requires-Dist: scipy
 Requires-Dist: matplotlib
@@ -28,6 +29,7 @@ Requires-Dist: interpax
 Requires-Dist: jax==0.8.1
 Requires-Dist: pytest
 Requires-Dist: optimistix==0.0.11
+Dynamic: license-file
 
 <h1 align="center">
 ABCMB<!-- omit from toc -->
@@ -35,6 +37,7 @@ ABCMB<!-- omit from toc -->
 <h4 align="center">
 
 [![License: MIT](https://img.shields.io/badge/License-MIT-red.svg)](https://opensource.org/licenses/MIT)
+[![arXiv](https://img.shields.io/badge/arXiv-2602.15104%20-green.svg)](https://arxiv.org/abs/2602.15104)
 [![Run Tests](https://github.com/TonyZhou729/ABCMB/actions/workflows/accuracy.yml/badge.svg)](https://github.com/TonyZhou729/ABCMB/actions/workflows/accuracy.yml)
 <!--[![arXiv](https://img.shields.io/badge/arXiv-2408.14538%20-green.svg)](https://arxiv.org/abs/2408.14538) -->
 
@@ -63,5 +66,20 @@ We have included several pedagogical jupyter notebooks to walk you through how t
 ## Issues
 Please feel free to open an issue if something is amiss in ABCMB!
 
+## Citation
+
+If you use ABCMB to publish scientific research, we suggest you cite
+```
+@misc{abcmb,
+      title={{ABCMB: A Python+JAX Package for the Cosmic Microwave Background Power Spectrum}}, 
+      author={Zilu Zhou and Cara Giovanetti and Hongwan Liu},
+      year={2026},
+      eprint={2602.15104},
+      archivePrefix={arXiv},
+      primaryClass={astro-ph.CO},
+      url={https://arxiv.org/abs/2602.15104}, 
+}
+```
+
 
 

{abcmb-0.2.1 → abcmb-0.2.3}/ABCMB.egg-info/SOURCES.txt

@@ -1,3 +1,4 @@
+LICENSE
 README.md
 pyproject.toml
 setup.cfg

abcmb-0.2.3/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 Zilu Zhou, Cara Giovanetti, Hongwan Liu
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

{abcmb-0.2.1 → abcmb-0.2.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ABCMB
-Version: 0.2.1
+Version: 0.2.3
 Summary: A fast, differentiable, and extensible CMB code
 Home-page: https://github.com/TonyZhou729/ABCMB
 Author: Zilu Zhou, Cara Giovanetti, Hongwan Liu
@@ -19,6 +19,7 @@ Classifier: Programming Language :: Python :: 3.13
 Classifier: Programming Language :: Python :: 3.14
 Requires-Python: >=3.8
 Description-Content-Type: text/markdown
+License-File: LICENSE
 Requires-Dist: numpy
 Requires-Dist: scipy
 Requires-Dist: matplotlib
@@ -28,6 +29,7 @@ Requires-Dist: interpax
 Requires-Dist: jax==0.8.1
 Requires-Dist: pytest
 Requires-Dist: optimistix==0.0.11
+Dynamic: license-file
 
 <h1 align="center">
 ABCMB<!-- omit from toc -->
@@ -35,6 +37,7 @@ ABCMB<!-- omit from toc -->
 <h4 align="center">
 
 [![License: MIT](https://img.shields.io/badge/License-MIT-red.svg)](https://opensource.org/licenses/MIT)
+[![arXiv](https://img.shields.io/badge/arXiv-2602.15104%20-green.svg)](https://arxiv.org/abs/2602.15104)
 [![Run Tests](https://github.com/TonyZhou729/ABCMB/actions/workflows/accuracy.yml/badge.svg)](https://github.com/TonyZhou729/ABCMB/actions/workflows/accuracy.yml)
 <!--[![arXiv](https://img.shields.io/badge/arXiv-2408.14538%20-green.svg)](https://arxiv.org/abs/2408.14538) -->
 
@@ -63,5 +66,20 @@ We have included several pedagogical jupyter notebooks to walk you through how t
 ## Issues
 Please feel free to open an issue if something is amiss in ABCMB!
 
+## Citation
+
+If you use ABCMB to publish scientific research, we suggest you cite
+```
+@misc{abcmb,
+      title={{ABCMB: A Python+JAX Package for the Cosmic Microwave Background Power Spectrum}}, 
+      author={Zilu Zhou and Cara Giovanetti and Hongwan Liu},
+      year={2026},
+      eprint={2602.15104},
+      archivePrefix={arXiv},
+      primaryClass={astro-ph.CO},
+      url={https://arxiv.org/abs/2602.15104}, 
+}
+```
+
 
 

{abcmb-0.2.1 → abcmb-0.2.3}/README.md

@@ -4,6 +4,7 @@ ABCMB<!-- omit from toc -->
 <h4 align="center">
 
 [![License: MIT](https://img.shields.io/badge/License-MIT-red.svg)](https://opensource.org/licenses/MIT)
+[![arXiv](https://img.shields.io/badge/arXiv-2602.15104%20-green.svg)](https://arxiv.org/abs/2602.15104)
 [![Run Tests](https://github.com/TonyZhou729/ABCMB/actions/workflows/accuracy.yml/badge.svg)](https://github.com/TonyZhou729/ABCMB/actions/workflows/accuracy.yml)
 <!--[![arXiv](https://img.shields.io/badge/arXiv-2408.14538%20-green.svg)](https://arxiv.org/abs/2408.14538) -->
 
@@ -32,5 +33,20 @@ We have included several pedagogical jupyter notebooks to walk you through how t
 ## Issues
 Please feel free to open an issue if something is amiss in ABCMB!
 
+## Citation
+
+If you use ABCMB to publish scientific research, we suggest you cite
+```
+@misc{abcmb,
+      title={{ABCMB: A Python+JAX Package for the Cosmic Microwave Background Power Spectrum}}, 
+      author={Zilu Zhou and Cara Giovanetti and Hongwan Liu},
+      year={2026},
+      eprint={2602.15104},
+      archivePrefix={arXiv},
+      primaryClass={astro-ph.CO},
+      url={https://arxiv.org/abs/2602.15104}, 
+}
+```
+
 
 

{abcmb-0.2.1 → abcmb-0.2.3}/abcmb/ABCMBTools.py

@@ -299,7 +299,7 @@ def fast_interp(x, xp_min, xp_max, fp):
     # to make this operation much faster.
     eps = 1.e-6
     n = fp.shape[-1]
-    i = (x - xp_min) / (xp_max - xp_min) * n
+    i = (x - xp_min) / (xp_max - xp_min) * (n - 1) # fix bug in JAX issue
     i = jnp.clip(i, eps, n - 1.0 - eps)  # Avoid index out of range
     i_lower = jnp.floor(i).astype(jnp.int32)
     i_upper = jnp.minimum(i_lower + 1, n - 1)

{abcmb-0.2.1 → abcmb-0.2.3}/abcmb/background.py

@@ -102,8 +102,6 @@ class Background(eqx.Module):
     tau_reion  : float 
     lna_rec    : float
     rA_rec     : float # Comoving angular diameter distance at recombination.
-    rs_d       : float # Sound horizon at baryon decoupling
-    z_d        : float # redshift of baryon decoupling
 
     # Transfer related
     lna_transfer_start : float # Time where transfer functions start integrating.
@@ -749,8 +747,16 @@ class Background(eqx.Module):
         float
             Baryon drag ratio (units: dimensionless)
         """
-        rho_b = self.species_list[-3].rho(lna,params)
-        rho_g = self.species_list[-2].rho(lna,params)
+
+        rho_b = 0.
+        rho_g = 0.
+
+        for s in self.species_list:
+            if s.name == "Photon":
+                rho_g += s.rho(lna, params)
+            elif s.name == "Baryon":
+                rho_b += s.rho(lna, params)
+
         return 3. * rho_b / (4 * rho_g)
 
     @jax.named_scope("tabulate kappa d")
@@ -771,7 +777,7 @@ class Background(eqx.Module):
         array
             Tabulated baryon optical depth values (units: dimensionless)
         """
-        integrand = lambda lna, y, args: jnp.float64(-1./self.tau_c(lna, params)/self.aH(lna, params)/(self.R_ratio_lna(lna)))
+        integrand = lambda lna, y, args: jnp.float64(-1./self.tau_c(lna, params)/self.aH(lna, params)/(self.R_ratio_lna(lna, params)))
         term = ODETerm(integrand)
         stepsize_controller = PIDController(pcoeff=0.4, icoeff=0.3, dcoeff=0, rtol=1.e-3, atol=1.e-6)
         adjoint=ForwardMode()
@@ -812,7 +818,7 @@ class Background(eqx.Module):
          # initial condition assuming cs**2 = 1/3 at early times
         rs0 = 1./jnp.sqrt(3) / (self.aH( self.lna_tau_tab[0], params ))
 
-        integrand = lambda lna, y, args: 1./jnp.sqrt(3*(1+self.R_ratio_lna(lna))) / (self.aH(lna, params))
+        integrand = lambda lna, y, args: 1./jnp.sqrt(3*(1+self.R_ratio_lna(lna, params))) / (self.aH(lna, params))
         term = ODETerm(integrand)
         stepsize_controller = PIDController(pcoeff=0.4, icoeff=0.3, dcoeff=0, rtol=1.e-3, atol=1.e-6)
         adjoint=ForwardMode()
@@ -850,7 +856,7 @@ class Background(eqx.Module):
         float
             Decoupling redshift (units: dimensionless)
         """
-        return self.find_z_at_kappad_equals_one(1/jnp.exp(self.lna_tau_tab) - 1, self._tabulate_kappa_d())
+        return self.find_z_at_kappad_equals_one(1/jnp.exp(self.lna_tau_tab) - 1, self._tabulate_kappa_d(params))
 
     def rs_d(self, params):
         """
@@ -868,7 +874,7 @@ class Background(eqx.Module):
         float
             Sound horizon at decoupling (units: Mpc)
         """
-        return self.interp_rs_at_z(1/jnp.exp(self.lna_tau_tab) - 1, self._tabulate_rs(), self.z_d())
+        return self.interp_rs_at_z(1/jnp.exp(self.lna_tau_tab) - 1, self._tabulate_rs(params), self.z_d(params))
 
 class ReionizationModel(eqx.Module):
     """

{abcmb-0.2.1 → abcmb-0.2.3}/abcmb/main.py

@@ -507,13 +507,13 @@ class Model(eqx.Module):
 
         # There is NO NEED to modify this list!!  This is to make sure any new
         # user-defined keys will not trigger recompilation by wrapping them in
-        # jnp.array, as is done manually above for all other keys.
+        # jnp.array, as is done manually above for all other keys.  LINX-
+        # related inputs are intentionally excluded from this list!
         expected_keys = {
             'h', 'H0', 'omega_cdm', 'omega_b', 'A_s', 'n_s', 'TCMB0',
             'tau_reion', 'z_reion', 'Delta_z_reion', 'z_reion_He', 'Delta_z_reion_He', 'exp_reion',
             'omega_Lambda', 'T_nu_massive', 'N_nu_massive', 'm_nu_massive',
-            'N_nu_massless', 'Neff', 'T_nu_massless',
-            'Delta_Neff_init', 'tau_n_fac', 'nuclear_rates_q', 'YHe',
+            'N_nu_massless', 'Neff', 'T_nu_massless', 'YHe',
             'omega_m', 'R_b', 'omega_r', 'R_nu', 'om'
         }
         

{abcmb-0.2.1 → abcmb-0.2.3}/abcmb/species.py

@@ -43,11 +43,13 @@ class Fluid(eqx.Module):
 
     delta_idx     : int = eqx.field(default=0)
     num_moments : int = eqx.field(default=0, static=True)
-    name          : str = eqx.field(default="")
+    name          : str = eqx.field(default="", static=True)
     is_matter     : bool = eqx.field(default=False) # Does the fluid contribute towards matter overdensity today.
 
     def __init__(self, delta_idx, specs):
         self.delta_idx = delta_idx
+        self.name = ""
+        self.is_matter = False
 
     def rho(self, lna, args):
         """
@@ -361,6 +363,7 @@ class DarkEnergy(BackgroundFluid):
 
     def __init__(self, delta_idx, specs):
         super().__init__(delta_idx, specs)
+        self.name = "DarkEnergy"
 
     def rho(self, lna, args):
         """
@@ -425,6 +428,8 @@ class ColdDarkMatter(StandardFluid):
 
     def __init__(self, delta_idx, specs):
         super().__init__(delta_idx, specs)
+        self.name = "ColdDarkMatter"
+        self.is_matter = True
 
     def rho(self, lna, args):
         """
@@ -535,6 +540,7 @@ class MasslessNeutrino(StandardFluid):
 
     def __init__(self, delta_idx, specs):
         super().__init__(delta_idx, specs)
+        self.name = "MasslessNeutrino"
         self.num_moments = specs["l_max_massless_nu"] + 1
 
     def rho(self, lna, args):
@@ -699,6 +705,8 @@ class MassiveNeutrino(Fluid):
     def __init__(self, delta_idx, specs):
 
         super().__init__(delta_idx, specs)
+        self.name = "MassiveNeutrino"
+        self.is_matter = True
         self.num_ells_per_bin = specs["l_max_massive_nu"] + 1
         self.num_moments = 3 * self.num_ells_per_bin
 
@@ -1003,6 +1011,8 @@ class Baryon(StandardFluid):
 
     def __init__(self, delta_idx, specs):
         super().__init__(delta_idx, specs)
+        self.name = "Baryon"
+        self.is_matter = True
 
     def rho(self, lna, args):
         """
@@ -1199,6 +1209,7 @@ class Photon(StandardFluid):
 
     def __init__(self, delta_idx, specs):
         super().__init__(delta_idx, specs)
+        self.name = "Photon"
         self.num_F_ell_modes = specs["l_max_g"] + 1
         self.num_G_ell_modes = specs["l_max_pol_g"] + 1
         self.num_moments = self.num_F_ell_modes + self.num_G_ell_modes

abcmb-0.2.3/abcmb/version.py

@@ -0,0 +1 @@
+__version__ = "0.2.3"

abcmb-0.2.1/abcmb/version.py

@@ -1 +0,0 @@
-__version__ = "0.2.1"