Skip to content

Enable cross entropy loss for xla autocast with FP32 precision#7992

Merged
JackCaoG merged 3 commits intor2.1_aws_neuronfrom autocast_bf16_neuronSep 26, 2024

Commits

Commits on Sep 4, 2024

Commits on Sep 11, 2024