AI Engineer
Required skills
- PyTorch
- DeepSpeed
- FSDP
- Ray
- Kubernetes
- TensorRT
- ONNX
- Triton Inference Server
- CUDA
- Python
Nice to have
- NeurIPS
- CVPR
- ICCV
- SIGGRAPH
- ICLR
- ICML
Role summary
LiveX AI is seeking an AI Engineer to train, fine-tune, and serve generative models for real-time interactive AI avatars. This role involves the full model lifecycle, from data curation and training diffusion, video, and multimodal models to optimizing for low-latency inference in live streaming. Responsibilities include designing data pipelines, optimizing models for real-time serving using techniques like distillation and quantization, and building training and deployment infrastructure on GPU clusters. The engineer will define and track quality metrics, collaborate with cross-functional teams, and stay current with AI research in generative video and neural rendering.
LiveX AI · AI Engineer (Machine Learning Engineer)Palo Alto, CA · Full-time, on-site · Department: EngineeringSource: https://livex.ai/careers/ai-engineer (posted May 27, 2026)
About the Role
LiveX AI is building realtime, interactive AI avatars: digital humans that see, listen, and respond with natural speech and expressive motion. This engineer trains, fine-tunes, and serves the generative models behind these avatars across the full model lifecycle, from data curation and training diffusion, video, and multimodal models to low-latency inference for live streaming.
Responsibilities
Train and fine-tune generative models for talking-head synthesis, audio-driven facial animation, and full-body avatar video (diffusion models, DiT, 3D Gaussian Splatting, GAN-based, transformer-based video generators).
Design data pipelines for large-scale video, audio, and multimodal datasets: collection, cleaning, labeling, alignment, augmentation.
Optimize models for realtime serving with sub-200ms end-to-end latency: distillation, quantization, pruning, caching, streaming inference, custom CUDA/Triton kernels where needed.
Build and maintain training, evaluation, and deployment infrastructure on GPU clusters (PyTorch, DeepSpeed, FSDP, Ray, Kubernetes).
Define and track quality metrics (lip-sync accuracy, identity preservation, expression naturalness, latency, throughput) and drive measurable gains.
Work with backend, product, and research teams to ship avatar features and iterate on user feedback.
Keep up with research in generative video, multimodal AI, and neural rendering; bring useful techniques into the stack.
Qualifications
3+ years of hands-on ML engineering, or strong research experience with production-quality code, with a record of shipping deep learning models.
Deep expertise in PyTorch and modern DL workflows; able to read, reproduce, and extend research papers.
Strong background in one or more of: diffusion models, video generation, generative AI, computer vision, multimodal (audio+vision) AI, speech synthesis, neural rendering.
Solid Python engineering; distributed training on multi-GPU / multi-node setups.
Proven realtime inference optimization: TensorRT, ONNX, Triton Inference Server, CUDA, mixed precision, streaming pipelines.
Comfortable in ambiguous, fast-changing environments; proactive, ownership-driven.
Fast-learning startup mentality; eager to push what interactive avatars can do.
Nice-to-Haves
Talking-head, audio-driven animation, lip-sync, or digital human systems; 3D Gaussian Splatting, NeRF, face reenactment, avatar synthesis.
Publications at top venues (NeurIPS, CVPR, ICCV, SIGGRAPH, ICLR, ICML) or notable open-source contributions.
Compensation
Competitive salary, equity, and benefits; access to substantial GPU compute. No range listed. Equal Opportunity Employer.
How to Apply (original posting)
Email resume and relevant supporting materials to career@livex.ai, subject "Candidate Application - AI Engineer".
Similar roles
- Member of Technical Staff — Agent Post-TrainingEmbedding VC · San Francisco, CA · Onsite
- Member of Technical Staff — RL & Agentic Model TrainingEmbedding VC · San Francisco, CA · Onsite
Backend/AI EngineerReducto · San Francisco, CA
Senior AI Engineer, IT Solutions 1 1 1Celestica · Toronto, ON · Hybrid- AI EngineerNucleo · San Francisco, CA · Hybrid
Applied AI Engineer InternParspec · San Mateo, CA · Hybrid
