Description

The Artificial General Intelligence (AGI) team is looking for a Senior Applied Scientist to work on pre-training methodologies for Generative Artificial Intelligence (GenAI) models. You will interact closely with our customers and with the academic and research communities. You will be at the heart of a growing and exciting focus area for Amazon, working with other acclaimed engineers and world-famous scientists.

Key job responsibilities

Join us to work as an integral part of a team that has diverse experience with GenAI models in this space. We work on these areas:

• Scaling laws

• Hardware-informed efficient model architecture, low-precision training

• Optimization methods, learning objectives, curriculum design

• Deep learning theories on efficient hyperparameter search and self-supervised learning

• Reasoning, self-reflection, interactive learning, data synthesis

• Distributed training methods and solutions

About the team

The AGI team has a mission to push the envelope in LLMs and multimodal systems, in order to provide the best-possible experience for our customers.

Basic Qualifications

• PhD, or Master's degree and 5+ years of applied research experience

• 3+ years of building machine learning models for business application experience

• Experience with neural deep learning methods and machine learning

• Experience programming in Java, C++, Python or related language

• Experience with deep learning frameworks such as Pytorch and JAX.

Preferred Qualifications

• PhD in engineering, technology, computer science, machine learning, robotics, operations research, statistics, mathematics or equivalent quantitative field, or Master's degree and 6+ years of industry or academic research experience

• Experience with patents or publications at top-tier peer-reviewed conferences or journals

• Relevant Generative Artificial Intelligence (GenAI) research experience with LLMs and multimodalities

• For system researchers, familiarity with deep learning compilers, auto-parallelization, and XLA/MLIR ecosystems

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 https://www.amazon.jobs/en/disability/us.

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 compensation reflects the cost of labor across several US geographic markets. The base pay for this position ranges from $150,400/year in our lowest geographic market up to $260,000/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.