Sentence embeddings in JAX, after Transformers v5 dropped it

Hugging Face Transformers v5 removed its TensorFlow and JAX code to focus on PyTorch. If you used...