I have been working on the following PR to add support for unified torch.accelerator API for Apple MPS:
Its been couple of months and I haven’t got a feedback on changes I implemented for incorporate initial review commments.
I feel slack would help me better coordinate with the reviewers and help take this PR to merge.