Member of Technical Staff, Kernels

About Us

Inception creates the world’s fastest, most efficient AI models. Our Mercury model is the world’s fastest reasoning LLM and first commercially available diffusion LLM, delivering 5x greater speed and efficiency than today’s LLMs, with best-in-class quality. We are the AI researchers and engineers behind breakthrough AI technologies such as diffusion models, flash attention, and DPO. Today’s autoregressive LLMs generate tokens sequentially, which makes them painfully slow and expensive. Inception’s diffusion-based LLMs (dLLMs) generate answers in parallel, making them 5x faster and more efficient while delivering best-in-class quality. Our team includes engineers from AWS, Google DeepMind, Meta AI, Microsoft, HashiCorp, and OpenAI. Based in Palo Alto, CA, we are backed by top-tier venture capitalists and tech luminaries.

About the Role

We're looking for engineers and scientists to design, optimize, and maintain the compute foundations that power large-scale language model training and inference. You will develop high-performance ML kernels, enable efficient low-precision arithmetic, and improve the distributed compute stack that makes training and serving large models possible.

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.

Qualifications

• BS/MS/PhD in Computer Science, Engineering, or a related field (or equivalent experience).
• Proficiency in CUDA, CuTe, Triton, or other GPU programming frameworks.
• Understanding of ML frameworks (PyTorch, TensorFlow) from a systems perspective.
• Background in performance optimization and profiling of ML systems.
• Experience implementing low-precision formats (FP8, INT8, block floating point) or contributing to related compiler stacks (XLA, TVM).
• Familiarity with distributed training techniques (data parallel, model parallel, pipeline parallel).
• Proficiency in Python and at least one systems programming language (C++/Rust/Go).
• Experience with containerization (Docker), orchestration (Kubernetes), and CI/CD pipelines.

Preferred Skills

• Experience building and maintaining large-scale language models with tens of billions of parameters or more.
• Experience with distributed systems and cloud computing platforms (AWS/GCP/Azure).
• Familiarity with distributed frameworks such as PyTorch/XLA, DeepSpeed, Megatron-LM.
• Prior contributions to open-source deep learning infrastructure such as PyTorch, DeepSpeed, or XLA.

Pay

The annual base salary range for this role is $200,000 – $350,000 USD. Final compensation is determined based on experience, skills, and qualifications. Equity and benefits are included in the total package.

Benefits

• Competitive salary and equity in a rapidly growing startup.
• Flexible vacation and paid time off (PTO).
• Health, dental, and vision insurance.
• 401k match.
• Catered meals (breakfast, lunch, & dinner).
• Commuter subsidies.
• A collaborative and inclusive culture.

Why Join Inception

• Work with World-Class Talent: Collaborate with the inventors of diffusion models and leading AI researchers.
• Shape Foundational Technology: Your decisions will influence how the next generation of AI products are built and used.
• Immediate Impact: Join at the ground floor where your contributions directly shape product direction and company trajectory.