Trax — Deep Learning with Clear Code and Speed
machine-learning
reinforcement-learning
deep-learning
numpy
deep-reinforcement-learning
transformer
jax
jax-fork
-
Updated
Dec 1, 2023 - Python