Advanced compilation

Note

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

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

• Ahead-of-time lowering and compilation

• Pallas: a JAX kernel language