def tr_adam
import bendtorch@1.3.0.0/main.bend as Bendtorchimport Base
3542def tr_adam(-P: Type, p: P, mm: P, vv: P, g: P, r: Array<F32> & F32, +off: U32) -> Train<P>:
3543 match r:
3544 case Tuple{xs, p1}: tr_adam2(P, p, mm, vv, g, p1, Array.get(F32, xs, off))