feat: added mps support for inference#81

Skylerwiernik wants to merge 2 commits into

facebookresearch:mainfacebookresearch/vjepa2:mainfrom

Skylerwiernik:mainSkylerwiernik/vjepa2:mainCopy head branch name to clipboard