L'auteur a construit eqx-zoo, une bibliothèque qui charge les checkpoints de Hugging Face Hub directement en tant que modules plain d'Equinox et vérifie chaque modèle contre la bibliothèque transformers. Le projet prend actuellement en charge Llama, Qwen2, Qwen3, Qwen3-MoE pour la génération, et BERT, RoBERTa, XLM-RoBERTa pour les embeddings.
- La vérification garantit que les sorties float32 correspondent à transformers couche par couche et que la génération greedy reproduit exactement les tokens.
- L'auteur a découvert que comparer directement les implémentations bf16 est défectueux en raison de différences d'arrondi, nécessitant une comparaison contre float32 par rapport à l'erreur de la source.
- Il a été constaté que la compilation JIT d'un decoder dans XXL est mesurablement moins précise que l'exécution eager, tandis que les encoders ne montrent pas ce problème.
- Un bug subtil de LayerNorm en bf16 n'a été détectable qu'au niveau du bloc, soulignant le besoin de tests granulaires.
Les checkpoints vérifiés sont disponibles dans une collection Hugging Face, et la bibliothèque est installable via pip.