Search results

Use Callback-Based Integration for Hugging Face Instead of Trainer Subclassing

documentationImprovements or additions to documentationImprovements or additions to documentation

Status: Open.

Task

#88 In traceopt-ai/traceml;

· abhinavsriva opened on Apr 25, 2026

Title: Add step-level total communication time sampler

enhancementNew feature or requestNew feature or request

Status: Open.

Feature

#86 In traceopt-ai/traceml;

· abhinavsriva opened on Apr 23, 2026

Add traceml.yaml for user-facing TraceML defaults

enhancementNew feature or requestNew feature or request

Status: Open.

Feature

#85 In traceopt-ai/traceml;

· abhinavsriva opened on Apr 23, 2026

Move TraceML runtime startup into user code via traceml.init(...)

enhancementNew feature or requestNew feature or request

Status: Open.

Feature

#84 In traceopt-ai/traceml;

· abhinavsriva opened on Apr 23, 2026

Standalone aggregator service + launcher support

enhancementNew feature or requestNew feature or request

Status: Open.

Feature

#83 In traceopt-ai/traceml;

· abhinavsriva opened on Apr 23, 2026

Add step-scoped H2D .to() timing

enhancementNew feature or requestNew feature or request

Status: Open.

Feature

#82 In traceopt-ai/traceml;

· abhinavsriva opened on Apr 23, 2026

Store batch size samples in DB for all ranks

enhancementNew feature or requestNew feature or request

help wantedExtra attention is neededExtra attention is needed

Status: Open.

Feature

#81 In traceopt-ai/traceml;

· abhinavsriva opened on Apr 23, 2026

Phase-level memory breakdown for forward / backward / optimizer

enhancementNew feature or requestNew feature or request

Status: Open.

Feature

#64 In traceopt-ai/traceml;

· abhinavsriva opened on Mar 23, 2026

TraceML Multi-Node Distributed Architecture design

Status: Open.

#37 In traceopt-ai/traceml;

· ppraneth opened on Feb 27, 2026

Verify TraceML runtime failures do not stop user training

help wantedExtra attention is neededExtra attention is needed

Status: Open.

Task

#24 In traceopt-ai/traceml;

· abhinavsriva opened on Feb 2, 2026

Validate correctness of forward_pre_hook + tensor backward hook for OOM attribution

bugSomething isn't workingSomething isn't working

good first issueGood for newcomersGood for newcomers

Status: Open.

#18 In traceopt-ai/traceml;

· abhinavsriva opened on Jan 6, 2026