ML Systems, LLM Post-Training

Posted 3 Days Ago
San Jose, CA, USA
In-Office
150K-350K Annually
Entry level
Artificial Intelligence • Semiconductor
The Role
Optimize large language model inference across multi-node GPU clusters for production AI agents. Responsibilities include implementing and benchmarking performance improvements, profiling GPU and memory bottlenecks, building evaluation harnesses, integrating model architectures, and applying distributed inference, batching, quantization, and systems-level optimization techniques.
Summary Generated by Built In
Position Overview

We are seeking an ML Systems Engineer to optimize the performance and efficiency of large language model inference powering our agentic AI platform. This is a technical role focused on low-level systems optimization. You will implement performance optimizations, build evaluation harnesses, and architect multi-node clusters for training and inference that push the limits of LLM throughput and latency. Your work will directly impact the responsiveness and cost-efficiency of AI agents used by leading semiconductor companies to design chips.

Key Responsibilities
  • Design, deploy, and optimize LLM inference systems across multi-node clusters, maximizing throughput and minimizing latency for production workloads.

  • Implement and benchmark concrete inference optimizations.

  • Profile and analyze inference bottlenecks at the systems level—from GPU kernel execution to memory bandwidth constraints.

  • Build robust evaluation harnesses and benchmarking frameworks that measure accuracy, throughput, latency, and resource utilization across various parallelism strategies.

  • Collaborate with research scientists to integrate new model architectures and optimizations into production inference infrastructure.

  • Investigate and apply emerging techniques from research papers and open-source projects to continuously improve inference performance.

Qualifications
  • B.S., M.S., or PhD in Computer Science, Electrical Engineering, or a related field (or equivalent experience).

  • Experience with large-scale ML systems, GPU computing, or high-performance inference optimization.

  • Strong proficiency in Python and C++/CUDA; hands-on experience with SGLang, vLLM, PyTorch, or similar inference frameworks.

  • Deep understanding of GPU architecture, memory hierarchies, and parallel computing paradigms.

  • Experience deploying and optimizing LLMs in production: model serving, batching strategies, distributed inference, or quantization.

  • Strong systems-level debugging and profiling skills; comfort working at multiple layers of the stack from CUDA kernels to application logic.

  • Familiarity with distributed computing frameworks (Ray, multi-node training/inference) is a plus.

  • Self-directed problem solver who is interested in working on ambitious optimization challenges.

Why Join Us
  • Work on cutting-edge LLM inference optimization problems with real-world production impact.

  • Access to substantial GPU compute resources for experimentation and benchmarking.

  • Collaborate with a world-class team spanning AI research, systems engineering, and EDA.

  • Shape the performance characteristics of AI systems used by leading semiconductor companies.

What we offer
  • $150K/yr – $350K/yr + Offers Equity. We are open to discuss above-scale compensation with exceptional candidates on a case-by-case basis.

  • Unlimited PTO and full benefits (medical, vision, dental, 401k).

  • Two engineering-centric offices with free parking, private gym, and free lunch, drinks and snacks.

Skills Required

  • Bachelor’s, master’s, or PhD in Computer Science, Electrical Engineering, or a related field, or equivalent experience
  • Experience with large-scale machine learning systems, GPU computing, or high-performance inference optimization
  • Strong proficiency in Python and C++/CUDA
  • Hands-on experience with SGLang, vLLM, PyTorch, or similar inference frameworks
  • Deep understanding of GPU architecture, memory hierarchies, and parallel computing paradigms
  • Experience deploying and optimizing LLMs in production, including model serving, batching, distributed inference, or quantization
  • Strong systems-level debugging and profiling skills across CUDA kernels and application logic
  • Familiarity with distributed computing frameworks such as Ray and multi-node training or inference
  • Self-directed problem-solving ability and interest in ambitious optimization challenges
Am I A Good Fit?
beta
Get Personalized Job Insights.
Our AI-powered fit analysis compares your resume with a job listing so you know if your skills & experience align.

The Company
HQ: Santa Barbara, CA

What We Do

ChipAgents is revolutionizing the chip design and verification process with agentic AI tools tailored for the semiconductor industry. Our flagship product, ChipAgents, leverages advanced AI agents to streamline and enhance the design, debugging, and verification of RTL (Register Transfer Level) code, accelerating time-to-market for chip developers. We are committed to empowering hardware design and verification engineers by providing cutting-edge solutions that improve efficiency, scalability, and accuracy in chip development. Based in Santa Barbara, CA, Alpha Design AI is at the forefront of AI innovation in electronic design automation (EDA).

Similar Jobs

Liberty Mutual Insurance Logo Liberty Mutual Insurance

Senior Underwriting Officer - Environmental Casualty Product Line Manager

Artificial Intelligence • Fintech • Insurance • Marketing Tech • Software • Analytics
Hybrid
6 Locations
40000 Employees
98K-298K Annually
Remote or Hybrid
US
15100 Employees

CDW Logo CDW

Senior Project Manager

Information Technology
Remote or Hybrid
CA, USA
15100 Employees
107K-150K Annually

Square Logo Square

Account Manager

eCommerce • Fintech • Hardware • Payments • Software • Financial Services
Hybrid
San Francisco, CA, USA
12000 Employees
108K-203K Annually

Similar Companies Hiring

Onshore Thumbnail
Artificial Intelligence • Fintech • Software • Financial Services
New York, New York
85 Employees
Blee Thumbnail
Artificial Intelligence • Marketing Tech • Software
New York, New York
33 Employees
Vega Thumbnail
Artificial Intelligence • Automotive • Insurance • Transportation
US
65 Employees

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account