AXLearn is a library for deep learning built upon Jax and GSPMD to support large-scale training.
ruomingp/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.