Research interests

JAX, Flax, TPU, 🤗

Team members 278

models 151