Skip to content

Commit 7999b9f

Browse files
committed
docs(readme): add diffraxtra
1 parent 90bf627 commit 7999b9f

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

README.md

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -63,7 +63,8 @@ If you found this library useful in academic research, please cite: [(arXiv link
6363

6464
**Always useful**
6565
[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.
66+
[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.
6768

6869
**Deep learning**
6970
[Optax](https://github.com/deepmind/optax): first-order gradient (SGD, Adam, ...) optimisers.

0 commit comments

Comments
 (0)