diff --git a/docs/developer_guides/torchax_model_development.md b/docs/developer_guides/torchax_model_development.md index a4273e1b9..810d7b332 100644 --- a/docs/developer_guides/torchax_model_development.md +++ b/docs/developer_guides/torchax_model_development.md @@ -27,7 +27,7 @@ and implementing all `torch` operators that this tensor is supposed to support. ![torchax](../assets/torchax.png) -See [this page](https://jax-torch-interop.readthedocs.io/en/latest) for more details on how torchax works. +See [this page](https://google.github.io/torchax/) for more details on how torchax works. ## TPU JAX worker