ABCMB 0.2.1__tar.gz → 0.2.2__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ABCMB
-Version: 0.2.1
+Version: 0.2.2
 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.2}/ABCMB.egg-info/SOURCES.txt

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

abcmb-0.2.2/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.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ABCMB
-Version: 0.2.1
+Version: 0.2.2
 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.2}/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.2}/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.
@@ -771,7 +769,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 +810,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 +848,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 +866,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.2}/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.2/abcmb/version.py

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

abcmb-0.2.1/abcmb/version.py

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