Eä : Un compilateur SIMD pour Python écrit en Rust

Ce que fait Eä

Eä est un compilateur pour les noyaux SIMD qui résout un problème courant de flux de travail : écrire du code Python, profiler pour trouver les boucles critiques, réécrire en C, lutter avec ctypes, déboguer les pointeurs, et finalement obtenir une accélération de 5×—pour recommencer la semaine suivante.

Comment ça marche

Avec Eä, vous :

• Écrivez un petit fichier .ea
• Exécutez une commande
• L'appelez depuis Python comme une fonction normale

Le noyau s'exécute à la vitesse vectorisée native. Exemple d'utilisation :

import ea
kernel = ea.load("fma.ea")
result = kernel.fma_f32x8(a, b, c, out)  # 6,6× plus rapide que NumPy

Détails techniques

Le compilateur génère :

• Une bibliothèque partagée
• Un wrapper Python
• Également des bindings Rust, C++, PyTorch et CMake

Architectures cibles :

• x86-64 (AVX2 / AVX-512)
• AArch64 (NEON)

Le compilateur lui-même :

• ~12 000 lignes de Rust
• 475 tests
• Aucun ctypes, fichier d'en-tête ou système de build requis

Contexte de développement

Le développeur a créé Eä avec l'aide de modèles d'IA, utilisant Claude pour les tâches lourdes tout en conservant le contrôle architectural. L'idée principale était que gérer tout le « code de liaison » importe plus que le SIMD lui-même, permettant aux développeurs de se concentrer uniquement sur le noyau.

Benchmarks

Les benchmarks montrent des performances 6,6× plus rapides que NumPy pour l'exemple fma_f32x8. Le développeur note que ceux-ci proviennent d'une configuration assez simple mais a essayé de garder les choses équitables et reproductibles.