Machine Learning Engineer – Deep Model Infrastructure 2985

San Jose, US-United States
Posted 3 months ago
About The Company

This company pioneers short-form video creation and social engagement, boasting a vast, engaged user base. Its platform empowers users with creative tools, filters, and effects. With a diverse content ecosystem, it’s a hub of creativity and expression. The proprietary algorithm ensures personalized content feeds, enhancing user engagement and satisfaction. This company wields significant influence on digital media, making it an invaluable partner for innovative collaborations and marketing endeavors.

About the team

The mission of our AML team is to push next-generation machine learning algorithms and platform for the recommendation system, ads ranking and search ranking in our company. We also drive substantial impact on core businesses of the company. Currently, we are looking for Software Engineer – Deep Model Infrastructure to join our team to support and advance that mission.


– Responsible for the design and implementation of a deep model infrastructure for feeds, ads and search ranking models.
– Responsible for profiling and optimizing both training and inference frameworks to ensure efficient use of resources.
– Work closely with the algorithm department to achieve joint optimization of the algorithm and system.

Minimum qualification

– Proficient in C++/CUDA/Python, and have solid programming skills.
– Familiar with deep learning frameworks (Pytorch/TensorFlow).
– Ability to work independently and complete projects from beginning to end and in a timely manner.
– Good communication and teamwork skills to clearly communicate technical concepts with other teammates.

Preferred Qualifications

– Experience in Deep Model training/serving infrastructure (e.g., Large Language Model).
– Experience in Model Parallel framework (e.g., Megatron/DeepSpeed).
– Experience contributing to an open sourced machine learning framework (Pytorch/TensorFlow).

Job Features

Job CategoryAI Engineering
SeniorityJunior / Mid IC
Base Salary$132,000 - $252,000

Apply Online