Refactor affinity and make it stateful (#2579)
* Move under initialized check
* One more
* Numa affinity
* Docs
* Import
* Add verbosity
* Apply suggestions from code review
Co-authored-by: Benjamin Bossan <BenjaminBossan@users.noreply.github.com>
* Improve import err
* Test + fix bug
* Update src/accelerate/utils/environment.py
Co-authored-by: Stas Bekman <stas00@users.noreply.github.com>
* Clean
---------
Co-authored-by: Benjamin Bossan <BenjaminBossan@users.noreply.github.com>
Co-authored-by: Stas Bekman <stas00@users.noreply.github.com>
