Job Details

Overview

Brahma is a specialty recruitment firm representing an innovative start-up developing solutions to accelerate chip design. We're looking for an ML engineer to push the frontier of AI in semiconductors. This is not a standard ML role; you'll be building and shipping state-of-the-art models, reinforcement learning systems, and multi-agent architectures that directly operate on real-world RTL, chip architecture, and physical design workflows. You'll be one of the first engineers, shaping the technical foundation of a company tackling to solve one of the missing pieces in the AI ecosystem. Location: San Francisco Bay Area.

What You'll Do

• Train, fine-tune, and evaluate cutting-edge models for RTL, architecture, and PD tasks.
• Design and deploy multi-agent systems for RTL code generation, design verification, and PD automation—leveraging MCP servers, toolchains, and real chip workflows.
• Own the entire ML lifecycle from raw data pipelines to live deployment inside our production IDE.
• Collaborate cross-functionally with hardware engineers, product, and design to build AI-first chip design workflows.
• Experiment at the frontier with LLMs, diffusion models, RL (especially test-time scaling), and graph-based approaches for EDA.

What We're Looking For

• Deep expertise in reinforcement learning, multi-agent systems, or large-scale model training—with a track record of shipping systems, not just papers.
• Strong engineering skills in Python + ML frameworks (PyTorch, JAX, TensorFlow) and distributed training/inference.
• Ability to move fast, prototype, and scale research into production.
• Obsession with pushing state-of-the-art performance in real-world constraints.

Seniority level

• Associate

Employment type

• Full-time

Job function

• Technology
• Information and Media
• Semiconductor Manufacturing