Sr Software Development Engineer

AWS AI is seeking exceptional software developers to join their Deep Learning cross-framework team. This role sits within AWS's pioneering cloud computing division, focusing on developing cutting-edge solutions for machine learning and AI infrastructure.

The position involves working with TensorFlow and PyTorch frameworks, developing solutions for training Deep Learning models at massive scale. You'll be part of the SageMaker Engines team, working alongside leading engineers and researchers in the field. The role combines deep technical expertise in machine learning frameworks with distributed systems engineering.

Key responsibilities include implementing sophisticated model parallelism methods, optimizing distributed training performance, and developing memory-efficient training techniques. You'll work on systems that handle thousands of accelerators and contribute to AWS's position as a thought leader in cloud-based AI infrastructure.

The ideal candidate brings strong software development experience, particularly in C++ and Python, with a deep understanding of machine learning frameworks and distributed systems. You'll need to demonstrate leadership experience and the ability to mentor others, as well as strong architectural design skills.

AWS offers a comprehensive benefits package, including competitive base pay ranging from $151,300 to $261,500 depending on location, plus equity and other benefits. The company strongly values diversity and inclusion, providing numerous employee-led affinity groups and ongoing learning opportunities.

This role offers unique opportunities to work on cutting-edge AI infrastructure, with access to AWS's vast resources and the chance to impact how the world's leading companies train their AI models. You'll be part of a culture that emphasizes continuous learning, work-life harmony, and professional growth.

The position is based in the San Francisco Bay Area and requires hands-on experience with machine learning frameworks, distributed systems, and high-performance computing. Join AWS to help shape the future of cloud-based AI infrastructure while working with some of the industry's most advanced technologies and talented professionals.