ML Research Engineer — HW Co-Design for Real-Time AI

Etched; Cupertino (CA); On-site; USD 120,000 - 160,000; Full time

Benefits offered by this job

Housing subsidy of $2,000/month

Daily lunch and dinner in the office

Relocation support for Cupertino

Job summary

A cutting-edge AI hardware company in Cupertino seeks a candidate to conduct research on custom AI chips, focusing on optimizing model performance and researching novel architectures. Responsibilities include collaborating on the software stack and interacting directly with hardware architects. Applicants should have a strong ML research background and experience with programming languages such as Python and Rust. This role emphasizes innovative solutions that are infeasible on traditional GPUs.

Qualifications

• Strong interest in HW co-design.
• Experience in distributed inference/training environments.
• Ability to conduct novel research.

Responsibilities

• Propose and conduct novel research on Sohu.
• Translate core mathematical operations into instruction sequences.
• Develop software performance for Sohu HW.

Skills

ML Research background

Experience with Python

Familiarity with transformer architectures

Tools

Pytorch

JAX

Rust

Job description

A cutting-edge AI hardware company in Cupertino seeks a candidate to conduct research on custom AI chips, focusing on optimizing model performance and researching novel architectures. Responsibilities include collaborating on the software stack and interacting directly with hardware architects. Applicants should have a strong ML research background and experience with programming languages such as Python and Rust. This role emphasizes innovative solutions that are infeasible on traditional GPUs.