Senior Applied Scientist, AGI Customization

Description

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.

• Contribute to the development of novel customization techniques such as 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 Sage Maker.
• 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.

• 3+ years of building machine learning models for business applications.
• PhD, or Master's degree and 6+ years of applied research experience.
• Experience programming in Java, C++, Python or related languages.
• Experience with neural deep learning methods and machine learning.

• 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.

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

Amazon offers comprehensive benefits including health insurance (medical, dental, vision, prescription, Basic Life & AD&D insurance and optional 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. Learn more about our benefits at (Use the "Apply for this Job" box below)..

USA, CA, Sunnyvale – $ – $ USD annually

USA, MA, Cambridge – $ – $ USD annually

USA, WA, BELLEVUE – $ – $ USD annually