Research Engineer (Foundation Model) - Machine Learning Systems

8 Months ago • 3 Years + • System Design

Job Summary

Job Description

ByteDance's Seed Foundation Machine Learning (ML) Systems team seeks a Research Engineer to optimize large-scale parallel training for cutting-edge deep learning algorithms. This role involves researching and developing ML systems, including accelerated computing architecture, management, and monitoring, as well as deploying distributed ML training and inference. The ideal candidate will have expertise in distributed and parallel computing principles, machine learning algorithms, and high-performance computing with a strong foundation in programming languages like C/C++, CUDA, and Python. The team's focus is on reinventing ML infrastructure for large-scale language models, with publications in top-tier conferences like SIGCOMM, NSDI, EuroSys, OSDI, SOSP, MLSys, and NeurIPS.
Must have:
  • Bachelor's degree in related field
  • 3+ years of experience
  • Familiarity with ML algorithms and frameworks
  • Understanding of GPU and/or ASIC
  • Expertise in C/C++, CUDA, and Python
Good to have:
  • GPU-based high-performance computing
  • RDMA high-performance network (MPI, NCCL, ibverbs)
  • Distributed training framework optimizations
  • AI compiler stacks (torch.fx, XLA, MLIR)
  • Large-scale data processing and parallel computing
  • Experience with cloud computing or ML systems
  • In-depth CUDA programming and performance tuning

Job Details

Responsibilities
ByteDance will be prioritizing applicants who have a current right to work in Singapore, and do not require ByteDance sponsorship of a visa. About ByteDance Founded in 2012, ByteDance's mission is to inspire creativity and enrich life. With a suite of more than a dozen products, including TikTok and Helo as well as platforms specific to the China market, including Toutiao, Douyin, and Xigua, ByteDance has made it easier and more fun for people to connect with, consume, and create content. Why Join Us Creation is the core of ByteDance's purpose. Our products are built to help imaginations thrive. This is doubly true of the teams that make our innovations possible. Together, we inspire creativity and enrich life - a mission we aim towards achieving every day. To us, every challenge, no matter how ambiguous, is an opportunity; to learn, to innovate, and to grow as one team. Status quo? Never. Courage? Always. At ByteDance, we create together and grow together. That's how we drive impact - for ourselves, our company, and the users we serve. Join us. About the Team The Seed Foundation Machine Learning (ML) Systems team provides end-to-end (E2E) machine learning experience and machine learning resources for the company. The team builds heterogeneous ML training and inference systems based on GPU and AI chips and advances the state-of-the-art of ML systems technology to accelerate models such as stable diffusion and LLM. The team is also responsible for research and development of hardware acceleration technologies for AI and cloud computing, via technologies such as distributed systems, compilers, HPC, and RDMA networking. The team is reinventing the ML infra for large scale language models. We have published papers at top tier conferences such as SIGCOMM, NSDI, EuroSys, OSDI, SOSP, MLSys, NeurIPS, etc. Responsibilities: - Optimizing large scale parallel training for state of the art deep learning algorithms such as large language models, multi-modality models, diffusion, reinforcement learning, etc - Research and develop our machine learning systems, including accelerated computing architecture, management and monitoring - Deploy the machine learning systems, distributed machine learning training and inference - Manage cross-layer optimization of system and AI algorithms and hardware for machine learning (GPU, ASIC)
Qualifications
Minimum Qualification: - Bachelor or above degree in distributed, parallel computing principles and know the recent advances in computing, storage, networking, and hardware technologies; - At least 3 years or more working experiences; - 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; Preferred Qualifications: The following experiences will be a big plus: - 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) 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. At ByteDance, our mission is to inspire creativity and enrich life. To achieve that goal, we are committed to celebrating our diverse voices and to creating an environment that reflects the many communities we reach. We are passionate about this and hope you are too.

Similar Jobs

bytedance - Experienced Software Engineer - Traffic Platform

bytedance

San Jose, California, United States (On-Site)
8 Months ago
Epic Games - Senior Engine Programmer

Epic Games

(On-Site)
4 Months ago
Unity - Senior Graphics Engineer

Unity

Helsinki, Uusimaa, Finland (On-Site)
2 Months ago
magic media games - C++ Game Developer

magic media games

(Remote)
1 Month ago
Riot Games - Senior Technical Artist

Riot Games

Singapore (On-Site)
1 Day ago
Cubic corporation - Lead Project System Engineer

Cubic corporation

Hyderabad, Telangana, India (On-Site)
3 Weeks ago
Tesla - PCB Design Engineer, Electronic Systems

Tesla

Pune, Maharashtra, India (On-Site)
4 Months ago
Apple - Embedded System Software Engineer

Apple

San Diego, California, United States (On-Site)
1 Month ago
Capgemini - System Engineer (DB2 DBA

Capgemini

Mumbai, Maharashtra, India (On-Site)
1 Month ago
Accenture - Application Developer

Accenture

Bengaluru, Karnataka, India (On-Site)
13 Hours ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

Figma - Software Engineer, Mobile Platform

Figma

San Francisco, California, United States (Remote)
2 Weeks ago
Scopely - Machine Learning Engineer

Scopely

Spain (Hybrid)
2 Days ago
flying wild hog - AI Programmer

flying wild hog

(Remote)
4 Months ago
bytedance - Senior Software Development Engineer, SDN-Traffic Intelligence & Control

bytedance

San Jose, California, United States (On-Site)
3 Months ago
Avalanche Studios Group - Systems Programmer

Avalanche Studios Group

Liverpool, England, United Kingdom (Hybrid)
3 Months ago
Scorewarrior - Senior Client C++ Developer (Framework)

Scorewarrior

Limassol, Limassol, Cyprus (On-Site)
3 Months ago
Ion - Technical Consultant - Endur

Ion

Uniondale, New York, United States (On-Site)
8 Months ago
Actian - Zen Quality Assurance Engineer - Bangalore/Pune

Actian

Bengaluru, Karnataka, India (On-Site)
8 Months ago
Ion - RPA Junior Developer

Ion

Chișinău, Chisinau, Moldova (Hybrid)
1 Year ago
bytedance - Research Scientist in Large Language Model, Graduates - Doubao (Seed) - 2024 Start (PhD)

bytedance

Seattle, Washington, United States (On-Site)
8 Months ago

Get notifed when new similar jobs are uploaded

Jobs in Singapore

Sabre India - Principal, Sales & Account Management

Sabre India

Singapore (On-Site)
1 Month ago
Thales - Discipline Manager (Functional Engineering Manager)

Thales

Singapore (On-Site)
1 Month ago
Riot Games - Associate Media Specialist, APAC (Contract)

Riot Games

Singapore (On-Site)
2 Months ago
Riot Games - Senior Technical Artist

Riot Games

Singapore (On-Site)
1 Day ago
bytedance - Solution Architect (Edge Cloud)

bytedance

Singapore (On-Site)
4 Months ago
bytedance - Senior Regional Compliance Counsel, APAC

bytedance

Singapore (On-Site)
2 Months ago
HoYoverse - Brand Marketing

HoYoverse

Singapore (On-Site)
2 Months ago
Interactive Brokers - Application Service Group Associate

Interactive Brokers

Singapore (On-Site)
3 Weeks ago
Garena - Site Reliability Engineer/Senior Site Reliability Engineer

Garena

Singapore (On-Site)
2 Months ago
Microsoft - Senior Strategic Account Executive, Public Sector

Microsoft

Singapore (On-Site)
2 Months ago

Get notifed when new similar jobs are uploaded

System Design Jobs

Open Systems Technologies - System Engineer Monitoring

Open Systems Technologies

Saarlouis, Saarland, Germany (On-Site)
6 Days ago
Apple - DC-DC Power System Design Engineer

Apple

Austin, Texas, United States (On-Site)
4 Weeks ago
adglobal 360 - System Engineer

adglobal 360

Shibuya, Tokyo, Japan (On-Site)
8 Months ago
NVIDIA - Senior Deep Learning Research Engineer, Advanced AI Systems

NVIDIA

Santa Clara, California, United States (On-Site)
3 Months ago
Crowd Strick - Senior Engineer, Backend - Channel File Systems

Crowd Strick

Canada (Remote)
1 Month ago
Intel  - BT System Integration Engineer

Intel

Bengaluru, Karnataka, India (On-Site)
3 Weeks ago
Take-Two Interactive - Application Security Architect

Take-Two Interactive

Toronto, Ontario, Canada (On-Site)
2 Months ago
Capgemini - System Engineer (SQL DBA)

Capgemini

Bengaluru, Karnataka, India (On-Site)
1 Month ago
bytedance - Optical System Engineer

bytedance

San Jose, California, United States (On-Site)
3 Months ago
CME Group - Systems Engineer II

CME Group

Bengaluru, Karnataka, India (On-Site)
1 Month ago

Get notifed when new similar jobs are uploaded

About The Company

Founded in 2012, ByteDance's mission is to inspire creativity and enrich life. With a suite of more than a dozen products, including TikTok as well as platforms specific to the China market, including Toutiao, Douyin, and Xigua, ByteDance has made it easier and more fun for people to connect with, consume, and create content.

San Jose, California, United States (On-Site)

San Jose, California, United States (On-Site)

Seattle, Washington, United States (On-Site)

Seattle, Washington, United States (On-Site)

San Jose, California, United States (On-Site)

View All Jobs

Get notified when new jobs are added by bytedance

Level Up Your Career in Game Development!

Transform Your Passion into Profession with Our Comprehensive Courses for Aspiring Game Developers.

Job Common Plug