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.