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

Boeing Logo Boeing

Engineering Manager

Aerospace • Information Technology • Software • Cybersecurity • Design • Defense • Manufacturing
In-Office
Long Beach, CA, USA
170000 Employees
151K-205K Annually

Boeing Logo Boeing

Industrial Security Specialist - Millennium Space Systems

Aerospace • Information Technology • Software • Cybersecurity • Design • Defense • Manufacturing
In-Office
El Segundo, CA, USA
170000 Employees
77K-139K Annually

Boeing Logo Boeing

Space Electronics Packaging and Mechanical Product Design and Analysis Engineer (Subject Matter Expert)

Aerospace • Information Technology • Software • Cybersecurity • Design • Defense • Manufacturing
In-Office
El Segundo, CA, USA
170000 Employees
146K-198K Annually

Boeing Logo Boeing

Operations Specialist

Aerospace • Information Technology • Software • Cybersecurity • Design • Defense • Manufacturing
In-Office
El Segundo, CA, USA
170000 Employees
86K-126K 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