-
Notifications
You must be signed in to change notification settings - Fork 330
Pull requests: apple/axlearn
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
Add DataflowRunnerJob to allow running Dataflow jobs via Bastion.
#1213
opened May 27, 2025 by
zachberger
•
Draft
4 tasks
Updated jax.tree_map to jax.tree_util.tree_map in update_transformation.py
#1207
opened May 22, 2025 by
m-ny
Loading…
[UPDATE JAX API] Update
jax.experimental.array_serialization.serialization
to latest JAX
#1198
opened May 21, 2025 by
Steboss
Loading…
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…
Previous Next
ProTip!
Exclude everything labeled
bug
with -label:bug.