Use MPS device (via Pytorch for Apple silicon) when available instead of cpu d9f299b Doron Adler commited on Jun 26