Software Engineer- AI/ML, AWS Neuron Distributed Training
¿Es esta oferta para usted?
Crear mi CV Cree su CV y descubra su porcentaje de coincidencia con este puesto — y con todos los demás.
El puesto
Design, implement, and optimize distributed training solutions for large-scale ML models running on Trainium instances.
Extend and optimize popular distributed training frameworks, including FSDP, torchtitan, and Hugging Face libraries for the Neuron ecosystem.
Profile, analyze, and tune end-to-end training models and pipelines to achieve optimal performance on Trainium hardware.
Partner with hardware, compiler, and runtime teams to influence system design and unlock new capabilities.
Work directly with solution architects and customers to deploy and optimize training workloads at scale.
Operate at the intersection of cutting-edge ML research and high-performance systems, contributing to training workloads such as LLMs, Dense and Mixture-of-Experts architectures, and multimodal models.
Extend and optimize popular distributed training frameworks, including FSDP, torchtitan, and Hugging Face libraries for the Neuron ecosystem.
Profile, analyze, and tune end-to-end training models and pipelines to achieve optimal performance on Trainium hardware.
Partner with hardware, compiler, and runtime teams to influence system design and unlock new capabilities.
Work directly with solution architects and customers to deploy and optimize training workloads at scale.
Operate at the intersection of cutting-edge ML research and high-performance systems, contributing to training workloads such as LLMs, Dense and Mixture-of-Experts architectures, and multimodal models.
Ver la oferta completa
Funciones, perfil, competencias y ventajas — crea tu cuenta gratis.
¿Ya tiene una cuenta? Iniciar sesión
Ofertas similares
Otros puestos que podrían encajar.
? Fellow, AI Performance Software Engineer ? Applied Machine Learning Engineer, Circuit Design - New College Grad 2026 ? Applied Machine Learning Engineer - VLSI Design ? Machine Learning Engineer II, Sponsored Products and Brands ? Applied Scientist, LLM Code Agents, Kiro Science ? Senior Data Engineer, Ops Data Platform