TorchLean API

NN.MLTheory.Proofs.StateSpace

State-space model proofs #

This entrypoint collects the proof layer for S4/Mamba-style state-space sequence models. We prove two complementary facts:

The CUDA and runtime implementations are free to use efficient scan schedules, but their semantic target is the spec-level recurrence captured here.

References: