Find your next role
Discover amazing opportunities across our network of companies committed to gender equality in the workplace.
Software Development Engineer II, Amazon SageMaker
Amazon
Software Engineering
Bellevue, WA, USA
Posted on May 22, 2025
DESCRIPTION
Do you want be part of a team shaping the future of artificial intelligence and machine learning at AWS? Are you passionate about building highly scalable, reliable, and efficient solutions that empower developers, data scientists, and enterprises to innovate?
Our team owns DeepLearning AMI and Deep Learning containers AWS products. We build base frameworks layer to integrate open source frameworks and related machine learning libraries to work on EC2, SageMaker, Graviton and Neuron. Our goal is to make AWS the best place to run Deep Learning.
Key job responsibilities
Deliver High-Impact Solutions: Collaborate with internal teams (e.g., Amazon SageMaker, EKS, EC2) to deliver seamless integrations that improve the customer experience and optimize performance for ML workloads.
Engage with Customers: Partner with customers to understand their needs, gather feedback, and ensure DLCs meet and exceed expectations for usability, performance, and reliability.-
Build and Lead Teams: Mentor and coach a diverse team of talented engineers. Foster a culture of technical excellence, collaboration, and innovation.
About the team
In this role, you will work on building Deep Learning images based on PyTorch, JAX, TensorFlow to AWS customers and internal SageMaker teams. You will work with partners like Nvidia, EC2, Graviton teams to build images and run performance benchmarks using LLM models. The role provides opportunity to influence across teams like HyperPod, SM Training and Notebooks. You will also work on contributing to AWS open source packages like DLAMI and DLC.