GitHub repository: pragnya-sridhar/axlearn
Description: An Extensible Deep Learning Library