mlx>=0.2
numpy
tqdm
scikit-learn
matplotlib