Applied Scientist, Neuron ARG, Annapurna ML
Core
Design, develop, and deploy analyzers for ML compiler stages and compiler IRs to optimize application models for deep learning and generative AI.
Role type
Senior Applied Scientist (Compiler & Program Analysis)
Builds
Optimized deep learning compiler stack and business-critical tooling for Trainium accelerators
Domain
Deep Learning Systems, Compiler Technology, Program Analysis
Deliverable
production ML models | product features
Required skills
Programming languages, compilers, program analyzers, theorem provers, program synthesis engines, algorithms and data structures, parsing, numerical optimization, parallel and distributed computing, high-performance computing
Preferred skills
Patents or publications at top-tier peer-reviewed conferences, experience with ML and AI accelerators
Technologies
Java, C++, Python, PyTorch, JAX
Responsibilities
Architect and implement business-critical tooling, publish research, mentor a team of scientists and engineers, tackle challenges in compiler technology and deep-learning systems software
Seniority
Senior, hands-on IC with mentorship