Tech Lead: AI Infra & DPU Research

ByteDance; San Jose (CA); On-site; USD 244,800 - 588,000; Full time

Benefits offered by this job

Medical, dental, and vision insurance

401(k) with company match

Paid parental leave

Disability coverage

Life insurance

Wellbeing benefits

Job summary

ByteDance is seeking an Tech Lead, Research Scientist for DPU and AI Infrastructure in San Jose. The role focuses on building foundational DPU-based infrastructure for ByteDance and Volcano Engine Public Cloud, spanning software-hardware co-design, high-performance networking, and AI/ML acceleration.

You will design DPU network software with low latency, collaborate with hardware teams, and drive architecture and long-term research directions in cloud-scale AI computing.

Qualifications

• B.S./M.S. in Computer Science, Computer Engineering, or related fields; or Ph.D. with strong research/publications.
• 5+ years of relevant industry experience (exception for Ph.D. with strong background).
• Proficiency in C/C++ development and debugging.
• Strong Linux systems development experience.
• Solid understanding of compute, network architecture, and operating systems.
• Background in at least one of: software-hardware co-design, distributed systems, high-performance networking, or AI/ML systems.

Responsibilities

• Design and develop DPU network software with a focus on high performance, low latency, and reliability.
• Collaborate with hardware teams to build software-hardware co-design solutions for networking and storage acceleration.
• Explore AI/ML infrastructure acceleration, leveraging DPUs, GPUs, and custom hardware to optimize distributed training and inference.
• Drive end-to-end performance optimization, from OS kernels and drivers to user-space runtime systems.
• Contribute to architecture design, technical proposals, and long-term research directions.

Skills

C/C++

Linux

Networking

AI/ML

Education

B.S./M.S. in Computer Science/Computer Engineering; Ph.D. with publications

Tools

DPDK

SR-IOV

eBPF

Job description

ByteDance is seeking an Tech Lead, Research Scientist for DPU and AI Infrastructure in San Jose. The role focuses on building foundational DPU-based infrastructure for ByteDance and Volcano Engine Public Cloud, spanning software-hardware co-design, high-performance networking, and AI/ML acceleration.

You will design DPU network software with low latency, collaborate with hardware teams, and drive architecture and long-term research directions in cloud-scale AI computing.