Hugging Face Transformers v5 removed its TensorFlow and JAX code to focus on PyTorch. If you used FlaxBertModel or FlaxAutoModel to compute sentence embeddings in JAX, those classes are gone. This post shows how to compute sentence embeddings in JAX with eqx-zoo , an open-source library that lo...
Source: [Dev.to](https://dev.to/xquantize/sentence-embeddings-in-jax-after-transformers-v5-dropped-it-137k)