About the job
We are looking for an AI/ML Engineer to build efficient, stable foundation models for long-horizon agentic workloads. The role combines model development with systems and hardware optimization.
Responsibilities
Design, train, fine-tune, and evaluate custom foundation models.
Improve model efficiency, stability, latency, throughput, and memory usage.
Build models that support planning, tool use, memory, self-correction, and multi-step task execution.
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, including AWS Trainium and other custom tensor hardware.
Develop evaluations for model reliability and long-horizon agent performance.
Qualifications
Minimum
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 any of the following areas: algorithms and data structures, parsing, numerical optimization, data mining, parallel and distributed computing, high-performance computing
Preferred
Experience using Unix/Linux
Experience in professional software development
Experience with AWS Trainium, Inferentia, Neuron SDK, TPUs, or other custom accelerators
PhD in machine learning, computer science, electrical engineering, or a related field