Jobs in the United States

25+ jobs

Save this search

Added 35 mins ago
Principal Engineer - AI Systems & Performance Lab
SanDisk
Milpitas, CA
We are seeking an experienced and visionary Principal Engineer to join our organization in Milpitas, United States. This role is director‑equivalent in scope and impact, with a strong…
Added 35 mins ago
Staff VLSI CAD Engineer
SanDisk
Milpitas, CA
Senior VLSI CAD Engineer We are looking for an experienced Senior VLSI CAD Engineer to drive design environment management and automation for Product and TEG chip development. This role is…
Added 15 hours ago
Software Development Engineer - AI Workload Optimization
Advanced Micro Devices, Inc
Austin, TX
…technical challenges into strategic roadmaps and influence senior leadership, while also engaging deeply with key customers to solve their most critical performance…
Added 15 hours ago
Principal Distributed Systems Software Engineer, Collectives
Advanced Micro Devices, Inc
San Jose, CA
…systems. You will engage with internal hardware, GPU library, compiler, runtime and ROCm teams as well as open-source maintainers to ensure seamless integration of optimizations, applying…
Added 15 hours ago
Senior GPU Performance Engineer - AI Workloads
Advanced Micro Devices, Inc
Santa Clara, CA
We are seeking a GPU Performance Engineer to optimize AI workloads and guide the evolution of next-generation AMD Instinct GPU architectures. In this role, you will work across the software…
Added 16 hours ago
AS
Senior Software Engineer, Swarm Systems
APIUM SWARMING TECHNOLOGIES INC
San Diego, CA
- $113,965 to $178,135 USD / year
ASR seeks a software engineer to join our software development team. This role targets a developer capable of maintaining the existing code base while developing new features within our…
Added 17 hours ago
RC
Senior Software Engineer, Swarm Systems
RED CAT HOLDINGS
San Diego, CA
- $113,965 to $178,135 USD / year
ASR seeks a software engineer to join our software development team. This role targets a developer capable of maintaining the existing code base while developing new features within our…
Added 1 day ago
Sr. Software Engineer- AI/ML, AWS Neuron Distributed Training - Performance Optimization
Amazon
Seattle, WA
learning accelerators and the Trn3/Trn2/Trn1 and Inf2/Inf1 servers that use them. This role is for a software engineer in the Distributed Training team for AWS Neuron. This role is…
Added 1 day ago
ASIC DFT Engineer - Acacia (Hybrid)
Cisco Systems, Inc.
San Jose, CA
We are looking for a Design-for-Testability (DFT) Engineer to support DFT implementation, validation, and signoff for complex ASIC / SoC designs. The ideal candidate will have hands-on…
Added 1 day ago
Senior Software Engineer, ML API – AI Inference
Cerebras Systems
Sunnyvale, CA
We are hiring a Senior Software Engineer to build and evolve the ML API layer that makes this heterogeneous serving system accessible, reliable, and easy to use. You will own significant…
Added 1 day ago
Compiler Engineer - New Grad
SambaNova
Austin, TX · San Jose, CA
As a new grad Compiler Engineer, you'll work on core areas of our compiler, including the IR and pass infrastructure, model lowering and partitioning, tensor tiling, memory management, and…
Added 1 day ago
Senior FPGA Engineer [644]
D-Wave Quantum
Remote, United States
D-Wave is seeking a Senior FPGA Engineer to join our Quantum Systems and Technologies team. In this role, you will design and implement the signal processing and real-time control systems…
Added 2 days ago
IC
Senior Infrastructure and DevOps Systems Engineer
Intel Corporation
OR · MA · CA · AZ
- $100,550 to $194,270 USD / year
Our customer-facing engineering team is seeking a senior, hands-on Infrastructure and DevOps Systems Engineer to design, deploy, automate, integrate, secure, and troubleshoot large-scale…
Added 2 days ago
Senior Software Engineer, Performance, SOC Architect
Google
Sunnyvale, CA
Share Senior Software Engineer, Performance, SOC Architect
Added 2 days ago
TI
Senior Machine Learning Engineer
TETRAMEM INC
San Jose, CA
- $200,000 to $280,000 USD / year
Understanding of ML compiler and runtime design.
Added 2 days ago
Software Engineer I – Compiler & Runtime
TETRAMEM INC
San Jose, CA
Develop, maintain, and optimize software components within TetraMem’s compiler, runtime, and SDK toolchain for analog compute-in-memory (CIM) hardware.
Added 2 days ago
Senior/Principal Physical Design Engineer (remote)
Chelsea Search Group
Remote, United States
We are seeking an experienced Physical Design / P&R Engineer to take a key role in delivering complex, high-performance ASICs from RTL/netlist through GDSII and tapeout. You’ll own critical…
Added 2 days ago
Senior/Principal Physical Design Engineer (remote)
Chelsea Search Group
Remote, United States
We are seeking an experienced Physical Design / P&R Engineer to take a key role in delivering complex, high-performance ASICs from RTL/netlist through GDSII and tapeout. You’ll own critical…
Added 2 days ago
Senior/Principal Physical Design Engineer (remote)
Chelsea Search Group
Remote, United States
We are seeking an experienced Physical Design / P&R Engineer to take a key role in delivering complex, high-performance ASICs from RTL/netlist through GDSII and tapeout. You’ll own critical…
Added 2 days ago
Senior/Principal Physical Design Engineer (remote)
Chelsea Search Group
Remote, United States
We are seeking an experienced Physical Design / P&R Engineer to take a key role in delivering complex, high-performance ASICs from RTL/netlist through GDSII and tapeout. You’ll own critical…
Added 2 days ago
Senior/Principal Physical Design Engineer (remote)
Chelsea Search Group
Remote, United States
We are seeking an experienced Physical Design / P&R Engineer to take a key role in delivering complex, high-performance ASICs from RTL/netlist through GDSII and tapeout. You’ll own critical…
Added 2 days ago
Senior/Principal Physical Design Engineer (remote)
Chelsea Search Group
Remote, United States
We are seeking an experienced Physical Design / P&R Engineer to take a key role in delivering complex, high-performance ASICs from RTL/netlist through GDSII and tapeout. You’ll own critical…
Added 2 days ago
Senior/Principal Physical Design Engineer (remote)
Chelsea Search Group
Remote, United States
We are seeking an experienced Physical Design / P&R Engineer to take a key role in delivering complex, high-performance ASICs from RTL/netlist through GDSII and tapeout. You’ll own critical…
Added 2 days ago
Senior/Principal Physical Design Engineer (remote)
Chelsea Search Group
Remote, United States
We are seeking an experienced Physical Design / P&R Engineer to take a key role in delivering complex, high-performance ASICs from RTL/netlist through GDSII and tapeout. You’ll own critical…
Added 2 days ago
Senior/Principal Physical Design Engineer (remote)
Chelsea Search Group
Remote, United States
We are seeking an experienced Physical Design / P&R Engineer to take a key role in delivering complex, high-performance ASICs from RTL/netlist through GDSII and tapeout. You’ll own critical…

Amazon

Sr. Software Engineer- AI/ML, AWS Neuron Distributed Training - Performance Optimization

Seattle, WA, United States

Apply About employer

Applications open in a new tab.

Job details

Posting language: English
Posted: May 23, 2025

Original job description

Description

Annapurna Labs designs silicon and software that accelerates innovation. Customers choose us to create cloud solutions that solve challenges that were unimaginable a short time ago - even yesterday. Our custom chips, accelerators, and software stacks enable us to take on technical challenges that have never been seen before, and deliver results that help our customers change the world.

AWS Neuron is the complete software stack for the AWS Trainium and Inferentia cloud-scale machine

learning accelerators and the Trn3/Trn2/Trn1 and Inf2/Inf1 servers that use them. This role is for a software engineer in the Distributed Training team for AWS Neuron. This role is responsible for development, enablement and performance tuning of a wide variety of ML model families, including massive scale multi-modal large language models like Llama, Qwen, gpt-oss, DeepSeek and beyond, as well as multi-modal generation models such as Stable Diffusion, Flux, WAN, and many more.

The Distributed Training team works side by side with chip architects, compiler engineers and runtime engineers to create, build and tune distributed training solutions with AWS Trainium, maximize training throughput, minimize time-to-convergence, and push the boundaries of training efficiency on Trainium. You will identify and resolve performance bottlenecks across the stack, from collective communications and memory utilization to compiler optimizations and kernel performance.

Key job responsibilities

This role will lead efforts to optimize distributed training performance on Trainium, with a primary focus on maximizing training throughput, model flops utilization, and efficiency across the Neuron software stack. You will work across PyTorch, JAX, and the Neuron compiler and runtime to enable and tune large-scale training workloads on the latest Trainium instances.

About the team

Our team is dedicated to supporting new members. We have a broad mix of experience levels and tenures, and we’re building an environment that celebrates knowledge-sharing and mentorship. Our senior members enjoy one-on-one mentoring and thorough, but kind, code reviews. We care about your career growth and strive to assign projects that help our team members develop your engineering expertise so you feel empowered to take on more complex tasks in the future.

Basic Qualifications

- 5+ years of non-internship professional software development experience
- 5+ years of programming with at least one software programming language experience
- 5+ years of leading design or architecture (design patterns, reliability and scaling) of new and existing systems experience
- 5+ years of full software development life cycle, including coding standards, code reviews, source control management, build processes, testing, and operations experience
- Experience as a mentor, tech lead or leading an engineering team

Preferred Qualifications

- Bachelor's degree in computer science or equivalent
- Machine Learning knowledge in frameworks and end to end model training.

Amazon is an equal opportunity employer and does not discriminate on the basis of protected veteran status, disability, or other legally protected status.

Our inclusive culture empowers Amazonians to deliver the best results for our customers. If you have a disability and need a workplace accommodation or adjustment during the application and hiring process, including support for the interview or onboarding process, please visit https://amazon.jobs/content/en/how-we-hire/accommodations for more information. If the country/region you’re applying in isn’t listed, please contact your Recruiting Partner.

The base salary range for this position is listed below. Your Amazon package will include sign-on payments and restricted stock units (RSUs). Final compensation will be determined based on factors including experience, qualifications, and location. Amazon also offers comprehensive benefits including health insurance (medical, dental, vision, prescription, Basic Life & AD&D insurance and option for Supplemental life plans, EAP, Mental Health Support, Medical Advice Line, Flexible Spending Accounts, Adoption and Surrogacy Reimbursement coverage), 401(k) matching, paid time off, and parental leave. Learn more about our benefits at https://amazon.jobs/en/benefits .

USA, WA, Seattle - 168,100.00 - 227,400.00 USD annually

Apply About employer

More about this job

Skills

ML
AWS
WAN
PyTorch
Coaching
Assign homework
Software development
Computer science
Machine learning
Recruitment
Mental health

Report this job