Skip to content

Commit

Permalink
Update references to JAX's GitHub repo
Browse files Browse the repository at this point in the history 
JAX has moved from https://github.com/google/jax to https://github.com/jax-ml/jax

PiperOrigin-RevId: 702886844
  • Loading branch information
@jakeharmon8 @copybara-github
jakeharmon8 authored and copybara-github committed Dec 5, 2024
1 parent c6de1e5 commit 74b1f15
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion README.md
View file
Original file line number Diff line number Diff line change
Expand Up @@ -98,7 +98,7 @@ pip install -r vit_jax/requirements.txt
pip install -r vit_jax/requirements-tpu.txt
```

For newer versions of [JAX](https://github.com/google/jax), follow the instructions
For newer versions of [JAX](https://github.com/jax-ml/jax), follow the instructions
provided in the corresponding repository linked here. Note that installation
instructions for CPU, GPU and TPU differs slightly.

Expand Down

0 comments on commit 74b1f15

Please sign in to comment.