Distributed ML Engineer for High-Performance AI Training

Ifm Us; Sunnyvale (CA); On-site; USD 140,000 - 210,000; Full time

Benefits offered by this job

Medical benefits

Dental benefits

Vision benefits

Bonus

401K Plan

Paid time off

Parental Leave

Employee Assistance Program

Life insurance

Disability insurance

Job summary

MBZUAI in Sunnyvale, CA seeks a Distributed ML Engineer to optimize ML software stacks for training and inference on state-of-the-art hardware. You will prototype kernels, build benchmarks, and help scale large‑scale models while collaborating with researchers and engineers.

The role emphasizes parallel computing, system‑level coding and hands-on ML experience, with visa sponsorship available. You will contribute to cutting‑edge foundational AI research and advanced HPC capabilities.

Qualifications

• Ph.D. in CS, EE or CSEE with 1+ years working experience, OR Master’s in CS, EE or CSEE or equivalent experience with 2+ year working experience.

Responsibilities

• Understand, analyze, profile, optimize, and provide guidance on deep learning workloads to improve efficiency.
• Design and implement performance benchmarks and testing methodologies.
• Build tools to automate workload analysis and optimization.
• Triage system issues and identify bottlenecks to enhance GPU utilization.
• Support kernels and systems for new model architectures and algorithms.
• Participate in or lead design reviews and decide among technologies.
• Review code for best practices and quality.
• Contribute to documentation and educational content.
• Represent MBZUAI at conferences and events to showcase HPC and DL capabilities.
• Perform other duties as directed by the line manager.

Skills

Parallel computing

System level coding

GPU utilization

Performance benchmarking

Code review

Education

Ph.D. in CS, EE or CSEE

Master’s in CS, EE or CSEE

Job description

MBZUAI in Sunnyvale, CA seeks a Distributed ML Engineer to optimize ML software stacks for training and inference on state-of-the-art hardware. You will prototype kernels, build benchmarks, and help scale large‑scale models while collaborating with researchers and engineers.

The role emphasizes parallel computing, system‑level coding and hands-on ML experience, with visa sponsorship available. You will contribute to cutting‑edge foundational AI research and advanced HPC capabilities.