About the job
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
- Develop and maintain high-performance runtime libraries and drivers for machine learning applications and AI accelerators
- Work on design, development, and deployment of Neuron Runtime and other Neuron components
- 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
Qualifications
Minimum
- 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
Preferred
- 3+ years of full software development life cycle, including coding standards, code reviews, source control management, build processes, testing, and operations experience
- Bachelor's degree in computer science or equivalent