Applied Scientist -- Foundation Models, SSO
Core
Design, train, fine-tune, and evaluate custom foundation models for long-horizon agentic workloads, combining model development with systems and hardware optimization.
Role type
Senior IC Applied Scientist (Foundation Models)
Builds
Efficient, stable foundation models supporting planning, tool use, memory, self-correction, and multi-step task execution
Domain
Cloud computing, AI/ML, High-performance computing
Deliverable
production ML models
Required skills
Machine learning, deep learning, transformers, foundation models, PyTorch, JAX, large-scale model training, GPU architecture, CUDA, Triton, custom kernels, distributed training, model parallelism
Preferred skills
AWS Trainium, Inferentia, Neuron SDK, TPUs, Unix/Linux, professional software development
Technologies
PyTorch, JAX, CUDA, Triton, AWS Trainium, Inferentia, Neuron SDK, TPUs
Responsibilities
Design, train, fine-tune, and evaluate custom foundation models; Improve model efficiency, stability, latency, throughput, and memory usage; Optimize training and inference using CUDA, Triton, custom GPU kernels, and tensor operations; Profile and resolve compute, memory, communication, and distributed-training bottlenecks; Work with GPUs and specialized AI accelerators; Develop evaluations for model reliability and long-horizon agent performance
Seniority
Senior, hands-on IC
