Loreon
Labs
Platform
Docs
Home
Ecosystems
Python
axlearn
An Extensible Deep Learning Library
Python
Emerging
GitHub
Stars
—
Forks
—
Contributors
8
Last push
18mo ago
Recent commits
Latest commits.
Update pyproject.toml
2840f61
Param Bole
18mo ago
Merge branch 'apple:main' into main
e1abde8
Param Bole
18mo ago
Introduce the scale enum flag in Embedding layer for LLM embedding. (#909)
2d1fb29
Dongseong Hwang
18mo ago
Flash2 and supports cross attention and dropout (#905)
6559036
Hanzhi Zhou
18mo ago
Allow parallel gpu tests (#910)
ee7d60d
Hanzhi Zhou
18mo ago
Optimize TPU Flash Attention (20x XLA compilation speed-up on 32k long context) (#908)
c40b39a
Dongseong Hwang
18mo ago
The codebook of the KmeansVectorQuantizer should be initialized with scale=1/sqrt(dim). (#907)
d83b450
Dongseong Hwang
18mo ago
`AdaptiveLayerNormModulation` now supports sequence conditions. (#906)
3433f24
Dongseong Hwang
18mo ago
Top contributors
Builders behind this project.
ruomingp
109 commits
apghml
47 commits
ds-hwang
35 commits
zhiyun
32 commits
gyin94
25 commits
tgunter
23 commits
kelvin-zou
18 commits
hanzhi713
17 commits
