O autor construiu eqx-zoo, uma biblioteca que carrega checkpoints do Hugging Face Hub diretamente como módulos plain de Equinox e verifica cada modelo contra a biblioteca transformers. O projeto atualmente suporta Llama, Qwen2, Qwen3, Qwen3-MoE para geração, e BERT, RoBERTa, XLM-RoBERTa para embeddings.
- A verificação garante que as saídas float32 correspondam ao transformers camada por camada e que a geração greedy reproduza os tokens exatamente.
- O autor descobriu que comparar implementações bf16 diretamente é falho devido a diferenças de arredondamento, exigindo comparação contra float32 em relação ao erro da fonte.
- Compilar JIT um decoder em XXL foi encontrado como mensuravelmente menos preciso do que a execução eager, enquanto os encoders não mostraram esse problema.
- Um bug sutil de LayerNorm em bf16 foi detectável apenas no nível de bloco, destacando a necessidade de testes granulares.
Os checkpoints verificados estão disponíveis em uma coleção do Hugging Face, e a biblioteca é instalável via pip.