Simplify Logo

Full-Time

GPU ML Engineer

Confirmed live in the last 24 hours

Apple

Apple

10,001+ employees

Designs and sells consumer electronics and software

VR & AR
Consumer Software
Fintech
Consumer Goods

Mid, Senior

Waltham, MA, USA

Category
Applied Machine Learning
AI & Machine Learning
Required Skills
Tensorflow
Pytorch
Machine Learning
Requirements
  • Proven programming and problem-solving skills.
  • Good understanding of machine learning fundamentals.
  • GPU compute programming models & optimization techniques.
  • GPU compute framework development, maintenance, and optimization.
  • Machine learning development using one or more ML frameworks (TensorFlow, PyTorch or JAX).
Responsibilities
  • Adding optimizations in machine learning computation graph.
  • Defining and implementing APIs in Metal Performance Shaders Graph, investigating new algorithms.
  • Develop/maintain/optimize ML training acceleration technologies.
  • Support adoption of GPU-accelerated training across 1st and 3rd party clients.
  • Tune GPU-accelerated training across products.
  • Performing in-depth analysis, compiler and kernel level optimizations to ensure the best possible performance across hardware families.
Desired Qualifications
  • Experience with adding computational graph support, runtime or device backend to Machine learning libraries (TensorFlow, PyTorch or JAX) support.
  • Experience with high performance parallel programming, GPU programming or LLVM/MLIR compiler infrastructure.
  • Experience with system level programming and computer architecture.
  • Background in mathematics, including linear algebra and numerical methods.
Apple

Apple

View
Website

Apple Inc. designs and sells a variety of technology products and services, including iPhones, iPads, Mac computers, Apple Watches, and Apple TVs. The company also offers services like the App Store, Apple Music, iCloud, and Apple Pay. Its products work together seamlessly, providing users with a cohesive experience across devices. Apple differentiates itself from competitors through its integrated approach, where it controls both hardware and software, ensuring high-quality performance and user satisfaction. The company's goal is to enhance customer loyalty and drive financial success through continuous innovation and a strong focus on user experience.

Company Stage

IPO

Total Funding

$619.1K

Headquarters

Cupertino, California

Founded

1976

Simplify Jobs

Simplify's Take

What believers are saying

  • Apple's sustainability efforts enhance its brand image among eco-conscious consumers.
  • Expansion into the smart home market could increase Apple's market share.
  • The Apple Store app launch in India targets a fast-growing smartphone market.

What critics are saying

  • AI News Summary feature spreading misinformation could damage Apple's reputation.
  • Delays in OLED MacBook Air plans may affect Apple's competitive edge.
  • Transition of Apple Card partners could disrupt financial services and customer experience.

What makes Apple unique

  • Apple's integrated hardware and software create a seamless user experience.
  • The Apple Card offers unique financial incentives, enhancing customer loyalty.
  • Apple's spatial computing innovations provide immersive experiences without physical controllers.

Help us improve and share your feedback! Did you find this helpful?

Benefits

Health Insurance

Dental Insurance

401(k) Retirement Plan

401(k) Company Match

Tuition Reimbursement

Performance Bonus

Relocation Assistance

Employee Stock Purchase Plan