trax
Trax — Deep Learning with Clear Code and Speed
8k stars
142 watching
814 forks
Language: Python
last commit: 18 days ago
Linked from 1 awesome list
deep-learningdeep-reinforcement-learningjaxmachine-learningnumpyreinforcement-learningtransformer
n2cholas/awesome-jax