Allow specifying datasets in configs, change default optimizer, improved CUDA support#1

aronson merged 6 commits intomainallura-org/mlx-trainer:mainfrom

fizzAI-patch-1allura-org/mlx-trainer:fizzAI-patch-1Copy head branch name to clipboard