Logo
Magic

Kernel Engineer

Magic, San Francisco, California, United States, 94102

Save Job

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 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. What You Might Work On:

Design, develop and optimize custom high-performance GPU kernels

Evaluate porting Magic's 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

What We're Looking For:

Experience with low-level programming of AI accelerators (e.g. CUDA, XLA)

Deep understanding of GPU, TPU, and/or CPU architecture

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 individuals

Focus.

Safely deploy AGI. Everything else is noise

Quality.

Magic should feel like magic

Compensation, Benefits And Perks (US):

Annual salary range: $100K - $550K

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