Senior AGI Customization Scientist - Medical Imaging

Amazon; Cambridge (MA); On-site; USD 167,000 - 226,000; Full time

Job summary

Amazon's Artificial General Intelligence customization team is seeking an experienced Applied Scientist in medical imaging for foundation model customization. You will work on supervised fine-tuning, RL, and knowledge distillation to enable enterprise-grade, application-specific models using Amazon Nova models.

Collaboration across teams will drive enterprise-ready tooling on SageMaker. The role requires 3+ years in ML model building, PhD or Master's + 6+ years, and hands-on coding in

Qualifications

• 3+ years of building machine learning 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 influence across domains.
• Experience with modeling tools such as R, scikit-learn, Spark MLLib, MxNet, TensorFlow, NumPy, SciPy.
• Experience with large-scale distributed systems such as Hadoop, Spark.

Responsibilities

• Contribute to the 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

Machine learning

Java

C++

Python

Education

PhD or Master’s + 6+ years applied research

Tools

R

scikit-learn

Spark MLLib

MxNet

TensorFlow

NumPy

SciPy

Job description

Amazon's Artificial General Intelligence customization team is seeking an experienced Applied Scientist in medical imaging for foundation model customization. You will work on supervised fine-tuning, RL, and knowledge distillation to enable enterprise-grade, application-specific models using Amazon Nova models.

Collaboration across teams will drive enterprise-ready tooling on SageMaker. The role requires 3+ years in ML model building, PhD or Master's + 6+ years, and hands-on coding in