Autograd engine #
This is the public umbrella for TorchLean's low-level eager autograd engine.
Engine.Coreis the pure CPU tape over shape-erasedRuntime.AnyTensorvalues.Engine.TapeMis aStateTconvenience wrapper around the pure tape.Engine.FastKernelsprovides opt-in runtime kernels for hot CPU/GPU paths.Engine.Cudacollects the CUDA float32 tape, FFI kernels, and proof-facing native contracts.
Higher-level APIs should usually import NN.Runtime.Autograd.Torch or
NN.Runtime.Autograd.TorchLean; this module is for code that intentionally works at the tape
engine boundary.