JAX

eqx-zoo：将 Hugging Face 预训练模型移植为经数值验证的 Equinox/JAX 模块

xquantize/eqx-zoo · Verified JAX/Equinox ports of Hugging Face LLMs and embedding models: numerically matched against PyTorch, forward and backward, and saved back to Hugging Face formats.

eqx-zoo 提供 Hugging Face 预训练语言模型和句向量模型的 Equinox 模块实现，加载权重后可与 Transformers 或 sentence-transformers 的参考结果进行逐层和端到端数值对齐验证；模型是标准 pytree，可直接使用 jax.jit、jax.grad 和 jax.vmap，并支持将微调后的模型保存回 Hugging Face 格式。