From 900ef1dda1a16d1d8f2c404bfd0dbb006f81eced Mon Sep 17 00:00:00 2001
From: Twice <twice@apache.org>
Date: Fri, 13 Dec 2024 03:58:06 +0800
Subject: [PATCH] [PJRT] Update README to align with the current status
 (#19457)

Some content in the README of PJRT plugin is outdated, so in this PR I
try to update them.

Since it is difficult for me to decide some of the content directly,
there will be some comments below to solicit people's suggestions : )

---------

Signed-off-by: PragmaTwice <twice@apache.org>
---
 integrations/pjrt/README.md | 24 +++++++++++++-----------
 1 file changed, 13 insertions(+), 11 deletions(-)

diff --git a/integrations/pjrt/README.md b/integrations/pjrt/README.md
index 04f8aaaa4c2a..9c018be70f58 100644
--- a/integrations/pjrt/README.md
+++ b/integrations/pjrt/README.md
@@ -13,12 +13,11 @@ most powerful).
 
 ## Install a compatible version of Jax and the IREE compiler
 
-```
+```shell
 pip install -r requirements.txt
 
-# Assume that you have the Jax repo checked out at JAX_REPO from
-# https://github.com/google/jax (must be paired with nightly jaxlib).
-pip install -e $JAX_REPO
+# a higher version of jax is highly recommended, e.g. 0.4.36
+pip install jax==0.4.36
 ```
 
 Verify that your Jax install is functional like:
@@ -78,14 +77,17 @@ The plugin `openxla_pjrt_artifacts` is in the `ctstools` directory and
 performs additional manipulation of the environment in order to save
 compilation artifacts, reproducers, etc.
 
-## Contacts
+## Communication channels
 
-* [GitHub issues](https://github.com/openxla/openxla-pjrt-plugin/issues):
-  Feature requests, bugs, and other work tracking
-* [OpenXLA discord](https://discord.gg/pvuUmVQa): Daily development discussions
-  with the core team and collaborators
+* Please submit feature requests and bug reports about the plugin in [GitHub Issues](https://github.com/iree-org/iree/issues).
+* Discuss the development of the plugin at `#jax` or `#pjrt-plugin` channel of [IREE Discord server](https://discord.gg/wEWh6Z9nMU).
+* Check the [OpenXLA/XLA](https://github.com/openxla/xla) repo and [its communication channels](https://github.com/openxla/community?tab=readme-ov-file#communication-channels) for PJRT APIs and clients.
 
 ## License
 
-OpenXLA PJRT plugin is licensed under the terms of the Apache 2.0 License with
-LLVM Exceptions. See [LICENSE](LICENSE) for more information.
+IREE PJRT plugin is licensed under the terms of the Apache 2.0 License with
+LLVM Exceptions. See [LICENSE](../../LICENSE) for more information.
+
+[PJRT C API](./third_party/pjrt_c_api) comes from
+[OpenXLA/XLA](https://github.com/openxla/xla) and is licensed under
+the Apache 2.0 License. See its own [LICENSE](./third_party/pjrt_c_api/LICENSE) for more information.