Trax — Deep Learning with Clear Code and Speed - View it on GitHub