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 90bf627 commit 7999b9fCopy full SHA for 7999b9f
1 file changed
README.md
@@ -63,7 +63,8 @@ If you found this library useful in academic research, please cite: [(arXiv link
63
64
**Always useful**
65
[Equinox](https://github.com/patrick-kidger/equinox): neural networks and everything not already in core JAX!
66
-[jaxtyping](https://github.com/patrick-kidger/jaxtyping): type annotations for shape/dtype of arrays.
+[jaxtyping](https://github.com/patrick-kidger/jaxtyping): type annotations for shape/dtype of arrays.
67
+[diffraxtra](https://github.com/GalacticDynamics/diffraxtra): extras for `diffrax`; OOP and vectorization.
68
69
**Deep learning**
70
[Optax](https://github.com/deepmind/optax): first-order gradient (SGD, Adam, ...) optimisers.
0 commit comments