jaxion 0.0.9__py3-none-any.whl → 0.0.10__py3-none-any.whl

jaxion/__init__.py

@@ -1,3 +1,20 @@
+from importlib.metadata import version, PackageNotFoundError
+
 from .simulation import Simulation as Simulation
 from .constants import constants as constants
 from .analysis import radial_power_spectrum as radial_power_spectrum
+
+"""
+jaxion: A JAX library for simulations of fuzzy dark matter, stars, gas + more!
+"""
+
+try:
+    __version__ = version("jaxion")
+except PackageNotFoundError:
+    __version__ = "unknown"
+
+__all__ = [
+    "Simulation",
+    "constants",
+    "radial_power_spectrum",
+]

jaxion/hydro.py

@@ -197,36 +197,42 @@ def slope_limiter(f, dx, f_dx, f_dy, f_dz):
     denom = adjust_denominator(orig_f_dx)
     num = (f - jnp.roll(f, 1, axis=0)) / dx
     ratio = num / denom
-    limiter = jnp.maximum(0.0, jnp.minimum(1.0, ratio))
-    f_dx = limiter * f_dx
+    limiter_left = jnp.maximum(0.0, jnp.minimum(1.0, ratio))
 
     num = -(f - jnp.roll(f, -1, axis=0)) / dx
     ratio = num / denom  # Use the same adjusted denominator
-    limiter = jnp.maximum(0.0, jnp.minimum(1.0, ratio))
-    f_dx = limiter * f_dx
+    limiter_right = jnp.maximum(0.0, jnp.minimum(1.0, ratio))
+
+    # Take minimum of both limiters and apply to original slope
+    limiter = jnp.minimum(limiter_left, limiter_right)
+    f_dx = limiter * orig_f_dx
 
     # For y-direction
     denom = adjust_denominator(orig_f_dy)
     num = (f - jnp.roll(f, 1, axis=1)) / dx
     ratio = num / denom
-    limiter = jnp.maximum(0.0, jnp.minimum(1.0, ratio))
-    f_dy = limiter * f_dy
+    limiter_left = jnp.maximum(0.0, jnp.minimum(1.0, ratio))
 
     num = -(f - jnp.roll(f, -1, axis=1)) / dx
     ratio = num / denom
-    limiter = jnp.maximum(0.0, jnp.minimum(1.0, ratio))
-    f_dy = limiter * f_dy
+    limiter_right = jnp.maximum(0.0, jnp.minimum(1.0, ratio))
+
+    # Take minimum of both limiters and apply to original slope
+    limiter = jnp.minimum(limiter_left, limiter_right)
+    f_dy = limiter * orig_f_dy
 
     # For z-direction
     denom = adjust_denominator(orig_f_dz)
     num = (f - jnp.roll(f, 1, axis=2)) / dx
     ratio = num / denom
-    limiter = jnp.maximum(0.0, jnp.minimum(1.0, ratio))
-    f_dz = limiter * f_dz
+    limiter_left = jnp.maximum(0.0, jnp.minimum(1.0, ratio))
 
     num = -(f - jnp.roll(f, -1, axis=2)) / dx
     ratio = num / denom
-    limiter = jnp.maximum(0.0, jnp.minimum(1.0, ratio))
-    f_dz = limiter * f_dz
+    limiter_right = jnp.maximum(0.0, jnp.minimum(1.0, ratio))
+
+    # Take minimum of both limiters and apply to original slope
+    limiter = jnp.minimum(limiter_left, limiter_right)
+    f_dz = limiter * orig_f_dz
 
     return f_dx, f_dy, f_dz

{jaxion-0.0.9.dist-info → jaxion-0.0.10.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: jaxion
-Version: 0.0.9
+Version: 0.0.10
 Summary: A differentiable simulation library for fuzzy dark matter in JAX
 Author-email: Philip Mocz <philip.mocz@gmail.com>
 License-Expression: Apache-2.0
@@ -77,13 +77,7 @@ Install with:
 pip install jaxion
 ```
 
-or, for GPU support, use:
-
-```console
-pip install jaxion[cuda12]
-```
-
-See the docs for more info on how to [build from source](https://jaxion.readthedocs.io/en/latest/pages/installation.html).
+or see the docs for how to [build from source](https://jaxion.readthedocs.io/en/latest/pages/installation.html).
 
 
 ## Examples
@@ -123,6 +117,11 @@ Check out the [`examples/`](https://github.com/JaxionProject/jaxion/tree/main/ex
 </p>
 
 
+## Try it out!
+
+Launch a live demo in Google Colab: [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/JaxionProject/jaxion/blob/main/notebooks/soliton_binary_merger/soliton_binary_merger.ipynb)
+
+
 ## High-Performance
 
 Jaxion is scalable on multiple GPUs!
@@ -139,6 +138,12 @@ Jaxion is scalable on multiple GPUs!
 Jaxion welcomes community contributions of all kinds. Open an issue or fork the code and submit a pull request. Please check out the [Contributing Guidelines](CONTRIBUTING.md)
 
 
+## TODO/Wishlist
+
+* cosmological initial condition generator
+* add cosmological factors to gas evolution
+
+
 ## Links
 
 * [Code repository](https://github.com/JaxionProject/jaxion) on GitHub (this page).
@@ -153,10 +158,10 @@ If you use this software, please cite it as below.
 @software{Mocz_Jaxion_2025,
    author = {Mocz, Philip},
       doi = {10.5281/zenodo.17438467},
-    month = oct,
+    month = dec,
     title = {{Jaxion}},
       url = {https://github.com/JaxionProject/jaxion},
-  version = {0.0.4},
+  version = {0.0.9},
      year = {2025}
 }
 ```

{jaxion-0.0.9.dist-info → jaxion-0.0.10.dist-info}/RECORD

@@ -1,17 +1,17 @@
-jaxion/__init__.py,sha256=Hdji1UQ47lG24Pqcy6UUq9L0-qy6m9Ax41L0vIYzBto,164
+jaxion/__init__.py,sha256=qUWqH7N8srFTTRM4MQf4A7OTbyE6mGx5mrBYloEOpv8,492
 jaxion/analysis.py,sha256=4YT9Z2dkFoXwft3fQM1HyynVPlIdtRd80VtI2vWTyq4,1568
 jaxion/constants.py,sha256=HyY2ktKQakv78jD1yQvFdM3sklUJcPgDMYlTsSPQTxI,512
 jaxion/cosmology.py,sha256=UC1McXNTXGoPRYXn0nI2-csVkJWL-ZBNoCa44oU1b4w,2681
 jaxion/defaults.json,sha256=MZraX_f5we3tLKBgpngy5hn0SUVznZkgmNEsGclgNUg,3093
 jaxion/gravity.py,sha256=2smqy_jjmr0VkMGzLPMYjLahHBMfZ2nNUx0gUiTWcDI,293
-jaxion/hydro.py,sha256=KoJ02tRpAc4V3Ofzw4zbHLRaE2GdIatbOBE04_LsSRw,6980
+jaxion/hydro.py,sha256=gYotvKh-bo86R9-2BwD_VDITI6Tm2RpHfI1Tih60l6s,7310
 jaxion/particles.py,sha256=AD3VlomC_VPOcHalAlZo8kcGFV5pui9tCnKkenm6DxE,7026
 jaxion/quantum.py,sha256=GWOpN6ipfEw-6Ah2zQpxS3oqeSt_iHMDSgnVYSjXY5E,3321
 jaxion/simulation.py,sha256=vpRLIb2EJdu0Df7ABgukgQ-ykhEP9nuGVIqu2ZzEKEw,20507
 jaxion/utils.py,sha256=OCpJ3crZqr5VFacymYzi5BkRqFCVBcneoS44wV9mZPg,4087
 jaxion/visualization.py,sha256=D0Nr5vhuqxL50MBIPD2ZMOdFDmG6zbCEiH1RemXtn_M,3346
-jaxion-0.0.9.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-jaxion-0.0.9.dist-info/METADATA,sha256=LN5zMcmFtvkFf5awRE9rPqrLUPFzysz0J7rjenzL2Iw,6468
-jaxion-0.0.9.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-jaxion-0.0.9.dist-info/top_level.txt,sha256=S1OV2VdlDG_9UwpKOIji4itQGOS-VWUOWUi3GeXWzt0,7
-jaxion-0.0.9.dist-info/RECORD,,
+jaxion-0.0.10.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+jaxion-0.0.10.dist-info/METADATA,sha256=9wCMOK5CoRldiRf9MLkwWdbJct0V5oRPlxO5FlJoA2s,6760
+jaxion-0.0.10.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+jaxion-0.0.10.dist-info/top_level.txt,sha256=S1OV2VdlDG_9UwpKOIji4itQGOS-VWUOWUi3GeXWzt0,7
+jaxion-0.0.10.dist-info/RECORD,,

{jaxion-0.0.9.dist-info → jaxion-0.0.10.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.9.0)
+Generator: setuptools (80.10.2)
 Root-Is-Purelib: true
 Tag: py3-none-any