Have you tried using this package on other optimizers?
And if so, do you get an error? From the docs I cannot see that it’s only limited to Adam or any other optimizer besides this special warmup for RAdam 485.