Full-Time

ML Compute Acceleration 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
Deep Learning
AI & Machine Learning
Required Skills
Tensorflow
Pytorch
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.
  • Experience with system level programming and computer architecture.
  • Experience with high performance parallel programming, GPU programming or LLVM/MLIR compiler infrastructure is a plus.
  • Background in mathematics, including linear algebra and numerical methods.
  • Strong communication and collaboration skills.
  • Strong background of building high performance, production quality software on schedule.
  • Experience with compiler technologies.
  • Experience with adding computational graph support, runtime or device backend to Machine learning libraries (TensorFlow, PyTorch or JAX) support is a plus.
Responsibilities
  • Adding optimizations in machine learning computation graph.
  • Defining and implementing APIs in Metal Performance Shaders Graph, investigating new algorithms.
  • Developing and maintaining MLIR dialect in Apple and open source with upgrades using latest LLVM.
  • Performing in-depth analysis, compiler and kernel level optimizations to ensure the best possible performance across hardware families.
  • Tune GPU and Neural Engine accelerated compute across products.
  • Tuning the cost model and optimizing runtime dispatch to multiple IPs to get best performance on Apple Silicon.

Apple Inc. designs, manufactures, 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. Apple's products work together seamlessly, providing users with a cohesive experience across devices. This integration is a key differentiator from competitors, as it allows for a smooth transition between hardware and software. Additionally, Apple has expanded into financial services with products like the Apple Card, which offers cash back on purchases, and a trade-in program that encourages device upgrades while promoting recycling. The company's goal is to maintain its leadership in the technology market by continuously innovating and enhancing user experiences, particularly in areas like spatial computing.

Company Stage

IPO

Total Funding

$619.1K

Headquarters

Cupertino, California

Founded

1976

Simplify Jobs

Simplify's Take

What believers are saying

  • Redesigned Magic Mouse could improve customer satisfaction and address user complaints.
  • Secure Vault feature may attract privacy-conscious users to Apple's ecosystem.
  • Project Titan could diversify Apple's offerings into the growing electric vehicle market.

What critics are saying

  • Legal issues in DR Congo may affect Apple's supply chain and reputation.
  • Competition from Jio Cloud may pressure Apple's iCloud to improve offerings.
  • Magic Mouse redesign could lead to increased production costs and potential delays.

What makes Apple unique

  • Apple's integrated hardware and software create a seamless user experience.
  • The Secure Vault feature in iOS 18.2 enhances privacy for sensitive photos.
  • Apple's foldable device plans show commitment to innovation in new tech segments.

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