Logo
Brahma Consulting Group

Machine Learning Engineer

Brahma Consulting Group, San Francisco

Save Job

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

#J-18808-Ljbffr