Talent.com
Machine Learning Co-Design Researcher

Machine Learning Co-Design Researcher

ETCHED LLCSan Jose, CA, United States
30+ days ago
Job type
  • Full-time
Job description

About Etched

Etched is building the world's first AI inference system purpose-built for transformers - delivering over 10x higher performance and dramatically lower cost and 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. Backed by hundreds of millions from top-tier investors and staffed by leading engineers, Etched is redefining the infrastructure layer for the fastest growing industry in history.

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.

    Create a job alert for this search

    Machine Learning Researcher • San Jose, CA, United States

    Related jobs
    • Promoted
    Machine Learning Researcher

    Machine Learning Researcher

    OpenReqCupertino, CA, United States
    Full-time
    With Etched ASICs, we have fundamentally different constraints than existing AI chips.We can parallelize workloads and digest. Sohu enables entirely new research directions and products.When our chi...Show moreLast updated: 30+ days ago
    • Promoted
    AIML - Machine Learning Researcher, MLR

    AIML - Machine Learning Researcher, MLR

    AppleCupertino, CA, United States
    Full-time
    Play a part in building the next revolution in machine learning technology.We’re looking for passionate researchers to work on ambitious, curiosity-driven, long-term research projects.In this role,...Show moreLast updated: 30+ days ago
    • Promoted
    Machine Learning Research Engineer - Robotics

    Machine Learning Research Engineer - Robotics

    Scale AI, Inc.San Francisco, CA, United States
    Full-time
    Scale's Robotics business unit is dedicated to solving the data bottleneck in Physical AI.This position will be a key contributor in conducting applied research in Robotics and developing ML pipeli...Show moreLast updated: 30+ days ago
    • Promoted
    • New!
    Senior Machine Learning Engineer / Researcher (SF)

    Senior Machine Learning Engineer / Researcher (SF)

    MedalSan Francisco, CA, United States
    Full-time
    Senior Machine Learning Engineer / Researcher.Highlight AI is a cutting-edge desktop assistant designed to enhance productivity by seamlessly integrating with your workflow.We empower users to inte...Show moreLast updated: 19 hours ago
    • Promoted
    Machine Learning Research Engineer, Enterprise ML Systems

    Machine Learning Research Engineer, Enterprise ML Systems

    Scale AI, Inc.San Francisco, CA, United States
    Full-time
    AI is becoming vitally important in every function of our society.At Scale, our mission is to accelerate the development of AI applications. For 9 years, Scale has been the leading AI data foundry, ...Show moreLast updated: 14 days ago
    • Promoted
    Machine Learning Research Engineer, Agents - Enterprise GenAI

    Machine Learning Research Engineer, Agents - Enterprise GenAI

    Scale AI, Inc.San Francisco, CA, United States
    Full-time
    AI is becoming vitally important in every function of our society.At Scale, our mission is to accelerate the development of AI applications. For 9 years, Scale has been the leading AI data foundry, ...Show moreLast updated: 11 days ago
    • Promoted
    Machine Learning Engineer, Applied Research and Model Development

    Machine Learning Engineer, Applied Research and Model Development

    LIGHTFIELD INCSan Francisco, CA, United States
    Full-time
    Machine Learning Engineer, Applied Research and Model Development.Get AI-powered advice on this job and more exclusive features. This range is provided by Lightfield.Your actual pay will be based on...Show moreLast updated: 30+ days ago
    • Promoted
    Machine Learning Researcher

    Machine Learning Researcher

    Cognition CorpSan Francisco, CA, United States
    Full-time
    We are an applied AI lab building end-to-end software agents.We're building collaborative AI teammates that enable engineers to focus on more interesting problems and empower engineering teams to s...Show moreLast updated: 30+ days ago
    • Promoted
    • New!
    Machine Learning Researcher

    Machine Learning Researcher

    EPM ScientificSan Francisco, CA, United States
    Full-time
    Machine Learning Researcher - Generative Modeling for Drug Discovery.A talent-dense AI4Biology team is applying frontier-scale AI to model biological systems and accelerate therapeutic development....Show moreLast updated: 19 hours ago
    • Promoted
    • New!
    Machine Learning Researcher

    Machine Learning Researcher

    Barrington JamesSan Francisco, CA, United States
    Full-time
    An innovative superintelligence start-up is seeking a talented ML Researcher to help build next?generation intelligence reasoning models, shaping the future of AI that reasons across knowledge, con...Show moreLast updated: 19 hours ago
    • Promoted
    Machine Learning Research Engineer - Robotics

    Machine Learning Research Engineer - Robotics

    Scale AISan Francisco, CA, United States
    Full-time
    Scale's Robotics business unit is dedicated to solving the data bottleneck in Physical AI.This position will be a key contributor in conducting applied research in Robotics and developing ML pipeli...Show moreLast updated: 30+ days ago
    • Promoted
    • New!
    Machine Learning Engineer, Foundation Model

    Machine Learning Engineer, Foundation Model

    StripeSan Francisco, CA, United States
    Full-time
    Stripes mission is to accelerate global economic and technological development.We offer financial infrastructure and a variety of services to serve the needs of a wide range of users, from startups...Show moreLast updated: 19 hours ago
    • Promoted
    Machine Learning Research Engineer - Large Language Models (LLMs), Siri Core Modeling

    Machine Learning Research Engineer - Large Language Models (LLMs), Siri Core Modeling

    Apple Inc.Cupertino, CA, United States
    Full-time
    Machine Learning Research Engineer - Large Language Models (LLMs), Siri Core Modeling.Cupertino, California, United States Machine Learning and AI. Join the Siri Planner team as a Machine Learning R...Show moreLast updated: 3 days ago
    • Promoted
    • New!
    Machine Learning Research Engineer

    Machine Learning Research Engineer

    KiddomSan Francisco, CA, United States
    Full-time +1
    This range is provided by Kiddom.Your actual pay will be based on your skills and experience talk with your recruiter to learn more. Kiddom is a groundbreaking educational platform that promotes stu...Show moreLast updated: 18 hours ago
    • Promoted
    Machine Learning Researcher / ML-Ops Engineer

    Machine Learning Researcher / ML-Ops Engineer

    Rivet Industries, Inc.Palo Alto, CA, United States
    Full-time
    Machine Learning Researcher / ML-Ops Engineer.Rivet is an American company building integrated task systems — fusing hardened hardware with software, sensors, AI, and networking — for industrial wo...Show moreLast updated: 23 days ago
    • Promoted
    • New!
    Machine Learning Research Engineer

    Machine Learning Research Engineer

    Coders ConnectSan Francisco, CA, United States
    Full-time
    We're partnering with a frontier AI startup that's redefining how models learn to understand.They're collaborating with leading AI labs and building new methods that help models reason about creati...Show moreLast updated: 18 hours ago
    • Promoted
    • New!
    2026 PhD Machine Learning / Artificial Intelligence Engineering Co-Op / Intern

    2026 PhD Machine Learning / Artificial Intelligence Engineering Co-Op / Intern

    Advanced Micro Devices, Inc.San Jose, CA, United States
    Full-time
    WHAT YOU DO AT AMD CHANGES EVERYTHING.At AMD, our mission is to build great products that accelerate next-generation computing experiences-from AI and data centers, to PCs, gaming and embedded syst...Show moreLast updated: 18 hours ago
    • Promoted
    Machine Learning Research Lead, Security & Policy Research Lab

    Machine Learning Research Lead, Security & Policy Research Lab

    Scale AI, Inc.San Francisco, CA, United States
    Full-time
    As the leading data and evaluation partner for frontier AI companies, Scale plays an integral role in understanding the capabilities and safeguarding AI models and systems.Building on this expertis...Show moreLast updated: 25 days ago