TorchLean IBP Workflow #
Small end-to-end workflow:
TorchLean forward model → lower to NN.IR.Graph → run Lean IBP (runIBP).
Run:
lake exe verify -- torchlean-ibp
lake exe verify -- torchlean-ibp --arithmetic ieee
Input dimension for the small MLP in this workflow.
Instances For
Hidden width for the small MLP in this workflow.
Instances For
Output dimension for the small MLP in this workflow.
Instances For
Input shape for the workflow model.
Instances For
Output shape for the workflow model.
Instances For
TorchLean model used in the workflow (a 2-layer ReLU MLP).
Instances For
Deterministically instantiate mkModel from initialization seed zero.
Instances For
Parameter shapes for model.
Instances For
def
NN.Verification.Builtin.IBPWorkflow.runMain
{α : Type}
[TorchLean.Storage α]
[Context α]
[ToString α]
[TorchLean.Runtime.FromFloat α]
[MLTheory.CROWN.BoundOps α]
[MLTheory.CROWN.NonlinearBoundOps α]
:
Runtime-selected typed runner used by the CLI entrypoint.