Eä: Un compilador SIMD para Python escrito en Rust

Qué hace Eä

Eä es un compilador para kernels SIMD que resuelve un problema común del flujo de trabajo: escribir código Python, perfilar para encontrar bucles críticos, reescribir en C, lidiar con ctypes, depurar punteros y finalmente obtener una aceleración de 5×, solo para repetir el proceso la semana siguiente.

Cómo funciona

Con Eä, tú:

• Escribes un pequeño archivo .ea
• Ejecutas un comando
• Lo llamas desde Python como una función normal

El kernel se ejecuta a velocidad vectorizada nativa. Ejemplo de uso:

import ea
kernel = ea.load("fma.ea")
result = kernel.fma_f32x8(a, b, c, out)  # 6.6× más rápido que NumPy

Detalles técnicos

El compilador genera:

• Biblioteca compartida
• Envoltorio de Python
• También enlaces para Rust, C++, PyTorch y CMake

Arquitecturas objetivo:

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

El compilador en sí:

• Aproximadamente 12,000 líneas de Rust
• 475 pruebas
• No requiere ctypes, archivos de cabecera ni sistema de compilación

Contexto de desarrollo

El desarrollador construyó Eä con la ayuda de modelos de IA, utilizando Claude para el trabajo pesado mientras mantenía el control arquitectónico. La idea principal fue que manejar todo el "código de conexión" importa más que el SIMD en sí, permitiendo a los desarrolladores enfocarse solo en el kernel.

Puntos de referencia

Los puntos de referencia muestran un rendimiento 6.6× más rápido que NumPy para el ejemplo fma_f32x8. El desarrollador señala que estos provienen de una configuración bastante simple pero intentó mantener las cosas justas y reproducibles.