Senior AI/ML Systems Engineer — Distributed Training
Amazon Inc.
Amazon in Cupertino, CA is seeking a Senior Software Engineer- AI/ML to help build and optimize distributed training for AWS Neuron on Trainium. You will work across PyTorch and JAX, shaping parallelism strategies and high-performance kernels, collaborating with compiler, runtime, and open source frameworks. The role blends ML, HPC, and distributed systems, with mentoring responsibilities and opportunities to influence future accelerator architecture. #J-18808-Ljbffr Amazon Inc.
Vacancy posted 3 days ago
Similar jobs that could be interesting for youBased on the Senior AI/ML Systems Engineer — Distributed Training in Cupertino, CA vacancy
$174.9k - $261.3k
...world!The Data Labeling Engineering team designs, builds,... ...data engineering, and AI/ML, defining the... ...that create reliable training data at scale. Our tools... ...and direct impact on systems that unblock the next... ...experience building robust distributed platforms and applications...SeniorTrainingFull timeLocal areaRemote workWork from homeRelocation packageFlexible hoursGeneral Motors
Sunnyvale, CA4 hours ago- General Motors is seeking a Senior Engineer for Embodied AI Scaling Foundations to measure and visualize AV model performance at scale. You will design... ...with modeling and data teams to connect signals to data, training, and launch decisions. You will help deliver end-to-end...SeniorTraining
$153.2k - $234.1k
...hardware and battery systems to intuitive design,... ...driving? Join the Embodied AI team at General... ...world scenarios. As a Senior ML Infra Engineer, you will work on the... ...dataset generation, training, evaluation and iteration... ...on large-scale distributed systems, applications...SeniorTrainingFull timeLocal areaRemote workWork from homeRelocation packageFlexible hoursGeneral Motors
Sunnyvale, CA2 days ago$130.7k - $261.3k
...scientists.THE OPPORTUNITYThis Senior Staff ML Ops Engineer position can work out of... ...Devices Digital (MDD) AI initiatives, driving the architecture... ..., feature engineering, training, validation, deployment,... ...architecture and distributed systems.Experience in reviewing and...SeniorTrainingShift workAbbott
Santa Clara, CA3 days ago$144.7k - $261.3k
...team, part of Embodied AI’s Scaling Foundations... ...development. We engineer high-performance tools... ...with data-intensive ML teams to drive rapid... ...generation autonomous systems. About the Role As a Senior Engineer in the Embodied... ...signals back to data, training, and launch decisions...SeniorTrainingLocal areaRemote workWork from homeFlexible hours- Wayve is building the leading AI platform for autonomous driving with a mapless, hardware... .... This role focuses on performance engineering across ML systems, GPU compute and distributed platforms, shaping optimisations for training and cloud inference. You will profile...SeniorTraining
- Mind Robotics is hiring a Model Systems Engineer in Palo Alto to build core systems enabling fast, reliable training of large ML models and scalable production deployment. You will design distributed training stacks, optimize memory and compute usage, and partner with modeling...Training
$200k - $400k
...ultra-scale GPU supercomputing systems to train next-generation foundation... ...communication performance, distributed reliability, and cross-layer... ...looking for a deeply technical engineer to co-design and optimize... ...use artificial intelligence (AI) tools to support parts of...SeniorTrainingVisa sponsorshipInstitute of Foundation Models
Sunnyvale, CA2 days ago$193.3k - $261.5k
...is looking for a Senior Software Development Engineer to own the... ...scale with large distributed models.This is a... ...kernels for a custom ML accelerator... ...C++- Strong Linux systems knowledge- Experience... ...architecture, training/inference... ...early adopter of AI-assisted development...SeniorTrainingLocal areaFlexible hoursAmazon
Cupertino, CA3 days ago- Institute of Foundation Models in Sunnyvale, CA seeks a deeply technical engineer to co-design and optimize the communication stack for large-scale distributed training, including hybrid parallelism and Mixture-of-Experts workloads. This role emphasizes performance engineering...SeniorTraining
- ...scientific discovery to powering AI and the technologies people... .... THE ROLE: We are hiring ML Systems Research Engineers to build the reinforcement... ..., metareasoning, and post-training methods for engineering... ...resource usage must be balanced. Distributed experimentation, job...Training
$144.7k - $261.3k
...infrastructure, and ML/AI GPU platforms for AV... ...GM is looking for a Senior Performance Engineer to join the AV Capacity... ...GM’s long-term GPU system strategy and "... ...reliability of large-scale ML training and inference... ...within large-scale distributed production environments...SeniorTrainingFull timeWork at officeLocal areaRemote workWork from homeFlexible hours3 days per weekGeneral Motors
Sunnyvale, CA3 days ago- ...areMoveworks is the Agentic AI Assistant platform... ...all of their business systems through natural... ...Moveworks’ Reasoning Engine and natural language capabilities... ...build cutting edge ML infrastructure for... ...responsibilities including distributed training and inference pipeline...SeniorTrainingPermanent employmentWork at officeRemote workFlexible hours
Moveworks
Mountain View, CA1 day ago - Cerebras Systems builds the world's largest AI chip, 56 times larger than GPUs. This architecture... ...Cerebras to deliver industry-leading training and inference speeds; over 10... ....Cerebras is seeking a Distributed Systems Security Engineer to help build out the next generation...Training
Cerebras Systems
Sunnyvale, CA5 days ago $174.72k - $295.68k
...integrating advanced AI and autonomous... ...time Machine Learning Engineer - AI Foundation, with... ...establishing a state-of-art ML infrastructure for training very large... ...performing large scale distributed training of models.A... ...with edge computing systems.Knowledge of disributed...SeniorTrainingFull timeXPENG Motors
Santa Clara, CA15 hours ago$150k - $200k
.../HybridPlusAI is a Physical AI company pioneering AI-based... ...highly skilled Machine Learning Engineer with deep expertise in... ...for deployment in production systems.Ensure that your work is performed... ....Experience with distributed training, high-performance computing,...SeniorTrainingFull timePlus.ai
Santa Clara, CA1 day ago- ...Silicon Valley Physical AI research team of... ...role Build the systems that make... ...experimentation. As our ML Systems owner, you... ...that foundation: a training platform that shortens... ...Performance engineering: profile compute,... ...time Responsible distributed execution: support...Training
$193.3k - $261.5k
...for AWS's custom ML accelerators. Working... ...boundary, our engineers craft high-performance... ...'s possible in AI acceleration.The AWS... ...ML inference and training performance.As part... ...computing, and distributed architectures, where... ...machine learning systems, you'll bring expertise...SeniorTrainingInternshipLocal areaWork from homeFlexible hoursAmazon
Cupertino, CA15 hours ago$173k - $253k
Matterport - Senior ML Ops Engineer Job Description CoStar Group is a leading... ...model architectures, training procedures, and deployment requirements... ...acceleration, and efficient AI.Contribute to the continuous... ...with version control systems (e.g., Git) and agile...SeniorTrainingFull timeWork at officeWork from homeMatterport
Sunnyvale, CA3 days ago$244.14k - $413.16k
...integrating advanced AI and autonomous... ...global driving. As a Senior Staff Machine Learning Engineer, you will architect... ...simulations for closed-loop training and evaluation.... ...PyTorch and expertise in distributed training (DeepSpeed,... ...to support your ML model development/research...SeniorTrainingFull timeOverseasXPENG Motors
Santa Clara, CA4 days ago- Apple Inc. is seeking a Senior Machine Learning Engineer to shape the Siri platform's next generation of... .... You will collaborate with ML researchers and product engineers to... ...translate research into deployable systems, build training/inference pipelines, and integrate...SeniorTraining
$130k - $220k
Santa Clara, CASoftware Engineering - Motion Planning /Full-time /HybridPlusAI is a Physical AI company pioneering AI-based virtual... ...environment construction, training, and modeling.Contribute to Reinforcement... ...to facilitate large-scale distributed trainingTrack model training,...SeniorTrainingFull timePlus.ai
Santa Clara, CA15 hours ago$174.72k - $295.68k
..., integrating advanced AI and autonomous driving... ...-time Machine Learning Engineer / Research Scientist to... ...end autonomous driving systems.You will work closely with... ...experts to design, train, and deploy large-scale... ...thousands of GPUs using distributed training frameworks (FSDP...SeniorTrainingFull timeXPENG Motors
Santa Clara, CA4 days ago- Moveworks is seeking a Senior Staff Machine Learning Engineer to design, build, and scale ML infrastructure powering LLMs and AI agents for enterprise customers. The role emphasizes... ..., performance, and reliability across training, evaluation, and deployment. You will...SeniorTraining
$174.72k - $295.68k
...integrating advanced AI and autonomous driving... ...seeking Machine Learning Engineers with strong expertise... ...-scale deep learning systems, along with solid software... ...learned simulator for training and evaluating driving... ...-on experience with distributed training (FSDP,...SeniorTrainingFull timeXPENG Motors
Santa Clara, CA15 hours ago$170.6k - $261.3k
...for a mathematically rigorous engineer to own model numerics: how... ...the same machinery to out-of-distribution inputs, and turn it into... ...engineers who are not you. Build training dynamics observability .... ...formulation for AV or ADAS systems. Published or applied work...SeniorTrainingLocal areaWork from homeFlexible hours$130k - $220k
...Clara, CASoftware Engineering - Motion Planning /... ...HybridPlusAI is a Physical AI company pioneering... ...engineer to train and deploy the latest generation of ML-based planning algorithms... ..., mapping, and systems teams to deploy... ...Experience with large-scale distributed training or cloud-...SeniorTrainingFull timePlus.ai
Santa Clara, CA15 hours ago$184k - $287.5k
NVIDIA is seeking a Senior MLOps Engineer to join our DSX Enablement... ...groundbreaking AI workloads. We partner... ...experts on the demands of ML workloads, help... ...full-stack AI and ML system problems, and work on... ...Partners (NCPs), including distributed training, inference...SeniorTrainingFull timeRemote workNvidia
Santa Clara, CA3 days ago$170k - $240.8k
...delivering-driven expert in ML Training Infrastructure with a... ...and high-performance AI/ML platform... ...development initiatives. As a Senior ML Engineer, you will collaborate... ....Raise the bar on system observability,... ...optimization solutions to scale distributed training workflows...SeniorTrainingFull timeLocal areaWork from homeRelocation packageFlexible hoursGeneral Motors
Sunnyvale, CA4 hours ago$160k - $200k
Santa Clara, CAData Engineering - ML Infrastructure /... ...is a Physical AI company pioneering... ...growing teams.As a Senior ML Infrastructure... ...for both training and inference phases... ...model versioning systems and experiment tracking... ....Implement distributed systems and storage...SeniorTrainingFull timePlus.ai
Santa Clara, CA4 days ago
Do you want to receive more vacancies?
Subscribe and receive similar vacancies to Senior AI/ML Systems Engineer — Distributed Training. Be the first to apply!
Related searches
- ai developer Cupertino, CA
- ai engineer Cupertino, CA
- machine learning engineer Cupertino, CA
- senior windows systems engineer Cupertino, CA
- system performance engineer Cupertino, CA
- software system engineer Cupertino, CA
- electronic systems engineer Cupertino, CA
- computer system validation engineer Cupertino, CA
- senior staff systems engineer Cupertino, CA
- operations support system engineer Cupertino, CA