Module training 

Structs

TrainingMetrics

Training metrics

Functions

save_training_plot

Save training plot

train_standard

Standard training loop with checkpoint support

train_with_improvements

Training with improvements (curriculum learning, improved rewards, domain extraction, etc.)