The author has built eqx-zoo, a library that loads Hugging Face Hub checkpoints directly as plain Equinox modules and verifies each model against the transformers library. The project currently supports Llama, Qwen2, Qwen3, Qwen3-MoE for generation, and BERT, RoBERTa, XLM-RoBERTa for embeddings.
- Verification ensures float32 outputs match transformers layer-by-layer and greedy generation reproduces tokens exactly.
- The author discovered that comparing bf16 implementations directly is flawed due to rounding differences, necessitating comparison against float32 relative to the source's error.
- JIT-compiling a decoder in XXL was found to be measurably less accurate than eager execution, while encoders did not show this issue.
- A subtle LayerNorm bug in bf16 was only detectable at the block level, highlighting the need for granular testing.
The verified checkpoints are available in a Hugging Face collection, and the library is installable via pip.