ByteDance
Student Researcher (Seed - Infra - Training) - 2026 Start (PhD)
ByteDance, Seattle, Washington, us, 98127
Student Researcher (Seed - Infra - Training) - 2026 Start (PhD)
Introduction 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. 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 basic understanding of how GPU and/or ASIC works. Expert in at least one or two programming languages in 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 The hourly rate range for this position is $85- $85. 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. EEO Statement ByteDance is committed to creating an inclusive space where employees are valued for their skills, experiences, and unique perspectives. Our platform connects people from across the globe and so does our workplace.
#J-18808-Ljbffr
Introduction 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. 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 basic understanding of how GPU and/or ASIC works. Expert in at least one or two programming languages in 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 The hourly rate range for this position is $85- $85. 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. EEO Statement ByteDance is committed to creating an inclusive space where employees are valued for their skills, experiences, and unique perspectives. Our platform connects people from across the globe and so does our workplace.
#J-18808-Ljbffr