Reviews
Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.
Search similar apps
License
Apache License 2.0
Related apps
diffrax
Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capab
Python1191apache-2.0
2 months ago
deep-learningdifferential-equationsdynamical-systems
equinox
Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.
Python1603apache-2.0
4 months ago
deep-learningequinoxjax