Skip to main content
Ctrl+K

Advanced compilation#

Note

This is a placeholder for a section in the new JAX tutorials.

For the time being, you may find some related content in the old documentation:

  • Ahead-of-time lowering and compilation

  • Custom operations for GPUs with C++ and CUDA

  • Pallas: a JAX kernel language