[Feature Request] Array API loss #4429
Description
Summary
Migrate PT losses to Array API losses.
Detailed Description
Need the consistent tests for loss.
Further Information, Files, and Links
No response
{{ message }}
Migrate PT losses to Array API losses.
Need the consistent tests for loss.
No response