srush / learns-dex
☆33Updated 3 years ago
Related projects ⓘ
Alternatives and complementary repositories for learns-dex
- ☆18Updated 2 years ago
- ☆30Updated 2 years ago
- A JAX implementation of stochastic addition.☆12Updated 2 years ago
- Silly twitter torch implementations.☆46Updated 2 years ago
- Discontinuous Hamiltonian Monte Carlo in JAX☆40Updated 4 years ago
- Inference Combinators in JAX☆43Updated 2 weeks ago
- ☆24Updated 5 years ago
- Dive into Jax, Flax, XLA and C++☆31Updated 4 years ago
- Inference on non-linear dynamical systems written in JAX☆12Updated 4 years ago
- Very deep VAEs in JAX/Flax☆45Updated 3 years ago
- ☆78Updated 3 years ago
- [NeurIPS'19] Deep Equilibrium Models Jax Implementation☆37Updated 4 years ago
- Framework-agnostic library for checking array/tensor shapes at runtime.☆47Updated 3 years ago
- Generic PyTorch implementation of einsum that supports different semirings☆46Updated 3 months ago
- Generic API for dispatch to Pyro backends.☆16Updated 2 years ago
- A differentiation API for PyTorch☆30Updated 4 years ago
- ☆12Updated 4 years ago
- ☆15Updated 3 years ago
- Code for Unbiased Implicit Variational Inference (UIVI)☆13Updated 5 years ago
- Classic papers from Edwin T. Jaynes converted to latex.☆32Updated 2 years ago
- Probabilistic Solution of Differential Equations☆14Updated 2 years ago
- Modelling epidemiological dynamics and performing inference in these models☆27Updated 3 years ago
- Probabilistic Programming eXecution protocol (PPX)☆73Updated 2 years ago
- Prototypes of differentiable differential equation solvers in JAX.☆27Updated 4 years ago
- ☆11Updated 4 years ago
- This repository provides open-source code for sparse continuous distributions and corresponding Fenchel-Young losses.☆16Updated last year
- Recognizing and exploiting conjugacy without a domain-specific language☆36Updated 4 years ago
- Documentation:☆120Updated last year
- ☆98Updated 4 months ago
- Code accompanying VarGrad: A Low-Variance Gradient Estimator for Variational Inference☆12Updated 4 years ago