← All roles

Software Engineer, AI Kernels & Performance Optimization — MTIA Software

Meta· Engineering
Apply on company site
Menlo Park, CA$154k - $217kFull-timePosted 2w ago

About the role

Senior kernel and performance engineering role within Meta's MTIA Software team (AI Kernels & Performance Optimization) focused on high-performance compute kernels, pre-silicon bring-up, and software-hardware co-design for MTIA accelerators.

Responsibilities

  • Design, implement, and optimize high-performance compute and communication kernels for MTIA accelerators, taking ownership from architectural analysis through production deployment
  • Profile and root-cause performance across the full stack — instruction scheduling, memory hierarchy and DMA behavior, on-chip interconnect, multi-device collectives — and drive fixes to the right layer, whether kernel, compiler, runtime, or hardware
  • Build and extend kernel authoring frameworks, templates, and libraries so that other engineers can reach high performance without deep architectural expertise; raise the ceiling and lower the floor at the same time
  • Deliver and maintain broad kernel coverage for PyTorch operators across recommendation, ranking, and generative AI workloads, in both eager and compiled execution paths
  • Partner with silicon architecture and design teams on hardware/software co-design: quantify the value of proposed features with real kernels, characterize rooflines pre-silicon, and advocate for changes the software stack actually needs
  • Work with compiler, runtime, framework, and product-facing teams to land end-to-end wins on production models rather than isolated microbenchmark improvements
  • Investigate numerics and precision trade-offs, and design software mitigations that recover performance or accuracy lost to hardware limitations
  • Set technical direction for a kernel domain, write design documents, and mentor engineers on performance methodology and accelerator programming

Qualifications

  • Bachelor's degree in Computer Science, Computer Engineering, a related technical field, or equivalent practical experience
  • 6+ years of professional experience in high-performance computing, accelerator kernel development, compiler backends, or systems performance engineering
  • Proficiency in C++ and Python, including low-level systems programming, templates and generic programming, and comfort reading and writing performance-critical code
  • Demonstrated experience writing and optimizing kernels for parallel architectures (GPU, TPU/AI ASICs, or SIMD/vector CPU targets)
  • Working knowledge of computer architecture as it applies to performance: memory hierarchies and bandwidth, latency hiding, occupancy and scheduling, vectorization, and synchronization
  • Rigorous, measurement-driven approach to performance: ability to build a roofline or analytical model, profile against it, and explain the residual gap

Skills

Required:High-performance kernel design and optimizationPerformance profiling and debuggingKernel framework developmentPyTorch operator coverage
Nice to have:Hardware/software co-design

Benefits

  • Benefits information as described in the posting (not enumerated here)
Last updated Oct 23, 2026. Interested? Apply on the company site.

About Meta

Shop the latest AI glasses and wearable technology from Meta.

Shop the latest AI glasses and wearable technology from Meta. Explore Ray-Ban Meta, Oakley Meta, display glasses, and cutting-edge VR headsets.

AI glassesOakley MetaRay-Ban MetaVR headsetsliquid-ai-recruiting-targetwearable technology
Company profileWebsite

More open roles at Meta