Use amp autocast in ZeRO3 linear (#990)
* Use amp autocast in ZeRO3 linear
* Fix typo
* Handle specific exceptions
* CI breaks on torch.distributed
* Add autocast unit test
* Format fixes
* Fix skip logic
Co-authored-by: Jeff Rasley <jerasley@microsoft.com>
