JIT decorators in Python

https://eli.thegreenplace.net/2025/decorator-jits-python-as-a-dsl/

Breaks down how different JIT decorators work, from AST based, to bytecode based, to tracers, which covers Triton, Numba and Jax as examples.

In both cases, the function decorated with jit doesn’t get executed by the Python interpreter in the normal sense. Instead, the code inside is more like a DSL (Domain Specific Language) processed by a special purpose compiler built into the library (JAX or Triton). Another way to think about it is that Python is used as a meta language to describe computations.

Discover more from Ian’s Blog

Subscribe now to keep reading and get access to the full archive.

Continue reading