ML Compute Acceleration Engineer
1 week ago
ML Compute Acceleration Engineer
Waltham,Massachusetts,United States
Software and Services
Apple’s Compute Frameworks team in GPU, Graphics and Displays org provides a suite of high-performance data parallel algorithms for developers inside and outside of Apple for iOS, macOS and Apple TV. Our efforts are currently focused in the key areas of linear algebra, image processing, machine learning, along with other projects of key interest to Apple. We are always looking for exceptionally dedicated individuals to grow our outstanding team to lay the foundation of technologies like Apple Intelligence.
Description
Our team is seeking extraordinary machine learning and GPU programming engineers who are passionate about providing robust compute solutions for accelerating machine learning networks on Apple Silicon using GPU and Neural Engine. Role has the opportunity to influence the design of compute and programming models in next generation GPU and Neural Engine architectures. Responsibilities: * Adding optimizations in machine learning computation graph. * Defining and implementing APIs in Metal Performance Shaders Graph, investigating new algorithms. * Developing and maintaining MLIRdialect 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. Intended deliverables: * GPU Compute acceleration technology. * Apple Intelligence implementation and acceleration. * Optimized compute graphs across products. If this sounds of interest, we would love to hear from you
Minimum Qualifications
-
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.
Key Qualifications
Preferred Qualifications
-
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.
Education & Experience
Additional Requirements
- Apple is an equal opportunity employer that is committed to inclusion and diversity. We take affirmative action to ensure equal opportunity for all applicants without regard to race, color, religion, sex, sexual orientation, gender identity, national origin, disability, Veteran status, or other legally protected characteristics.Learn more about your EEO rights as an applicant. (https://www.eeoc.gov/sites/default/files/2023-06/22-088_EEOC_KnowYourRights6.12ScreenRdr.pdf)
Apple Footer
Apple is an equal opportunity employer that is committed to inclusion and diversity. We take affirmative action to ensure equal opportunity for all applicants without regard to race, color, religion, sex, sexual orientation, gender identity, national origin, disability, Veteran status, or other legally protected characteristics. Learn more about your EEO rights as an applicant (Opens in a new window) .
Apple will not discriminate or retaliate against applicants who inquire about, disclose, or discuss their compensation or that of other applicants. United States Department of Labor. Learn more (Opens in a new window) .
Apple will consider for employment all qualified applicants with criminal histories in a manner consistent with applicable law. If you’re applying for a position in San Francisco, review the San Francisco Fair Chance Ordinance guidelines (opens in a new window) applicable in your area.
Apple participates in the E-Verify program in certain locations as required by law. Learn more about the E-Verify program (Opens in a new window) .
Apple is committed to working with and providing reasonable accommodation to applicants with physical and mental disabilities. Reasonable Accommodation and Drug Free Workplace policy Learn more (Opens in a new window) .
Apple is a drug-free workplace. Reasonable Accommodation and Drug Free Workplace policy Learn more (Opens in a new window) .
-
GPU Acceleration Engineer
4 weeks ago
Watertown, Massachusetts, United States Apple Full timeJoin us at Apple as a GPU Acceleration Engineer and be part of our innovative team that provides GPU acceleration for popular Machine Learning libraries such as TensorFlow, PyTorch, and JAX using Metal runtime and device backend.The successful candidate will have experience in designing and developing compiler-based optimizations for Metal backend in ML...
-
GPU Accelerated AI Engineer
3 weeks ago
Watertown, Massachusetts, United States Apple Full timeRole Summary:\We're seeking a highly skilled GPU Accelerated AI Engineer to join our team at Apple in Waltham, Massachusetts. As a key member of our Compute Frameworks team, you'll be responsible for developing high-performance data parallel algorithms for iOS, macOS, and Apple TV.\Your primary responsibilities will include:\\Adding optimizations in machine...
-
GPU ML Engineer Expert
3 weeks ago
Watertown, Massachusetts, United States Apple Full timeJob Description:\We are seeking a talented GPU ML Engineer to join our team at Apple in Waltham, Massachusetts. As a key member of our Compute Frameworks team, you will contribute to the development of high-performance data parallel algorithms for iOS, macOS, and Apple TV.\Your primary responsibilities will include adding optimizations in machine learning...
-
GPU ML Engineer
1 week ago
Watertown, United States Apple Full timeGPU ML Engineer Waltham,Massachusetts,United States Software and Services Apple’s Compute Frameworks team in GPU, Graphics and Displays org provides a suite of high-performance data parallel algorithms for developers inside and outside of Apple for iOS, macOS and Apple TV. Our efforts are currently focused in the key areas of linear algebra, image...
-
ML Framework
1 week ago
Watertown, United States Apple Full timeML Framework (MetalLM) Engineer Waltham,Massachusetts,United States Software and Services Apple’s ML Frameworks team in GPU, Graphics and Displays org provides GPU acceleration for popular Machine learning libraries such as TensorFlow, PyTorch and JAX using Metal runtime and device backend. It optimizes compute performance with kernels and computational...
-
GPU Acceleration Engineer
3 weeks ago
Watertown, Massachusetts, United States Apple Full timeJob DescriptionWe are looking for an exceptional Machine Learning Compiler Developer to join our team at Apple. In this role, you will work on designing and developing compiler-based optimizations for Metal backend in ML frameworks such as torch.compile for PyTorch.You will be responsible for working on cutting-edge ML inference framework project and...
-
ML Framework Developer
4 weeks ago
Watertown, Massachusetts, United States Apple Full timeJob Title: ML Framework DeveloperEstimated Salary: $180,000 - $240,000 per yearWe are seeking a skilled ML Framework Developer to join our team at Apple in Waltham, Massachusetts. As an ML Framework Developer, you will be responsible for designing and developing compiler-based optimizations for Metal backend in ML frameworks such as PyTorch, working on...
-
High-Performance Computing Specialist
3 weeks ago
Watertown, Massachusetts, United States Apple Full timeKey ResponsibilitiesWe are looking for a highly skilled Machine Learning and GPU Programming Engineer to join our team. Key responsibilities include:Developing Optimizations in ML Computation Graph: Design and implement efficient computation graph optimizations for machine learning workloads.Defining and Implementing APIs in Metal Performance Shaders Graph:...
-
Machine Learning Engineer
4 weeks ago
Watertown, Massachusetts, United States Apple Full timeJob Title: Machine Learning EngineerEstimated Salary: $140,000 - $200,000 per yearWe are seeking a skilled Machine Learning Engineer to join our team at Apple in Waltham, Massachusetts. As a Machine Learning Engineer, you will be responsible for designing and developing compiler-based optimizations for Metal backend in ML frameworks such as PyTorch.About the...
-
GPU Compute Specialist
3 weeks ago
Watertown, Massachusetts, United States Apple Full timeJob Title: GPU Compute SpecialistEstimated Salary: $160,000 - $220,000 per yearWe are seeking a skilled GPU Compute Specialist to join our team at Apple in Waltham, Massachusetts. As a GPU Compute Specialist, you will be responsible for working on cutting-edge ML inference framework projects, optimizing code for efficient and scalable ML inference using...
-
GPU Engineer
4 weeks ago
Watertown, Massachusetts, United States Apple Full timeJob Title: GPU EngineerAbout the Role:We are seeking an exceptional GPU engineer to join our team at Apple. As a GPU engineer, you will be responsible for designing and developing high-performance data parallel algorithms for developers inside and outside of Apple.About the Team:The Compute Frameworks team in GPU, Graphics and Displays org provides a suite...
-
Machine Learning Engineer
3 weeks ago
Watertown, Massachusetts, United States Apple Full timeJob SummaryWe are seeking a talented MACHINE LEARNING ENGINEER to join our team at Apple in Waltham, Massachusetts. As a key member of our GPU acceleration team, you will design and develop compiler-based optimizations for Metal backend in ML frameworks such as torch.compile for PyTorch.About the RoleThe ideal candidate will have 3+ years of experience in...
-
Compute Solutions Developer
4 weeks ago
Watertown, Massachusetts, United States Apple Full timeWe are seeking a highly motivated Compute Solutions Developer to join our team at Apple.The successful candidate will have experience in designing and developing compiler-based optimizations for Metal backend in ML frameworks such as torch.compile for PyTorch, working on cutting-edge ML inference framework project, optimizing code for efficient and scalable...
-
GPU Software Engineer
3 weeks ago
Watertown, Massachusetts, United States Apple Full timeAbout Apple's Compute Frameworks TeamWe are a team of talented engineers passionate about providing high-performance data parallel algorithms for developers worldwide. Our efforts focus on key areas like linear algebra, image processing, machine learning, and more.Job DescriptionWe are seeking an exceptional Machine Learning and GPU Programming Engineer to...
-
Machine Learning Architect
4 weeks ago
Watertown, Massachusetts, United States Apple Full timeWe are seeking a highly skilled Machine Learning Architect to join our team at Apple.The successful candidate will have experience in designing and developing compiler-based optimizations for Metal backend in ML frameworks such as torch.compile for PyTorch. Key responsibilities include working on cutting-edge ML inference framework project, optimizing code...
-
Advanced Computing Engineer
4 weeks ago
Watertown, Massachusetts, United States Flexcompute Inc. Full timeFlexcompute Inc., an early-stage technology startup, is revolutionizing the field of scientific computing by developing ultra-fast simulation technology. Our award-winning products are utilized by companies worldwide to design and optimize technology products.We are seeking an experienced Senior Scientific Visualization Engineer who can help us create a...
-
Senior/Staff Software Engineer
4 weeks ago
Watertown, United States Dyno Therapeutics Full timeThe Company Dyno Therapeutics is reshaping the gene therapy landscape through AI-powered vectors. Through the application of our transformative technologies and strategic partnerships with leaders in gene therapy, we believe a future with life changing gene therapies for millions of people is within reach. Our team includes world-class molecular and...
-
2025 Summer Software Engineering Internship
2 weeks ago
Watertown, United States Dyno Therapeutics Full timeThe Company Dyno Therapeutics is reshaping the gene therapy landscape through AI-powered vectors. Through the application of our transformative technologies and strategic partnerships with leaders in gene therapy, we believe a future with life changing gene therapies for millions of people is within reach. Our team includes world-class molecular and...
-
2025 Summer Software Engineering Internship
1 week ago
Watertown, United States Dyno Therapeutics Full timeThe Company Dyno Therapeutics is reshaping the gene therapy landscape through AI-powered vectors. Through the application of our transformative technologies and strategic partnerships with leaders in gene therapy, we believe a future with life changing gene therapies for millions of people is within reach. Our team includes world-class molecular and...
-
Senior Machine Learning Developer
4 weeks ago
Watertown, Massachusetts, United States Apple Full timeJob Description:We are seeking a Senior Machine Learning Developer to join our team at Apple. As a Senior Machine Learning Developer, you will be responsible for developing high-performance machine learning algorithms and software for various Apple products.About the Team:The Compute Frameworks team in GPU, Graphics and Displays org provides a suite of...