Reviews
Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/
Search similar apps
License
Apache License 2.0
Related apps
torchtyping
Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.
Python1318apache-2.0
11 months ago
named-tensorspython-typingpytorch
equinox
Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.
Python1846apache-2.0
21 hours ago
deep-learningequinoxjax