Last released Sep 13, 2023
Visualize, create, and operate on JAX PyTree in the most intuitive way possible.
Last released Jul 27, 2023
Finite difference tools in JAX
Last released Apr 20, 2023
JAX NN library.
Last released Jun 9, 2023
Stencil computations in JAX.
Supported by
