Kernel Engineer
About The Position
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.
Requirements
- Experience with low-level programming of AI accelerators (e.g. CUDA, XLA)
- Deep understanding of GPU, TPU, and/or CPU architecture
Responsibilities
- Design, develop and optimize custom high-performance GPU kernels
- Evaluate porting Magic’ compute kernels to alternative hardware options
- Think beyond the kernel level to the broader scheme of how we train and inference our models and suggest improvements
Benefits
- 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
Stand Out From the Crowd
Upload your resume and get instant feedback on how well it matches this job.
What This Job Offers
Job Type
Full-time
Career Level
Mid Level
Education Level
No Education Listed
Number of Employees
11-50 employees
