probml / dynamax
State Space Models library in JAX
☆661Updated last week
Related projects: ⓘ
- BlackJAX is a Bayesian Inference library designed for ease of use, speed and modularity.☆797Updated 3 weeks ago
- Gaussian processes in JAX.☆436Updated last week
- Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/☆299Updated 3 weeks ago
- Simulation-based inference toolkit☆571Updated this week
- Statistical Rethinking (2nd ed.) with NumPyro☆443Updated 7 months ago
- Hardware accelerated, batchable and differentiable optimizers in JAX.☆918Updated this week
- ☆529Updated 2 months ago
- Uncertainty quantification with PyTorch☆311Updated last month
- Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/☆1,373Updated this week
- Optimal transport tools implemented with the JAX framework, to get differentiable, parallel and jit-able computations.☆499Updated this week
- Structural Time Series in JAX☆179Updated 4 months ago
- Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax☆340Updated 2 weeks ago
- Bayesian learning and inference for state space models☆556Updated last month
- The tiniest of Gaussian Process libraries☆282Updated last week
- Compositional Linear Algebra☆390Updated 2 weeks ago
- PyTorch-based library for Riemannian Manifold Hamiltonian Monte Carlo (RMHMC) and inference in Bayesian neural networks☆407Updated 3 weeks ago
- A Python library for amortized Bayesian workflows using generative neural networks.☆304Updated this week
- Bayes-Newton—A Gaussian process library in JAX, with a unifying view of approximate Bayesian inference as variants of Newton's method.☆220Updated 8 months ago
- More than a hundred strange attractors☆391Updated this week
- Oryx is a library for probabilistic programming and deep learning built on top of Jax.☆211Updated 3 weeks ago
- ☆759Updated 2 months ago
- Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.☆2,132Updated this week
- Turn SymPy expressions into trainable JAX expressions.☆313Updated 4 months ago
- Express & compile probabilistic programs for performant inference on CPU & GPU. Powered by JAX.☆325Updated 5 months ago
- Examples of PyMC models, including a library of Jupyter notebooks.☆272Updated this week
- Notes for the Numerics of Machine Learning Lecture Course at the University of Tübingen☆182Updated 5 months ago
- Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/☆2,038Updated this week
- This library would form a permanent home for reusable components for deep probabilistic programming. The library would form and harness a…☆300Updated 6 months ago
- Functional tensors for probabilistic programming☆234Updated last year
- Optax is a gradient processing and optimization library for JAX.☆1,640Updated this week