About Rain:

At Rain AI, we’re creating a future with abundant and scalable artificial intelligence. Were 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!