Simple Learning to Optimize in PyTorch - View it on GitHub