Research Scientist in ML Systems

$184,300 - $337,250/Yr

Bytedance - Seattle, WA

Full-time - Mid Level

Seattle, WA

Professional, Scientific, and Technical Services

About the position

The Research Scientist in ML Systems at ByteDance is responsible for advancing machine learning systems through research and development, focusing on heterogeneous computing architecture, deployment, and optimization of AI algorithms. This role is pivotal in driving technological advancements within the Doubao (Seed) Team, which is dedicated to pioneering AI foundation models and enhancing the capabilities of ByteDance's products.

Responsibilities

• Research and develop machine learning systems, including heterogeneous computing architecture, management, and monitoring.
• Deploy machine learning systems, including distributed task scheduling, training, and inference.
• Manage cross-layer optimization of system and AI algorithms and hardware for machine learning (GPU, FPGA, ASIC).

Requirements

• Master's degree or above in distributed, parallel computing principles with knowledge of recent advances in computing, storage, networking, and hardware technologies.
• Familiarity with machine learning algorithms and platforms.
• Basic understanding of GPU, FPGA, and ASIC operations.
• Expertise in at least one or two programming languages in a Linux environment: C/C++, CUDA, Python.

Nice-to-haves

• Experience with GPU-based high-performance computing and RDMA high-performance networks (NCCL).
• Familiarity with TensorFlow, JAX, PyTorch, or other deep learning frameworks.
• Experience in large-scale data processing and parallel computing.
• Experience in designing and operating large-scale systems in cloud computing or machine learning.

Benefits

• 100% premium coverage for employee medical insurance, approximately 75% for dependents, and a Health Savings Account (HSA) with company match.
• Dental, Vision, Short/Long term Disability, Basic Life, Voluntary Life, and AD&D insurance plans.
• 10 paid holidays per year plus 17 days of Paid Personal Time Off (PPTO) and 10 paid sick days per year.
• 12 weeks of paid Parental leave and 8 weeks of paid Supplemental Disability.
• Mental and emotional health benefits through EAP and Lyra.
• 401K company match, gym, and cellphone service reimbursements.