{bc}
company_site

Research Engineer - LLM/VLM Inference Optimization (Seed Infra)

ByteDance
Seattle, USA
Full-time
Mid
Onsite
Discovered 1 weeks ago
CC++PythonPyTorchTensorFlowCUDA
Free

Job Fit Check

Base Career helps you apply smarter for this job.

?%
Ready to Scan

Key skills for this role

CC++Python
Smart Apply

Full Job Posting

About the Team The Seed Infrastructures team oversees the distributed training, reinforcement learning framework, high-performance inference, and heterogeneous hardware compilation technologies for AI foundation models.

Responsibilities

  1. Design, develop, and optimize high-performance inference systems for large-scale LLMs and VLMs, covering inference engines, serving frameworks, and end-to-end deployment pipelines. 2. Build state-of-the-art model inference engines through advanced performance optimization techniques such as compiler-level optimizations, parallel computing, graph fusion, efficient CUDA kernel development, low-precision computation, streaming inference, speculative decoding, and high-concurrency request optimization. 3. Collaborate closely with other research teams to identify performance bottlenecks, conduct in-depth performance analysis, and optimize large models; contribute to the development of model toolchains and the broader technical ecosystem.

Minimum Qualifications: 1.

Bachelor's degree or above in Computer Science, Electrical Engineering, Software Engineering, or a related field. 2.

Strong proficiency in C/C++ and Python; solid foundations in algorithms, data structures, and systems programming; familiarity with containerization and server-side debugging. 3.

Hands-on experience with at least one mainstream machine learning framework (e.g., PyTorch, TensorFlow). 4.

Experience

deploying or optimizing LLM/VLM inference at production scale, with demonstrated impact on latency, throughput, or serving cost. 5.

Familiarity with GPU architecture and experience optimizing compute-intensive operators (e.g., FlashAttention, GEMM, GEMV, Conv2D).

Preferred Qualifications

Experience

with large-scale LLM serving infrastructure or equivalent production LLM deployment experience. 2.

Experience

in GPU programming (CUDA/OpenCL) and familiarity with frameworks such as TensorRT, Triton, or CUTLASS. 3.

Experience

in performance modeling, profiling, and optimization, or strong knowledge of CPU/GPU architectures. 4.

Familiarity with model/data parallelism frameworks for distributed inference.

Apply for this job in 1 click

Skip the repetitive application forms

Install the Base Career Chrome Extension and autofill job applications across major job boards with your profile.

Sarah M.James T.Maya R.

Trusted by over 500,000 job seekers on Base Career

Start Free Today

More from this employer

More jobs at ByteDance