We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 2999e72 commit 450b4dcCopy full SHA for 450b4dc
docs/source/features/pallas.md
@@ -110,3 +110,7 @@ the proper JAX:
110
``` bash
111
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
112
```
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