GPU Kernel Engineer

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 is to combine frontier-scale pre-training, domain-specific RL, ultra-long context, and test-time compute to achieve this goal.


About the role: 

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

What you might work on:

  • Design, develop and optimize custom high-performance GPU kernels
  • Optimize inference for our novel architectures, both by writing more efficient code and thinking about how we can trade off accuracy and speed
  • Think beyond the kernel level to the broader scheme of how we train our models and suggest improvements

What we’re looking for:

  • Experience working with CUDA, NCCL is a plus
  • Feeling comfortable staring at SASS and writing PTX
  • Deep understanding of how GPUs and other AI accelerators work

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, benefits and perks (US):

  • Annual salary range: $150K - $1M
  • 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 dependants
  • Unlimited paid time off
  • Option to work in-person in SF or remotely
  • Visa sponsorship and relocation stipend to bring you to SF
  • A small, fast-paced, highly focused team
Apply for this job
logo Magic GPU Kernel Engineer FullTime Flexible 📍 San Francisco Apply Now
Your subscription could not be saved. Please try again.
Your subscription has been successful.

Newsletter

Subscribe and stay updated.

Your subscription could not be saved. Please try again.
Your subscription has been successful.

Join our newsletter