Software Engineer- AI/ML, AWS Neuron Distributed Training
Is deze vacature iets voor u?
Mijn cv maken Maak uw cv en ontdek uw matchpercentage met deze functie — en met alle andere.
De functie
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.
Bekijk de volledige vacature
Taken, profiel, vaardigheden en voordelen — maak gratis een account aan.
Al een account? Inloggen
Vergelijkbare vacatures
Andere functies die kunnen passen.
? 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