We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 7999b9f commit 850025eCopy full SHA for 850025e
1 file changed
docs/index.md
@@ -52,6 +52,7 @@ Have a look at the [Getting Started](./usage/getting-started.md) page.
52
**Always useful**
53
[Equinox](https://github.com/patrick-kidger/equinox): neural networks and everything not already in core JAX!
54
[jaxtyping](https://github.com/patrick-kidger/jaxtyping): type annotations for shape/dtype of arrays.
55
+[diffraxtra](https://github.com/GalacticDynamics/diffraxtra): extras for `diffrax`; OOP and vectorization.
56
57
**Deep learning**
58
[Optax](https://github.com/deepmind/optax): first-order gradient (SGD, Adam, ...) optimisers.
0 commit comments