Stax

Just-in-time compilation as program specialization: guards, specialization keys, tracing-time proxy capture, data-dependent control flow, and graph rewrites.

A systematic tour of Stax, TensorPlay’s compiler backend: specialization, graph capture, transformations, autodiff, loop-level IR, scheduling, and codegen.