Skip to content

Commit 450b4dc

Browse files
authored
Add Pallas tutorials to the Pallas guide at torch_xla. (#8904)
1 parent 2999e72 commit 450b4dc

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

docs/source/features/pallas.md

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -110,3 +110,7 @@ the proper JAX:
110110
``` bash
111111
pip install torch_xla[pallas] -f https://storage.googleapis.com/jax-releases/jax_nightly_releases.html -f https://storage.googleapis.com/jax-releases/jaxlib_nightly_releases.html
112112
```
113+
114+
## Write your own Pallas kernels
115+
116+
You can find the authoritative guide on how to write a Pallas kernel at https://docs.jax.dev/en/latest/pallas/index.html.

0 commit comments

Comments
 (0)