Starting Integration of Transformers with JAX 🚀

We are collaborating with Google’s Flax team to integrate the Flax functional API on top of all the very cool features from JAX. It’s just the beginning of this collaboration with Google Flax team yet, you should be able to use BERT & RoBERTa models right away.

This integration will come with the same unified API you see in our PyTorch and TensorFlow models. Expect more to come in the following weeks and do not hesitate to let us know if you would like to see other models supported 🔥

Hugging Face  



Events & TalksEvents & Talks