Hugging Face Unveils Trackio

Trackio is an open-source Python library for tracking and visualizing machine learning experiment metrics, parameters, and hyperparameters.

Key features include:

• wandb compatibility: Provides an API compatible with wandb.init, wandb.log, and wandb.finish, so it can be used with almost no modification to existing code.
• Easy sharing and embedding: Offers a Gradio-based local dashboard, which can be synced to Hugging Face Spaces to share experiment results via a single URL or embed them as an iframe in blogs and documentation.
• Transparency and data accessibility: Directly measures GPU energy usage via nvidia-smi, enabling quantification of environmental impact for model cards, and allows data to be easily extracted rather than locked into a proprietary API.
• Lightweight design: Enables flexible experimentation, such as adjusting the timing of tensor transfers so as not to affect training performance.