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.
GitHub repository with 7 stars and 0 forks.
Language: Python
Topics: bert, deep-learning, embeddings, equinox, fine-tuning, huggingface, jax, llama, llm, machine-learning