Software Engineer- AI/ML, AWS Neuron Distributed Training
Ist diese Stelle etwas für Sie?
Lebenslauf erstellen Erstellen Sie Ihren Lebenslauf und entdecken Sie Ihre Übereinstimmung mit dieser Stelle — und mit allen anderen.
Die Stelle
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.
Die vollständige Anzeige sehen
Aufgaben, Anforderungen, Kompetenzen und Vorteile — mit Ihrem kostenlosen Konto.
Bereits ein Konto? Anmelden
Ähnliche Stellen
Weitere Positionen, die passen könnten.
? 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