Module train 

Structs

StepResult

Training step result.

Functions

train_step

Train an MLP (or any differentiable graph) for one step. forward_fn builds the computation graph on the tape and returns (loss_id, param_ids). The training loop runs backward, extracts gradients, and updates params.