uniperceiver/optim/__init__.py

from .build import build_optimizer, create_moe_param_groups, create_seperate_moe_param_groups, create_group_moe_param_groups


from .adamw import AdamW
from .lamb import LAMB