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