Make env variables optional for FSDP (#2998)
* Bookmark
* Tests pass!
* Fix imports
* Try with raw dict
* Make diff easier
* Add defaults to all relevent areas
* Rest of refactor
* Fix all of benjamin's nits
* Adjust logic based on Benjamin's feedback
* Adjust for new logic
