GitHub repository: lkolluru05/axlearn
Description: An Extensible Deep Learning Library