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