Machine Learning Engineer, AGI

$129,300 - $223,600/Yr

Amazon - New York, NY

Full-time - Mid Level

New York, NY

Sporting Goods, Hobby, Musical Instrument, Book, and Miscellaneous Retailers

About the position

The Artificial General Intelligence (AGI) team at Amazon is seeking a passionate and inventive Software Development Engineer/Machine Learning Engineer (SDE/MLE) to play a pivotal role in the development of industry-leading multi-modal and multi-lingual large language models (LLM). In this role, you will lead the charge in developing cutting-edge algorithms and modeling techniques that will push the boundaries of large model training using advanced hardware such as AWS Trainium. Your groundbreaking work will have a direct impact on our customers' lives through innovative products and services powered by your Generative AI breakthroughs. As an SDE/MLE, you will dive into Amazon's vast and diverse data sources, harnessing the immense power of our large-scale computing resources to accelerate the development of multi-modal Large Language Models (LLMs) and other Generative Artificial Intelligence (Gen AI) applications. Your expertise will be invaluable in defining data strategies, enrichment processes, model optimizations, and evaluation methods that will set new industry standards. This position offers a unique opportunity to push the limits of AI in a fast-paced and innovative environment, allowing you to make a lasting impact on the world. In your day-to-day responsibilities, you will lead the development of novel algorithms and modeling techniques to advance the state of the art in large model training. Your work will directly influence the products and services that utilize Generative AI innovations, leveraging Amazon's heterogeneous data sources and large-scale computing resources. As a key player in the AGI team, you will shape the future direction of AGI at Amazon, driving the system architecture and championing best practices to ensure an unparalleled infrastructure of the highest quality. You will work in an Agile/Scrum environment, moving fast to deliver high-quality software.

Responsibilities

• Quickly learn cutting-edge technologies and algorithms in the field of Generative AI to contribute to building the best LLMs.
• Develop and maintain key platforms needed for developing, evaluating, and deploying LLM for real-world applications.
• Investigate design approaches, prototype new technology, and evaluate technical feasibility in collaboration with team members.
• Process massive data and scale machine learning models while optimizing, in close collaboration with Applied Scientists.

Requirements

• 3+ years of non-internship professional software development experience.
• 2+ years of non-internship design or architecture experience (design patterns, reliability, and scaling) of new and existing systems.
• Experience in distributed training for large language models.
• Experience programming with at least one software programming language.
• Bachelor's degree in computer science or equivalent.

Nice-to-haves

• 3+ years of full software development life cycle experience, including coding standards, code reviews, source control management, build processes, testing, and operations.
• Master's or PhD in Machine Learning and its practical applications.

Benefits

• Competitive salary based on market location and experience.
• Equity and sign-on payments as part of total compensation package.
• Full range of medical, financial, and other benefits.