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.