AI/ML Software Engineer - Trainium Distributed Training
Amazon Locker
Annapurna Labs (AWS) is hiring engineers to optimize distributed training for Trainium, focusing on throughput, FLOPs utilization, and convergence time. You’ll work across PyTorch, JAX, and the Neuron stack to enable and tune large-scale workloads, identify missing operators, and design effective parallelism strategies.
You will mentor a team of engineers, collaborate with customers, and contribute to open source ecosystems, shaping performance across the stack from frameworks to kernels.
#J-18808-LjbffrVacancy posted 1 day ago
Similar jobs that could be interesting for youBased on the AI/ML Software Engineer - Trainium Distributed Training in Seattle, WA vacancy
$165.2k - $223.6k
...Labs designs silicon and software that accelerates innovation... ...software stack for the AWS Trainium and Inferentia cloud-scale... ...role is for a software engineer in the Distributed Training team for AWS Neuron. This... ...tuning of a wide variety of ML model families, including...TrainingInternshipFlexible hoursAmazon
Seattle, WA8 days ago- ...Veeda AI is seeking a highly skilled engineer to advance distributed training and high-performance inference for multimodal world models. You will own multi-node throughput, profile GPU utilization, and optimize kernels and CUDA/Triton code across NVLink fabrics. Candidates...Training
Veeda AI
Seattle, WA4 days ago $142.3k - $263.3k
...ML Software EngineerOur team builds the ML-inference stack... ...powers generative AI for Apple Intelligence... ...Silicon in the datacenter, distributing work across on-SoC... ...the team you will help engineer continuous improvements... ...models (you won't be training them).Solid grasp of concurrency...TrainingRelocationApple
Seattle, WA20 hours ago- ...Xaira Therapeutics is seeking a Senior Software Engineer to join our Platform team to design, build, and deploy the AI infrastructure that powers our world-class research... ...and engineers to enable thousands of GPUs for training and inference of biology foundation models in...Training
Xaira Therapeutics, Inc.
Seattle, WA2 days ago $191k - $253k
...powered by Lattice OS, an AI-powered operating system that... ....ABOUT THE TEAMAnduril's Software Platform organization is seeking a Software Engineer to join the Data Distribution team. The Data Distribution... ...experience, education and/or training, critical skills, and/or...TrainingFull timeWork experience placementImmediate startAnduril Industries
Seattle, WA8 days ago$157.7k - $213.8k
...world's best data and AI infrastructure platform... ...business. Founded by engineers — and customer obsessed... ...query engines. As a software engineer on the Runtime... ...building the next generation distributed data storage and... ...certifications and training, and specific work location...TrainingLocal areaWorldwideDataBricks
Bellevue, WA8 days ago$165.2k - $223.6k
...of AWS innovation. Trainium delivers best-in-class ML training performance with the... ...by the AWS Neuron Software Development Kit (SDK... ...'s most talented engineers. Our team covers multiple... ...Machine Learning, AI accelerators, or... ..., embedded or distributed software applications...TrainingInternshipLocal areaRelocationFlexible hoursDay shiftAmazon
Seattle, WA7 days ago$150.4k - $277.6k
...Senior Software ML Engineer - Watch Software Cupertino, California, United States Machine Learning and AI Our team is working toward a future where our devices are aware of... ...who can work across disciplines — from training advanced models to rapid on-device prototyping...TrainingLive inWorldwideRelocationApple Inc.
Seattle, WA4 days ago$146.83k - $192.72k
...company for yoga, running, training, and other athletic... ...The Enterprise Data & AI team is a strategic... ...responsibilities As an AI/ML Engineer, you will contribute... ...pipelines using distributed training frameworks (PyTorch... ...best-practice software engineering including...TrainingPermanent employmentFull timePart timeWork visaLululemon Athletica
Seattle, WA5 days ago$320k
...Staff+ Software Engineer, Infrastructure (Distributed Systems)San Francisco, CA | New York City,... ...interpretable, and steerable AI systems. We want AI to be... ...distributed systems that train, serve, and secure our AI... ..., such as GPUs, TPUs, or Trainium, and associated...TrainingWork at officeVisa sponsorshipFlexible hoursAnthropic
Seattle, WA4 days ago$168.1k - $227.4k
...Neuron is the complete software stack for AWS Inferentia and Trainium, AWS purpose-built... ...senior software engineering role is part of... ...parallel computation, distributed KV cache,... ...their architecture, training and inference lifecycles... ..., TPUs, or other AI accelerator...TrainingWork experience placementInternshipLocal areaFlexible hoursAmazon
Seattle, WA9 days ago$165.2k - $223.6k
...AWS Neuron, the software development kit... ...on AWS Trainium, Amazon's custom... ...Neuron includes an ML compiler, runtime... ...customers can train frontier-scale... ...their stack.The Distributed Training team is... ...boundary, our engineers build the infrastructure... ...direction of AI acceleration...TrainingInternshipLocal areaWork from homeFlexible hoursAmazon
Seattle, WA8 days ago$143.7k - $194.4k
...Neuron is the complete software stack for AWS Inferentia and Trainium — AWS purpose-built... ...Neuron chips.As an engineer on this team, you'... ...quantization, and distributed inference on Neuron... ...architecture, training/inference lifecycles... ...Neuron, TPU or other AI acceleration hardware...TrainingInternshipFlexible hoursAmazon
Seattle, WA7 days ago$143.7k - $194.4k
...AWS Neuron, the software development kit... ...Inferentia and Trainium.The AWS Neuron... ...Inferentia and Trainium ML accelerators.... ...inference and training performance.The... ...boundary, our engineers build... ...'s possible in AI acceleration.As... ...computing, and distributed architectures,...TrainingWork experience placementInternshipFlexible hoursAmazon
Seattle, WA5 days ago$168.1k - $227.4k
...Neuron is the complete software stack for the AWS Inferentia and Trainium cloud-scale... ...Software Development Engineer for the Neuron... ...applications and AI accelerators. You... ...Improving performance of ML Kernels and ML... ...build massive-scale distributed training and inference solutions...TrainingInternshipWork from homeFlexible hoursAmazon
Seattle, WA8 days ago$168.1k - $227.4k
Shape the Future of AI Accelerators at... ...Neuron, the software development kit... ...Inferentia and Trainium.As a Senior Software Engineer on our Machine Learning... ...computing, and distributed architectures,... ...programming and ML knowledge are... ...transformer architecture, training/inference...TrainingInternshipWork from homeFlexible hoursAmazon
Seattle, WA8 days ago$143.7k - $194.4k
At AWS SageMaker AI, we're making it... ...platform for building, training, customizing, and... ...data prep, distributed training, evaluation... ...of GPU and AWS Trainium devices, Model Factory... ...re looking for a Software Development Engineer to help design,... ...closely with the ML scientists and...TrainingContract workInternshipFlexible hoursAmazon
Bellevue, WA6 days ago$143.7k - $194.4k
...are looking for a Software Development Engineer II (SDE-2) to join... ...across Amazon Linux distributions, implementing automated... ...focus on enabling AI and ML workloads by... ...services (SageMaker, Trainium, Inferentia), and open... ...to cutting edge AI training and inference.Mentorship...TrainingInternshipWork from homeFlexible hoursAmazon
Seattle, WA8 days ago$191k - $253k
...powered by Lattice OS, an AI-powered operating... ...years.ABOUT THE TEAM The Software Integration... ...internally by Anduril engineers and externally by our... ...yours.Experience with distributed simulation and interoperability... ...experience, education and/or training, critical skills, and/...TrainingFull timeWork experience placementImmediate startAnduril Industries
Seattle, WA7 days ago$197.3k - $313.7k
...SalesforceSalesforce is the #1 AI CRM, where humans with... ...built on PHP/Hack.The Distributed Data Services TeamWe... ...the judgment to weigh engineering vs. product trade-offs... ...and releasing software with Docker containersKubernetesExperience... ...promotion, benefits, training, assessment of job...TrainingPermanent employmentFull timeSalesforce
Seattle, WA5 days ago$117.2k - $313.7k
...DetailsAbout SalesforceSalesforce is the #1 AI CRM, where humans with agents drive... ...you are the future of Salesforce.Distributed Systems Software Engineer - Public Cloud (Senior/Lead/... ...compensation, promotion, benefits, training, assessment of job performance, discipline...TrainingFull timeSalesforce
Bellevue, WA8 days ago$182k - $242k
...The Essential Cloud for AI™. Built for pioneers by... ...to successfully training self-improving agents.... ...’re looking for strong engineers with great taste. The most... ...experience with modern ML frameworks such as PyTorch... ...learning Experience with distributed training, GPU...TrainingFull timeTemporary workCasual workWork at officeFlexible hoursCoreWeave
Bellevue, WA4 days ago$175k - $308.5k
...Machine Learning and AI Description... ...understanding. This includes training and optimizing... ...into production software systems. You... ...with large‑scale distributed training and model... ...curation and pipeline engineering Experience with... ...integrating ML models into performance...TrainingRelocationApple Inc.
Seattle, WA4 days ago$125.7k - $334.6k
...Distributed Systems Software Engineer - Public Cloud (Mid/Senior/Lead/Principal) Get AI-powered advice on this job and more exclusive features. Pay found in job post Retrieved... ..., compensation, promotion, benefits, training, assessment of job performance, discipline,...TrainingFull timeSalesforce
Bellevue, WA21 hours ago- Summary As a Senior Software Engineer in Apple AI and Data Governance at Apple, you will be instrumental... ...across Apple's machine learning training and post-training lifecycle. Architect... ...identity. Develop backend services and distributed applications in Java, Scala, Kotlin,...Training
- ...is where embodied AI meets real robots,... ...combining rigorous engineering with learning... ...are seeking a Staff ML Systems Engineer to... ...architect and build the distributed infrastructure... ...processing, model training, evaluation, and... ...Our teams span AI, software, robotics engineering...TrainingLocal area
- ...envision a future where AI fundamentally reshapes... ...seeking a Staff Research Engineer with a strong track... ...algorithms for LLM post-training to advance personalized... ...professional experience. Strong ML and Generative Modeling... .... Familiarity with distributed training. Proficiency...Training
Apple
Seattle, WA3 days ago $172.5k - $260.1k
...Job Category Software Engineering Job Details About Salesforce Salesforce is the #1 AI CRM, where humans with agents drive customer... ...U.S. Government CAC/PIV. Distributed Systems Software Engineer - GovCloud... ..., promotion, benefits, training, assessment of job performance...TrainingLocal area100 Salesforce, Inc.
Bellevue, WA2 days ago$168.1k - $227.4k
...that powers the next generation of AI workloads? As a Senior Software Development Engineer on the EC2 UltraServer... ...provisioning and availability of AWS Trainium-based AI servers. You will architect... ...and datacenter teams to manage AI/ML infrastructure at scale.Key job responsibilities...InternshipFlexible hoursShift workDay shiftAmazonWebServices
Seattle, WA7 days ago$159.8k - $235k
...abstractions that DoorDash Engineering depends on: reliable,... .... Within Storage, the Distributed Caching team owns every... ..., including Taulu, ML Platform, and product engineering... ...proficiency in using AI coding tools (e.g.,... ..., Cursor) in the full software development lifecycle,...Hourly payFull timeWork at officeLocal areaRemote workFlexible hoursDoordash
Seattle, WA5 days ago
Do you want to receive more vacancies?
Subscribe and receive similar vacancies to AI/ML Software Engineer - Trainium Distributed Training. Be the first to apply!
Related searches
- machine learning ai engineer Seattle, WA
- ai developer Seattle, WA
- ai engineer remote Seattle, WA
- ai ml engineer Seattle, WA
- ai prompt engineer Seattle, WA
- senior ai engineer Seattle, WA
- ai engineer Seattle, WA
- senior ml engineer Seattle, WA
- data scientist machine learning engineer Seattle, WA
- machine learning software engineer Seattle, WA