Skip to content

Commit 850025e

Browse files
committed
docs(index): add diffraxtra
1 parent 7999b9f commit 850025e

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

docs/index.md

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -52,6 +52,7 @@ Have a look at the [Getting Started](./usage/getting-started.md) page.
5252
**Always useful**
5353
[Equinox](https://github.com/patrick-kidger/equinox): neural networks and everything not already in core JAX!
5454
[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.
5556

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

0 commit comments

Comments
 (0)