jettify / pytorch-optimizer Python 3.2K 307 torch-optimizer -- collection of optimizers for Pytorch pytorch optimizer diffgrad adamod