Member of Technical Staff, Kernels

San Francisco on site Until 8/22/2026 First posted March 21, 2025 Last posted March 21, 2025
Job description

Magic’s mission is to build safe AGI that accelerates humanity’s progress on the world’s most important problems. We believe the most promising path to safe AGI lies in automating research and code generation to improve models and solve alignment more reliably than humans can alone. Our approach combines frontier-scale pre-training, domain-specific RL, ultra-long context, and inference-time compute to achieve this goal.

About the role

As a Kernel Engineer, you will design, implement, and maintain high-performance kernels to optimize throughput and latency during training and inference.

Magic's long-context windows create distinct kernel optimization challenges around memory utilization, data movement, and sustained throughput.

What you'll work on

  • Design and implement kernels that support high-performance long-context behavior

  • Ownership of kernel design, implementation, deployment, and production reliability

  • Focus on robustness, extensive testing, and functional correctness, while pushing on performance

  • Evaluate porting Magic’s compute kernels to alternative hardware options

  • Co-design kernels with understanding and interaction with training, inference, and RL teams

  • For a sample of our work, see Magic-Attention, presented at GTC 2026

What we’re looking for

  • Low-level programming experience targeting AI accelerators such as NVIDIA Blackwell or Google TPUs

  • Develop and optimize GPU kernels in frameworks such as NCCL, MSCCLPP, CUTLASS, CuTeDSL, Triton, Quack, Flash-Attention, and similar frameworks

  • Experience in other kernel authoring frameworks such as Pallas/Mosaic (GPU or TPU), or Mojo also maps well to the work on Magic's kernel team

  • Strong depth over shallow breadth: for kernel engineering, we prefer candidates with deep expertise in computer architecture, low-level machine optimizations, and code generation, with breadth across ML

  • Agility, ownership mindset, and grit

Compensation, benefits and perks (US)

  • Annual salary ranges between $225K - $550K based on experience

  • Equity is a significant part of total compensation, in addition to salary

  • 401(k) plan with 6% salary matching

  • Generous health, dental and vision insurance for you and your dependents

  • Unlimited paid time off

  • Visa sponsorship and relocation stipend to bring you to SF, if possible

  • A small, fast-paced, highly focused team

Magic strives to be the place where high-potential individuals can do their best work. We value quick learning and grit just as much as skill and experience.

Our culture

  • Integrity. Words and actions should be aligned

  • Hands-on. At Magic, everyone is building

  • Teamwork. We move as one team, not N individuals

  • Focus. Safely deploy AGI. Everything else is noise

  • Quality. Magic should feel like magic

Compensation (from employer):
$225K – $550K

About this role

Summary

Design and optimize high-performance kernels for training and inference in AI models.

Job title

Member of Technical Staff, Kernels

Experience level

Industry

software

Location requirements

San Francisco; remote work possible with sponsorship and relocation

Salary

$225K – $550K

Management role

No

Skills & keywords

Required skills

low-level programmingGPU kernelsNCCLMSCCLPPCUTLASSCuTeDSLTritonQuackFlash-Attentioncomputer architecture

Preferred skills

low-level programmingGPU kernelsAI acceleratorsML frameworks

Specializations

kernel engineeringAI acceleratorsGPUperformance optimization
Locations

Structured locations inferred from the posting.

San Francisco, CA, USA

On-site City