the half bank parallel to the parameter bank (element i's half at byte
2 i), so AdamW writes each parameter's half beside its update
(Realization.Nvidia.SM86.AdamWHalfSM86): matrix index's copy at half its
entry's offset
356def cgHalfOffset =
357 (lambda unrestricted index : Nat .
358 (naturalAdd cgHalfStart (naturalDivideUnchecked (cgEntryOffset (cgMatrixEntry index)) 2)))The compiler supplied declaration spans and resolved links from this source snapshot. This page does not assert that this file belongs to a checked closure.