[Feature Request] Array API loss

[njzjz]

Summary

Migrate PT losses to Array API losses.

Detailed Description

Need the consistent tests for loss.

Further Information, Files, and Links

No response