Applied Scientist, Machine Learning Accelerator - Annapurna Labs

Amazon • Cupertino, CA

2d • $142,800 - $222,200 • Onsite

About The Position

Our team is involved with pre-silicon design verification for custom IP. A critical requirement of the verification flow is the requirement of legal and realistic stimulus of a custom Machine Learning Accelerator Chip. Content creation is built using formal methods that model legal behavior of the design and then solving the problem to create the specific assembly tests. The entire framework for creating these custom tests is developed using a SMT solver and custom software code to guide the solution space into templated scenarios. This highly visible and innovative role requires the design of this solving framework and collaborating with design verification engineers, hardware architects and designers to ensure that interesting content can be created for the projects needs.

Requirements

• 3+ years of building models for business application experience
• PhD, or Master's degree and 4+ years of CS, CE, ML or related field experience
• Experience in patents or publications at top-tier peer-reviewed conferences or journals
• Experience programming in Java, C++, Python or related language
• Experience in any of the following areas: algorithms and data structures, parsing, numerical optimization, data mining, parallel and distributed computing, high-performance computing

Nice To Haves

• Experience using Unix/Linux
• Experience in professional software development

Responsibilities

• Develop an understanding for a custom machine learning instruction set architecture.
• Model correctness of instruction streams using first order logic.
• Create custom API's to allow control over scheduling and randomness.
• Deploy algorithms to ensure concurrent code is safely constructed.
• Create coverage metrics to ensure solution space coverage.
• Use novel methods like machine learning to automate content creation.

Benefits

• health insurance (medical, dental, vision, prescription, Basic Life & AD&D insurance and option for Supplemental life plans, EAP, Mental Health Support, Medical Advice Line, Flexible Spending Accounts, Adoption and Surrogacy Reimbursement coverage)
• 401(k) matching
• paid time off
• parental leave
• sign-on payments
• restricted stock units (RSUs)

What This Job Offers

• Job Type: Full-time
• Career Level: Senior