dpiras/cosmopower-jax
Differentiable cosmological emulators: the JAX version of CosmoPower
This tool helps cosmologists and astrophysicists analyze complex cosmological data more efficiently. It takes in cosmological parameters, such as the density of matter or the expansion rate of the universe, and quickly produces predictions of cosmological power spectra, which describe the distribution of matter and energy in the universe. This allows researchers to rapidly test hypotheses and explore a wide range of cosmic models.
Available on PyPI.
Use this if you need to perform high-dimensional Bayesian inference or generate quick, differentiable predictions of cosmological power spectra for research or analysis.
Not ideal if you need to train new JAX-based neural network models from scratch on custom data, as this functionality is not currently provided.
Stars
45
Forks
5
Language
Jupyter Notebook
License
GPL-3.0
Category
Last pushed
Nov 27, 2025
Commits (30d)
0
Dependencies
4
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/ml-frameworks/dpiras/cosmopower-jax"
Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.
Related frameworks
explosion/thinc
🔮 A refreshing functional take on deep learning, compatible with your favorite libraries
google-deepmind/optax
Optax is a gradient processing and optimization library for JAX.
patrick-kidger/diffrax
Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable....
google/grain
Library for reading and processing ML training data.
patrick-kidger/equinox
Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/