The equations for the loss function do not render appropriately: https://github.com/mlverse/torch/blob/7ff1cdd49854569d227b53dab8dd410ff4564dc0/R/nn-loss.R#L28 <img width="560" alt="Image" src="https://github.com/user-attachments/assets/92b3e48f-47cf-421a-838d-40dfed21d07b" /> Additionally, it is not clear how to use this function. It would be great if there was a minimal example in the docs as well.