Skip to main content

Module autograd

Module autograd 

Source
Expand description

Tape-based reverse-mode autograd (roadmap v4, Stage 8).

Define-by-run: a training forward records one Node per op on the Tape; Tape::backward sweeps the nodes in reverse, composing each backward pass out of the already-verified forward ops (matmul with trans_a/trans_b, add, sum_rows) plus the dedicated backward kernels (gelu_bwd, layernorm_bwd, softmax_bwd, scatter_add). One implementation serves both backends because every rule goes through ops.

Structsยง

TVar
A tensor tracked on the tape.
Tape