Model training

This module contains all code related to model training.

• Training Loop
• Training Loop Config
• Training state
• Training step
• Optimizer
• Exponential Moving Average (EMA)
• IO handling during training
• Model evaluation
• Fine-tuning utilities