Jaxion: A JAX package for Fuzzy Dark Matter
Jaxion: A JAX package for Fuzzy Dark Matter - Published in JOSS (2026)
Science Score: 92.0%
This score indicates how likely this project is to be science-related based on various indicators:
-
✓CITATION.cff file
Found CITATION.cff file -
○codemeta.json file
-
○.zenodo.json file
-
✓DOI references
Found 4 DOI reference(s) in README and JOSS metadata -
✓Academic publication links
Links to: zenodo.org -
○Committers with academic emails
-
○Institutional organization owner
-
✓JOSS paper metadata
Published in Journal of Open Source Software
Keywords
Repository
A simple JAX-powered simulation library for numerical experiments of fuzzy dark matter, stars, gas + more!
Basic Info
- Host: GitHub
- Owner: JaxionProject
- License: apache-2.0
- Language: Jupyter Notebook
- Default Branch: main
- Homepage: https://jaxion.readthedocs.io
- Size: 46 MB
Statistics
- Stars: 47
- Watchers: 1
- Forks: 4
- Open Issues: 1
- Releases: 11
Topics
Metadata Files
README.md
jaxion
A simple JAX-powered simulation library for numerical experiments of fuzzy dark matter, stars, gas + more!
Author: Philip Mocz (@pmocz)
Jaxion is built for multi-GPU scalability and is fully differentiable. It is a high-performance JAX-based simulation library for modeling fuzzy dark matter alongside stars, gas, and cosmological dynamics. Being differentiable, Jaxion can seamlessly integrate with pipelines for inverse-problems, inference, optimization, and coupling to ML models.
Jaxion is the simpler companion project to differentiable astrophysics code Adirondax
Install Jaxion
Install with:
console
pip install jaxion
or see the docs for how to build from source.
Examples
Check out the examples/ directory for demonstrations of using Jaxion.
Try it out!
Launch a live demo in Google Colab: 
High-Performance
Jaxion is scalable on multiple GPUs!
Contributing
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
TODO/Wishlist
- cosmological initial condition generator
- add cosmological factors to gas evolution
Links
- Code repository on GitHub (this page).
- Documentation for up-to-date information about installing and using Jaxion.
Cite this repository
If you use this software, please cite it as below.
bibtex
@software{Mocz_Jaxion_2025,
author = {Mocz, Philip},
doi = {10.5281/zenodo.17438467},
month = apr,
title = {{Jaxion}},
url = {https://github.com/JaxionProject/jaxion},
version = {0.0.11},
year = {2026}
}
Owner
- Name: Jaxion Project
- Login: JaxionProject
- Kind: organization
- Repositories: 1
- Profile: https://github.com/JaxionProject
JOSS Publication
Jaxion: A JAX package for Fuzzy Dark Matter
Authors
Center for Computational Astrophysics, Flatiron Institute, 162 5th Avenue, New York, NY 10010, USA
Editor
Arfon Smith
Tags
JAX astronomy cosmology dark matterCitation (CITATION.cff)
cff-version: 1.2.0 message: "If you use this software, please cite it as below." authors: - family-names: "Mocz" given-names: "Philip" orcid: "https://orcid.org/0000-0001-6631-2566" title: "Jaxion" version: 0.0.3 doi: 10.5281/zenodo.17438467 date-released: 2025-10-24 url: "https://github.com/JaxionProject/jaxion"
GitHub Events
Total
- Release event: 2
- Issues event: 2
- Watch event: 15
- Issue comment event: 1
- Push event: 78
Last Year
- Release event: 2
- Issues event: 2
- Watch event: 15
- Issue comment event: 1
- Push event: 78
Issues and Pull Requests
Last synced: 2 months ago
All Time
- Total issues: 0
- Total pull requests: 0
- Average time to close issues: N/A
- Average time to close pull requests: N/A
- Total issue authors: 0
- Total pull request authors: 0
- Average comments per issue: 0
- Average comments per pull request: 0
- Merged pull requests: 0
- Bot issues: 0
- Bot pull requests: 0
Past Year
- Issues: 0
- Pull requests: 0
- Average time to close issues: N/A
- Average time to close pull requests: N/A
- Issue authors: 0
- Pull request authors: 0
- Average comments per issue: 0
- Average comments per pull request: 0
- Merged pull requests: 0
- Bot issues: 0
- Bot pull requests: 0
Top Authors
Issue Authors
Pull Request Authors
Top Labels
Issue Labels
Pull Request Labels
Packages
- Total packages: 1
-
Total downloads:
- pypi 141 last-month
- Total dependent packages: 0
- Total dependent repositories: 0
- Total versions: 13
- Total maintainers: 1
pypi.org: jaxion
A differentiable simulation library for fuzzy dark matter in JAX
- Homepage: https://github.com/JaxionProject/jaxion
- Documentation: https://jaxion.readthedocs.io
- License: Apache-2.0
-
Latest release: 0.0.13
published 24 days ago