ML Systems, LLM Post-Training

San Josefull time

Apply on Employer Site

On Site

$150k

H1B: no

Alpha Design AI Inc ML Systems Engineer role focused on optimizing large language model inference systems for an agentic AI platform. Responsible for performance optimizations, benchmarking, and multi-node cluster architecture.

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.

Qualification

Experience with large-scale ML systemsDeep understanding of GPU architecture

Required

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.

Preferred

Familiarity with distributed computing frameworks such as Ray and multi-node training/inference

Benefits

Dental
Vision
401(k)
PTO
Equity/RSU

Company

Alpha Design AI Inc

View company page

Source: ashby

Active within the past hour

More roles at Alpha Design AI Inc

Technical Recruiter

San Jose · Remote

Forward Deployed Engineer

San Jose

STA/Timing Closure Lead

San Jose

Senior AI Engineer, Agents

San Jose