.. _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