Integrations
Add TraceML training diagnostics to the tools you already use.
Hugging Face Trainer
Add training diagnostics to Transformers Trainer.
Read docsPyTorch Lightning
Measure training steps with a Lightning callback.
Read docsRay Train
Inspect PyTorch training workloads on Ray Train.
Read docsMONAI
Diagnose medical imaging training with MONAI’s SupervisedTrainer.
Read docsRF-DETR
Measure RF-DETR training time and compare runs.
Read docsCustom PyTorch loops
Add TraceML directly to your training script.