From 0a613272e86d8f55cc0fca5f8293258cd8289240 Mon Sep 17 00:00:00 2001
From: zofia <110436990+zufangzhu@users.noreply.github.com>
Date: Wed, 6 Sep 2023 10:39:06 +0800
Subject: [PATCH] refine readme (#58)

---
 README.md | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/README.md b/README.md
index e6bb2ec..b5a92bf 100644
--- a/README.md
+++ b/README.md
@@ -66,6 +66,12 @@ source /opt/intel/oneapi/mkl/2023.1.0/env/vars.sh
 source /opt/intel/oneapi/tbb/2021.9.0/env/vars.sh
 ```
 
+### Install Jax and Jaxlib
+
+```bash
+pip install jax==0.4.13 jaxlib==0.4.13
+```
+
 ## 3. Install
 
 ### Install via PyPI wheel
@@ -78,7 +84,6 @@ pip install --upgrade intel-extension-for-openxla
 
 ```bash
 git clone https://github.com/intel/intel-extension-for-openxla.git
-pip install jax==0.4.13 jaxlib==0.4.13
 ./configure        # Choose Yes for all.
 bazel build //xla/tools/pip_package:build_pip_package
 ./bazel-bin/xla/tools/pip_package/build_pip_package ./