Metadata-Version: 2.4
Name: jaxion
Version: 0.0.1
Summary: A differentiable simulation library for fuzzy dark matter in JAX
Author-email: Philip Mocz <philip.mocz@gmail.com>
License-Expression: Apache-2.0
Project-URL: Documentation, https://jaxion.readthedocs.io
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: tensorflow
Requires-Dist: orbax-checkpoint==0.11.18
Requires-Dist: optax==0.2.5
Requires-Dist: numpy
Requires-Dist: matplotlib
Requires-Dist: setuptools>=70.1.1
Provides-Extra: cuda12
Requires-Dist: jax[cuda12]==0.5.3; extra == "cuda12"
Dynamic: license-file

# jaxion

[![Repo Status][status-badge]][status-link]
[![PyPI Version Status][pypi-badge]][pypi-link]
[![Test Status][workflow-test-badge]][workflow-test-link]
[![Readthedocs Status][docs-badge]][docs-link]
[![License][license-badge]][license-link]

[status-link]:         https://www.repostatus.org/#active
[status-badge]:        https://www.repostatus.org/badges/latest/active.svg
[pypi-link]:           https://pypi.org/project/jaxion
[pypi-badge]:          https://img.shields.io/pypi/v/jaxion?label=PyPI&logo=pypi
[workflow-test-link]:  https://github.com/JaxionProject/jaxion/actions/workflows/test-package.yml
[workflow-test-badge]: https://github.com/JaxionProject/jaxion/actions/workflows/test-package.yml/badge.svg?event=push
[docs-link]:           https://jaxion.readthedocs.io
[docs-badge]:          https://readthedocs.org/projects/jaxion/badge
[license-link]:        https://opensource.org/licenses/Apache-2.0
[license-badge]:       https://img.shields.io/badge/License-Apache_2.0-blue.svg

A simple JAX-powered simulation library for numerical experiments of fuzzy dark matter, stars, gas + more!

Author: [Philip Mocz (@pmocz)](https://github.com/pmocz/)

⚠️ Jaxion is currently being developed and is not yet ready for use. Check back later ⚠️

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 seemlessly integrate with piplines for inverse-problems, inference, optimization, and coupling to ML models.


## Getting started

Install with:

```console
pip install jaxion
```

or, for GPU support use:

```console
pip install jaxion[cuda12]
```

Check out the `examples/` directory for demonstrations of using Jaxion.


## Links

* [Code repository](https://github.com/JaxionProject/jaxion) on GitHub (this page).
* [Documentation](https://jaxion.readthedocs.io) for up-to-date information about installing and running jaxion.


## Cite this repository

