DifferentiableUniverseInitiative / jax_cosmo
A differentiable cosmology library in JAX
☆178Updated 3 months ago
Related projects ⓘ
Alternatives and complementary repositories for jax_cosmo
- Probabilistic Programming and Nested sampling in JAX☆147Updated last week
- Differentiable and accelerated spherical transforms with JAX☆136Updated this week
- Normalizing-flow enhanced sampling package for probabilistic inference in Jax☆200Updated 2 months ago
- A system for scientific simulation-based inference at scale.☆161Updated 7 months ago
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆80Updated last week
- Interpolation and function approximation with JAX☆135Updated 3 weeks ago
- Density Estimation Likelihood-Free Inference with neural density estimators and adaptive acquisition of simulations☆106Updated last year
- Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation☆61Updated 4 months ago
- Gravitational-wave data analysis tools in Jax☆59Updated 3 weeks ago
- Numerical quadrature with JAX☆44Updated 3 weeks ago
- 🐈 Automatic differentiable spectrum modeling of exoplanets/brown dwarfs using JAX, compatible with NumPyro and JAXopt☆57Updated this week
- The tiniest of Gaussian Process libraries☆296Updated last week
- Machine learning assisted marginal likelihood (Bayesian evidence) estimation for Bayesian model selection☆56Updated this week
- ☆107Updated 3 weeks ago
- Differentiable Gravitational Waveforms with JAX☆54Updated 3 months ago
- JAX-powered Cosmological Particle-Mesh N-body Solver☆28Updated last week
- MCHMC: sampler from an arbitrary differentiable distribution☆70Updated this week
- Using neural networks to extract sufficient statistics from data by maximising the Fisher information☆30Updated last year
- Differentiable Cosmological Forward Model☆70Updated last week
- nessai: Nested Sampling with Artificial Intelligence☆36Updated last week
- Likelihood-free AMortized Posterior Estimation with PyTorch☆119Updated 3 months ago
- Probabilistic modeling of tabular data with normalizing flows.☆55Updated 5 months ago
- A package to describe amortized (conditional) normalizing-flow PDFs defined jointly on tensor products of manifolds with coverage control…☆42Updated 9 months ago
- Stainless neural networks in JAX☆31Updated last month
- Dingo: Deep inference for gravitational-wave observations☆66Updated this week
- Stencil computations in JAX☆66Updated last year
- Community-sourced list of papers and resources on neural simulation-based inference.☆95Updated 5 months ago
- Next generation Bayesian analysis tools for efficient posterior sampling and evidence estimation.☆29Updated 2 years ago
- Add a tqdm progress bar to your JAX scans and loops.☆96Updated 3 weeks ago
- A miscellaneous set of helper functions, custom distributions, and other utilities that I find useful when using NumPyro in my work☆21Updated 2 weeks ago