Software Development Engineer II, Amazon SageMaker

Join Amazon's SageMaker team and shape the future of artificial intelligence and machine learning at AWS. This role focuses on building and maintaining the DeepLearning AMI and Deep Learning containers AWS products, working at the intersection of open source frameworks and AWS services. You'll be responsible for developing the base frameworks layer that integrates various machine learning libraries to work seamlessly across EC2, SageMaker, Graviton, and Neuron platforms.

As a Software Development Engineer II, you'll collaborate with multiple AWS teams to optimize ML workloads and improve customer experience. The position offers unique opportunities to work with cutting-edge technologies, including PyTorch, JAX, and TensorFlow, while building solutions that power AWS's machine learning infrastructure. You'll also contribute to performance benchmarking of LLM models and influence the direction of various AWS ML products.

The role combines technical expertise with leadership responsibilities, as you'll be expected to mentor team members while driving technical excellence and innovation. Working at Amazon means joining a company at the forefront of AI/ML development, with competitive compensation including base pay ranging from $129,300 to $223,600, plus equity and comprehensive benefits.

This position is perfect for someone who is passionate about machine learning infrastructure, has strong software development skills, and wants to make a significant impact on how AI/ML workloads run on AWS. You'll be part of a team that's making AWS the premier platform for deep learning, working with the latest technologies and solving complex challenges at scale.