1,433 open roles
Tech Lead, Research Scientist - DPU & AI Infra
Job description
About the Team
The ByteDance DPU (Data Processing Unit) team is building the foundational computing infrastructure for ByteDance and Volcano Engine Public Cloud. Our mission is to advance the architecture, development, and research of next-generation software-hardware technologies across compute, networking, and storage for cloud and AI computing. Our technology stack spans
- Cloud virtualization & hypervisors
- High-performance user-space network protocols (DPDK, RDMA, etc.)
- High-speed interconnect and virtual switching
- Distributed storage acceleration
- GPU virtualization and scheduling for AI/ML workloads We work at the intersection of software systems, distributed infrastructure, and custom hardware acceleration, shaping the next wave of cloud-scale computing.
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.
Qualifications
Minimum 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.
Preferred Qualifications
- Ph.D. in related fields with research training and publications.
- Experience with software-hardware co-design (networking, storage, or distributed compute).
- Hands-on experience with network virtualization (OVS, SR-IOV, eBPF).
- Familiarity with DPDK and high-performance user-space networking.
- Bonus points for hardware acceleration experience, FPGA/ASIC/GPU/CUDA
- Bonus points for experience with NCCL Collectives along with AI communication patterns and parallelization techniques
- Proven experience designing and building AI/ML infrastructure related but not limited to inference kv cache system, data preprocessing system.
Description copied from ByteDance's careers page. Read the full posting before you apply.
More jobs at ByteDance
Talent Acquisition Partner (Global AI To B) - BytePlus
ByteDance· San Jose, California, United StatesTalent Acquisition Partner (Global AI To B) - BytePlus
ByteDance· London, England, United KingdomCorporate Finance Associate - Group FP&A
ByteDance· Hong Kong (China), Hong Kong Island, Hong Kong, ChinaBackend Software Engineer (SRE) - Cloud Infrastructure
ByteDance· SingaporeCustomer Success Manager - Lark Japan
ByteDance· Tokyo, Japan
More jobs in San Jose
Advanced AI for Industry & Society Staff TA - College of Engineering - Integrated Innovation Institute
Carnegie Mellon University· Silicon Valley, CAWarehouse Part Time Overnight
Lowe's· San Jose, CA (S San Jose) 1756· $40k – $41kRetail Manager - CLUB Membership
Bass Pro Shops· San Jose, CA· $70k – $83kSoftware Architect - PNS AI Governance
TikTok· San Jose, California, United StatesLead Organizer
Californians for Justice· San Jose, CA 95133· $82k