AXLearn is a library for deep learning built upon Jax and GSPMD to support large-scale training.
Gullak2008/axlearn
Folders and files
| Name | Name | Last commit date | ||
|---|---|---|---|---|
| Name | Name | Last commit date | ||
|---|---|---|---|---|
AXLearn is a library for deep learning built upon Jax and GSPMD to support large-scale training.