Research Engineer - ML Sys & Infra

This range is provided by Storm3. Your actual pay will be based on your skills and experience — talk with your recruiter to learn more.

Base pay range

$200,000.00 / yr - $350,000.00 / yr

Direct message the job poster from Storm3

⚡ Research Engineer - ML Sys, Infra Optimization and Scaling

Come join a revolutionary AI research lab in SF Bay Area that is poised to develop & publish high-impact breakthroughs across LLMs, RL and Multimodal AI.

Opportunity to join as an early member in a rapidly growing team, led by A-listers in tech & academia. Seeking Engineers / Researchers skilled in distributed training, inference optimization and scaling laws for large-scale deep learning clusters and infrastructure .

Responsibilities

• Research & implement SOTA methods in high performance parallel computing
• Set up distributed training frameworks and extend AI optimizer frameworks
• Write production-quality code for ML infra; ensure reliability at scale
• Work collaboratively in a early-stage team to enhance multi-GPU training set-up

Requirements

Experience working with CUDA and deep learning optimization in HPC environment

Why apply

📧 Interested in applying? Please click on the ‘Easy Apply’ button or alternatively email me your resume at anir.gantugs@storm3.com

#J-18808-Ljbffr