Search results

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

check_numerics doesn't work inside repeat.py

Status: Open.

#815 In apple/axlearn;

· ds-hwang opened on Nov 5, 2024

fsdp=16 model=16 gbs=16 should work on 256 chips

Status: Open.

#773 In apple/axlearn;

· samos123 opened on Oct 22, 2024

cannot install on Apple Silicon

Status: Open.

#762 In apple/axlearn;

· chunyang-wen opened on Oct 20, 2024

Golden Logit tests to ensure Fuji v2 70B matches Llama 2 70B

Status: Open.

#687 In apple/axlearn;

· samos123 opened on Sep 5, 2024