01
recipe = Float4()02
with te.autocast(enabled=True, recipe=recipe):03
logits = model(tokens)04
quality = validate(logits, reference)