Senior AI Kernel & Performance Engineer

Meta; Bellevue, Northern (WA, KY); Hybrid; USD 154,000 - 217,000; Full time

Job summary

Meta seeks an experienced Software Engineer for AI Kernels and Performance Optimization on the MTIA Software team in Bellevue, WA. You will own performance of workloads across MTIA accelerators, from fused attention kernels to numeric decisions influencing model convergence.

The role requires deep knowledge of C++, Python, and high-performance kernel development, with experience profiling on modern architectures and collaboration with silicon teams. Competitive compensation and equity offered.

Qualifications

• Bachelor's degree or equivalent practical experience in Computer Science, Computer Engineering, or related field.
• 6+ years of professional experience in high-performance computing, accelerator kernel development, or systems performance engineering.
• Strong proficiency in C++ and Python with low-level systems programming experience.

Responsibilities

• Design, implement, and optimize high-performance compute and communication kernels for MTIA accelerators.
• Profile and root-cause performance across the full stack including memory hierarchy and DMA behavior.
• Build and extend kernel authoring frameworks, templates, and libraries for broad kernel coverage.
• Deliver production-ready kernel coverage for PyTorch operators across AI workloads.
• Collaborate with silicon architecture teams on hardware/software co-design and rooflines.
• Mentor engineers on performance methodology and accelerator programming.

Skills

C++

Python

Kernel dev

HPC

Performance profiling

Debugging

Education

BSc in CS or related

Equivalent practical experience

Tools

CUDA

LLVM

MLIR

PyTorch

Job description

Meta seeks an experienced Software Engineer for AI Kernels and Performance Optimization on the MTIA Software team in Bellevue, WA. You will own performance of workloads across MTIA accelerators, from fused attention kernels to numeric decisions influencing model convergence.

The role requires deep knowledge of C++, Python, and high-performance kernel development, with experience profiling on modern architectures and collaboration with silicon teams. Competitive compensation and equity offered.