pytorch_lightning
loralib
fairscale