Software Engineer- AI/ML, AWS Neuron Distributed Training
职位介绍
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.
查看完整职位
工作职责、任职要求、技能与福利 — 免费创建账号即可查看。
已有账户？ 登录
相似职位
其他可能适合您的职位。
? 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