Search results

changes added to test orbax

Status: Draft (not ready).

apple/axlearnnumber 1282

#1282 In apple/axlearn;

· lkolluru05 opened on Jul 2, 2025

Revert "Upgrade orbax checkpointer to 0.11.15"

Status: Open (in progress).

apple/axlearnnumber 1281

#1281 In apple/axlearn;

· findmyway opened on Jul 1, 2025

[Do not review] Upgrade Python to 3.12 and use Ubuntu 24.04

Status: Draft (not ready).

apple/axlearnnumber 1280

#1280 In apple/axlearn;

· andersensam opened on Jul 1, 2025

Workload Hang & Rolling Window Goodput Monitoring Support

Status: Open (in progress).

apple/axlearnnumber 1278

#1278 In apple/axlearn;

· dipannita08 opened on Jun 27, 2025

Add sliding window fast compilation path

Status: Open (in progress).

apple/axlearnnumber 1276

#1276 In apple/axlearn;

· hanzhi713 opened on Jun 26, 2025

Add opensource MoE switch model for TPU v6e testing

Status: Open (in progress).

apple/axlearnnumber 1268

#1268 In apple/axlearn;

· yiping-ma opened on Jun 19, 2025

Update NCCL configs for A3 Ultra and A4

Status: Open (in progress).

apple/axlearnnumber 1265

#1265 In apple/axlearn;

· andersensam opened on Jun 17, 2025

Update vm image.

Status: Open (in progress).

apple/axlearnnumber 1262

#1262 In apple/axlearn;

· markblee opened on Jun 14, 2025

Adding basic elastic training (pause-and-resume)

Status: Draft (not ready).

apple/axlearnnumber 1256

#1256 In apple/axlearn;

· lukebaumann opened on Jun 12, 2025

<code>CompositeLossMetrics</code> now performs a weighted sum of losses.

Status: Open (in progress).

apple/axlearnnumber 1251

#1251 In apple/axlearn;

· ds-hwang opened on Jun 10, 2025

Pathways: reuse setup_spmd for pathways init

Status: Open (in progress).

apple/axlearnnumber 1248

#1248 In apple/axlearn;

· samos123 opened on Jun 9, 2025

orbax to reg checkpointer conversion

Status: Draft (not ready).

apple/axlearnnumber 1246

#1246 In apple/axlearn;

· lkolluru05 opened on Jun 6, 2025