jaxdl (pypi): JAX (Flax) Deep Learning Library
Keywords: deep|machine|learning|reinforcement|research