Apply on Employer Site

MBZUAI (Mohamed bin Zayed University of Artificial Intelligence) · 5 months ago

Distributed Machine Learning Engineer

Sunnyvale, CA

Full-time

Onsite

Entry, Mid Level

$150K/yr - $450K/yr

2+ years exp

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.

Artificial Intelligence (AI)Higher EducationUniversities

H1B Sponsored

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

Qualification

Parallel ComputingSystem Level CodingLarge-scale Machine LearningDeep Learning OptimizationPerformance BenchmarkingDebug MethodologiesWorkload AnalysisGPU UtilizationDesign ReviewsCode ReviewDocumentation Contribution