Amazon
Posted 2mo ago

Software Dev Engineer, Machine Learning Compilers

Amazon
Sunnyvale, California, United States
$165k-$224k/yrOnsiteFull Time
Responsibilities
  • designing software
  • developing compilers
  • optimizing performance
Requirements
  • 3+ years professional software development
  • 2+ years system design/architecture
  • Proficiency in Java/C++/C#
  • Experience deploying LLMs on GPUs/Neuron/TPU or other AI acceleration hardware
  • Bachelor's degree preferred
  • Embedded/C++ and compiler experience preferred
Technical tools mentioned
JavaC++C#GPUNeuronTPU

Job description

Description

Amazon Devices is an inventive research and development company that designs and engineers high-profile consumer products like the Kindle family, Fire Tablets, Fire TV, Health & Wellness devices, Amazon Echo, and Astro. We are building the next generation of edge AI capabilities through our advanced compression platform, compiler and custom neural accelerator silicon. Come join us to accelerate deep learning networks on edge processors and beyond.

We are looking for a talented and passionate software engineer to be part of an exciting technology creation team at Amazon. You will have an enormous opportunity to make a large impact on the design, architecture, and implementation of deep learning technologies embedded into consumer products used every day, by people you know. The position provides an unique opportunity to contribute and make an impact from hardware design stage followed by pre and post silicon development as well as productizing it on consumer devices.

In this role you will be work along side partner science teams to develop the compiler infrastructure and lower deep learning workloads to heterogeneous device backends. You will also partner up with peer science teams to innovate on model quantization and compression techniques for efficient execution on hardware.

Key job responsibilities
Design and develop software stack for deep learning accelerator
Develop Compiler passes for graph ingestions, optimizations and partitioning.
Develop backend code generation capabilities across heterogeneous platforms
Profile, analyze and optimize system level performance, develop new tooling where necessary
Participate in design reviews, API development, and documentation
Successfully collaborate with hardware, software, applied science and product teams to onboard more and more user experiences to be powered by Deep Learning accelerator.
Mentor and provide guidance to junior engineers

A day in the life
You join a small team building the compiler that brings large AI models to a new generation of custom silicon. The chip has a fraction of the memory of a phone, and the compiler is what makes language models run on it at all. The team is small enough that each engineer owns a meaningful piece of the system end to end. There is no layer between you and the problem.
The morning starts with results from an overnight run. A piece of the compiler you own just produced its tightest result yet on a real model. You ship the change for hardware validation.
You spend the afternoon directing AI agents through the codebase, reviewing their changes, and steering the design.
Before lunch, you load your compiled model onto the chip and run it through a demo app you wrote yourself, watching tokens stream out of silicon you helped make work. Later, you meet with the research team. They depend on your component. You sketch a cleaner interface together.

About the team
We sit at the intersection of AI models and custom silicon, and our work decides what is possible at the edge.
Engineers here bring deep experience across compilers and program analysis, optimization algorithms, computer architecture, machine learning systems, and the practical craft of getting large software to run reliably under tight constraints. People have shipped production code generators, tuned schedulers for novel hardware, and worked at every layer from the model down to the bare metal.
Because the team is small, you work alongside that experience daily, not at a distance. You partner directly with researchers shaping the models, hardware engineers shaping the silicon, and firmware engineers shaping the runtime. You learn how each layer constrains and unlocks the others, and you see your decisions land end to end.
This is a place to build technical depth quickly and own work that matters from day one.

Basic Qualifications

- 2+ years of non-internship design or architecture (design patterns, reliability and scaling) of new and existing systems experience
- 3+ years of non-internship professional software development experience
- 3+ years of programming using a modern programming language such as Java, C++, or C#, including object-oriented design experience
- Experience in developing and deploying LLMs in production on GPUs, Neuron, TPU or other AI acceleration hardware

Preferred Qualifications

- 3+ years of full software development life cycle, including coding standards, code reviews, source control management, build processes, testing, and operations experience
- Bachelor's degree in computer science or equivalent
- Experience in embedded development in C/C++
- Experience building compiler for application specific accelerators or custom instruction set

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

Los Angeles County applicants: Job duties for this position include: work safely and cooperatively with other employees, supervisors, and staff; adhere to standards of excellence despite stressful conditions; communicate effectively and respectfully with employees, supervisors, and staff to ensure exceptional customer service; and follow all federal, state, and local laws and Company policies. Criminal history may have a direct, adverse, and negative relationship with some of the material job duties of this position. These include the duties and responsibilities listed above, as well as the abilities to adhere to company policies, exercise sound judgment, effectively manage stress and work safely and respectfully with others, exhibit trustworthiness and professionalism, and safeguard business operations and the Company’s reputation. Pursuant to the Los Angeles County Fair Chance Ordinance, we will consider for employment qualified applicants with arrest and conviction records.

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, CA, Sunnyvale - 165,200.00 - 223,600.00 USD annually

About Amazon

Multinational technology focused on e-commerce and cloud computing.

Similar jobs

Software Development Engineer roles near Sunnyvale, California
12h
Save
Mark Applied
Hide
Senior Staff Software Development Engineer, ZIA Core Datapath
San Jose, California, United States
$158k-$225k/yr HybridFull Time
Zscaler
ZscalerNASDAQ: ZS: Cloud-native Zero Trust cybersecurity platform for digital transformation.
12+ YOE12+ years of experience building Python test automation frameworks; expertise in networking, cloud security testing, debugging, and L3-L7 technologies including firewalls, tunneling, load balancers, IPS/IDS, and routing.
Python, AI/ML, TCP/IP, HTTP(S), TLS, PKI, DNS, DHCP, VPN, HA, IPsec, GRE, IPS/IDS
1d
Save
Mark Applied
Hide
Sr. Software Development Engineer
Washington or San Jose or Spokane Valley
$180k-$269k/yr HybridFull Time
F5
F5NASDAQ: FFIV: Delivering and securing applications across any multi-cloud environment.
5+ YOERequires 5+ years of software engineering experience with a bachelor's degree in computer science, computer engineering, or similar, or 7+ years equivalent experience; strong Go and Python skills and technical leadership experience.
Go, Python, Bash, Linux, Docker, Docker Compose, Git, GitLab, CI/CD, Visual Studio Code, GoLand
3d
Save
Mark Applied
Hide
Sr. Software Development Engineer, AI & Data Platforms (AiDP)
Sunnyvale or North America
OnsiteFull Time
Apple
AppleNASDAQ: AAPL: Designing and manufacturing consumer electronics, software, and digital services.
10+ YOE10+ years building production-grade full-stack and backend services with Node.js, ReactJS, TypeScript, and Java; experience with Kafka, Flink, AI products, Docker, Kubernetes, AWS, and SQL/NoSQL databases; bachelor's degree or equivalent.
Node.js, ReactJS, TypeScript, Java, Kafka, Flink, Docker, Kubernetes, AWS, SQL, NoSQL, CI/CD, DevOps
3d
Save
Mark Applied
Hide
GenAI Software Development Engineer
Santa Clara or Austin or Seattle or Secaucus
$204k-$306k/yr OnsiteFull Time
AMD
AMDNASDAQ: AMD: Leader in high-performance computing, graphics, and visualization technologies.
Production software development experience, LLM and RAG systems expertise, proficiency in a modern programming language, cloud infrastructure experience, and a BS or MS in computer science, electrical engineering, or a related field.
Python, TypeScript, Node.js, Go, Java, C#, Rust, AWS, Azure, GCP, Claude Code, GitHub Copilot, Cursor, Slack, GitHub, Jira, Confluence, LightRAG, Neo4j, React, Next.js, Angular, Vue, Terraform, Model Context Protocol (MCP), LangGraph, CrewAI, AutoGen, Slack API, GitHub API, Atlassian REST APIs
4d
Save
Mark Applied
Hide
Software Development Engineer 2 (Java 17+)
Portland or Boston or Chicago or Dallas or San Francisco or Seattle
$96k-$116k/yr RemoteFull Time
WEX
WEXNYSE: WEX: Public fintech and payments helping businesses manage fleet fuel, employee benefits, and corporate payments.
3+ YOERequires 3–6 years of backend software development experience, strong Java and Spring expertise, REST API design, AWS, SQL, PostgreSQL, messaging, Docker, Kubernetes, Git, and Maven or Gradle experience.
Java, Spring Boot, Spring Data, Spring Security, Salesforce, Apex, GraphQL, gRPC, AWS, AWS Lambda, Amazon EC2, Amazon RDS, SQL, PostgreSQL, Kafka, RabbitMQ, Docker, Kubernetes, Git, Maven, Gradle, JUnit, Mockito
6d
Save
Mark Applied
Hide
Senior Software Development Engineer
San Francisco, California, United States
$229k-$331k/yr OnsiteFull Time
Adobe
AdobeNASDAQ: ADBE: Empowering everyone to create through innovative digital experiences.
12+ YOERequires 12+ years of software engineering experience, full-stack expertise, production and platform engineering maturity, CI/CD, AWS, infrastructure-as-code, AI systems, security, technical leadership, and strong communication.
JavaScript, TypeScript, React, Web Components, Ruby, Ruby on Rails, Python, Go, Terraform, Chef, Kubernetes, Helm, AWS, Amazon EC2, Auto Scaling Groups (ASG), AWS Identity and Access Management (IAM), AWS Lambda, Amazon API Gateway, Amazon CloudFront, Jenkins, SageMaker, OpenSearch, Elasticsearch, Vault, CyberArk, Lit, MCP
1w
Save
Mark Applied
Hide
Software Development Engineer
San Jose or United States
$187k-$229k/yr HybridFull Time
Zoom
ZoomNasdaq Global Select Market: ZM: American publicly traded communications platform serving businesses and individuals with AI-assisted video, voice, chat, and phone services.
2+ YOEMaster's degree in a related technical field and 2 years of experience in JavaScript, Node.js, frontend frameworks, MVC backend technologies, Web APIs, OAuth APIs, and REST APIs.
Java, Python, JavaScript, TypeScript, Node.js, React, Vue.js, AWS, OAuth 2.0, OIDC, REST APIs, SDK, LLM, Express, Ruby on Rails, Django, Angular, CI/CD, CDN
1w
Save
Mark Applied
Hide
Software Development Engineer - Document Intelligence
Pleasanton, California, United States
$148k-$222k/yr HybridFull Time
Workday
WorkdayNASDAQ: WDAY: Enterprise cloud applications for finance and human resources.
5+ YOERequires 5+ years software development experience, 2+ years Python, scalable data pipelines, cloud services, containerization, MLOps, and a relevant bachelor's degree or equivalent practical experience.
Python, Kubernetes, AWS, Azure, GCP, Docker, PostgreSQL, Redis, REST, NLP, RAG, LLMs, CI/CD