Research interests

JAX, Flax, TPU, 🤗

Team members 265

models 152