Amazon
Applied Scientist -- Foundation Models, SSO
$143K–$193K
Houston, TX · Full-time · Posted Aug 17, 2026
- Experience
- 3+ years building models for business application; PhD, or Master's degree and 4+ years of CS, CE, ML or related field experience.
- Education
- PhD, or Master's degree and 4+ years of CS, CE, ML or related field experience.
About the job
Build and optimize foundation models for long-horizon agentic workloads, combining model development with systems and hardware optimization.
What you'll do
- Design, train, fine-tune, and evaluate foundation models; develop evaluations for model reliability and long-horizon agent performance.
- Improve model efficiency, stability, latency, throughput, and memory use; optimize training and inference and resolve compute, memory, communication, and distributed-training bottlenecks.
- Build models for planning, tool use, memory, self-correction, and multi-step task execution, using GPUs and specialized accelerators.
What they ask for
- 3+ years of building models for business application experience.
- PhD, or Master's degree and 4+ years of CS, CE, ML or related field experience.
- Experience in patents or publications at top-tier peer-reviewed conferences or journals.
- Experience programming in Java, C++, Python or related language; experience in at least one listed area, including algorithms and data structures, numerical optimization, or parallel and distributed computing.
Nice to have
- Experience using Unix/Linux.
- Experience in professional software development.
Pay
142,800.00 - 193,200.00 USD annually (base salary)
Tools
PyTorch, JAX, CUDA, Triton, AWS Trainium, Inferentia, Neuron SDK, TPUs
AI focus
Develop efficient, stable foundation models for long-horizon agentic workloads, including planning, tool use, memory, self-correction, and multi-step execution; evaluate reliability and agent performance.
Model-written summary · the original posting is authoritative