Backend Engineer (Model Inference), Machine Learning Systems

1 Month ago • All levels • Artificial Intelligence • Research & Development

About the job

Job Description

ByteDance's Large Model Team seeks a Backend Engineer (Model Inference) to contribute to its advanced AI large model technology. This role involves designing and developing large-scale ML system architecture, delivering end-to-end solutions for deep model inference, optimizing model performance, and managing GPU computing power. The ideal candidate possesses strong C/C++ and Python skills, experience with machine learning frameworks (TensorFlow/PyTorch/MxNet), and knowledge of GPU programming, model quantization, and cluster scheduling.
Must have:
  • Proficient in C/C++ and Python
  • Experience in large-scale ML systems
  • Familiar with TensorFlow/PyTorch/MxNet
  • Knowledge of GPU programming, model quantization
Good to have:
  • Experience with recommendation/advertising/search systems
  • Understanding of GPU hardware architecture and CUDA/cuDNN
Not hearing back from companies?
Unlock the secrets to a successful job application and accelerate your journey to your next opportunity.
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 ByteDance Large Model Team is committed to developing the most advanced AI large model technology in the industry, becoming a world-class research team, and contributing to technological and social development. The Large Model Team has a long-term vision and determination in the field of AI, with research directions covering NLP, CV, speech, and other areas. Relying on the abundant data and computing resources of the platform, the team has continued to invest in relevant fields and has launched its own general large model, providing multi-modal capabilities. The Machine Learning (ML) System sub-team combines system engineering and the art of machine learning to develop and maintain massively distributed ML training and inference system/services around the world, providing high-performance, highly reliable, scalable systems for LLM/AIGC/AGI. As a vital AI infrastructure for the company, our machine learning system integrates our most up-to-date R&D results in AI algorithms and systems. Come and join us, you will get the chance of building large-scale machine learning systems, and working with the best AI system and algorithm researchers and engineers. Responsibilities - Responsible for the design and development of large-scale ML system architecture such as solving technical system problems on high concurrency, reliability, scalability, etc - Develop end-to-end solutions on deep model inference for internal business units such as Search, Recommendation, and relevant Large Language Model (LLM) based systems etc - Provide highly automated and extremely performant model optimization solutions for frameworks such as PyTorch and TensorFlow. Some technical solutions includes subgraph matching, compilation optimization, model quantization, heterogeneous hardware, etc.; - Manage the large-scale GPU computing power cluster for our global businesses by improving utilization rates of the computing power through methods such as elastic scheduling, GPU overselling, and task orchestration; - Engage in cross functional collaboration with the algorithm department to conduct joint optimization of algorithms and systems.
Qualifications
Minimum Qualifications: - Proficient in C/C ++ and Python languages in the Linux environment, with relevant experience in large-scale machine learning systems or search and promote recommendation systems; - Familiar with at least one machine learning framework (Tensorflow/PyTorch/MxNet or other self-developed frameworks). - Background knowledge and experience in at least one of the following: GPU programming, compiler, model quantization, and GPU cluster scheduling. - Have the ability to solve problems independently, a good team spirit, and excellent ability to disassemble complex problems. - Have a strong sense of responsibility, good learning ability, communication skills, and self-drive. - Have good working documentation habits and write updated workflow and technical documents as required in a timely manner. Preferred Qualifications: - Experience with recommendation/advertising/search off-line inference system architecture; - Understand the GPU hardware architecture, understand the GPU software stack (CUDA, cuDNN), and have experience in GPU performance analysis. 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.
View Full Job Description

Add your resume

80%

Upload your resume, increase your shortlisting chances by 80%

About The Company

Where imagination meets innovation, delivering limitless gaming experiences.

View All Jobs

Get notified when new jobs are added by ByteDance

Similar Skill Jobs

Williams-Sonoma,  Inc  - Sr Software Engineer (Machine Learning)

Williams-Sonoma, Inc , India (On-Site)

Orion Innovation - Data Engineer-AI,ML

Orion Innovation, India (On-Site)

The Walt Disney Company - Senior Software Engineer, Machine Learning - Ad Platforms

The Walt Disney Company, United States (On-Site)

Passive Logic - AI Control Theory & Optimization Scientist

Passive Logic, United States (On-Site)

PropReturns - Senior Data Scientist

PropReturns, India (On-Site)

Get notifed when new similar jobs are uploaded

Jobs in Singapore

Virtuos - HQ accountant

Virtuos, Singapore (On-Site)

ByteDance - Backend Engineer - BytePlus

ByteDance, Singapore (On-Site)

Razer - Legal Counsel

Razer, Singapore (On-Site)

Razer - Digital Marketing Manager

Razer, Singapore (On-Site)

Get notifed when new similar jobs are uploaded