Join Our Innovative Engineering Team at Amazon

Are you passionate about technology, innovation, and making a lasting impact on the industry? Become a part of an innovative team of engineers and scientists at Amazon, where we leverage terabytes of data and state-of-the-art algorithms to push the boundaries of AI.

About the Role

We are at the forefront of building the foundational behavioral model for Amazon Stores using Large Language Models (LLM) and large model training techniques. Our aim is to fuse general world knowledge with Amazon e-commerce domain knowledge to enhance customer experience.

Key Responsibilities

• Quickly learn cutting-edge technologies and algorithms in the field of Generative AI and LLMs.
• Develop and maintain key platforms for the development, evaluation, and deployment of LLMs for real-world applications.
• Collaborate with team members to investigate design approaches, prototype new technologies, and assess technical feasibility.
• Work closely with applied scientists to process massive data and scale machine learning models while optimizing performance.

Location

We are open to hiring candidates to work out of the following locations:

• San Diego, CA, USA
• Seattle, WA, USA

Basic Qualifications

• 3+ years of non-internship professional software development experience
• 2+ years of non-internship design or architecture experience (design patterns, reliability, and scaling)
• Experience programming with at least one software programming language
• Bachelor's degree in computer science or equivalent
• Experience with ML frameworks such as PyTorch, Spark MMLib, Tensorflow
• Experience in developing machine learning models and ML model training/inference optimization

Preferred Qualifications

• 3+ years of full software development life cycle, including coding standards, code reviews, source control management, build processes, testing, and operations experience
• Experience with LLM training/inference optimization at the algorithm level
• Experience with LLM training/inference optimization at the system level: KV cache optimization, Quantization, etc.
• Experience with PyTorch operations, memory, communication, and/or data loading optimization at scale
• Master's degree or PhD in Computer Science or a related technical field

Commitment to Diversity

Amazon is committed to a diverse and inclusive workplace. We are an equal opportunity employer and do not discriminate based on race, national origin, gender, gender identity, sexual orientation, protected veteran status, disability, age, or any other legally protected status. For individuals with disabilities who need accommodation, please visit our .

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 geographical market up to $223,600/year in our highest geographic market. Pay is based on various factors including market location and job-related knowledge, skills, and experience. Amazon offers a total compensation package, which may include equity, sign-on payments, and other forms of compensation, as well as a comprehensive range of medical, financial, and other benefits. For more information, please visit our .

This position will remain posted until filled. Applicants should apply via our internal or external career site.

Apply Now

If you're ready to revolutionize the world of AI with us, apply today!

Company Name: Amazon.com

Job Title: SDE-II (Machine Learning)