Sr. Software Development Engineer, HPC/ML Networking Engineer

Join Annapurna Labs, a crucial part of AWS, where we're revolutionizing cloud computing through custom chips and software development. We're seeking a Senior Software Engineer to join our Elastic Collectives team, focusing on optimizing networking solutions for ML and HPC workloads. You'll work with cutting-edge AI/ML technologies, designing systems that scale network-intensive workloads across thousands of processors.

Our team specializes in building the collective operations layer for distributed machine learning, working with both Trainium and Nvidia stacks. You'll collaborate with principal and senior principal engineers daily, hunting for performance bottlenecks and optimizing customer ML/AI workloads. Your work will directly impact every AWS customer using large model training and inference.

We offer a supportive environment that celebrates knowledge-sharing and mentorship, with team members of various experience levels. You'll benefit from one-on-one mentoring, thorough code reviews, and projects that enhance your engineering expertise. We value work-life harmony and provide flexible working hours to ensure long-term success both personally and professionally.

At AWS, we embrace diversity and foster an inclusive culture through employee-led affinity groups and ongoing learning experiences. Our comprehensive benefits package includes medical, financial, and other support systems to help you thrive. If you're passionate about solving complex infrastructure problems, working with HPC and ML customers, and delivering meaningful solutions at scale, this role offers an exciting opportunity to shape the future of cloud computing.

The position requires strong expertise in low-latency networking, collective operations, and kernel-level programming. You'll need excellent problem-solving abilities and strong communication skills to work effectively in our collaborative environment. Join us in pushing the boundaries of what's possible in cloud computing while growing your career with one of technology's most innovative companies.