Research interests

JAX, Flax, TPU, 🤗

Team members 279

models 151