GitHub repository: stjordanis/trax
Description: Trax — your path to advanced deep learning