github.com/google/trax (go): Trax — Deep Learning with Clear Code and Speed
Keywords: deep-learning|deep-reinforcement-learning|jax|machine-learning|numpy|reinforcement-learning|transformer