Transformers v5 移除 JAX 支持后，如何用 JAX 计算句子嵌入

Hugging Face Transformers v5 删除了 TensorFlow 和 JAX 代码以专注 PyTorch，原有的 FlaxBertModel、FlaxAutoModel 等类已不存在。文章介绍用开源库 eqx-zoo 在 JAX 中计算句子嵌入：该库将 Hugging Face checkpoint 作为普通 Equinox 模块加载，直接读取 safetensors 文件，无需 PyTorch。文中称嵌入结果与 sentence-transformers 的差异在 float32 舍入误差范围内，并给出验证方法，同时覆盖英文与多语言模型、一个小型语义搜索示例以及基于语言模型的 Qwen3-Embedding。

AI推荐理由 针对 Hugging Face Transformers v5 移除 TensorFlow/JAX 代码后 FlaxBertModel、FlaxAutoModel 类消失的具体迁移问题，给出用 eqx-zoo 加载 Hugging Face checkpoint 为 Equinox 模块、无需 PyTorch 的替代方案，并称嵌入结果与 sentence-transformers 在 float32 舍入误差内一致且文末给出验证方式；对使用 JAX 的开发者有直接实用价值，但属个人博客教程，验证细节仅由作者自述，未提供独立复现。