Etched is hiring : Machine Learning Co-Design Researcher in San Jose

Join to apply for the Machine Learning Co-Design Researcher role at Etched

About Etched : Etched is building AI chips that are hard-coded for individual model architectures. Our first product (Sohu) only supports transformers, but has an order of magnitude more throughput and lower latency than a B200. With Etched ASICs, you can build products that would be impossible with GPUs, like real-time video generation models and extremely deep & parallel chain-of-thought reasoning agents.

Key Responsibilities :

Translate core mathematical operations from transformer models into optimized operation sequences for Sohu

Develop and leverage a deep understanding of Sohu to co-design both HW instructions and model architecture operations to maximize model performance

Implement high-performance software components for the Model Toolkit

Collaborate with hardware engineers to maximize chip utilization and minimize latency

Implement efficient batching strategies and execution plans for inference workloads

Design and implement cutting edge inference time compute scaling methods

Alter and fine-tune model architectures or inference time compute algorithms

Contribute to the evolution of our system architecture and programming model

Representative projects :

Optimize operation sequences to maximize Sohu's computational resources for specific transformer architectures such as Llama 4.

Research and implement efficient memory management for KV cache sharing and prefix optimization

Develop algorithms for continuous batching and batch interleaving to improve throughput and / or latency

Research and implement model-specific inference-time acceleration algorithms such as speculative decoding, tree search, KV cache sharing, priority scheduling, etc by interacting with the rest of the inference serving stack

Research and implement structured decoding and novel sampling algorithms for reasoning models

You may be a good fit if you have :

Co-design expertise across both SW and HW domains

Strong software engineering skills with systems programming experience

Deep knowledge of transformer model architectures and / or inference serving stacks (vLLM, SGLang, etc.)

Strong mathematical skills, esp. in linear algebra

Ability to reason about performance bottlenecks and optimization opportunities

Experience working cross-functionally in diverse software and hardware organizations

Strong Candidates May Also Have Experience With :

Experience with hardware accelerators, ASICs, or FPGAs

Experience with Rust programming language

Deep expertise in ML systems engineering and hardware / software co-design with demonstrated impact (contributions to open-source projects or published papers)

Track record of optimizing large co-designed SW / HW systems

Benefits :

Full medical, dental, and vision packages, with generous premium coverage

Housing subsidy of $2,000 / month for those living within walking distance of the office

Daily lunch and dinner in our office

Relocation support for those moving to West San Jose

Compensation Range : $150,000 - $275,000

How We’re Different : Etched believes in the Bitter Lesson. We think most of the progress in the AI field has come from using more FLOPs to train and run models, and the best way to get more FLOPs is to build model-specific hardware. Larger and larger training runs encourage companies to consolidate around fewer model architectures, which creates a market for single-model ASICs.

We are a fully in-person team in West San Jose, and greatly value engineering skills. We do not have boundaries between engineering and research, and we expect all of our technical staff to contribute to both as needed.

#J-18808-Ljbffr