Files
ai-toolkit/toolkit/optimizers/optimizer_utils.py