GPU Machine Learning Engineer

  • Apple, Inc.
  • Cupertino, CA, USA
  • Nov 09, 2020

Job Description

GPU Machine Learning Engineer Santa Clara Valley (Cupertino),California,United States Software and Services + Excellent programming and problem-solving skills + Good understanding of machine learning fundamentals + Experience with system-level programming and computer architecture + Background in mathematics, including linear algebra and numerical methods, is a plus + Experience with high-performance parallel programming, GPU programming experience a plus + Experience with machine learning libraries (TensorFlow. PyTorch); adding runt time and computation graph support is a plus **Description** Our team is seeking extraordinary machine learning and GPU programming engineers who are passionate about providing robust compute solutions for accelerating machine learning networks on Apple's GPUs. Work includes defining and implementing APIs in Metal Performance Shaders, investigating new algorithms, adding support for high-level machine learning libraries on GPUs, adding optimizations in machine learning computation graph, performing in-depth analysis and kernel-level optimizations to ensure the best possible performance across GPU families. The role has the opportunity to influence the design of compute and programming models in next-generation GPU architectures. **Education & Experience** Technical BS/MS degree. PhD is a plus. **Additional Requirements** + Strong communication and collaboration skills + Strong track record of building high performance, production quality software on schedule. **Apple Footer** 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 at (Opens in a new window) at . Apple will not discriminate or retaliate against applicants who inquire about, disclose, or discuss their compensation or that of other applicants. United States Department of Labor. at Learn more at (Opens in a new window) at . Apple will consider for employment all qualified applicants with criminal histories in a manner consistent with applicable law. If you're applying for a position in San Francisco, review the San Francisco Fair Chance Ordinance at Poster Set All Languages 10 01 18.pdf guidelines at Poster Set All Languages 10 01 18.pdf (opens in a new window) at Poster Set All Languages 10 01 18.pdf applicable in your area. Apple participates in the E-Verify program in certain locations as required by law. Learn more about the E-Verify program at (Opens in a new window) at . Apple is committed to working with and providing reasonable accommodation to applicants with physical and mental disabilities. Apple is a drug-free workplace. Reasonable Accommodation and Drug Free Workplace policy at Learn more at (Opens in a new window) at .
Associated topics: algorithm, backend, c++, c#, develop, devops, matlab, php, python, sdet