Exponential families for JAX
☆76Mar 13, 2026Updated last week
Alternatives and similar repositories for efax
Users that are interested in efax are comparing it to the libraries listed below
Sorting:
- Tools for JAX☆50Updated this week
- Oryx is a library for probabilistic programming and deep learning built on top of Jax.☆304Updated this week
- Lightweight library of stochastic gradient MCMC algorithms written in JAX.☆105Oct 23, 2023Updated 2 years ago
- Mathematical operations for JAX pytrees☆210Dec 5, 2024Updated last year
- Stencil computations in JAX☆71Nov 15, 2025Updated 4 months ago
- AeMCMC is a Python library that automates the construction of samplers for Aesara graphs representing statistical models.☆39Oct 23, 2023Updated 2 years ago
- Python package to heuristically solve optimization problems on quasi-2d lattices using approximate tensor network contractions.☆12Sep 24, 2021Updated 4 years ago
- ☆16Apr 8, 2025Updated 11 months ago
- A dead simple Python package for creating custom JAX pytree objects☆66Nov 14, 2024Updated last year
- Bayes-Newton—A Gaussian process library in JAX, with a unifying view of approximate Bayesian inference as variants of Newton's method.☆242Dec 22, 2023Updated 2 years ago
- BlackJAX is a Bayesian Inference library designed for ease of use, speed and modularity.☆1,033Feb 3, 2026Updated last month
- Probabilistic Programming and Nested sampling in JAX☆228Mar 15, 2026Updated last week
- A package for efficient simulation and differentiation of parametrized quantum circuits.☆14Sep 26, 2025Updated 5 months ago
- Dive into Jax, Flax, XLA and C++☆32Apr 1, 2020Updated 5 years ago
- Gaussian processes in JAX and Flax.☆602Mar 14, 2026Updated last week
- Tools for an Aesara-based PPL.☆67Oct 28, 2024Updated last year
- Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python☆519Mar 2, 2026Updated 2 weeks ago
- Pytrees + dataclasses ❤️☆75Dec 19, 2025Updated 3 months ago
- Bayesian model reduction for probabilistic machine learning☆11Jul 3, 2025Updated 8 months ago
- Documentation:☆130May 22, 2023Updated 2 years ago
- ☆627Jan 23, 2026Updated last month
- Probabilistic deep learning using JAX☆15Feb 8, 2025Updated last year
- A Python package for probabilistic state space modeling with JAX☆941Jan 6, 2026Updated 2 months ago
- Normalizing Flows using JAX☆87Nov 30, 2023Updated 2 years ago
- A generic library for linear and non-linear Gaussian smoothing problems. The code leverages JAX and implements several linearization algo…☆13Dec 4, 2024Updated last year
- ☆80Sep 4, 2021Updated 4 years ago
- A package for creating a matrix product state (MPS) from a free fermion (Gaussian) state.☆10Dec 6, 2021Updated 4 years ago
- Code for the paper "Tensor Networks for Maching Learning"☆17Nov 7, 2019Updated 6 years ago
- ☆19Sep 15, 2021Updated 4 years ago
- An implementation of "Supervised Learning with Quantum-Inspired Tensor Networks" using TensorFlow☆11May 13, 2018Updated 7 years ago
- Quantum GAN algorithm based on MPS, realized by Julia.☆11Apr 5, 2019Updated 6 years ago
- A Pytree Module system for Deep Learning in JAX☆212Feb 26, 2023Updated 3 years ago
- Continuous Matrix Product States in Julia☆30Sep 24, 2025Updated 5 months ago
- Self-explaining Matrix Product States library in Python☆22Sep 2, 2023Updated 2 years ago
- Inference Combinators in JAX☆52May 17, 2025Updated 10 months ago
- A metrics library for the JAX ecosystem☆41Mar 16, 2023Updated 3 years ago
- Discontinuous Hamiltonian Monte Carlo in JAX☆42Feb 24, 2020Updated 6 years ago
- Jax SSM Library☆48Nov 24, 2022Updated 3 years ago
- An HMC/NUTS implementation in Aesara☆31Jun 22, 2023Updated 2 years ago