type EMOM
import bendtorch-gpu@1.3.0.0/main.bend as BendtorchGpuimport Base
217type EMOM is Data:
218 EMOM{n: U32, lr: F32, mu: F32}
import bendtorch-gpu@1.3.0.0/main.bend as BendtorchGpuimport Base
217type EMOM is Data:
218 EMOM{n: U32, lr: F32, mu: F32}