Eä: A SIMD Compiler for Python Written in Rust

What Eä Does

Eä is a compiler for SIMD kernels that solves a common workflow problem: writing Python code, profiling to find hot loops, rewriting in C, fighting ctypes, debugging pointers, and finally getting a 5× speedup—only to repeat the process the next week.

How It Works

With Eä, you:

• Write a small .ea file
• Run one command
• Call it from Python like a normal function

The kernel runs at native vectorized speed. Example usage:

import ea
kernel = ea.load("fma.ea")
result = kernel.fma_f32x8(a, b, c, out)  # 6.6× faster than NumPy

Technical Details

The compiler generates:

• Shared library
• Python wrapper
• Also Rust, C++, PyTorch, and CMake bindings

Target architectures:

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

The compiler itself:

• ~12,000 lines of Rust
• 475 tests
• No ctypes, header files, or build system required

Development Context

The developer built Eä with the help of AI models, using Claude for heavy lifting while maintaining architectural control. The main insight was that handling all the "glue code" matters more than SIMD itself, allowing developers to focus only on the kernel.

Benchmarks

Benchmarks show 6.6× faster performance than NumPy for the fma_f32x8 example. The developer notes these are from a fairly simple setup but tried to keep things fair and reproducible.