Add more optimizers

[simi2525]

Integrate https://github.com/jettify/pytorch-optimizer into the project. This is highly valuable in order to directly use sometimes superior state-of-the-art optimizers, such as shampoo optimizer, accelerated SGD or others.