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