Loreon
Labs
Platform
Docs
Home
Ecosystems
Python
axlearn
An Extensible Deep Learning Library
Python
Emerging
GitHub
Stars
—
Forks
—
Contributors
8
Last push
9mo ago
Recent commits
Latest commits.
Make Bazel cover tensorflow-datasets and seqio
029cd03
Yi Wang
9mo ago
Allow overwrite minute based xla flag like megascale_graph_hang_threshold
0f98bf3
Shang-Chen Wu
9mo ago
Limit the gcs_grpc change to only pathways jobs
af3e178
Muyang Yu
9mo ago
Fix bug issues/1755
9cb0ad6
Yi Wang
9mo ago
Temporarily support both JAX 0.5.3 and 0.6.2
948c3e9
Yi Wang
9mo ago
Add py_test rules: attention_bias_test and base_layer_test
f4a4aad
Yi Wang
9mo ago
Update
f03876d
Yi Wang
9mo ago
Optimize pathways ckpt loading performace
d3a6a5a
Muyang Yu
9mo ago
Top contributors
Builders behind this project.
ruomingp
116 commits
ds-hwang
91 commits
hanzhi713
67 commits
changlan
52 commits
apghml
48 commits
wangkuiyi
35 commits
zhiyun
33 commits
Ethanlm
29 commits
