Details about Jax-Cosmo
| Category | Information |
|---|---|
| Purpose | Differentiable cosmology library thanks to JAX. The user takes benefit of JAX to produce Just-In-Time compilated code adapted to the device (eg. CPU/GPU) and can use paralellized code on single as well as multi-devices. |
| Related community |
Action Dark Energy
|
| Tool type | Software |
| URL | https://github.com/DifferentiableUniverseInitiative/jax_cosmo |
| Author | A nice Team :) |
| Additional contact | francois.lanusse@cea.fr |
| Publication: | |
| Language | Python 3.x |
| Keywords | None given
|
Comments:
pip install jax-cosmo