Sr. Software Development Engineer, AWS Neuron SDK- Containers
Core
Develop and maintain high-performance monitoring and profiling tools (Neuron Profiler) for AWS Inferentia and Trainium ML accelerators to optimize AI workloads and ML frameworks.
Role type
Senior Software Development Engineer (Infrastructure/Tooling)
Builds
Neuron Profiler and Neuron Foundation Tools for Trainium and Inferentia devices
Domain
Cloud-scale machine learning accelerators and AI hardware
Deliverable
production ML models | infrastructure
Required skills
C++, system design, architecture, mentoring, full software development life cycle, code reviews, source control, build processes, testing, operations
Preferred skills
Compiler development, runtime development, distributed training systems
Technologies
AWS Inferentia, AWS Trainium, Trn1, Inf1, C++ compiler, runtime
Responsibilities
Design and develop Neuron Profiler tools, collaborate with cross-functional teams to optimize ML kernels and frameworks, manage full development life cycle of tools ensuring scalability and reliability, work with technical leaders to define product directions
Seniority
Senior, hands-on IC with leadership responsibilities
