feat: add trackio as experiment tracking integration (#3253)

* feat: add trackio as experiment tracking integration

- Add TrackioConfig to integrations schema with project_name, run_name, and space_id
- Create trackio_.py module for environment setup
- Add is_trackio_available() utility function
- Integrate trackio with report_to in trainer builder
- Add trackio callback for experiment tracking
- Add trackio config keys to gpt-oss example YAMLs
- Trackio runs locally by default, syncs to HF Space if space_id provided

* changes

* changes

* changes

* changes

* changes

* changes

* changes

* Update requirements.txt

* don't allow pydantic 2.12 for now

---------

Co-authored-by: Abubakar Abid <aaabid93@gmail.com>
Co-authored-by: Wing Lian <wing@axolotl.ai>
This commit is contained in:
Abubakar Abid
2025-12-23 05:49:07 -08:00
committed by GitHub
parent 92ee4256f7
commit f2155eaf79
16 changed files with 134 additions and 7 deletions

View File

@@ -20,15 +20,16 @@ deepspeed>=0.17.0
trl==0.25.0
hf_xet==1.2.0
kernels>=0.9.0
trackio
trackio>=0.13.0
typing_extensions>=4.14.0
optimum==1.16.2
hf_transfer
sentencepiece
gradio==5.49.1
gradio>=6.2.0,<7.0
modal==1.0.2
pydantic>=2.10.6
pydantic>=2.10.6,<2.12
addict
fire
PyYAML>=6.0