Imported from https://github.com/jettify/pytorch-optimizer