# Training-only dependencies (trainer.py, convert2hf_dataset.py). # Install with: pip install -e ".[train]" datasets==3.4.1 pytorch_lightning==2.5.1 matplotlib==3.10.1 tensorboard tensorboardX