Processor Performance and Modeling Architect

About Rain: 

At Rain AI, we’re creating a future with abundant and scalable artificial intelligence. We're building the world’s most cost and energy efficient hardware for AI. Our products achieve an order of magnitude improvement over the status quo by co-designing every layer of the AI stack, from circuits to algorithms. Our ultimate goal is to become the dominant hardware company of the AI era.  

Rain AI is currently a Series A stage startup and backed by world leaders in AI. Our seed round was led by Sam Altman (OpenAI). In addition, our current VC partners include Y Combinator, Daniel Gross, Jaan Tallinn, Founders X Fund, Airbus Ventures, Liquid 2 Ventures and Deepwater. 

About the role: 

At this role, you will be responsible for configuring and customizing processor cores that are used in Rain AI’s accelerator SoCs. This role has multiple aspects, including architecting/micro-architecting internal processor pipelines, developing models for customized features, analyzing the performance of the micro-kernels utilizing custom features, as well as documentation and architecture specification development. You will work in conjunction with other members of the architecture team, as well as Rain’s hardware and software teams to ensure seamless integration of processor cores into Rain’s hardware and software stack. 

This is a remote role – you can work from anywhere in the United States.  

Responsibilities: 

• Propose, model, implement and verify custom extensions to processor cores that accelerate AI workloads and kernels 

• Develop models and environments that streamline implementation and testing of customized features 

• Closely collaborate with hardware and software team members to ensure proper integration and implementation of the customized processors in the SoC’s hardware and software stack 

• Develop architectural specification and documentation for processor cores and customized features to serve as reference manuals for hardware and software teams 

Qualifications:  

• MS with 5+ years of experience or PhD in Electrical or Computer Engineering, with familiarity with processor architecture and AI/ML accelerators 

• Familiarity with functional and performance modeling concepts and environments at the IP and SoC level 

• Familiarity with AI/ML workloads and their key operators 

• Deep understanding of processor architecture and micro-architecture 

• Familiarity with RISC-V, as well as its standard and custom extensions 

• Familiarity with C++ and SystemC code development 

• Familiarity with parallel architectures, such as SIMD, vector, SIMT, multi-threading, etc. 

• Strong communication skills and fluency in specification development 

• High degree of motivation and independence 

Preferred Qualifications: 

• Previous start up experience 

• Experience in design and development of AI/ML accelerators 

Our Benefits: 

• Medical Insurance with 100% coverage of employee premiums 

• Dental and Vision Insurance 

• 401k match  

• Unlimited PTO + all federal holidays  

• Company wide time off: 

• Two weeks off around Christmas and New Years 

• Summer “shutdown”: one week-off for all employees 

• Work from anywhere in the United States  

• $500 of office equipment per year 

• And more!