Skip to content

Add Trackio rollout trace logging#450

Open
abidlabs wants to merge 2 commits into
alibaba:mainfrom
abidlabs:add-trackio-rollout-traces
Open

Add Trackio rollout trace logging#450
abidlabs wants to merge 2 commits into
alibaba:mainfrom
abidlabs:add-trackio-rollout-traces

Conversation

@abidlabs
Copy link
Copy Markdown

@abidlabs abidlabs commented May 22, 2026

Hi folks! This PR adds trace logging via Trackio, the free, local-first experiment tracking library from Hugging Face 🤗

This PR builds on ROLL's existing Trackio metrics tracker. Specifically, I

  • added Trackio trace logging support to the tracker abstraction
  • added RLVR rollout generations as trackio.Trace records with step/sample metadata
  • added agentic rollout trajectories as trackio.Trace records with trajectory/reward metadata
  • documented Trackio trace logging and the trackio_max_traces_per_step cap
  • added tests with mocked Trackio

This is what it looks like:

image

AI assistance was used to prepare this PR.

@CLAassistant
Copy link
Copy Markdown

CLAassistant commented May 22, 2026

CLA assistant check
All committers have signed the CLA.

@abidlabs abidlabs marked this pull request as ready for review May 22, 2026 21:13
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants