guilhem_lavaux/JaxPM
0
0
Fork 1
mirror of https://github.com/DifferentiableUniverseInitiative/JaxPM.git synced 2025-02-22 09:37:11 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Adding very simple install instructions

Browse source
This commit is contained in:
Francois Lanusse 2024-12-21 11:32:58 -05:00
parent 953c36f582
commit def149499a
1 changed files with 9 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
README.md
View file

@ -4,6 +4,15 @@
<!-- ALL-CONTRIBUTORS-BADGE:END -->
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: