GitHub repository: pyyush/trax
Description: Trax — Deep Learning with Clear Code and Speed