Exponential families for JAX
☆76Feb 20, 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☆50Feb 20, 2026Updated last week
- Oryx is a library for probabilistic programming and deep learning built on top of Jax.☆302Updated this week
- Mathematical operations for JAX pytrees☆209Dec 5, 2024Updated last year
- Stencil computations in JAX☆71Nov 15, 2025Updated 3 months ago
- Lightweight library of stochastic gradient MCMC algorithms written in JAX.☆105Oct 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
- Bayes-Newton—A Gaussian process library in JAX, with a unifying view of approximate Bayesian inference as variants of Newton's method.☆241Dec 22, 2023Updated 2 years ago
- Dive into Jax, Flax, XLA and C++☆32Apr 1, 2020Updated 5 years ago
- AeMCMC is a Python library that automates the construction of samplers for Aesara graphs representing statistical models.☆39Oct 23, 2023Updated 2 years ago
- A package for efficient simulation and differentiation of parametrized quantum circuits.☆14Sep 26, 2025Updated 5 months ago
- Probabilistic Programming and Nested sampling in JAX☆221Feb 18, 2026Updated last week
- A dead simple Python package for creating custom JAX pytree objects☆66Nov 14, 2024Updated last year
- BlackJAX is a Bayesian Inference library designed for ease of use, speed and modularity.☆1,021Feb 3, 2026Updated 3 weeks ago
- Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python☆516Feb 23, 2026Updated last week
- Gaussian processes in JAX and Flax.☆593Feb 20, 2026Updated last week
- Pytrees + dataclasses ❤️☆75Dec 19, 2025Updated 2 months ago
- Tools for an Aesara-based PPL.☆67Oct 28, 2024Updated last year
- Normalizing Flows using JAX☆86Nov 30, 2023Updated 2 years ago
- ☆16Apr 8, 2025Updated 10 months ago
- Documentation:☆129May 22, 2023Updated 2 years ago
- Self-explaining Matrix Product States library in Python☆22Sep 2, 2023Updated 2 years ago
- Code for the paper "Tensor Networks for Maching Learning"☆17Nov 7, 2019Updated 6 years ago
- Probabilistic deep learning using JAX☆15Feb 8, 2025Updated last year
- Groundstate Reconstructive Energies using Neural Autoregressive Distribution Estimators (GreNADE)☆10Apr 1, 2022Updated 3 years ago
- ☆19Sep 15, 2021Updated 4 years ago
- ☆620Jan 23, 2026Updated last month
- A Pytree Module system for Deep Learning in JAX☆213Feb 26, 2023Updated 3 years ago
- Continuous Matrix Product States in Julia☆30Sep 24, 2025Updated 5 months 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 projector QMC implementation of the transverse field Ising model☆11Oct 6, 2020Updated 5 years ago
- A package for creating a matrix product state (MPS) from a free fermion (Gaussian) state.☆10Dec 6, 2021Updated 4 years ago
- A differentiation API for PyTorch☆30Jun 21, 2020Updated 5 years ago
- A Python package for probabilistic state space modeling with JAX☆929Jan 6, 2026Updated last month
- Extending JAX with custom C++ and CUDA code☆403Aug 18, 2024Updated last year
- ☆80Sep 4, 2021Updated 4 years ago
- Notebook for illustrating the use of CMPSKit.jl and to reproduce the results from arXiv:2006.01801.☆13Nov 9, 2021Updated 4 years ago
- Jax SSM Library☆48Nov 24, 2022Updated 3 years ago
- Code for Hutch++: Optimal Stochastic Trace Estimation☆13Aug 30, 2021Updated 4 years ago