Search results

Proposal: Remove param_converter.py

Status: Open.

#1331 In apple/axlearn;

· wangkuiyi opened on Aug 8, 2025

Sharding Datasets Outside of TFDS

Status: Open.

#1329 In apple/axlearn;

· carlesoctav opened on Aug 5, 2025

fuji v2 70b AssertionError during attention in latest main

Status: Open.

#1328 In apple/axlearn;

· samos123 opened on Aug 3, 2025

Terminate <code>output-uploader</code> voluntarily after main container stops

Status: Open.

#1327 In apple/axlearn;

· linxiulei opened on Aug 1, 2025

fuji with tiktoken ModuleNotFoundError: No module named 'tokenizers'

Status: Open.

#1322 In apple/axlearn;

· samos123 opened on Jul 28, 2025

WandBWriter.log_config invoked before jax initialized, interferes with processor_zero_only

Status: Open.

#1287 In apple/axlearn;

· dmarx opened on Jul 8, 2025

Docs request: how to do inference with fuji

Status: Open.

#1192 In apple/axlearn;

· samos123 opened on May 20, 2025

Single-host config maker rejected by <code>config_class_for_function</code>

Status: Open.

#1151 In apple/axlearn;

· yhtang opened on May 2, 2025

Support for JAX versions &gt; 0.5.4 (current AXLearn pinned to 0.4.38)

Status: Open.

#1126 In apple/axlearn;

· Steboss opened on Apr 23, 2025

unit test for tpu_attention_test.py fails

Status: Open.

#1115 In apple/axlearn;

· lkolluru05 opened on Apr 18, 2025

Osier.id

Status: Open.

#996 In apple/axlearn;

· ezbud opened on Feb 17, 2025

Licensing clashes

Status: Open.

#970 In apple/axlearn;

· hyandell opened on Feb 6, 2025