обновлено 6 дней назад
CUDA Engineer (AI)
Мэтч & Сопровод
Для мэтча с этой вакансией нужен Plus
Описание вакансии
Текст:
TL;DR
CUDA Engineer (AI): Developing and optimizing low-level GPU code for high-performance inference workloads with an accent on custom CUDA kernel design and memory hierarchy management. Focus on maximizing throughput and minimizing latency for GPU-dense compute infrastructure at the intersection of energy and AI.
Location: Remote (UK)
Company
is a renewable energy startup building high-performance compute infrastructure to optimize power-dense GPU workloads for AI.
What you will do
- Write and optimize custom CUDA kernels for core transformer inference operations.
- Profile kernels to identify and eliminate bottlenecks in occupancy and memory throughput.
- Apply kernel fusion to reduce memory round-trips and launch overhead.
- Implement quantization-aware kernels and mixed-precision arithmetic to reduce latency.
- Build and tune caching mechanisms for efficient autoregressive decoding.
- Benchmark kernels against baselines to drive measurable performance improvements.
Requirements
- 4+ years of production CUDA experience with a track record of shipping performance-critical kernels.
- Deep understanding of GPU microarchitecture, warps, occupancy, and memory hierarchy.
- Strong CUDA C++ skills, including streams and asynchronous execution.
- Hands-on experience profiling compute-bound vs. memory-bound bottlenecks.
- Experience with kernel fusion, memory coalescing, and avoiding warp divergence.
- Solid grasp of parallel algorithm design and numerical precision tradeoffs.
Nice to have
- Experience with transformer/attention-style kernels or autoregressive decoding.
- Background in HPC or other latency-critical performance engineering.
- Exposure to multi-GPU or multi-node kernel-level optimization.
- Comfortable reading PTX/SASS to validate kernel efficiency.
Culture & Benefits
- Competitive salary and equity sign-on bonus.
- Biannual bonus scheme.
- Fully expensed tech setup.
- Breakfast and dinner allowance for office-based employees.
Будьте осторожны: если работодатель просит войти в их систему, используя iCloud/Google, прислать код/пароль, запустить код/ПО, не делайте этого - это мошенники. Обязательно жмите "Пожаловаться" или пишите в поддержку. Подробнее в гайде →
Похожие вакансии
6 дней назад
AI Performance Engineer
75 000 - 100 000$
Yandex
11 дней назад
GPU Performance Engineer (AI)
Resolution
6 дней назад
Research Engineer (AI Safety)
230 000 - 930 000$
Re.Source
16 часов назад
ML Engineer (NLP)
3 000$
6 дней назад
Edge AI Engineer
70 000 - 100 000$
11 дней назад
Founding Research Engineer (AI & Defence)
200 000 - 300 000$