NIFTy-PPL / JAXbind
Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation
☆61Updated 2 months ago
Related projects: ⓘ
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆77Updated last week
- Numerical quadrature with JAX☆39Updated last month
- Interpolation and function approximation with JAX☆117Updated 2 weeks ago
- Probabilistic Programming and Nested sampling in JAX☆141Updated 3 weeks ago
- Differentiable and accelerated spherical transforms with JAX☆123Updated this week
- Probabilistic programming framework for signal inference algorithms that operate regardless of the underlying grids and their resolutions☆16Updated this week
- A differentiable cosmology library in JAX☆172Updated last month
- 🐈 Automatic differentiable spectrum modeling of exoplanets/brown dwarfs using JAX, compatible with NumPyro and JAXopt☆45Updated this week
- Normalizing-flow enhanced sampling package for probabilistic inference in Jax☆188Updated last week
- MCHMC: sampler from an arbitrary differentiable distribution☆69Updated this week
- JAX bindings for the NVIDIA cuDecomp library☆24Updated 2 weeks ago
- Gravitational-wave data analysis tools in Jax☆53Updated this week
- Proximal Nested Sampling for high-dimensional Bayesian model selection☆20Updated 5 months ago
- Differentiable Gravitational Waveforms with JAX☆53Updated 3 weeks ago
- Density Estimation Likelihood-Free Inference with neural density estimators and adaptive acquisition of simulations☆102Updated last year
- Simulation-based inference in JAX☆18Updated last month
- ☆85Updated this week
- A system for scientific simulation-based inference at scale.☆158Updated 5 months ago
- ☆18Updated 10 months ago
- Differentiable Cosmological Forward Model☆69Updated 2 weeks ago
- Stencil computations in JAX☆66Updated 11 months ago
- JAX-powered Cosmological Particle-Mesh N-body Solver☆23Updated last month
- Evaluate and transform D matrices, 3-j symbols, and (scalar or spin-weighted) spherical harmonics☆45Updated 9 months ago
- Machine learning assisted marginal likelihood (Bayesian evidence) estimation for Bayesian model selection☆54Updated last month
- JAX + Units☆41Updated last week
- nessai: Nested Sampling with Artificial Intelligence☆35Updated this week
- Using neural networks to extract sufficient statistics from data by maximising the Fisher information☆29Updated last year
- Community-sourced list of papers and resources on neural simulation-based inference.☆84Updated 3 months ago
- A JAX package for weak gravitational lensing☆27Updated 4 months ago
- Probabilistic modeling of tabular data with normalizing flows.☆54Updated 3 months ago