Drimva

Amazon

Senior Applied Scientist, AWS Neuron Science - Core Algorithm

$192K–$260K

Cupertino, CA · Senior · Full-time · Posted Oct 6, 2026

Experience
6+ years of applied research experience with a Master's degree; alternatively, a PhD. Also requires 3+ years building machine-learning models for business applications.
Education
PhD, or Master's degree and 6+ years of applied research experience.

About the job

Researches and productionizes hardware-aware machine-learning algorithms for AWS Trainium and Inferentia, coordinating model and algorithm decisions with custom-silicon teams.

What you'll do

  • Lead research and experimentation through production, evaluating complex problems at scale.
  • Develop production-quality ML code and integrate it into large-scale training and inference systems.
  • Collaborate across foundation-model, engineering, and hardware teams; mentor colleagues and publish research.

What they ask for

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

Nice to have

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

Pay

192,200.00 - 260,000.00 USD annually (base salary)

Tools

Trainium, Inferentia, PyTorch, JAX, Java, C++, Python, NKI

AI focus

Hardware-aware machine learning for LLMs and foundation models, including low-precision training and inference, hardware-friendly architectures, distributed optimizers, accuracy validation, and RL post-training for NKI kernel generation.

Model-written summary · the original posting is authoritative

First seen Oct 15, 2026 · Last verified Nov 5, 2026 (UTC)