-
Notifications
You must be signed in to change notification settings - Fork 313
Pull requests: apple/axlearn
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
Differentiate SUCCEED and FAILED at the end of the Flink-Beam job
#1150
opened May 2, 2025 by
jinglu1
Loading…
Fallback to standard mesh on neuron backend for incompatible multi-granule meshes
#1146
opened May 1, 2025 by
apoorvtintin
Loading…
repro PR for test_spmd_evaler fail with jax 0.5.3
#1114
opened Apr 18, 2025 by
matthew-e-hopkins
•
Draft
Migrate from Legacy JAX APIs jax.tree_util to jax.tree
#986
opened Feb 12, 2025 by
apivovarov
Loading…
Use Accuracy from cross_entropy in causal_lm.py::CrossEntropyLossMetrics
#983
opened Feb 12, 2025 by
apivovarov
Loading…
Previous Next
ProTip!
Filter pull requests by the default branch with base:main.