Member of Technical Staff, Kernels

Inception is an AI company developing efficient diffusion-based language models. The Member of Technical Staff, Kernels will design, optimize, and maintain high-performance ML kernels and compute primitives for large-scale language model training and inference, while improving distributed compute infrastructure, precision support, scalability, and resource utilization.

Responsibilities:

• Design and implement custom ML kernels (CUDA, CuTe, Triton) for core dLLM operations such as attention, matrix multiplication, gating, and normalization, optimized for modern GPU architectures
• Design compute primitives to reduce memory bandwidth bottlenecks and improve kernel efficiency
• Contribute to infrastructure stability and scalability, ensuring reproducibility, consistency across precision formats, and high utilization of compute resources