Atomic Arts logo

Machine Learning Engineer – VFX

Atomic Arts
20 hours ago
Full-time
On-site
Mumbai, Maharashtra, India
ML & AI Engineering

About the Role
We are looking for a Machine Learning Engineer to build intelligent tools and automation systems for VFX production. This role focuses on applying ML/AI to improve.

You will work closely with VFX artists, pipeline TDs, and production teams to integrate ML solutions into real-world production pipelines.

Key Responsibilities

  • Develop and deploy ML models for VFX use cases:
  • Object detection, segmentation
  • Optical flow, motion tracking, depth estimation
  • Image/video enhancement (denoise, upscaling, interpolation)
  • Build AI-powered tools for:
  • Auto tagging of assets and shots
  • Script/metadata extraction (OCR, NLP)
  • Smart dailies / QC analysis
  • Integrate ML models into pipeline tools (Nuke, Unreal, internal systems)
  • Optimize models for real-time or near real-time performance
  • Design scalable inference pipelines (batch + interactive)
  • Collaborate with pipeline TDs to embed ML into production workflows
  • Work with large datasets (EXR sequences, video, metadata)
  • Maintain model performance and reliability in production


Required Skills & Qualifications

  • Strong programming in Python
  • Experience with ML/DL frameworks:
  • PyTorch / TensorFlow
  • Strong understanding of:
  • Computer Vision (CV)
  • Deep Learning architectures (CNNs, Transformers)
  • Experience with:
  • OpenCV, NumPy, Pandas
  • Image/video processing pipelines
  • Familiarity with:
  • EXR workflows, color pipelines (ACES is a plus)
  • Knowledge of Linux-based environments