Kernel Engineer

November 12

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

November 12

Fire Protection Engineer managing standards development at the National Fire Protection Association.

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