Trackio is an open-source Python library for experiment tracking, designed to be lightweight, transparent, and easy to integrate. The library is under 1,000 lines of code, making it hackable and extensible. Logs persist locally in SQLite and are automatically backed up to Parquet datasets on Hugging Face every five minutes when synced. Trackio provides local dashboards by default and can sync with Hugging Face Spaces for sharing and collaboration. The API is compatible with wandb for rapid migration and integrates with transformers and accelerate for minimal-setup logging. Trackio directly tracks GPU energy via nvidia-smi and can include results in model cards, supporting reproducibility and environmental reporting.
Hugging Face has introduced Trackio, a new open-source Python library for experiment tracking designed to be lightweight, transparent, and easy to integrate. Built as a drop-in replacement for Weights & Biases (wandb), Trackio offers local dashboards by default and seamless syncing with Hugging Face Spaces for sharing and collaboration. The library is under 1,000 lines of code, making it hackable and extensible. Logs persist locally in SQLite and are automatically backed up to Parquet datasets on Hugging Face every five minutes when synced.
Key features include: API compatibility with wandb, enabling rapid migration. Local-first design: logs and dashboards run and persist locally by default, with the option to host on Hugging Face Spaces. Transparency: direct tracking of GPU energy usage via nvidia-smi, with results easily included in model cards. Trackio emphasizes reproducibility and accessibility, providing researchers with a straightforward way to log and share experiments without relying on proprietary services.
Experiment tracking is a routine part of machine learning workflows, but the Hugging Face team argues that lowering barriers to entry is crucial for wider adoption and reproducibility. Several researchers on the launch team pointed to the importance of transparency. They noted that Trackio's ability to log GPU energy consumption and add it directly to model cards could set a baseline for reporting environmental impact across ML projects.
Collection
[
|
...
]