M1 GPU `mps` device integration (#596)
* fixing metric computation
* refactoring
* Mac M1 GPU `mps` device support
* Update state.py
* reverting the `nlp_example.py` changes from the copied branch
* resolve comments
Co-Authored-By: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>
* docs quality
* Update docs/source/usage_guides/mps.mdx
Co-authored-by: Zachary Mueller <muellerzr@gmail.com>
* resolving comments
* resolving comments
Co-Authored-By: Zachary Mueller <7831895+muellerzr@users.noreply.github.com>
* resolving comments
* resolving comments
* resolving comments on docs
Co-Authored-By: Zachary Mueller <7831895+muellerzr@users.noreply.github.com>
Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>
Co-authored-by: Zachary Mueller <muellerzr@gmail.com>
Co-authored-by: Zachary Mueller <7831895+muellerzr@users.noreply.github.com>
