ByteDance · 3 days ago

Research Scientist Intern - Machine Learning System

San Jose, CA

Internship

Onsite

Intern Level

$60/hr - $60/hr

Wonder how qualified you are to the job?

Maximize your interview chances

ContentData Mining

Comp. & Benefits

Insider Connection @ByteDance

Discover valuable connections within the company who might provide insights and potential referrals, giving your job application an inside edge.

Responsibilities

Research and develop 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 general purpose training framework features and model specific optimizations.

Improve efficiency and stability for extremely large scale distributed training jobs.

Qualification

Find out how your skills align with this job's requirements. If anything seems off, you can easily click on the tags to select or unselect skills to reflect your actual expertise.

Machine Learning AlgorithmsPyTorchJaxGPUASICProgramming LanguagesLinux EnvironmentC/C++CUDAPythonWork AuthorizationHigh Performance ComputingRDMA NetworkMPINCCLIbverbsDistributed TrainingDeepSpeedFSDPMegatronGSPMDAI Compiler StacksTorch.fxXLAMLIRLarge Scale Data ProcessingParallel ComputingCloud ComputingMachine LearningCUDA Programming

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 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.