Эа: SIMD-компилятор для Python, написанный на Rust

Что делает Eä

Eä — это компилятор для SIMD-ядер, который решает распространённую проблему рабочего процесса: написание кода на Python, профилирование для поиска горячих циклов, переписывание на C, борьба с ctypes, отладка указателей и, наконец, получение ускорения в 5 раз — только чтобы повторить процесс на следующей неделе.

Как это работает

С Eä вы:

• Пишете небольшой файл .ea
• Запускаете одну команду
• Вызываете его из Python как обычную функцию

Ядро работает на нативной векторизованной скорости. Пример использования:

import ea
kernel = ea.load("fma.ea")
result = kernel.fma_f32x8(a, b, c, out)  # В 6,6 раз быстрее, чем NumPy

Технические детали

Компилятор генерирует:

• Общую библиотеку
• Python-обёртку
• А также привязки для Rust, C++, PyTorch и CMake

Целевые архитектуры:

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

Сам компилятор:

• ~12 000 строк на Rust
• 475 тестов
• Не требует ctypes, заголовочных файлов или системы сборки

Контекст разработки

Разработчик создал Eä с помощью AI-моделей, используя Claude для тяжёлой работы, сохраняя архитектурный контроль. Основная идея заключалась в том, что обработка всего «связующего кода» важнее самого SIMD, что позволяет разработчикам сосредоточиться только на ядре.

Бенчмарки

Бенчмарки показывают производительность в 6,6 раз выше, чем у NumPy, для примера fma_f32x8. Разработчик отмечает, что эти результаты получены в довольно простой настройке, но старался сохранить честность и воспроизводимость.