google/trax: Trax — Deep Learning with Clear Code and Speed
Category: ml-dl
Language: Python
Topics: deep-learning, deep-reinforcement-learning, jax, machine-learning, numpy, reinforcement-learning, transformer