El autor ha creado eqx-zoo, una biblioteca que carga checkpoints de Hugging Face Hub directamente como módulos plain de Equinox y verifica cada modelo contra la biblioteca transformers. El proyecto actualmente soporta Llama, Qwen2, Qwen3, Qwen3-MoE para generación, y BERT, RoBERTa, XLM-RoBERTa para embeddings.
- La verificación asegura que las salidas float32 coincidan con transformers capa por capa y que la generación greedy reproduzca los tokens exactamente.
- El autor descubrió que comparar implementaciones bf16 directamente es defectuoso debido a diferencias de redondeo, lo que obliga a comparar contra float32 en relación con el error del código fuente.
- Se encontró que compilar JIT un decoder en XXL es mediblemente menos preciso que la ejecución eager, mientras que los encoders no mostraron este problema.
- Un bug sutil de LayerNorm en bf16 solo fue detectable a nivel de bloque, destacando la necesidad de pruebas granulares.
Los checkpoints verificados están disponibles en una colección de Hugging Face, y la biblioteca es instalable vía pip.