Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.
updated at May 10, 2024, 10:41 a.m.
Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/
updated at May 11, 2024, 4:48 p.m.
Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
updated at May 12, 2024, 12:38 p.m.