DifferentiableUniverseInitiative / JaxPM
JAX-powered Cosmological Particle-Mesh N-body Solver
☆34Updated 2 weeks ago
Alternatives and similar repositories for JaxPM:
Users that are interested in JaxPM are comparing it to the libraries listed below
- Curated list of astronomy/astrophysics code packages using JAX☆37Updated 2 months ago
- Unitful Quantities in JAX☆30Updated this week
- Astronomy Research + JAX Meeting 2024☆13Updated 3 months ago
- JAX-based linear Einstein-Boltzmann solver for cosmology☆14Updated 2 months ago
- Simulation-based (likelihood-free) inference customized for astronomical applications☆25Updated 3 months ago
- Package for Neural Posterior Estimation and Importance Sampling focused on Astronomical Applications☆36Updated 8 months ago
- Differentiable Likelihood for CMB Analysis☆15Updated last week
- Coverage tests to check the quality of your posterior estimators.☆30Updated 6 months ago
- A python package for Weak Lensing Implicit Inference with Differentiable Simulator☆13Updated 11 months ago
- JAX port of GalSim, for parallelized, GPU accelerated, and differentiable galaxy image simulations.☆29Updated 2 weeks ago
- Robust ML in Astro☆51Updated this week
- Machine Learning - accelerated Bayesian inference☆66Updated 2 months ago
- JAX Tutorial notebooks : basics, crash & tips, usage of optax/JaxOptim/Numpyro☆14Updated 3 months ago
- Differentiable Cosmological Forward Model☆78Updated 2 months ago
- ☆17Updated this week
- Normalizing flow models allowing for a conditioning context, implemented using Jax, Flax, and Distrax.☆15Updated 11 months ago
- A fast, differentiable, and extensible public BBN code☆11Updated last week
- ☆13Updated last year
- A python package for making arbitrarily structured, arbitrary-dimension boxes☆25Updated this week
- Accelerated Bayesian SED modeling using Amortized Neural Posterior Estimation☆23Updated last week
- Galactic and Gravitational Dynamics in Python (+ GPU and autodiff)☆36Updated this week
- Gradient Based Nested Sampling☆19Updated last year
- CPU/GPU agnostic gravitational-wave population inference☆42Updated 2 weeks ago
- Beyond-2pt blind data challenge☆16Updated 6 months ago
- A cosmological tool for emulation of various basic statistics of dark matter haloes based on Dark Quest project (https://darkquestcosmolo…☆22Updated last year
- ☆16Updated 4 years ago
- ☆22Updated last week