Switch JAX pjrt-plugin link. (#15923)
Also drop 'active' from 'active development' to better reflect commit
volume and set expectations...?
Relates to https://github.com/openxla/iree/issues/5454 (which I had some
drafts for but never finished... hmm...)
diff --git a/docs/website/docs/guides/ml-frameworks/jax.md b/docs/website/docs/guides/ml-frameworks/jax.md
index 76da6fb..b074cf4 100644
--- a/docs/website/docs/guides/ml-frameworks/jax.md
+++ b/docs/website/docs/guides/ml-frameworks/jax.md
@@ -10,7 +10,7 @@
# JAX integration
!!! note
- IREE's JAX support is under active development. This page is still under
+ IREE's JAX support is under development. This page is still under
construction.
## :octicons-book-16: Overview
@@ -23,7 +23,7 @@
[iree-org/iree-jax repository](https://github.com/iree-org/iree-jax).
* A PJRT plugin that adapts IREE as a native JAX backend for online / just in
time (JIT) use. This plugin is being developed in the
- [openxla/openxla-pjrt-plugin repository](https://github.com/openxla/openxla-pjrt-plugin).
+ [`integrations/pjrt` directory](https://github.com/openxla/iree/tree/main/integrations/pjrt).
<!-- TODO: Expand on interface differences -->
<!-- TODO: Add quickstart instructions -->
