Skip to content

markblee/axlearn

 
 

Repository files navigation

The AXLearn Library for Deep Learning

AXLearn is a library for deep learning built upon Jax and GSPMD to support large-scale training.

About

No description, website, or topics provided.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 95.1%
  • Jupyter Notebook 4.9%