jaxion 0.0.9__tar.gz → 0.0.10__tar.gz

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

@@ -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 → 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.9 → 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.9 → jaxion-0.0.10}/jaxion.egg-info/PKG-INFO

@@ -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 → 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]

jaxion-0.0.9/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