Add support for MPS Backend [without torch.amp.autocast ] + CI
#4818
| Job | Run time |
|---|---|
| 1m 44s | |
| 1m 39s | |
| 1m 59s | |
| 2m 3s | |
| 1m 43s | |
| 1m 52s | |
| 1m 45s | |
| 1m 49s | |
| 3m 50s | |
| 3m 22s | |
| 21m 46s |