Full-Time

ML Framework

Metallm, Engineer

Posted on 12/11/2024

Apple

Apple

10,001+ employees

Designs and sells consumer electronics and software

No salary listed

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
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.
Desired Qualifications
  • 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

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 easy interaction and functionality among its offerings. Additionally, Apple has expanded into financial services with products like the Apple Card, which rewards users with 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 enhancing its product offerings and user experiences, while also focusing on sustainability.

Company Size

10,001+

Company Stage

IPO

Headquarters

Cupertino, California

Founded

1976

Simplify Jobs

Simplify's Take

What believers are saying

  • Apple's collaboration with Air India enhances customer satisfaction with AirTag luggage tracking.
  • The CA-1M dataset positions Apple as a leader in spatial computing.
  • Testing Samsung OLED panels for iPad mini suggests improved display technology.

What critics are saying

  • Smart Home Hub delay until 2026 could hinder competitiveness in the smart home market.
  • DMA complaint over iOS app restrictions may lead to regulatory scrutiny.
  • Reliance on Samsung for OLED panels poses a supply chain risk.

What makes Apple unique

  • Apple's integrated hardware and software create a seamless user experience across devices.
  • The Apple Card enhances customer loyalty with up to 3% Daily Cash back.
  • Apple's trade-in program supports sustainability and incentivizes device upgrades.

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

Company News

5wnaz
Apr 8th, 2025
Apple is building a $1 billion campus in Austin

Apple is building a $1 billion campus in Austin.

SmartHouse
Apr 8th, 2025
Apple Smart Home Hub Delayed Until 2026 Amid Siri Setbacks

Gurman also notes that Apple is already working on a follow-up model, codenamed J595, which could include an "AI personality", added sensors, and even a robotic arm that moves the screen to follow users around a room - a move that would push Apple deeper into the futuristic smart home space.

MacRumors
Apr 7th, 2025
Apple Releases New Immersive Video App for Mac and Vision Pro

Apple today released a new app for managing an immersive video library on a Mac.

Exhibit
Apr 5th, 2025
Air India Collaborates with Apple to Provide AirTag-Based Lost Luggage Tracking

Air India collaborates with Apple to provide airtag-based lost luggage tracking.

A1 Coach
Apr 4th, 2025
The Rise of Mobile Gaming: Will It Replace Console?

Even Apple has launched its A-series chips to assist in mobile gaming using iOS.

INACTIVE