drjobs AIML - Staff Machine Learning Engineer - ML Efficiency, ML Platform & Technology

AIML - Staff Machine Learning Engineer - ML Efficiency, ML Platform & Technology

Employer Active

1 Vacancy
drjobs

Job Alert

You will be updated with latest job alerts via email
Valid email field required
Send jobs
Send me jobs like this
drjobs

Job Alert

You will be updated with latest job alerts via email

Valid email field required
Send jobs
Job Location drjobs

Santa Clara - USA

Monthly Salary drjobs

Not Disclosed

drjobs

Salary Not Disclosed

Vacancy

1 Vacancy

Job Description

We are seeking highly motivated and experienced engineers to join our team. The ideal candidate will have a deep understanding of machine learning systems and cloud computing infrastructure. Key responsibilities in this role are: - Engage with ML researchers to optimize end-to-end performance of large scale distributed ML workloads - Analyze workload metrics to identify sources of inefficiencies and work with users to understand and optimize ML workloads - Conduct workload analysis based on benchmarking key workloads on deployed systems - Improve large scale training resiliency by optimizing applications and frameworks for improved recovery from failures and preemptions - Influence architecture design development and operations of next generation ML accelerator systems based on workload insights


  • Experience working with large scale parallel and distributed accelerator-based systems
  • Experience optimizing performance and AI workloads at scale
  • Experience developing code in one or more of training frameworks (such as PyTorch TensorFlow or JAX)
  • Strong communicator with ability to analyze complex and ambiguous problems
  • Programming and software design skills (proficiency in C/C and/or Python)
  • Experience working in a high-level collaborative environment and promoting a teamwork mentality
  • Bachelors degree in Computer Science and 7 years of work experience


  • Deep understanding of computer systems and the interactions between HW and SW
  • Experience in performance analysis and optimization experience in Cloud accelerators
  • Advanced degree in CS

Required Experience:

Staff IC

Employment Type

Full-Time

Company Industry

About Company

Report This Job
Disclaimer: Drjobpro.com is only a platform that connects job seekers and employers. Applicants are advised to conduct their own independent research into the credentials of the prospective employer.We always make certain that our clients do not endorse any request for money payments, thus we advise against sharing any personal or bank-related information with any third party. If you suspect fraud or malpractice, please contact us via contact us page.