jaxion 0.0.8__tar.gz → 0.0.10__tar.gz

{jaxion-0.0.8 → jaxion-0.0.10}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: jaxion
-Version: 0.0.8
+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
@@ -9,16 +9,17 @@ Project-URL: Homepage, https://github.com/JaxionProject/jaxion
 Requires-Python: >=3.11
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: jax==0.5.3
-Requires-Dist: jaxdecomp==0.2.7
+Requires-Dist: jax==0.6.0
+Requires-Dist: jaxdecomp==0.2.9
 Requires-Dist: tensorflow
-Requires-Dist: orbax-checkpoint==0.11.18
-Requires-Dist: optax==0.2.5
+Requires-Dist: orbax-checkpoint==0.11.25
+Requires-Dist: optax==0.2.6
 Requires-Dist: numpy
 Requires-Dist: matplotlib
 Requires-Dist: setuptools>=70.1.1
+Requires-Dist: chex==0.1.90
 Provides-Extra: cuda12
-Requires-Dist: jax[cuda12]==0.5.3; extra == "cuda12"
+Requires-Dist: jax[cuda12]==0.6.0; extra == "cuda12"
 Dynamic: license-file
 
 <p align="center">
@@ -76,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
@@ -122,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!
@@ -138,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).
@@ -152,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.8 → jaxion-0.0.10}/README.md

@@ -53,13 +53,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
@@ -99,6 +93,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!
@@ -115,6 +114,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).
@@ -129,10 +134,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.10/jaxion/__init__.py

@@ -0,0 +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-0.0.8 → jaxion-0.0.10}/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.8 → jaxion-0.0.10}/jaxion.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: jaxion
-Version: 0.0.8
+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
@@ -9,16 +9,17 @@ Project-URL: Homepage, https://github.com/JaxionProject/jaxion
 Requires-Python: >=3.11
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: jax==0.5.3
-Requires-Dist: jaxdecomp==0.2.7
+Requires-Dist: jax==0.6.0
+Requires-Dist: jaxdecomp==0.2.9
 Requires-Dist: tensorflow
-Requires-Dist: orbax-checkpoint==0.11.18
-Requires-Dist: optax==0.2.5
+Requires-Dist: orbax-checkpoint==0.11.25
+Requires-Dist: optax==0.2.6
 Requires-Dist: numpy
 Requires-Dist: matplotlib
 Requires-Dist: setuptools>=70.1.1
+Requires-Dist: chex==0.1.90
 Provides-Extra: cuda12
-Requires-Dist: jax[cuda12]==0.5.3; extra == "cuda12"
+Requires-Dist: jax[cuda12]==0.6.0; extra == "cuda12"
 Dynamic: license-file
 
 <p align="center">
@@ -76,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
@@ -122,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!
@@ -138,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).
@@ -152,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.10/jaxion.egg-info/requires.txt

@@ -0,0 +1,12 @@
+jax==0.6.0
+jaxdecomp==0.2.9
+tensorflow
+orbax-checkpoint==0.11.25
+optax==0.2.6
+numpy
+matplotlib
+setuptools>=70.1.1
+chex==0.1.90
+
+[cuda12]
+jax[cuda12]==0.6.0

{jaxion-0.0.8 → jaxion-0.0.10}/pyproject.toml

@@ -10,7 +10,7 @@ license = "Apache-2.0"
 dynamic = ["version", "dependencies"]
 
 [tool.setuptools.packages.find]
-include = ["jaxio*"]
+include = ["jaxion*"]
 exclude = ["paper"]
 
 [tool.setuptools.package-data]
@@ -25,7 +25,7 @@ dependencies = {file = ["requirements.txt"]}
 
 [project.optional-dependencies]
 cuda12 = [
-    "jax[cuda12]==0.5.3",
+    "jax[cuda12]==0.6.0",
 ]
 
 [tool.setuptools-git-versioning]

jaxion-0.0.10/requirements.txt

@@ -0,0 +1,9 @@
+jax==0.6.0
+jaxdecomp==0.2.9
+tensorflow
+orbax-checkpoint==0.11.25
+optax==0.2.6
+numpy
+matplotlib
+setuptools>=70.1.1
+chex==0.1.90

jaxion-0.0.8/jaxion/__init__.py

@@ -1,3 +0,0 @@
-from .simulation import Simulation as Simulation
-from .constants import constants as constants
-from .analysis import radial_power_spectrum as radial_power_spectrum

jaxion-0.0.8/jaxion.egg-info/requires.txt

@@ -1,11 +0,0 @@
-jax==0.5.3
-jaxdecomp==0.2.7
-tensorflow
-orbax-checkpoint==0.11.18
-optax==0.2.5
-numpy
-matplotlib
-setuptools>=70.1.1
-
-[cuda12]
-jax[cuda12]==0.5.3

jaxion-0.0.8/requirements.txt

@@ -1,8 +0,0 @@
-jax==0.5.3
-jaxdecomp==0.2.7
-tensorflow
-orbax-checkpoint==0.11.18
-optax==0.2.5
-numpy
-matplotlib
-setuptools>=70.1.1