py-lightning-lite (spack): LightningLite enables pure PyTorch users to scale their existing code on any kind of device while retaining full control over their own loops and optimization logic. 
Keywords: ai|artificial-intelligence|data-science|deep-learning|machine-learning|python|pytorch