GPU ML Engineer Expert
4 days ago
Job 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 computation graphs, defining and implementing APIs in Metal Performance Shaders Graph, investigating new algorithms, and developing/maintaining/optimizing ML training acceleration technologies.
\You will work closely with our team to support adoption of GPU-accelerated training across 1st and 3rd party clients, tune GPU-accelerated training across products, and perform in-depth analysis, compiler and kernel level optimizations to ensure the best possible performance across hardware families.
\In this role, you will have the opportunity to influence the design of compute and programming models in next-generation GPU architectures.
\We offer an estimated salary of $120,000 - $180,000 per year based on experience, as well as excellent benefits and opportunities for professional growth.
\Key Responsibilities:
\- \
- Add optimizations in machine learning computation graphs\
- Define and implement APIs in Metal Performance Shaders Graph\
- Investigate new algorithms\
- Develop/maintain/optimize ML training acceleration technologies\
- Support adoption of GPU-accelerated training across 1st and 3rd party clients\
- Tune GPU-accelerated training across products\
- Perform in-depth analysis, compiler and kernel level optimizations\
- \
Requirements:
\- \
- 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\
- Machine learning development using one or more ML frameworks (TensorFlow, PyTorch or JAX)\
- \
About Us:
\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.
-
GPU Engineer
2 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...
-
GPU Acceleration Engineer
2 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 Software Engineer
4 days 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...
-
GPU Accelerated AI Engineer
4 days 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 Acceleration Engineer
4 days 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...
-
GPU Compute Specialist
1 week 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...
-
ML Framework Developer
2 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...
-
Machine Learning Engineer
2 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...
-
Machine Learning Engineer
1 week 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...
-
LLVM-Based Compiler Optimizations Expert
2 weeks ago
Watertown, Massachusetts, United States Apple Full timeJob Overview: LLVM-Based Compiler Optimizations ExpertLocation: Waltham, Massachusetts, United StatesAbout the Role:As a highly skilled expert in compiler optimizations, you will be responsible for researching, designing, and implementing new compiler optimizations to improve performance and functionality across all Apple devices. You will work closely with...
-
High-Performance Computing Specialist
1 week 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 Architect
2 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...
-
Compute Solutions Developer
2 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...
-
Senior Machine Learning Developer
2 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...
-
Apple Silicon Compiler Engineer
4 days ago
Watertown, Massachusetts, United States Apple Full timeAbout the RoleAs a Backend Compiler Specialist at Apple, you will play a critical role in developing optimized GPU binaries for future Apple Silicon GPUs. You will work closely with various hardware and software teams to prototype compiler feature support, code generation, and optimization techniques. The successful candidate will have a strong background in...
-
Data Visualization Expert
1 week ago
Watertown, Massachusetts, United States Flexcompute Inc. Full timeFundamental to our mission at Flexcompute Inc. is the pursuit of innovation through the development of revolutionary simulation technology. By pushing the boundaries of what is possible, we empower companies worldwide to create groundbreaking technological advancements.We seek a highly skilled Senior Scientific Visualization Engineer to drive the creation of...
-
Watertown, Massachusetts, United States Dyno Therapeutics Full timeAbout Dyno TherapeuticsWe are a pioneering company in the field of gene therapy, leveraging AI-powered vectors to transform the industry.Job SummaryWe are seeking a highly skilled full stack developer to join our team as a Senior/Staff Software Engineer - Full Stack. In this role, you will lead the development of new products for genetic sequence design and...
-
Advanced Computing Engineer
2 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...
-
Cloud Engineer
1 week ago
Watertown, Massachusetts, United States Bosch USA Full timeJob OverviewWe are seeking an experienced Cloud Engineer - AWS Expert to join our team at Bosch USA. This is a full-time opportunity for a highly motivated individual who wants to work with the latest technologies in cloud computing.The estimated salary for this role is $120,000 - $180,000 per year, depending on experience and qualifications.About UsBosch...
-
Scientific Computing Engineer
1 week ago
Watertown, Massachusetts, United States ZipRecruiter Full timeJob DescriptionWe are seeking a skilled Scientific Computing Engineer with extensive data visualization expertise to join our team. The ideal candidate will be responsible for developing state-of-the-art data visualization algorithms, implementing large-scale rendering capabilities, and exploring novel ways to interact with complex Computational Fluid...