dfm / numpyro-ext
A miscellaneous set of helper functions, custom distributions, and other utilities that I find useful when using NumPyro in my work
☆24Updated last month
Alternatives and similar repositories for numpyro-ext:
Users that are interested in numpyro-ext are comparing it to the libraries listed below
- Curated list of astronomy/astrophysics code packages using JAX☆37Updated 3 months ago
- Astronomy Research + JAX Meeting 2024☆13Updated 4 months ago
- nessai: Nested Sampling with Artificial Intelligence☆38Updated 2 weeks ago
- Proximal Nested Sampling for high-dimensional Bayesian model selection☆23Updated 11 months ago
- Package for Neural Posterior Estimation and Importance Sampling focused on Astronomical Applications☆37Updated last week
- pocoMC: A Python implementation of Preconditioned Monte Carlo for accelerated Bayesian Computation☆96Updated last month
- Neural Network-Boosted Importance Nested Sampling for Bayesian Statistics☆83Updated last month
- Gravitational-wave data analysis tools in Jax☆66Updated last week
- Next generation Bayesian analysis tools for efficient posterior sampling and evidence estimation.☆29Updated 2 years ago
- 🐈 Automatic differentiable spectrum modeling of exoplanets/brown dwarfs using JAX, compatible with NumPyro and Optax/JAXopt☆60Updated this week
- Simulation-based (likelihood-free) inference customized for astronomical applications☆26Updated 4 months ago
- An experiment: emcee implemented in JAX☆25Updated 2 years ago
- Unitful Quantities in JAX☆32Updated this week
- MCHMC: sampler from an arbitrary differentiable distribution☆71Updated 2 weeks ago
- Robust ML in Astro☆51Updated last month
- Nested Sampling post-processing and plotting☆57Updated last week
- Density Estimation Likelihood-Free Inference with neural density estimators and adaptive acquisition of simulations☆108Updated last year
- Coverage tests to check the quality of your posterior estimators.☆30Updated 7 months ago
- JAX-powered Cosmological Particle-Mesh N-body Solver☆34Updated 3 weeks ago
- Galactic and Gravitational Dynamics in Python (+ GPU and autodiff)☆38Updated last week
- ☆16Updated 4 years ago
- Using neural networks to extract sufficient statistics from data by maximising the Fisher information☆31Updated last year
- Differentiable Gravitational Waveforms with JAX☆55Updated 7 months ago
- JAX-based linear Einstein-Boltzmann solver for cosmology☆14Updated last week
- Corner plots, LaTeX tables and plotting walks.☆83Updated last week
- Combines Bayesian analyses from many datasets.☆22Updated 10 months ago
- PyAutoFit: Classy Probabilistic Programming☆63Updated this week
- CPU/GPU agnostic gravitational-wave population inference☆45Updated 2 weeks ago
- Code to replicate posterior probability distributions with bijectors/KDEs and perform marginal KL/bayesian dimensionality calculations.☆13Updated last month