ML Compiler Engineer, Neuron Kernel Interface
Core
Design and build compiler features for the Neuron compilers to handle large ML workloads on AWS Trainium, spanning frontend, IR, optimization, and code generation.
Role type
Senior IC ML Compiler Engineer (hardware/software stack)
Builds
Compiler stack features for AWS Trainium and Neuron Kernel Interface (NKI)
Domain
Cloud computing + Machine Learning hardware acceleration
Deliverable
production ML models
Required skills
C++, C#, Java, Perl, Object Oriented Design, large-scale multi-threaded embedded/distributed software, system design/architecture, compiler development
Preferred skills
Full software development life cycle, coding standards, code reviews, source control, build processes, testing, operations
Technologies
PyTorch, JAX, AWS Trainium, Neuron SDK, NKI
Responsibilities
Design and implement compiler features; own features from design through production; work with ML teams to unblock workloads; provide hardware feedback; review code and mentor engineers
Seniority
Senior, hands-on IC