From 9c9635d4f323d2f0841971a9fab6079a7e00fe0e Mon Sep 17 00:00:00 2001 From: Patrick Kidger <33688385+patrick-kidger@users.noreply.github.com> Date: Fri, 6 Oct 2023 15:51:22 +0100 Subject: [PATCH] Add Orbax to ecosystem list. --- README.md | 2 ++ 1 file changed, 2 insertions(+) diff --git a/README.md b/README.md index 39a9919..3ec2f64 100644 --- a/README.md +++ b/README.md @@ -55,6 +55,8 @@ Available at [https://docs.kidger.site/jaxtyping](https://docs.kidger.site/jaxty [BlackJAX](https://github.com/blackjax-devs/blackjax): probabilistic+Bayesian sampling. +[Orbax](https://github.com/google/orbax): checkpointing (async/multi-host/multi-device). + [sympy2jax](https://github.com/google/sympy2jax): SymPy<->JAX conversion; train symbolic expressions via gradient descent. [Eqxvision](https://github.com/paganpasta/eqxvision): computer vision models.