Hugging Face Transformers v5 drops JAX; open-source library eqx-zoo offers alternative
Hugging Face's Transformers v5 release has removed its TensorFlow and JAX code, focusing exclusively on PyTorch. This eliminated classes like FlaxBertModel previously used for computing sentence embeddings in JAX. An open-source library called eqx-zoo now provides a method to compute these embeddings in JAX using Hugging Face model checkpoints. The library loads checkpoints as Equinox modules and replicates the pooling and normalization steps defined for each model to ensure correct embeddings.
This is an AI-generated summary. ShortSingh links to the original source for the complete article.
Discussion (0)
Log in to join the discussion and vote.
Log in