PyYAML>=6.0

[train]
mlx-lm[train]>=0.31.0
