Student Researcher (Seed - Infra - Inference) - 2026 Start (PhD) jobs in United States
cer-icon
Apply on Employer Site
company-logo

ByteDance · 6 hours ago

Student Researcher (Seed - Infra - Inference) - 2026 Start (PhD)

ByteDance is a leading technology company dedicated to pioneering advanced AI foundation models. They are seeking talented individuals for a PhD internship to actively contribute to their products and research, focusing on machine learning systems and distributed training.

ContentData MiningInternetSocial Media
check
Comp. & Benefits
check
H1B Sponsor Likelynote

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

Qualification

Distributed computingMachine learning algorithmsProgramming languagesGPU architectureHigh performance computingData processingCollaborationProblem solvingCommunication

Required

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

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)

Benefits

Interns have day one access to health insurance
Life insurance
Wellbeing benefits and more
Interns also receive 10 paid holidays per year
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

Company

ByteDance

company-logo
ByteDance is a technology company that develops content creation platforms and services.

H1B Sponsorship

ByteDance has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (1350)
2024 (1123)
2023 (775)
2022 (487)
2021 (417)
2020 (245)

Funding

Current Stage
Late Stage
Total Funding
$9.8B
Key Investors
Capital TodayG42Tiger Global Management
2025-11-20Secondary Market· $300M
2024-07-25Secondary Market
2023-03-14Secondary Market· $100M

Leadership Team

leader-logo
Jochen Bischoff
Head of Global Business Solutions - Africa
linkedin
leader-logo
Matty Lin
General Manager, Global Business Solutions, KR
linkedin
Company data provided by crunchbase