Senior Machine Learning Engineer (AI Effects and Editing)

TL;DR

Senior Machine Learning Engineer (AI Effects and Editing): Designing and deploying ML models for next-generation video effects and AI editing with an accent on generative models and real-time rendering. Focus on building LoRA-based training pipelines, optimizing on-device inference, and integrating computer vision into rendering pipelines.

Location: Beijing, China

Company

A global design platform empowering users to tell compelling visual stories through intuitive and powerful video tools.

What you will do

• Design, fine-tune, and deploy ML models for AI-powered visual effects across image and video modalities.
• Build robust pipelines for LoRA-based model training, post-training quantization, and inference optimisation.
• Develop and integrate on-device ML capabilities (MediaPipe, CoreML, NNAPI) to enhance mobile runtime efficiency.
• Scale backend services for effect model serving, asset management, and real-time rendering.
• Prototype novel AI effects using diffusion, computer vision, and shader-driven approaches.
• Collaborate with design and engineering teams to transform creative ideas into user-facing features.

Requirements

• Academic or practical background in computer graphics, computer vision, or computer animation (e.g., GAN, VAE, geometric processing).
• Proven experience fine-tuning and deploying CV models, particularly for real-time or streaming use cases.
• Hands-on skills with LoRA training, model post-processing (quantization, pruning), and on-device inference optimisation.
• Proficiency in backend development using Python or Go and experience with model-serving architectures.
• Practical knowledge of computer graphics APIs (OpenGL/OpenGLES) and shader development (GLSL/HLSL).
• Full proficiency in English communications.

Culture & Benefits

• Opportunity to ship creative ML features at scale, powering real-time video editing for millions of users.
• Work within an interdisciplinary team bridging product, AI/ML, and design.
• Influence the future of intelligent creative tools through high-impact research and engineering.