Job Details

? Research Engineer - ML Sys, Infra Optimization and Scaling

?? Foundation Models, AI Research Institute

?? San Francisco Bay Area, USA

?? $250,000 - $400,000 salary + bonus

Come join a frontier AI lab in the Bay Area that's developing large-scale foundation models, and already with 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 ML Systems, Inference Optimization and HPC
• Set up / improve distributed training infrastructure and optimizer frameworks
• Write production-quality code and ensure reliability at scale
• Work collaboratively in a tight-knit team to compete on the global AI leaderboards

Requirements:

• MS or PhD in Comp Sci, Comp Engineering, or related
• Expertise in distributed ML frameworks (e.g., DeepSpeed, FSDP, vLLM)
• Multi-node (Ray, Kubernetes) and distributed inference optimization experience
• Experience working with CUDA and deep learning optimization in HPC environment
• Proficiency in leveraging high compute GPU clusters

Why apply:

• Opportunity to build out a new division at the forefront of AI innovation
• FAANG competitive salary & package
• Work alongside superstars from FAANG labs & leading AI companies
• Medical, Dental and Vision Insurance
• Relocation package available

?? San Francisco Bay Area, USA

?? Interested in applying? Please click on the 'Easy Apply' button or alternatively email me your resume at ...@storm3.com