☆80Sep 4, 2021Updated 4 years ago
Alternatives and similar repositories for fax
Users that are interested in fax are comparing it to the libraries listed below
Sorting:
- Documentation:☆129May 22, 2023Updated 2 years ago
- A differentiation API for PyTorch☆30Jun 21, 2020Updated 5 years ago
- Python package to heuristically solve optimization problems on quasi-2d lattices using approximate tensor network contractions.☆12Sep 24, 2021Updated 4 years ago
- Variational Walkback, NIPS'17☆28Oct 18, 2017Updated 8 years ago
- ☆34Oct 5, 2020Updated 5 years ago
- ☆16Jun 3, 2025Updated 9 months ago
- ☆624Jan 23, 2026Updated last month
- Visualize, create, and operate on pytrees in the most intuitive way possible.☆46Jan 11, 2025Updated last year
- Dump a quantum circuit in Yao to a tensor network graphical model☆11Dec 11, 2020Updated 5 years ago
- A tool for weighted model counting through tensor network contraction☆22Nov 13, 2021Updated 4 years ago
- Mathematical operations for JAX pytrees☆209Dec 5, 2024Updated last year
- JAXChem is a JAX-based deep learning library for complex and versatile chemical modeling☆81Jul 15, 2020Updated 5 years ago
- An implementation of "Supervised Learning with Quantum-Inspired Tensor Networks" using TensorFlow☆11May 13, 2018Updated 7 years ago
- [ICML 2022] Learning Efficient and Robust Ordinary Differential \\ Equations via Invertible Neural Networks☆10Apr 14, 2023Updated 2 years ago
- Fastidious accounting of entropy streams into and out of optimization and sampling algorithms.☆33Apr 19, 2016Updated 9 years ago
- Exponential families for JAX☆76Feb 20, 2026Updated 2 weeks ago
- Ab-initio simulation of interacting fermions with equivariant normalizing flow.☆34Oct 23, 2022Updated 3 years ago
- Inference on non-linear dynamical systems written in JAX☆11Aug 20, 2020Updated 5 years ago
- Stencil computations in JAX☆71Nov 15, 2025Updated 3 months ago
- Tree-structured recurrent switching linear dynamical systems☆38Jul 13, 2020Updated 5 years ago
- A Pytree Module system for Deep Learning in JAX☆213Feb 26, 2023Updated 3 years ago
- Weekly humanitarian AI reading group at Mila.☆21Mar 20, 2019Updated 6 years ago
- Code for the paper "Learning Differential Equations that are Easy to Solve"☆287Nov 25, 2021Updated 4 years ago
- A package for efficient simulation and differentiation of parametrized quantum circuits.☆14Sep 26, 2025Updated 5 months ago
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆121Updated this week
- Extending JAX with custom C++ and CUDA code☆403Aug 18, 2024Updated last year
- Matrix-free linear algebra in JAX.☆160Dec 2, 2025Updated 3 months ago
- JAX code for the paper "Control-Oriented Model-Based Reinforcement Learning with Implicit Differentiation"☆44Jun 14, 2021Updated 4 years ago
- Implementation of iterative inference in deep latent variable models☆43Aug 21, 2019Updated 6 years ago
- madjax☆14Nov 18, 2024Updated last year
- Differential Programming and Natural Evolutationary Strategies optimization algorithms for transport Majorana Zero Modes.☆14Nov 27, 2020Updated 5 years ago
- Implicit generative models and related stuff based on the MMD, in PyTorch☆16Sep 24, 2020Updated 5 years ago
- Matrix exponential in cuda for pytorch and tensorflow☆17Nov 26, 2018Updated 7 years ago
- Forward mode laplacian implemented in JAX tracer☆30Jan 7, 2026Updated last month
- JAX Arrays for human consumption☆110Oct 23, 2025Updated 4 months ago
- A Julia package for n-dimensional sparse tensors.☆27Dec 6, 2021Updated 4 years ago
- Normalizing Flows using JAX☆86Nov 30, 2023Updated 2 years ago
- A differentiable cosmology library in JAX☆225Jun 27, 2025Updated 8 months ago
- A collection of lectures and tutorials☆24Jun 14, 2024Updated last year