LionW - Improve quality for your training Lion x AdamW How To Use? from lionw import LionW #and use it normally model_net_g = LionW(model_parameters,...)