Loss.max_step: def of bendtorch-core@0.1.0.0
Import: import bendtorch-core@0.1.0.0/src/autograd/loss.bend as Loss
Lines 74-77 of bendtorch-core@0.1.0.0/src/autograd/loss.bend: https://jend.leeeet.dev/read.txt?path=bendtorch-core@0.1.0.0/src/autograd/loss.bend&start_line=74&end_line=77

import Base

# PyTorch's maximum: NaN when either word is NaN, else the larger word, and
# on a tie the later word, as `_mm512_max_ps(m, v)` keeps its second operand.
def max_step(+m: F32, +v: F32) -> F32:
  Bool.pick(F32, Bool.or(F32.is_gt(m, v), F32.is_ne(m, m)), m, v)

Uses (5):
  https://jend.leeeet.dev/definition.txt?path=Base/base.bend&name=Bool.or
  https://jend.leeeet.dev/definition.txt?path=Base/base.bend&name=Bool.pick
  https://jend.leeeet.dev/definition.txt?path=Base/base.bend&name=F32
  https://jend.leeeet.dev/definition.txt?path=Base/base.bend&name=F32.is_gt
  https://jend.leeeet.dev/definition.txt?path=Base/base.bend&name=F32.is_ne