Neuron Runtime Software Development Engineer , Neuron Runtime
About the role
AWS Neuron is the complete software stack for the AWS Inferentia and Trainium cloud-scale machine learning accelerators and the servers that use them. As the Software Development Engineer for the Neuron Runtime Team, you will be responsible for working alongside a team of engineers to develop and maintain high-performance runtime libraries and drivers for machine learning applications and AI accelerators. You will work on design, development, and deployment of Neuron Runtime and other Neuron components.
Responsibilities
- Work with the executive leadership and other senior management and technical leaders to define product directions and deliver them to customers.
- Build massive-scale distributed training and inference solutions.
- Build the full stack of software, servers, and chips to accelerate at the highest scale.
- Manage the full development life cycle of the Neuron Runtime, ensuring scalability, reliability, and usability.
- Collaborate with cross-functional teams to ensure that the our C++ compiler generates key information so customers can understand and optimize the performance of our custom hardware.
- Drive innovations that allow the profiler to support multiple frameworks, such as PyTorch, JAX, and XLA.
Requirements
- 3+ years of non-internship professional software development experience
- 2+ years of non-internship design or architecture (design patterns, reliability and scaling) of new and existing systems experience
- Experience programming with at least one software programming language
Qualifications
- Bachelor's degree in computer science or equivalent
Skills
- Hands-on experience with AWS services (e.g., EC2, ECS, CloudWatch, S3, Lambda) in production environments
- Track record in owning services end-to-end including deployment, monitoring, alarming, on-call, and post-incident review
Benefits
Our inclusive culture empowers Amazonians to deliver the best results for our customers. If you have a disability and need a workplace accommodation or adjustment during the application and hiring process, including support for the interview or onboarding process, please visit here.
Pay
The base salary range for this position is $143,700.00 - $194,400.00 annually.
Schedule
Learn more about our benefits at here.