Job Description

Verified Job On Employer Career Site

Job Summary:

TikTok is the leading destination for short-form mobile video, dedicated to inspiring creativity and bringing joy. The Trust and Safety engineering team is looking for a graduate to optimize algorithm applications and develop machine learning infrastructure, focusing on high concurrency and scalability challenges.

Responsibilities:

• Work closely with business teams to optimize the integration plan for algorithm applications, improve efficiency in evaluating and using algorithm applications across various business scenarios, and reduce the cost of managing and optimizing algorithm applications in different business scenarios.

• Be responsible for the architectural design, development, and performance tuning of algorithm applications, solving technical challenges such as high concurrency, high reliability, and high scalability. Work includes multiple sub-areas: ML model training and evaluation, model optimization, model inference, model management, dataset management, workflow orchestration, etc.

• Responsible for the design and development of Machine Learning infrastructure for LLM/AIGC, etc

• Be responsible for researching and implementing cutting-edge engineering technologies related to LLM, NLP, CV.

Qualifications:

Required:

• Familiar with one or two programming languages, such as C++, Go, or Python, and knowledgeable about CUDA or deep learning frameworks (such as PyTorch, Deepspeed, Megatron, vllm, etc.).

• Understanding of the principles of distributed systems, large-scale data processing, and parallel computing.

• Interested and experienced in one or more of the following areas: machine learning, deep learning, computational acceleration, and performance optimization.

• Familiar with the ML Infrastructure of Large Model training and inference.

Preferred:

• Excellent programming skills, data structure and algorithm skills, proficient in C/C++ or Python programming language, candidates with awards in ACM/ICPC, NOI/IOI, Top Coder, Kaggle and other competitions are preferred.

• Research or industry experience in the field of machine learning, especially in large language models (LLMs) and generative artificial intelligence.

• Distributed training framework optimizations such as DeepSpeed, FSDP, Megatron, GSPMD.

• Experiences in in-depth CUDA programming and performance tuning (cutlass, triton).

• Experience with evaluation of ML models, LLM application & agent development is desirable.

• Understanding cutting-edge LLM research and engineering (e.g., long context, multi modality, active learning, alignment research, agent ecosystem, etc.) and possess practical expertise in effectively implementing these advanced systems.

Company:

TikTok is a short-form video entertainment app and social network platform. It is a sub-organization of ByteDance. Founded in 2003, the company is headquartered in Los Angeles, California, USA, with a team of 10001+ employees. The company is currently Late Stage. TikTok has a track record of offering H1B sponsorships.

Job Tags

Similar Jobs