Skip to main content
Junior Distributed Machine Learning Engineer Join to apply for the Junior Distributed Machine Learning Engineer role at Jobright.aiJunior Distributed Machine Learning Engineer1 day ago Be among the first 25 applicants Join to apply for the Junior Distributed Machine Learning Engineer role at Jobright.aiGet AI-powered advice on this job and more exclusive features. Jobright is an AI-powered career platform that helps job seekers discover the top opportunities in the US. We are NOT a staffing agency. Jobright does not hire directly for these positions. We connect you with verified openings from employers you can trust.
Job Summary: Mohamed bin Zayed University of Artificial Intelligence is dedicated to research, innovation, and empowering brilliant minds in AI. The Distributed Machine Learning Engineer will optimize performance for machine learning software stacks, develop new systems, and work alongside researchers to tackle challenges in AI development.
Responsibilities:
  • Understand, analyze, profile, optimize, and provide guidance to the team on deep learning workloads on state-of-the-art hardware and software platforms to improve their efficiency with different levels of optimization
  • Design and implement performance benchmarks and testing methodologies to evaluate application performance
  • Build tools to automate workload analysis, workload optimization, and other critical workflows
  • Triage system issues and identify bottleneck and inefficiencies by analyzing the sources of issues and the impact on hardware, network and propose solutions to enhance GPU utilization
  • Support the team to develop appropriate kernels and systems for new model architectures and algorithms
Participate in, or lead design reviews with peers and stakeholders to decide amongst available technologies. Review code developed by other developers and provide feedback to ensure best practices (e.g., style guidelines, checking code in, accuracy, testability, and efficiency).Contribute to existing documentation or educational content and adapt content based on product/program updates and user feedback. Represent MBZUAI at industry conferences and events, showcasing the institution's cutting-edge HPC and deep learning capabilities and establishing MBZUAI as a global leader in AI research and innovation. Perform all other duties as reasonably directed by the line manager that are commensurate with these functional objectives.
Qualifications
Required:
  • Ph.D. in CS, EE or CSEE with 1+ years working experience, ORMasters in CS, EE or CSEE or equivalent experience with 2+ year working experience
  • Strong background in parallel computing
  • Hands-on experience in system level coding
  • Large-scale machine learning experience
Company: Official account of Mohamed bin Zayed University of Artificial Intelligence. Dedicated to research, innovation, and empowering brilliant minds in AI. Founded in 2019, the company is headquartered in Abu Dhabi, Abu Dhabi, ARE, with a team of 51-200 employees. The company is currently Growth Stage. MBZUAI (Mohamed bin Zayed University of Artificial Intelligence) has a track record of offering H1B sponsorships. Seniority level Entry level Employment type Full-time Job function IndustriesSoftware Development Referrals increase your chances of interviewing at Jobright.ai by 2xInferred from the description for this job Medical insurance Vision insurance 401(k) Get notified when a new job is posted. Sign in to set job alerts for "Machine Learning Engineer" roles. AI/ML Engineer (Multiple roles and seniority levels) Software Development Engineer In Test (SDET) San Jose, CA $87,600.00-$186,000.00 5 days agoSunnyvale, CA $117,000.00-$234,000.00 3 weeks agoSan Jose, CA $137,500.00-$236,500.00 1 week agoSan Jose, CA $120,700.00-$228,600.00 1 week agoNew Grads 2025 - Software Engineer - Computer Vision/Deep Learning San Jose, CA $120,000.00-$165,000.00 9 months agoSan Francisco Bay Area $155,000.00-$265,500.00 2 weeks agoSunnyvale, CA $117,000.00-$234,000.00 4 days agoSan Jose, CA $120,700.00-$228,600.00 1 week agoSan Jose, CA $137,500.00-$236,500.00 1 month agoNew Grads 2025 - Software Engineer, Algorithm San Jose, CA $120,000.00-$165,000.00 10 months agoSan Jose, CA $137,500.00-$236,500.00 3 weeks ago Software Engineer, AI Platform - New Grad Mountain View, CA $145,000.00-$170,000.00 1 week agoMachine Learning Engineer (I, II, or Sr.)Sunnyvale, CA $158,200.00-$185,000.00 4 weeks agoMachine Learning Engineer, Early Stage Project Mountain View, CA $136,000.00-$185,000.00 2 weeks agoSan Jose, CA $120,700.00-$228,600.00 1 week agoSan Jose, CA $130,000.00-$200,000.00 4 days agoSan Jose, CA $169,500.00-$291,500.00 1 week agoSunnyvale, CA $167,000.00-$185,500.00 1 month agoSunnyvale, CA $167,000.00-$185,500.00 4 weeks agoSan Jose, CA $120,000.00-$240,000.00 5 months agoSan Jose, CA $130,000.00-$182,000.00 11 months ago Redwood City, CA $167,200.00-$250,800.00 2 weeks agoSan Jose, CA $118,657.00-$187,200.00 2 weeks agoSan Jose, CA $137,500.00-$236,500.00 4 months agoNew Grads 2025 - General Software Engineer San Jose, CA $120,000.00-$165,000.00 6 months agoSan Jose, CA $142,700.00-$257,600.00 1 week ago We're unlocking community knowledge in a new way. Experts add insights directly into each article, started with the help of AI.

Also on the board Same function, level within a rung

Level

Senior

Location

Sunnyvale, CA

Occupation

Software Developers

Industry

Software Publishers

Posted

today

Apply for this role (opens in a new tab)