Full-Time
ML Framework
Metallm, Engineer
Confirmed live in the last 24 hours
Apple
10,001+ employees
Designs and sells consumer electronics and software
Mid, Senior
Waltham, MA, USA
- 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
- 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.
- 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 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'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