Search results

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

[Enhancement] Add PPO/GRPO

Status: Open.

#964 In apple/axlearn;

· sbhavani opened on Feb 3, 2025

tensorflow-text fails to build on MacOS 15.1.1 arm64 when following axlearn instructions

Status: Open.

#892 In apple/axlearn;

· ben-albrecht opened on Dec 14, 2024

Remove usage of jax.core and jax.interpreters

Status: Open.

#874 In apple/axlearn;

· samos123 opened on Dec 9, 2024

AdamW optimizer details between jittor/optim and axlearn/common/optimizers.

Status: Open.

#863 In apple/axlearn;

· linjing-lab opened on Dec 1, 2024

Version Conflict Between Torch and JAX for NVIDIA cuDNN-cu12

Status: Open.

#858 In apple/axlearn;

· apivovarov opened on Nov 26, 2024

[Q] Resolving CUDA OOM Errors When Running run_tests.sh on GPUs

Status: Open.

#857 In apple/axlearn;

· apivovarov opened on Nov 25, 2024