Job Specifications
Responsibilities
Team Intro:
The Seed Infrastructures team oversees the distributed training, reinforcement learning framework, high-performance inference, and heterogeneous hardware compilation technologies for AI foundation models.
We are looking for talented individuals to join us for an internship in 2026. PhD Internships at ByteDance aim to provide students with the opportunity to actively contribute to our products and research, and to the organization's future plans and emerging technologies.
PhD internships at ByteDance provide students with the opportunity to actively contribute to our products and research, and to the organization's future plans and emerging technologies. Our dynamic internship experience blends hands-on learning, enriching community-building and development events, and collaboration with industry experts.
Applications will be reviewed on a rolling basis - we encourage you to apply early. Please state your availability clearly in your resume (Start date, End date).
Summer Start Dates:
May 11th, 2026
May 18th, 2026
May 26th, 2026
June 8th, 2026
June 22nd, 2026
Responsibilities
Research and develop our machine learning systems, including heterogeneous computing architecture, management, scheduling, and monitoring.
Manage cross-layer optimization of system and AI algorithms and hardware for machine learning (GPU, ASIC).
Implement both general purpose training framework features and model specific optimizations (e.g. LLM, diffusions).
Improve efficiency and stability for extremely large scale distributed training jobs.
Qualifications
Minimum Qualifications
Currently in PhD program in distributed, parallel computing principles and know the recent advances in computing, storage, networking, and hardware technologies.
Familiar with machine learning algorithms, platforms and frameworks such as PyTorch and Jax.
Have a basic understanding of how GPU and/or ASIC works.
Expert in at least one or two programming languages in a Linux environment: C/C++, CUDA, Python.
Must obtain work authorization in country of employment at the time of hire, and maintain ongoing work authorization during employment.
Preferred Qualifications
GPU based high performance computing, RDMA high performance network (MPI, NCCL, ibverbs).
Distributed training framework optimizations such as DeepSpeed, FSDP, Megatron, GSPMD.
AI compiler stacks such as torch.fx, XLA and MLIR.
Large scale data processing and parallel computing.
Experiences in designing and operating large scale systems in cloud computing or machine learning.
Experiences in in-depth CUDA programming and performance tuning (cutlass, triton).
Job Information
For Pay TransparencyCompensation Description (Hourly) - Campus Intern
The hourly rate range for this position in the selected city is $57- $57.
Benefits may vary depending on the nature of employment and the country work location. Interns have day one access to health insurance, life insurance, wellbeing benefits and more. Interns also receive 10 paid holidays per year and paid sick time (56 hours if hired in first half of year, 40 if hired in second half of year). Interns who are not working 100% remote may also be eligible for housing allowance.
The Company reserves the right to modify or change these benefits programs at any time, with or without notice.
For Los Angeles County (unincorporated) Candidates:
Qualified applicants with arrest or conviction records will be considered for employment in accordance with all federal, state, and local laws including the Los Angeles County Fair Chance Ordinance for Employers and the California Fair Chance Act. Our company believes that criminal history may have a direct, adverse and negative relationship on the following job duties, potentially resulting in the withdrawal of the conditional offer of employment:
Interacting and occasionally having unsupervised contact with internal/external clients and/or colleagues;
Appropriately handling and managing confidential information including proprietary and trade secret information and access to information technology systems; and
Exercising sound judgment.
About Doubao (Seed)
Founded in 2023, the ByteDance Doubao (Seed) Team, is dedicated to pioneering advanced AI foundation models. Our goal is to lead in cutting-edge research and drive technological and societal advancements.
With a strong commitment to AI, our research areas span deep learning, reinforcement learning, Language, Vision, Audio, AI Infra and AI Safety. Our team has labs and research positions across China, Singapore, and the US.
Why Join ByteDance
Inspiring creativity is at the core of ByteDance's mission. Our innovative products are built to help people authentically express themselves, discover and connect – and our global, diverse teams make that possible. Together, we create value for our communities, inspire creativity and enrich life - a mission we work towards every day.
As ByteDancers, we strive to do great things wit
About the Company
ByteDance is a global incubator of platforms at the cutting edge of commerce, content, entertainment and enterprise services - over 2.5bn people interact with ByteDance products including TikTok.
Creation is the core of ByteDance's purpose. Our products are built to help imaginations thrive. This is doubly true of the teams that make our innovations possible.
Together, we inspire creativity and enrich life - a mission we aim towards achieving every day. At ByteDance, we create together and grow together. That's how we dri...
Know more