danieljtait / jax_xla_adventuresLinks
Dive into Jax, Flax, XLA and C++
☆31Updated 5 years ago
Alternatives and similar repositories for jax_xla_adventures
Users that are interested in jax_xla_adventures are comparing it to the libraries listed below
Sorting:
- ☆80Updated 3 years ago
- Prototypes of differentiable differential equation solvers in JAX.☆27Updated 5 years ago
- Discontinuous Hamiltonian Monte Carlo in JAX☆41Updated 5 years ago
- Turning SymPy expressions into JAX functions☆45Updated 4 years ago
- ☆12Updated 4 years ago
- Inference on non-linear dynamical systems written in JAX☆11Updated 4 years ago
- Visualize, create, and operate on pytrees in the most intuitive way possible.