google/trax
trax
Trax — Deep Learning with Clear Code and Speed
PythonUnknownAIdeep-learningdeep-reinforcement-learningjaxmachine-learningnumpyreinforcement-learningtransformer
8307 GitHub stars
820 forks
Python
Unknown
Starred 02/25/2020
Last pushed 09/26/2025