Trax — your path to advanced deep learning - View it on GitHub