mirror of
https://github.com/ROCm/jax.git
synced 2025-04-16 20:06:05 +00:00
21 lines
378 B
ReStructuredText
21 lines
378 B
ReStructuredText
.. _pallas:
|
|
|
|
Pallas: a JAX kernel language
|
|
=============================
|
|
Pallas is an extension to JAX that enables writing custom kernels for GPU and TPU.
|
|
This section contains tutorials, guides and examples for using Pallas.
|
|
|
|
.. toctree::
|
|
:caption: Guides
|
|
:maxdepth: 2
|
|
|
|
design
|
|
quickstart
|
|
|
|
.. toctree::
|
|
:caption: Platform Features
|
|
:maxdepth: 2
|
|
|
|
tpu/index
|
|
|