Machine Learning Engineer

DeepMotion is hiring a Machine Learning Engineer to create the server-side ML engine behind interactive AI motion capture and motion synthesis.

Responsibilities

• Develop and train algorithms and models for motion perception and motion generation
• Design, build, and maintain large-scale text, video, and motion datasets for multi-model training
• Develop, deploy, and optimize inference frameworks for motion perception and generation in a cloud environment
• Research, analyze, develop, and test machine learning components that support product strategies and the technical roadmap

Requirements

• BS or MS in Computer Science or Engineering, specializing in computer vision, machine learning, robotics, or artificial intelligence
• Solid knowledge of at least one machine learning research framework (e.g., PyTorch, Tensorflow)
• Solid knowledge of at least one high-performance inference framework (e.g., TensorRT, Apache TVM)
• Experience profiling and optimizing deep neural networks, including GPU profiling tools such as NVIDIA Nsight

Nice to Have

• Familiarity with Python-based image and video manipulation/encoding/decoding frameworks such as OpenCV and PyAV
• Experience with cloud orchestration systems such as Kubernetes and cloud providers such as AWS, GCP, and Azure
• Knowledge of transformers, diffusion models, and multimodal discriminative and generative models
• Ability to build robust, maintainable client-server architectures and APIs

Tech Stack

• PyTorch, Tensorflow, TensorRT, Apache TVM
• NVIDIA Nsight, OpenCV, PyAV
• Kubernetes, AWS, GCP, Azure
• Python

Location: San Francisco Bay Area, CA (onsite)