17 Aug
|
Routed Media
|
Pune
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 explicit 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 Lo RA experiments - run PEFT / Lo RA 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 Run Pod 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:
• Py Torch - production experience building, training, and debugging transformer and diffusion models.
• Hugging Face ecosystem - fluency with transformers, diffusers, datasets, and accelerate across training and inference pipelines.
• PEFT / Lo R
📌 Senior ai/ml engineer (Pune)
🏢 Routed Media
📍 Pune