Are you passionate about building the infrastructure that powers the future of AI? Annapurna Labs is seeking a talented Software Development Engineer to join the Neuron Runtime team in Seattle, WA. This is a unique opportunity to work at the intersection of high-performance software and cutting-edge custom hardware.
Job Overview:
- Role: Software Development Engineer, Neuron Runtime
- Company: Annapurna Labs (part of Amazon.com)
- Location: Seattle, WA (On-Site)
- Salary Range: $143,700 – $194,400 per year
- Type: Full-time
In this role, you will be responsible for the full development life cycle of high-performance runtime libraries and drivers for AWS Inferentia and Trainium accelerators. Your contributions will directly impact how internal and external customers optimize massive-scale machine learning workloads. You will drive innovation in performance profiling, ensuring that our C++ compilers and ML frameworks like PyTorch, JAX, and XLA operate at peak efficiency.
Primary Responsibilities:
- Design, develop, and deploy Neuron Runtime components.
- Optimize ML Kernels and framework performance.
- Create deep insights into system behavior and performance bottlenecks.
- Own services end-to-end, from deployment and monitoring to on-call and incident review.
- Collaborate with senior management to define product direction.
Eligibility Criteria:
To qualify, candidates must have at least 3 years of non-internship professional software development experience. You should also possess 2+ years of experience in system design or architecture, specifically focusing on reliability, scaling, and design patterns. Proficiency in at least one programming language is essential for success in this role.
For preferred candidates, a Bachelor’s degree in Computer Science or a related field is desired. Additionally, we value those with a complete grasp of the software development life cycle, including coding standards, rigorous code reviews, and operational excellence in production environments. Hands-on experience with AWS services like EC2, ECS, or Lambda will set your application apart.
At Annapurna Labs, we believe that innovation thrives in an inclusive environment. We offer a culture that celebrates diverse perspectives and provides extensive mentorship and career growth opportunities. Our team prioritizes a healthy work-life balance, giving you the flexibility to achieve both personal and professional fulfillment while working on some of the most complex distributed systems in the cloud industry.
Why Choose Annapurna Labs?
Joining Annapurna Labs, a part of Amazon, means being at the forefront of cloud-scale AI innovation. We offer a comprehensive benefits package, including health insurance (medical, dental, vision), 401(k) matching, paid time off, and parental leave, ensuring you and your family are supported.
Our commitment to inclusion is reflected in our 16 Leadership Principles and our employee-led affinity groups. We foster an environment where learning is continuous and where team members are encouraged to seek diverse perspectives and “learn and be curious” every single day.
If you are ready to tackle high-scale distributed training challenges and define the future of ML hardware, we want to hear from you. We encourage you to apply and join a team that values your technical expertise and your commitment to operational excellence. By joining our team, you will help us deliver superior performance to customers globally while advancing your own career in a high-growth, high-impact organization.
| Attribute | Details |
| Company | Annapurna Labs (part of Amazon.com) |
| Salary | $143,700 – $194,400 per year |
| Key Requirement | 3+ years software dev, 2+ years system design |
| Location | Seattle, WA |
| Application Link | Apply on Company Site |
Explore More Opportunities:
Thank you for your interest in the Neuron Runtime Software Development Engineer role. We are excited to see the impact you can make on our infrastructure and look forward to reviewing your application. Good luck with your career search!
