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. A key aspect of Apple's business model is its focus on both hardware and software, which allows for better integration and user satisfaction. Unlike many competitors, Apple emphasizes customer loyalty through offerings like the Apple Card, which rewards users with cash back, and a trade-in program that encourages device upgrades while promoting recycling. Apple's goal is to maintain its leadership in the technology market by continuously enhancing its product offerings and user experiences.

Company Size

10,001+

Company Stage

IPO

Headquarters

Cupertino, California

Founded

1976

Simplify Jobs

Simplify's Take

What believers are saying

  • Project Mulberry could revolutionize digital health, increasing Apple's ecosystem engagement.
  • Siri's AI enhancements in iOS 19 may improve user satisfaction and device attractiveness.
  • Partnership with Home Credit Philippines may boost iPhone 16e sales in Southeast Asia.

What critics are saying

  • Project Mulberry may face competition from Google Health and Amazon Care.
  • Privacy concerns over AI health initiatives could lead to backlash.
  • Vanessa Trigub's appointment may cause internal restructuring challenges in retail operations.

What makes Apple unique

  • Apple's Project Mulberry integrates AI into health monitoring, enhancing user engagement.
  • AI-powered Siri overhaul in iOS 19 boosts user interaction and device appeal.
  • Focus on spatial computing offers unique experiences, setting Apple apart from competitors.

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

AppleInsider
Mar 30th, 2025
Health app's future AI assistant will tell you how to keep fit

According to Mark Gurman in Bloomberg's newsletter on Sunday, Apple is working on Project Mulberry.

Softonic
Mar 30th, 2025
Apple's Project Mulberry wants to change medicine forever: AI is the key

Apple is preparing to make a major leap in digital health with the launch of Project Mulberry, a new AI-powered health initiative expected in iOS 19.4.

9to5Mac
Mar 29th, 2025
Here's everything we're expecting with Siri and Apple Intelligence in iOS 19

Bloomberg's mark Gurman reported last year that Apple was working on a new Siri backend for iOS 19 - which he referred to as 'LLM Siri.'

NoypiGeeks
Mar 28th, 2025
Home Credit teams up with major retailers to offer exclusive deals for Apple iPhone 16e

Home Credit Philippines (HCPH), one of the top consumer finance providers in the country, has partnered with top Apple retailers to host a simultaneous midnight launch of the Apple iPhone 16e.

Gulf Articles
Mar 28th, 2025
Apple Announces WWDC 2025 Dates: Key Highlights and What to Expect

Apple has been working on a major AI overhaul for Siri, enabling:

INACTIVE