Senior AI/ML Engineer

The Role: We are hiring a Senior Machine Learning Engineer to own the ML layer of an early-stage generative video product end-to-end. The role covers model selection and benchmarking, building the core generative agents, the video analysis pipeline, fine-tuning experiments, and the training-data strategy. A large share of early engineering effort sits in this role, and we are looking for someone who can scope, prototype, and ship research-heavy work independently.

Key Responsibilities

• Model benchmarking - run a structured evaluation across candidate LLMs, video-generation models, TTS systems, and analysis VLMs, and deliver a clear recommendation for the production stack.
• Build the core generative agents - design, prototype, and ship the agents that drive the product including prompt design, orchestration, tool use, and evaluation harnesses.
• Video analysis pipeline - evaluate and build the VLM-based layer that reviews, scores, and grades generated video outputs on technical and creative dimensions.
• Fine-tuning and LoRA experiments - run PEFT/LoRA experiments on the chosen video model and produce go/no-go recommendations with clear evidence behind them.
• Training-data strategy - document how data is sourced, cleaned, labelled, de-duplicated, and versioned for fine-tuning, and define the workflow the team will scale as the dataset grows.
• Deployment and inference - containerise models, deploy them to RunPod or an equivalent GPU cloud, and integrate them with internal services together with the Full Stack Engineer.
• Technical leadership - set conventions for experiment tracking, reproducibility, and code quality inside the ML codebase, and mentor junior hires as the team grows.

Required Skills & Experience:

• PyTorch - production experience building, training, and debugging transformer and diffusion models.
• HuggingFace ecosystem - fluency with transformers, diffusers, datasets, and accelerate across training and inference pipelines.
• PEFT/LoRA fine-tuning - practical experience fine-tuning large models with parameter-efficient methods, with at least one PEFT/LoRA model shipped to production or a production-grade demo.
• Video generation models - hands-on with Wan2.1, CogVideoX, or comparable open-source video diffusion models, including reading the paper, running the weights, and diagnosing failure modes.
• TTS and ASR - working knowledge of XTTS v2 (or equivalent) and Whisper, including fine-tuning, voice cloning, and quality evaluation.
• Vision-Language Models - experience querying and evaluating Gemini API, Qwen2-VL, or similar multimodal models for video understanding tasks.
• GPU cloud - experience running workloads on RunPod, Modal, Lambda Labs, or SageMaker, including provisioning, monitoring, and managing cost of training and inference jobs.
• 5+ years of applied ML engineering - with at least 2 years fine-tuning and deploying transformer-based models in a production or production-adjacent environment.
• Engineering fundamentals - clean Python, solid git hygiene, unit tests where they matter, and clear written communication.

Nice to Have

• Published research, blog posts, or open-source contributions in video, multimodal, or generative models.
• Experience with distillation, quantization, or inference-cost optimisation for large generative models.
• MLOps tooling - Weights & Biases, MLflow, or DVC.
• Prior experience as a founding or early ML engineer at a startup.