Senior Software Engineer - Generative AI, AGI Inference Engine

Amazon; New York (NY); On-site; USD 151,000 - 262,000; Full time

Job summary

Join a forward-thinking company as a Senior Software Engineer focused on Generative AI. In this dynamic role, you'll design and implement cutting-edge inference capabilities, collaborating with a talented team of engineers and scientists. Your expertise in software development and machine learning will drive the creation of state-of-the-art solutions that enhance customer experiences across various Amazon businesses. This position offers a unique opportunity to influence the future of AI technology in a supportive and innovative environment, where your contributions will have a significant impact on the industry.

Benefits

Health care plans

Equity options

Flexible working hours

Employee discounts

Retirement plans

Qualifications

• 5+ years of software development experience and programming.
• Experience leading design or architecture of systems.

Responsibilities

• Design, develop, test, and deploy high-performance inference capabilities.
• Collaborate with engineers and scientists to define the team's roadmap.

Skills

Software Development

Programming (Python, C++)

Machine Learning

Deep Learning

Performance Optimization

Education

Bachelor's degree in Computer Science

Tools

PyTorch

TensorRT-LLM

Amazon AI chip programming

Job description

Senior Software Engineer - Generative AI, AGI Inference Engine

Job ID: 2967553 | Amazon.com Services LLC

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

Key job responsibilities
As a Senior Software Development Engineer, you will be responsible for designing, developing, testing, and deploying high performance inference capabilities, including but not limited to multi-modality, SOTA model architectures, latency, throughput, and cost. 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
You will read papers and consult with scientists to get inspiration of emerging techniques, and blend those into our roadmap; You will design and experiment with new algorithms, benchmark the latency and accuracy of your implementations; Most importantly you will implement production grade solutions, and see them through the deployments swiftly; You may need to collaborate with other science and engineering teams to get things done properly; You will hold highest bar in operational excellence and support production systems, and constantly create solutions to minimize the ops 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

- 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
- Experience as a mentor, tech lead or leading an engineering team
- Prior experience with software performance optimization Or Knowledge of Machine Learning and Deep Learning

PREFERRED QUALIFICATIONS

- Bachelor's degree in computer science or equivalent
- Experience with Large Language Model inference
- Experience with GPU programming (e.g. TensorRT-LLM) or Amazon AI chip programming (Trainium)
- Experience with Python, PyTorch, and C++ programming, particularly performance optimization

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.

Our compensation reflects the cost of labor across several US geographic markets. The base pay for this position ranges from $151,300/year in our lowest geographic market up to $261,500/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. For more information, please visit https://www.aboutamazon.com/workplace/employee-benefits . This position will remain posted until filled. Applicants should apply via our internal or external career site.

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