Kernel Engineer

2 days ago

Apply Now

Description

• Design, implement and maintain high-performance kernels to optimize throughput and latency during training and inference. • Design, develop and optimize custom high-performance GPU kernels • Evaluate porting Magic’ compute kernels to alternative hardware options • Suggest improvements to how we train and inference our models

Requirements

• Experience with low-level programming of AI accelerators (e.g. CUDA, XLA) • Deep understanding of GPU, TPU, and/or CPU architecture

Benefits

• 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

Apply Now

Similar Jobs

2 days ago

Provide post sales support for Agilent GC and GCMS products.

🇺🇸 United States – Remote

💵 $82.5k - $128.9k / year

💰 Post-IPO Debt on 2019-09

⏰ Full Time

🟢 Junior

🟡 Mid-level

2 days ago

GE

10,000+

Remote tuning on heavy duty gas turbines using data streaming technology.

2 days ago

Provide technical service and support for Dynex Technologies products.

Built by Lior Neu-ner. I'd love to hear your feedback — Get in touch via DM or lior@remoterocketship.com