Metadata-Version: 2.4
Name: jax-sn
Version: 0.1.0
Summary: JAX-based Discrete Ordinates
Author: Timo Bogaarts
License-Expression: MIT
Project-URL: Documentation, https://timobogaarts.github.io/jax-sn/
Project-URL: Source, https://github.com/timobogaarts/jax-sn
Classifier: Development Status :: 3 - Alpha
Classifier: Intended Audience :: Science/Research
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3.12
Classifier: Programming Language :: Python :: 3.13
Classifier: Topic :: Scientific/Engineering
Requires-Python: >=3.12
Description-Content-Type: text/markdown
License-File: LICENSE
Requires-Dist: jax>=0.4.0
Requires-Dist: jaxlib>=0.4.0
Requires-Dist: numpy
Requires-Dist: scipy
Requires-Dist: fenics-basix>=0.7.0
Requires-Dist: equinox
Requires-Dist: lineax
Provides-Extra: warp
Requires-Dist: warp-lang>=1.16.0; extra == "warp"
Provides-Extra: geometry
Requires-Dist: jax_sbgeom; extra == "geometry"
Provides-Extra: viz
Requires-Dist: plotly; extra == "viz"
Provides-Extra: test
Requires-Dist: pytest; extra == "test"
Requires-Dist: sympy; extra == "test"
Requires-Dist: networkx; extra == "test"
Requires-Dist: trimesh; extra == "test"
Provides-Extra: docs
Requires-Dist: sphinx; extra == "docs"
Requires-Dist: furo; extra == "docs"
Requires-Dist: sphinx-autodoc-typehints; extra == "docs"
Requires-Dist: myst-nb; extra == "docs"
Requires-Dist: ipykernel; extra == "docs"
Requires-Dist: matplotlib; extra == "docs"
Requires-Dist: plotly; extra == "docs"
Provides-Extra: dev
Requires-Dist: pre-commit; extra == "dev"
Requires-Dist: nbstripout; extra == "dev"
Dynamic: license-file

# jax-sn

JAX-based discrete ordinates for differentiable solution of neutron transport.

Documentation at https://timobogaarts.github.io/jax-sn/

## Quickstart

Install with:

```bash
pip install jax-sn
```

Optional extras (`geometry`, `warp`, `viz`, ...) and a GPU-enabled JAX install are described in the
[documentation](https://timobogaarts.github.io/jax-sn/).

## License

MIT
