diff --git a/README.md b/README.md index 16fcfed..47041ab 100644 --- a/README.md +++ b/README.md @@ -4,6 +4,15 @@ JAX-powered Cosmological Particle-Mesh N-body Solver +## Install + +Basic installation can be done using pip: +```bash +pip install jaxpm +``` +For more advanced installation for optimized distribution on gpu clusters, please install jaxDecomp first. See instructions [here](https://github.com/DifferentiableUniverseInitiative/jaxDecomp). + + ## Goals Provide a modern infrastructure to support differentiable PM N-body simulations using JAX: