v0.0.2

Added

• add ddp that works with oss with reduce() not all_reduce() (#19)
• support for PyTorch v1.6
• add mixed precision Adam (#40)
• Adam optimizer state scaling (#44)

Fixed

• properly restore a sharded optim state (#39)
• OSS restore state to proper device (#46)
• optim/oss: support optimizers with additional step kwargs (#53)
• optim/oss: fix state cast (#56)
• fix eval for oss_ddp (#55)
• optim/oss: work correctly with LRScheduler (#58)