Member of Technical Staff, Kernels

About the Company

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.

Responsibilities

• Design and implement kernels that support high-performance long-context behavior
• Own kernel design, implementation, deployment, and production reliability
• Focus on robustness, extensive testing, and functional correctness while pushing on performance
• Evaluate porting compute kernels to alternative hardware options
• Co-design kernels in interaction with training, inference, and RL teams

Requirements

• Low-level programming experience targeting AI accelerators (e.g., NVIDIA Blackwell, Google TPUs)
• Experience developing and optimizing GPU kernels in frameworks like NCCL, MSCCLPP, CUTLASS, CuTeDSL, Triton, Quack, or Flash-Attention
• Deep expertise in computer architecture, low-level machine optimizations, and code generation
• Experience with kernel authoring frameworks such as Pallas/Mosaic or Mojo is a plus

Benefits

• Annual salary between $225K - $550K based on experience
• Significant equity component
• 401(k) plan with 6% salary matching
• Generous health, dental, and vision insurance
• Unlimited paid time off
• Visa sponsorship and relocation stipend available