lockwo / distreqxLinks
Distrax, but in equinox. Lightweight JAX library of probability distributions and bijectors.
☆31Updated last month
Alternatives and similar repositories for distreqx
Users that are interested in distreqx are comparing it to the libraries listed below
Sorting:
- A dataloader, but for JAX☆21Updated last year
- Add a tqdm progress bar to your JAX scans and loops.☆109Updated 3 weeks ago
- Jaxpr Visualisation Tool☆28Updated 5 months ago
- ☆161Updated 3 weeks ago
- JAX Arrays for human consumption☆91Updated last year
- diffusionjax is a simple and accessible diffusion models package in JAX☆45Updated 4 months ago
- Stainless neural networks in JAX☆34Updated 2 months ago
- Multiple dispatch over abstract array types in JAX.☆124Updated this week
- Probabilistic solvers for differential equations in JAX. Adaptive ODE solvers with calibration, state-space model factorisations, and cus…☆48Updated 2 months ago
- Visualize, create, and operate on pytrees in the most intuitive way possible.☆45Updated 4 months ago
- Turn jitted jax functions back into python source code☆22Updated 5 months ago
- Pytorch-like dataloaders for JAX.☆83Updated last week
- Amortized Probabilistic Conditioning for Optimization, Simulation and Inference (Chang et al., AISTATS 2025)☆16Updated last month
- Differentiable Principal Component Analysis (PCA) implementation in JAX☆27Updated last month
- Gradient-based constrained optimization for JAX☆32Updated 2 years ago
- Myriad is a real-world testbed that aims to bridge trajectory optimization and deep learning.☆66Updated last year
- A small library of paramaterizations and parameter constraints for PyTrees.☆31Updated 3 weeks ago
- Matrix-free linear algebra in JAX.☆119Updated last week
- ☆52Updated 2 years ago
- Minimal Implementation of Bayesian Optimization in JAX☆95Updated last month
- Minimal Gaussian process library in JAX with a simple (custom) approach to state management.☆12Updated last year
- A Python package of computer vision models for the Equinox ecosystem.☆104Updated 10 months ago
- Simple tools to mix and match PyTorch and Jax - Get the best of both worlds!☆30Updated last month
- ☆34Updated 3 years ago
- Interpolation and function approximation with JAX☆190Updated 3 weeks ago
- Flow-matching algorithms in JAX☆92Updated 9 months ago
- Bayesian inference with Python and Jax.☆32Updated 2 years ago
- Minimal JAX implementation of k-nearest neighbors using a k-d tree.☆37Updated this week
- Lightweight library of stochastic gradient MCMC algorithms written in JAX.☆103Updated last year
- Automatic Functional Differentiation in JAX☆72Updated 2 months ago