-
Notifications
You must be signed in to change notification settings - Fork 317
Pull requests: apple/axlearn
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
Use lambda instead of functools.partial to create single-host config maker
#1152
opened May 2, 2025 by
yhtang
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!
Mix and match filters to narrow down what you’re looking for.