From 7999b9f5d04e121bfec4d31d3c66ee3cb7391261 Mon Sep 17 00:00:00 2001 From: Nathaniel Starkman Date: Wed, 5 Feb 2025 20:11:00 -0500 Subject: [PATCH 1/2] docs(readme): add diffraxtra --- README.md | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/README.md b/README.md index a09ad532..35c5c1c7 100644 --- a/README.md +++ b/README.md @@ -63,7 +63,8 @@ If you found this library useful in academic research, please cite: [(arXiv link **Always useful** [Equinox](https://github.com/patrick-kidger/equinox): neural networks and everything not already in core JAX! -[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. +[diffraxtra](https://github.com/GalacticDynamics/diffraxtra): extras for `diffrax`; OOP and vectorization. **Deep learning** [Optax](https://github.com/deepmind/optax): first-order gradient (SGD, Adam, ...) optimisers. From 850025e19b63b6f169eca29679159f9fb823f1f9 Mon Sep 17 00:00:00 2001 From: Nathaniel Starkman Date: Wed, 5 Feb 2025 20:11:58 -0500 Subject: [PATCH 2/2] docs(index): add diffraxtra --- docs/index.md | 1 + 1 file changed, 1 insertion(+) diff --git a/docs/index.md b/docs/index.md index 3f26c539..7c3c9a40 100644 --- a/docs/index.md +++ b/docs/index.md @@ -52,6 +52,7 @@ Have a look at the [Getting Started](./usage/getting-started.md) page. **Always useful** [Equinox](https://github.com/patrick-kidger/equinox): neural networks and everything not already in core JAX! [jaxtyping](https://github.com/patrick-kidger/jaxtyping): type annotations for shape/dtype of arrays. +[diffraxtra](https://github.com/GalacticDynamics/diffraxtra): extras for `diffrax`; OOP and vectorization. **Deep learning** [Optax](https://github.com/deepmind/optax): first-order gradient (SGD, Adam, ...) optimisers.