Software Development Engineer III, Annapurna Labs
Core
Design and implement distributed execution solutions for the Neuron compiler to enable efficient cloud execution of large-scale ML workloads.
Role type
Senior Software Development Engineer (Compiler Infrastructure)
Builds
Distributed build, test, and release mechanisms for the AWS Neuron compiler stack
Domain
Cloud infrastructure, Deep Learning compilers, Hardware-software co-design
Deliverable
production ML models | infrastructure
Required skills
Distributed systems, Compiler design, Software architecture, Build automation, Testing frameworks, Code review leadership, System reliability, Cross-functional collaboration
Preferred skills
Master's degree in Computer Science
Technologies
TensorFlow, PyTorch, Jax, AWS Neuron, C++, Python
Responsibilities
Design and implement distributed execution for the Neuron compiler, create metrics and implement automation for build/test/release processes, resolve root causes of software defects, collaborate with compiler, hardware, and ML engineers
Seniority
Senior, hands-on IC with mentorship responsibilities
