lm-eval passes `implementation` to `fused_cross_entropy_loss_and_logsumexp_penalty` from `levanter.models.loss` here: https://github.com/marin-community/marin/blob/e78e1dc780f6495fa37a946eacf54f5ae21af9fc/lib/levanter/src/levanter/eval_harness.py#L284 and this isn't supported in any of the overloads of `fused_cross_entropy_loss_and_logsumexp_penalty` of [levanter/models/loss.py#L160-L216](https://github.com/marin-community/marin/blob/e78e1dc780f6495fa37a946eacf54f5ae21af9fc/lib/levanter/src/levanter/models/loss.py#L160-L216).
lm-eval passes
implementationtofused_cross_entropy_loss_and_logsumexp_penaltyfromlevanter.models.losshere:marin/lib/levanter/src/levanter/eval_harness.py
Line 284 in e78e1dc
and this isn't supported in any of the overloads of
fused_cross_entropy_loss_and_logsumexp_penaltyof levanter/models/loss.py#L160-L216.