H-1B Job Board

Finding companies that sponsor visas is a lot of work.
We've made your life easier by compiling top companies and startups that hire foreign nationals.

AIML - Machine Learning Engineer, MLR

Apple

Apple

Software Engineering
Cupertino, CA, USA
Posted on Jul 30, 2024

Summary

Posted:
Weekly Hours: 40
Role Number:200559789
As part of Apple’s Machine Learning Research organization, we do world-class scientific research and build the technologies that will power future products at Apple. The ML Research Team does world-class research and development across a wide range of domains including understanding and improving ML, addressing bias and fairness in algorithms, privacy and federated learning, embodied and interactive ML systems, computer vision, and more. The techniques and tools we create will impact ML solutions across Apple, which in turn power most of the features we deliver to billions of consumers worldwide. The vast diversity of humans that interact with our technology expect and deserve features that work for them, not just for a select few — ML fairness is how we work to ensure that. A ML Engineer in this position will work with researchers across Machine Learning Research to build scalable, distributed training and research pipelines as well as implement research ideas into algorithms and conduct experiments. Candidates will need strong communication skills to work with a diverse set of partners and customers to determine the right solutions for the problems. They will learn industry standard and custom tools/infrastructure to build these solutions. This work will directly benefit the researchers across the our team and all of Apple. We are looking for highly motivated, result-oriented engineers with a strong background in system engineering and software development to join our team. In this position, you will help build and deploy software to solve a variety of high-performance computing needs including large distributed ML training and research into large-scale ML training.

Description

We have more ideas than we have people, and it is critical for our team to rally together to grow a shared research agenda. The primary objective of a Research Engineer is to accelerate this shared research agenda. As our organization matures, this shared agenda will build out a set of core competencies and provide a foundation for future research projects.

Minimum Qualifications

  • We have more ideas than we have people, and it is critical for our team to rally together to grow a shared research agenda. The primary objective of a Research Engineer is to accelerate this shared research agenda. As our organization matures, this shared agenda will build out a set of core competencies and provide a foundation for future research projects.
  • Expert in Python programming and experience with PyTorch, JAX and/or TensorFlow.
  • Experience with distributed computing
  • Experience maintaining an open-source project
  • Excellent communication skills
  • Experience and passion to circumvent unexpected roadblocks
  • Past research experience is a plus
  • Experience with CUDA programming, and/or High-Performance Computing is a plus

Key Qualifications

Preferred Qualifications

  • Develop and maintain machine learning infrastructure for our ongoing efforts. Most importantly, we need to support a wide variety of datasets, develop a scalable and flexible training infrastructure and support new features as they increase in priority during the course of research.
  • Accelerate our research efforts through experimentation and comprehensive evaluation of model performance.
  • Become our resident expert on the machine learning infrastructure at Apple. Investigate and understand various machine learning infrastructures that we may wish to adopt for MLR.
  • Advise teammates within MLR about new technologies both within and external to Apple for performing machine learning research.

Education & Experience

Additional Requirements

Pay & Benefits

  • Apple is an equal opportunity employer that is committed to inclusion and diversity. We take affirmative action to ensure equal opportunity for all applicants without regard to race, color, religion, sex, sexual orientation, gender identity, national origin, disability, Veteran status, or other legally protected characteristics. Learn more about your EEO rights as an applicant.