DeepSpeed

df724e71 - Add a very simple PyTorch Lightning test! (#1726)

Add a very simple PyTorch Lightning test! (#1726) * Add a very simple PyTorch Lightning test * Run for just one epoch * Swap to using the plugin API till the strategy API makes it to 1.6