Full-Time

ML Framework

Metallm, 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
C/C++
Requirements
  • 3+ years of programming and problem-solving experience with C/C++/ObjC
  • Experience with Distributed training or inference techniques
  • GPU compute programming models & optimization techniques
  • Experience with system level programming and computer architecture
  • Contributions to an AI framework such as PyTorch, JAX or Tensorflow is a plus
  • Experience with graph compilers such as Triton, OpenXLA or LLVM/MLIR is a plus
  • Good understanding of machine learning fundamentals
Responsibilities
  • Design and develop compiler based optimizations for Metal backend in ML frameworks such as torch.compile for PyTorch
  • Work on cutting-edge ML inference framework project and optimize code for efficient and scalable ML inference using distributed techniques
  • Implement features of Metal device backend for ML training acceleration technologies
  • Work with Core teams of PyTorch, JAX or Tensorflow to provide Metal runtime and device backend support
  • Tune GPU-accelerated training across products.
  • Performing in-depth analysis, compiler and kernel level optimizations to ensure the best possible performance across hardware families.

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?