AI Engineer

Embedding VC
Palo Alto, CA, United States
12 days ago
Apply on www.indeed.com
Prepare application

Role details

Contract type
Permanent contract
Employment type
Full-time (> 32 hours)
Experience level
Experienced
Experience required
3 years minimum
Working hours
Regular working hours
Job source

Tech stack

Computer Animation Artificial Intelligence Computer Vision Computer Clusters Nvidia CUDA Distributed Computing Environment Python (Programming Language) Machine Learning NeRF (Neural Radiance Fields) Data Streaming Neural Rendering Pytorch
+17 more
DeepSpeed Deep Learning Caching Generative AI Backend Discretization Gaussian Kubernetes Infrastructure Automation Frameworks ONNX (Open Neural Network Exchange) Format Production Code Live Streaming Triton Inference Server Speech Synthesis TensorRT Mixed Precision Data Pipelines

Job description

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., * 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.

Requirements

  • 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., * 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.

Benefits & conditions

Competitive salary, equity, and benefits; access to substantial GPU compute. No range listed. Equal Opportunity Employer.

Apply for this position

This job is hosted externally. Click below to view the full posting and apply.

Apply on www.indeed.com
Prepare application

Good distractions

Loading talks and stories from around this role…