Research interests

JAX, Flax, TPU, 🤗

Team members 275