Machine Learning Research Engineer
EtchedCupertino, California, United States
Posted February 19, 2025
Full Time
About the Company
Etched specializes in developing cutting-edge servers by integrating transformer architecture directly into their chips. This innovative approach results in unparalleled processing power for transformer-based inference tasks. The company's core value proposition lies in its ability to deliver superior performance and efficiency, making it a leader in the AI hardware sector.
Job Description
Etched is building AI chips for high-performance, low-latency processing. The company focuses on democratizing generative AI through hardware-coded chips and is seeking a Machine Learning Research Engineer to contribute to research on Sohu and potentially other transformer-based models. The role involves translating model operations, developing software performance, and co-designing with HW architects.
Requirements
- An ML Research background with interests in HW co-design
- Experience with Python, Pytorch, and/or JAX
- Familiarity with transformer model architectures and/or inference serving stacks
- Experience working cross-functionally in diverse software and hardware organizations
Benefits
- Full medical, dental, and vision packages
- Housing subsidy
- Daily lunch and dinner
- Relocation support
