Senior Software Engineer - Generative AI

Job Description

Are you interested in advancing Amazon's Generative AI capabilities? Join a talented team of engineers and scientists in a highly collaborative and friendly environment. We are building state-of-the-art Generative AI technology that will benefit all Amazon businesses and customers.

Key Responsibilities

As a Software Development Engineer, you will be responsible for designing, developing, testing, and deploying high-performance inference capabilities. This includes multi-modality, state-of-the-art model architectures, latency, throughput, and cost optimization. You will collaborate closely with a team of engineers and scientists to influence our overall strategy and define the team’s roadmap. You will drive system architecture, spearhead best practices, and mentor junior engineers.

A Day in the Life

• Read papers and consult with scientists to get inspiration from emerging techniques and integrate them into our roadmap.
• Design and experiment with new algorithms, benchmark the latency and accuracy of your implementations.
• Implement production-grade solutions and oversee their deployment.
• Collaborate with other science and engineering teams to ensure proper execution.
• Maintain high standards in operational excellence and support production systems, constantly creating solutions to minimize operational load.

About The Team

Our mission is to build best-in-class, fast, accurate, and cost-efficient large language model inference solutions and infrastructure that will enable Amazon businesses to deliver more value to their customers.

Basic Qualifications

• 3+ years of non-internship professional software development experience.
• 2+ years of non-internship design or architecture experience of new and existing systems.
• Experience programming with at least one software programming language.
• Prior experience with software performance optimization or knowledge of Machine Learning and Deep Learning.

Preferred Qualifications

• 3+ years of full software development life cycle experience, including coding standards, code reviews, source control management, build processes, testing, and operations.
• Bachelor's degree in computer science or equivalent.
• Experience with Large Language Model inference.
• Experience with Trainium and Inferentia Development.
• Experience with GPU programming (TensorRT-LLM).
• Experience with Python, PyTorch, and C++ programming and performance optimization.

Amazon is committed to a diverse and inclusive workplace. Amazon is an equal opportunity employer and does not discriminate on the basis of race, national origin, gender, gender identity, sexual orientation, protected veteran status, disability, age, or other legally protected status. For individuals with disabilities who would like to request an accommodation, please visit Amazon's disability accommodations page.

Compensation

Our compensation reflects the cost of labor across several US geographic markets. The base pay for this position ranges from $129,300/year in our lowest geographic market up to $223,600/year in our highest geographic market. Pay is based on a number of factors including market location and may vary depending on job-related knowledge, skills, and experience. Amazon is a total compensation company. Dependent on the position offered, equity, sign-on payments, and other forms of compensation may be provided as part of a total compensation package, in addition to a full range of medical, financial, and/or other benefits.