Eä: Um Compilador SIMD para Python Escrito em Rust

O que o Eä faz

O Eä é um compilador para kernels SIMD que resolve um problema comum de fluxo de trabalho: escrever código Python, fazer profiling para encontrar loops críticos, reescrever em C, lutar com ctypes, depurar ponteiros e finalmente obter uma aceleração de 5×—só para repetir o processo na semana seguinte.

Como funciona

Com o Eä, você:

• Escreve um pequeno arquivo .ea
• Executa um comando
• Chama a partir do Python como uma função normal

O kernel roda na velocidade nativa vetorizada. Exemplo de uso:

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

Detalhes técnicos

O compilador gera:

• Biblioteca compartilhada
• Wrapper Python
• Também bindings para Rust, C++, PyTorch e CMake

Arquiteturas suportadas:

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

O compilador em si:

• Aproximadamente 12.000 linhas de Rust
• 475 testes
• Não requer ctypes, arquivos de cabeçalho ou sistema de build

Contexto de desenvolvimento

O desenvolvedor construiu o Eä com a ajuda de modelos de IA, usando o Claude para o trabalho pesado enquanto mantinha o controle arquitetural. A principal percepção foi que lidar com todo o "código de ligação" importa mais do que o SIMD em si, permitindo que os desenvolvedores foquem apenas no kernel.

Benchmarks

Os benchmarks mostram desempenho 6,6× mais rápido que o NumPy para o exemplo fma_f32x8. O desenvolvedor observa que estes são de uma configuração bastante simples, mas tentou manter as coisas justas e reproduzíveis.