gehring / faxView external linksLinks
☆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:☆127May 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 8 months ago
- ☆622Jan 23, 2026Updated 3 weeks ago
- 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☆21Nov 13, 2021Updated 4 years ago
- Mathematical operations for JAX pytrees☆208Dec 5, 2024Updated last year
- JAXChem is a JAX-based deep learning library for complex and versatile chemical modeling☆81Jul 15, 2020Updated 5 years ago
- [ICML 2022] Learning Efficient and Robust Ordinary Differential \\ Equations via Invertible Neural Networks☆10Apr 14, 2023Updated 2 years ago
- An implementation of "Supervised Learning with Quantum-Inspired Tensor Networks" using TensorFlow☆11May 13, 2018Updated 7 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 1, 2026Updated last week
- 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 2 months ago
- Tree-structured recurrent switching linear dynamical systems☆38Jul 13, 2020Updated 5 years ago
- A Pytree Module system for Deep Learning in JAX☆214Feb 26, 2023Updated 2 years ago
- Code for the paper "Learning Differential Equations that are Easy to Solve"☆286Nov 25, 2021Updated 4 years ago
- A package for efficient simulation and differentiation of parametrized quantum circuits.☆14Sep 26, 2025Updated 4 months ago
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆115Feb 3, 2026Updated last week
- Extending JAX with custom C++ and CUDA code☆403Aug 18, 2024Updated last year
- Matrix-free linear algebra in JAX.☆159Dec 2, 2025Updated 2 months ago
- Implementation of iterative inference in deep latent variable models☆43Aug 21, 2019Updated 6 years ago
- JAX code for the paper "Control-Oriented Model-Based Reinforcement Learning with Implicit Differentiation"☆44Jun 14, 2021Updated 4 years ago
- Matrix exponential in cuda for pytorch and tensorflow☆17Nov 26, 2018Updated 7 years ago
- Implicit generative models and related stuff based on the MMD, in PyTorch☆16Sep 24, 2020Updated 5 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
- Forward mode laplacian implemented in JAX tracer☆30Jan 7, 2026Updated last month
- JAX Arrays for human consumption☆111Oct 23, 2025Updated 3 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☆222Jun 27, 2025Updated 7 months ago
- A collection of lectures and tutorials☆24Jun 14, 2024Updated last year
- ☆31Mar 10, 2020Updated 5 years ago
- Dive into Jax, Flax, XLA and C++☆32Apr 1, 2020Updated 5 years ago