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.
Share the post "Neuron Runtime Software Development Engineer (On-Site)"
