Senior Applied Scientist, AGI Customization

About the role

The Artificial General Intelligence (AGI) Customization Team is seeking a highly skilled and experienced Applied Scientist to support adoption and enable customization of Amazon Nova. The role focuses on developing state-of-the-art services and tools for model customization, including supervised fine-tuning, reinforcement learning, and knowledge distillation across large language models.

Responsibilities

• Contribute to the development of novel customization techniques including extended post-training, continued pre-training, and advanced knowledge distillation
• Collaborate with cross-functional teams to design and implement enterprise-ready tooling for various training techniques on Amazon SageMaker
• Design and execute experiments to optimize model accuracy, latency, and cost across different customization approaches (SFT, DPO, PPO)
• Develop and enhance preference learning algorithms and training curricula for customer-specific applications
• Create robust evaluation frameworks for assessing model performance across different domains and use cases
• Contribute to the development of the Responsible AI toolkit, including creating training and evaluation datasets for model alignment
• Design and implement secure access mechanisms for early model checkpoints and weights
• Communicate technical insights and results to both technical and non-technical stakeholders through presentations and documentation

Qualifications

The basic qualifications for this role include 3+ years of building machine learning models for business application experience, and a PhD, or Master's degree and 6+ years of applied research experience. Experience programming in Java, C++, Python or related language, and experience with neural deep learning methods and machine learning are also required.

• Preferred qualifications include experience with modeling tools such as R, scikit-learn, Spark MLLib, MxNet, Tensorflow, numpy, scipy, etc.
• Experience with large scale distributed systems such as Hadoop, Spark, etc.

Pay

The base salary range for this position is $192,200.00 - $260,000.00 USD annually in Sunnyvale, CA, $167,100.00 - $226,100.00 USD annually in Cambridge, MA, and $167,100.00 - $226,100.00 USD annually in Bellevue, WA. The final compensation will be determined based on factors including experience, qualifications, and location, and will include sign-on payments and restricted stock units (RSUs).

Benefits

Amazon also offers comprehensive benefits including health insurance (medical, dental, vision, prescription, Basic Life & AD&D insurance and option for Supplemental life plans, EAP, Mental Health Support, Medical Advice Line, Flexible Spending Accounts, Adoption and Surrogacy Reimbursement coverage), 401(k) matching, paid time off, and parental leave.