Bendtorch.OptState: type of bendtorch@1.3.0.0
Import: import bendtorch@1.3.0.0/main.bend as Bendtorch
Lines 62-67 of bendtorch@1.3.0.0/main.bend: https://jend.leeeet.dev/read.txt?path=bendtorch@1.3.0.0/main.bend&start_line=62&end_line=67

import Base

# Owned optimizer buffers: SGD has none, momentum one, Adam/AdamW two (Packed: two, in packed BF16, see PTree).
type OptState<-P: Type> is Type:
  NoMom{}
  OneMom{m: P}
  TwoMom{m: P, v: P}
  Packed{t: PTree}

Uses (1):
  https://jend.leeeet.dev/definition.txt?path=bendtorch@1.3.0.0/main.bend&name=PTree