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