Software Development Engineer, Distributed Systems, Annapurna Labs

You will build and improve the Neuron compiler engineering infrastructure, focusing on efficient, reliable build, test, and release processes. You will also help design distributed execution mechanisms to run customer workloads more efficiently on AWS hardware.

• Identify and design solutions for efficient and reliable build, test, and release mechanisms
• Participate in design and implementation of distributed execution for the Neuron compiler
• Create metrics and implement automation to improve performance and quality
• Collaborate in design discussions, code review, and stakeholder communication
• Research and implement solutions that improve service performance, durability, cost, and security

The role is a distributed systems and software engineering position supporting a deep learning compiler stack. You will work with systems and tooling around compilation, deployment, and quality (metrics/automation/root-cause), and the stack targets frameworks including TensorFlow, PyTorch, and Jax running on AWS Neuron hardware.

The ideal candidate is a mid-level Software Development Engineer with 3+ years of non-internship professional software development experience, strong distributed systems experience, and a track record of designing end-to-end build, test, and release mechanisms. They are hands-on with software architecture, automation, metrics, and root-cause defect resolution, and can partner effectively with cross-functional teams across compiler and ML services.