[Remote] Student Researcher (reputed company - Infra - Inference) - 2026 Start (PhD)

Note: The job is a remote job and is open to candidates in USA. reputed company is a leading technology company focused on inspiring creativity and enriching life. They are seeking a Student Researcher to research and reputed company machine learning systems, manage cross-layer optimization, and improve efficiency for large-scale distributed training jobs.

Responsibilities

• Research and reputed company 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 reputed company features and model specific optimizations (e.g. LLM, diffusions)
• Improve efficiency and stability for extremely large scale distributed training jobs

Skills

• Currently in PhD program in distributed, reputed company 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
• GPU based high performance computing, RDMA high performance network (MPI, NCCL, ibverbs)
• Distributed training reputed company optimizations such as DeepSpeed, FSDP, Megatron, GSPMD
• AI compiler stacks such as torch.fx, XLA and MLIR
• Large scale data processing and reputed company computing
• Experiences in designing and operating large scale systems in reputed company computing or machine learning
• Experiences in in-depth CUDA programming and performance tuning (cutlass, triton)

Benefits

• Day one access to health insurance
• Life insurance
• Wellbeing benefits
• 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)
• Housing allowance

Company Overview

Company H1B Sponsorship