TorchLean API

NN.Proofs.Backend.Attention

Maintained CUDA attention selection #

The default profile selects a capsule declaring Lean-composed attention over LibTorch primitives. Its declared VJP ownership matches the TorchLean tape policy; reduction order remains implementation-defined. This does not select or verify an ATen flash, efficient, cuDNN, or math implementation.

A successful default attention plan retains exactly the selected Lean-composed capsule.