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