Senior Applied Scientist: Medical Imaging AGI Customization

Amazon; Bellevue (WA); On-site; USD 167,000 - 226,000; Full time

Benefits offered by this job

Health insurance

401(k) matching

Paid time off

Parental leave

Job summary

Amazon’s Artificial General Intelligence (AGI) Customization Team is seeking an experienced Applied Scientist in medical imaging to advance customization of foundation models. You will work on supervised fine-tuning, RL, and knowledge distillation across large language models and enterprise applications.

Responsibilities include developing novel techniques, building enterprise-ready tooling on SageMaker, and evaluating model performance across domains.

Qualifications

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

Responsibilities

• Contribute to development of novel customization techniques in medical imaging domain, 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

Skills

Java

C++

Python

Deep learning

ML research

Education

PhD or MS + 6+ yrs

Tools

R

scikit-learn

Spark MLlib

MxNet

TensorFlow

NumPy

SciPy

Job description

Amazon’s Artificial General Intelligence (AGI) Customization Team is seeking an experienced Applied Scientist in medical imaging to advance customization of foundation models. You will work on supervised fine-tuning, RL, and knowledge distillation across large language models and enterprise applications.

Responsibilities include developing novel techniques, building enterprise-ready tooling on SageMaker, and evaluating model performance across domains.