readme

zinccat · Oct 22, 2024 · 039058a · 039058a
1 parent 9487828
commit 039058a
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -2,7 +2,10 @@
 
 Porting [FlexAttention](https://github.com/pytorch-labs/attention-gym) to pure JAX.
 
-Please install Jax nightly: pip install -U --pre jax jaxlib "jax-cuda12-plugin[with_cuda]" jax-cuda12-pjrt -f https://storage.googleapis.com/jax-releases/jax_nightly_releases.html
+Please install Jax nightly: 
+```bash
+pip install -U --pre jax jaxlib "jax-cuda12-plugin[with_cuda]" jax-cuda12-pjrt -f https://storage.googleapis.com/jax-releases/jax_nightly_releases.html
+```
 
 Example usage: