AI/ML Distributed Training Engineer
Jobleads-US
Amazon Inc. in Cupertino is seeking a Software Engineer- AI/ML to help build and optimize AWS Neuron Training for large-scale model training on Trainium. You will work across PyTorch and Neuron to implement distributed training components and apply parallelism strategies.
You will profile workloads to locate bottlenecks and collaborate with compiler, runtime, and libraries teams to land fixes, while building internal tooling and tests that ensure reproducibility.
#J-18808-Ljbffr Jobleads-USVacancy posted 11 hours ago
Similar jobs that could be interesting for youBased on the AI/ML Distributed Training Engineer in Cupertino, CA vacancy
$140k - $215k
...world’s most advanced AI-native platform. We work on large scale distributed systems, processing almost... ...'ll work closely with engineering teams to expand test... ...:Experience testing AI/ML systems, LLM applications... ...development lifecycle, model training, evaluation, and...TrainingFull timeContract workWork experience placementWork at officeLocal areaCrowdStrike
Sunnyvale, CA1 day ago$192k - $278k
...strategy for the power grid distribution and power integrity team, accurately... ...'s degree in Electrical Engineering, Computer Engineering,... ...work to shape the future of AI/ML hardware acceleration. You will... ..., and relevant education or training. US: $192000 - $278000 (USD)...TrainingWorldwideGoogle
Sunnyvale, CA3 days ago$90k - $180k
...WorkflowsBuild agentic AI services (planning, tool... ...Ray (or similar) for distributed compute, batch/stream processing... ...microservices for training/inference, vector... ...to design reviews and engineering best practices. Mentor... ...services (preferably in AI/ML contexts). Proficiency...TrainingFull timeTemporary workPart timeWalmart
Sunnyvale, CA2 days ago$207k - $300k
...of solutions in specialized ML areas, optimize ML infrastructure... ....8 years of experience in Distributed Systems, Software Architecture... ...:Master’s degree or PhD in Engineering, Computer Science, or a related... ..., and relevant education or training. US: $207000 - $300000 (USD)...TrainingRemote workGoogle
Sunnyvale, CA4 days ago- ...Technologist, Private Cloud AI - Applied & Agentic... ...and partner with engineering to take POCs into scalable... ...operationalizing large‑scale AI/ML solutions on at least... ...sequence models),.Pre‑training, fine‑tuning,... ...mission‑critical, highly distributed, large‑scale platforms...TrainingFull timeWork experience placementWork at officeLocal areaImmediate start2 days per week
Hewlett Packard Enterprise
San Jose, CA2 days ago $106k - $205k
Title: Staff System Engineers Engineer, AI/MLAbout GlobalFoundries:GlobalFoundries is a leading full-... ...We are looking for a seasoned Staff AI/ML Systems Engineer to lead workload-driven... ...including recruitment, selection, training, utilization, promotion, compensation,...TrainingFull timeWork at officeLocal areaGlobalfoundries
Santa Clara, CA3 days ago$210k - $350k
...GEICO is seeking a Distinguished Engineer (AI Platforms and Applications)... ...engineering, data science, ML, product, and design organizations... ...expertise in large-scale distributed systems. Search and... ...work experience, education and training, the work location as well as...TrainingHourly payWork experience placementLocal area$150k - $300k
...Great Careers. The Service Engineering team is transitioning disparate... .... We are building an AI-powered CRM platform using agentic... ...capabilities across AI/ML, distributed systems, and operational excellence... ...experience, education and training, the work location as well...TrainingHourly payWork experience placementLocal areaFlexible hoursShift work$183k - $247.6k
...to shape the future of AI? Join the team building... ...advanced cloud for AI training and inference — where multi... ...innovation in AI/ML and HPC workloads. If you... ...hardware, and network engineers, supply chain specialists... ...networking, or large-scale distributed systems experience ~...TrainingLocal areaFlexible hoursAmazon
Cupertino, CA3 days ago$184k - $287.5k
...is seeking talented Deep Learning Scientists / AI Researchers / Machine Learning Engineers to join our rapidly growing AI Safety and Responsibility... ...Safety, Content Safety, Hallucinations, and ML Fairness).Model Pre-Training, Mid-Training, Post-Training: Develop datasets...TrainingFull timeWork experience placementNvidia
Santa Clara, CA2 days ago$215k - $250k
...available to any downstream query engine and use case (from traditional analytics to real-time AI / ML). We are a team of self-... ...data systems and globally distributed platforms that sit at the heart... ..., relevant certifications and training, business needs, market demands...TrainingOdd jobWork at officeLocal areaRemote workRelocationRelocation packageOnehouse
Sunnyvale, CAa month ago$124k - $195.5k
...unique position: we are developing AI-based products across multiple... ...for a talented Machine Learning Engineer to work on Product Security, Content Safety, ML Fairness and Robustness efforts for... ...Develop the datasets and models for training and evaluating models and end-to-...TrainingFull timeWork experience placementNvidia
Santa Clara, CA2 days ago$174k - $252k
...amongst available technologies. Engineer a sophisticated toolkit for the automated model training lifecycle, deploying AI agents to automate model... ...implement end-to-end (E2E) ML architectures for Ads... ...including information retrieval, distributed computing, large-scale...TrainingShift workGoogle
Mountain View, CA20 hours ago- ...and scientific discovery to powering AI and the technologies people rely on... ...are seeking a highly motivated AI Training Systems & Performance Engineering PhD Intern/Co-Op to join our team.... ...efficiency, and scalability across distributed multi-GPU environments. We will...TrainingFull timeSummer workInternshipSummer internshipWorldwide
AMD
Santa Clara, CA4 days ago - ...based MTIA Software Team is hiring a Software Engineer, Systems ML specializing in compilers and kernels. You will help develop the AI compiler stack, contribute to PyTorch core... ..., and advance deployment velocity for training and inference on MTIA platforms. #J-18808...Training
$160k - $198k
...artificial intelligence (“AI”) solutions, and other... ...a Senior AI Systems Engineer, you will architect, deploy... ...large-scale AI model training and inference. You will... ...services tailored for distributed AI model training and low... ...dedicated focus on AI/ML systems, high-...TrainingLocal areaWorldwideVisa sponsorshipArcher Aviation
San Jose, CA4 days ago$150k - $230k
...researchers and veteran systems engineers who share a vision for redefining the foundations of distributed computing. As AI workloads grow increasingly... ...distributed GPU training. You'll work at the intersection... ...networking (RDMA, InfiniBand) ML framework or runtime...TrainingClockwork.io
Palo Alto, CA21 days ago$174k - $252k
...Senior Software Engineer, Generative AI, Ads Safety Share Senior Software... ...software development, applied ML, autonomous systems, and... ...including information retrieval, distributed computing, large-scale... ...and relevant education or training. US: $174000 - $252000 (...TrainingShift work- ...Distinguished Technologist - Pre-Sales AI & Data Center Networking This... ...of innovation. Our Sales Engineering team empowers customers and... ...architecture and solution design for AI/ML Networking Infrastructure,... ..., work experience, education/training, and/or skill level. - United...TrainingFull timeWork experience placementLocal areaImmediate startRemote workWork from home
Hewlett Packard Enterprise
San Jose, CA4 days ago $197.8k - $271.98k
...combines analog, digital, AI, and software... ...are seeking a Staff AI Engineer in Time-Series & Sensor... ...represent ADI at major ML and signal-processing venues... ...hands on experience of training or fine-tuning one or more... ...using cloud or distributed systems (AWS, GCP, etc....TrainingPermanent employmentFull timeWork at officeDay shiftAnalog Devices
San Jose, CA3 days ago- ...Systems builds the world's largest AI chip, 56 times larger than GPUs. This... ...Cerebras to deliver industry-leading training and inference speeds; over 10 times... ...speed inference.Cerebras is seeking a Distributed Systems Security Engineer to help build out the next...Training
Cerebras Systems
Sunnyvale, CA3 days ago $133.5k - $183.5k
...a global leader in materials engineering solutions used to produce virtually... ...connect our world - like AI and IoT. If you want to push... ...by defining Design For Distribution (DFD) requirements and influencing... ...Supply Base.Provide advanced training and support to Packaging Engineer...TrainingFull timeApplied Materials
Santa Clara, CA3 days ago$183k - $247.6k
...the world's largest fleet of GPU-accelerated servers powering AI/ML training and inference at cloud scale. Our team defines the server... ...this is the role.We are seeking a Cloud Hardware Development Engineer to define server architectures based on workload demand, translate...TrainingLocal areaWorldwideFlexible hoursDay shiftAmazon
Cupertino, CA20 hours ago- ...Systems builds the world's largest AI chip, 56 times larger than... ...to deliver industry-leading training and inference speeds; over 10... ...test, validate, and stress distributed hardware systems during development... ...closely with hardware engineers to isolate and resolve system...Training
Cerebras Systems
Sunnyvale, CA1 day ago $184k - $287.5k
We are now looking for a Senior Robotics Research Engineer (Robotics & AI for Drug Discovery)! NVIDIA is at the forefront of the AI and robotics... ...learning frameworks such as PyTorch and JAX, as well as training deep learning models on GPU clusters.Significant experience...TrainingFull timeNvidia
Santa Clara, CA4 days ago$170k - $234k
...leader in materials science and engineering solutions that are at the... ...service is essential to advancing AI and accelerating the... ...materials science expertise with AI/ML, simulation, and data-driven modeling... ...for large-scale model training and simulations.Additional InformationTime...Training- ...build the backbone of Generative AI cloud at AWS? Do you want to... ...future of the cloud for AI training and inference? Want to do industry... ...and scalability in AI/ML and HPC workloads. You are... ...like you. The AWS Hardware Engineering team creates server designs for...TrainingInternshipLocal area
$275k - $382k
...center hardware spanning ML TPU, GPU, Networking,... ...s degree in Electrical Engineering, Computer Engineering,... ...across large, globally distributed engineering organizationsAs... ...volume production.The AI and Infrastructure team... ...relevant education or training. US: $275000 - $382000...TrainingWorldwideGoogle
Sunnyvale, CA1 day ago$132k - $189k
...abuse vectors such as malware distribution, cloaking, and policy... ...cross-functionally with Google engineers and product managers to identify... ...apply the latest advancement in AI/ML and develop scaled solutions... ...experience, and relevant education or training. US: $132000 - $189000 (USD)...TrainingGoogle
Sunnyvale, CA2 days ago$174k - $253k
...Review code developed by other engineers and provide feedback to... ...including information retrieval, distributed computing, large-scale system... ...fastest experience possible.The AI and Infrastructure team is... ...experience, and relevant education or training. US: $174000 - $253000 (USD)...TrainingWorldwideGoogle
Sunnyvale, CA20 hours ago
Do you want to receive more vacancies?
Subscribe and receive similar vacancies to AI/ML Distributed Training Engineer. Be the first to apply!
Related searches
- ai developer Cupertino, CA
- ai engineer Cupertino, CA
- computer vision machine learning engineer Cupertino, CA
- machine learning software engineer Cupertino, CA
- senior ml engineer Cupertino, CA
- machine learning engineer Cupertino, CA
- data engineer machine learning Cupertino, CA
- machine learning intern Cupertino, CA
- internship machine learning Cupertino, CA
- machine learning Cupertino, CA