Applied Scientist (AI)

TL;DR

Applied Scientist (Neuron ARG, Annapurna ML): Designing, developing, and deploying analyzers for ML compiler stages and compiler IRs with an accent on artificial intelligence and program analysis in deep learning compiler stack. Focus on architecting business-critical tooling for Trainium accelerator, publishing research, and innovating on generative AI capabilities.

Location: USA, WA, Seattle or USA, CA, Cupertino

USA, CA, Cupertino - $171,600 - $222,200 USD annually; USA, WA, Seattle - $142,800 - $193,200 USD annually

Company

hirify.global AI's Automated Reasoning Group within the Neuron team at Annapurna Labs, building state-of-the-art deep learning compiler stack for Trainium ML accelerator.

What you will do

• Design and develop analyzers for ML compiler stages and intermediate representations (IRs).
• Architect and implement business-critical tooling for compiler technology.
• Publish research and mentor scientists and engineers.
• Tackle challenges in programming languages, compilers, and deep-learning systems software.
• Collaborate with engineering team on Trainium accelerator innovations for generative AI.

Requirements

• PhD or Master's degree + 4+ years in CS, CE, ML or related field
• Experience in patents or publications at top-tier conferences/journals
• Programming in Java, C++, Python or related
• Experience in algorithms/data structures, parsing, numerical optimization, data mining, parallel/distributed computing, high-performance computing
• Location: Must be in USA (Seattle, WA or Cupertino, CA)

Nice to have

• PhD in computer science, computer engineering, or related field
• Background in machine learning and AI accelerators
• Experience in programming languages, compilers, program analyzers, theorem provers, program synthesis

Culture & Benefits

• Comprehensive benefits: health insurance (medical, dental, vision), 401(k) matching, paid time off, parental leave.
• Sign-on payments and restricted stock units (RSUs).
• Inclusive culture empowering high performance and customer focus.
• Accommodations for disabilities during hiring process.