Training Module

The training module contains the necessary components and configurations for training the Brain Foundation Models.

Key features:

• Training Setups: Defines model architectures and training paradigms for brain foundation models.
• Optimizers: Provides support for custom optimization algorithms and learning rate schedules.
• Custom Components: Easy integration of custom components for specific research needs through the training_setup interface and the registry system.