Sr Software Development Engineer

AWS AI is seeking exceptional software developers for their Deep Learning cross-framework team. This role focuses on developing extensions for TensorFlow and PyTorch machine learning frameworks and creating cross-framework solutions for large-scale Deep Learning model training. As part of the SageMaker Engines team, you'll work on implementing model parallelism, memory optimization techniques, and network communication optimizations for AWS infrastructure. The position offers the opportunity to work with leading engineers and researchers in a fast-paced environment, contributing to innovative solutions that support training of Deep Learning models across thousands of accelerators. AWS values diverse experiences and maintains an inclusive culture through employee-led affinity groups and ongoing learning experiences. The role offers competitive compensation, comprehensive benefits, and emphasizes work-life harmony while providing opportunities for mentorship and career growth. Join AWS to be part of a team that's continuously innovating in cloud computing and trusted by companies from startups to Global 500 enterprises.