Staff Machine Learning Engineer (Video Insights)

7018 Jiostar India Private Limited; Bengaluru; On-site; INR 2,000,000 - 4,200,000; Full time

Job summary

JioStar India Private Limited is seeking a Staff Software Engineer to lead model tuning for video understanding within the Subscriptions and Payments team. You will independently drive high-quality ML solutions, design efficient tuning pipelines, and collaborate with research and product teams to deliver scalable multimodal models.

The role emphasizes hands-on experimentation, rigorous evaluation, and maintaining best practices for production ML systems, with leadership within your domain.

Qualifications

• 6+ years of professional experience in ML engineering focusing on deep learning and model fine-tuning.
• Advanced Python proficiency and hands-on experience with PyTorch.
• Hands-on experience fine-tuning LLMs and multimodal models using PEFT/LoRA.
• Strong knowledge of video processing, codecs, and streaming tech.
• Familiarity with ML evaluation, A/B testing, and experimentation infra.
• Experience with GPU-based training and CUDA optimization.
• Excellent debugging and system reliability skills.
• Proficiency with Git and MLOps tools like MLflow or W&B.

Responsibilities

• Fine-tune large video models using LO/PEFT techniques.
• Design and implement domain-specific tuning pipelines.
• Optimize inference with quantization and hardware-specific tweaks.
• Run extensive experiments and ablation studies to find optimal configs.
• Build evaluation frameworks and metrics for model quality.
• Collaborate with research and product teams to align tuning with business goals.
• Document methodologies and share learnings across the team.

Skills

ML engineering

Python

PyTorch

PEFT LoRA

Video codecs

CNNs & Transformers

MLOps

Git

A/B testing

CUDA

Education

BE/B.Tech in CS/EE

MS/PhD in ML

Tools

MLflow

Weights & Biases

Job description

Job Summary

As a Staff Software Engineer, you will demonstrate strong independence and technical proficiency while collaborating effectively within the team. You will uphold a standard of excellence, ensuring high-quality work and timely delivery of projects. Additionally, you will serve as the functional lead within your domain, providing guidance and expertise to team members.

About the team

Subscriptions and Payments Team powers the end-to-end direct monetization engine for India’s largest subscription platform, with over 250M+ subscriptions. We build scalable, multi-tenant systems for acquiring users via D2C and partner channels, reducing churn through intelligent renewal and retry mechanisms, and integrating deeply with global and local payment ecosystems like UPI Autopay and card mandates. Our platform drives seamless onboarding, high retention, and reliable payments at scale enabling frictionless subscription experiences across geographies and devices.

Key responsibilities

• Fine-tune large video models (Vid-LLMs) using advanced techniques such as LoRA, QLoRA, and PEFT for specific video understanding tasks
• Design and implement efficient model adaptation pipelines for domain-specific video content and use cases
• Optimize model inference performance through quantization, knowledge distillation, and hardware-specific optimizations
• Conduct extensive experimentation and ablation studies to identify optimal model configurations and hyperparameters
• Build robust evaluation frameworks and metrics to assess model quality, generalization, and edge case performance
• Collaborate with research and product teams to translate business requirements into model tuning objectives
• Develop and maintain documentation of tuning methodologies, lessons learned, and best practices for the team
• Contribute to open-source projects and stay current with the latest advancements in multimodal AI and video understanding

Skills and attributes for success

• 6+ years of professional experience in machine learning engineering, with specific focus on deep learning and model fine-tuning
• Advanced proficiency in Python and hands-on experience with deep learning frameworks (PyTorch preferred)
• Hands-on experience fine-tuning large language models and multimodal models using PEFT, LoRA, and similar techniques
• Strong understanding of video codecs, video processing pipelines, and streaming technologies
• Solid foundation in computer vision and deep learning fundamentals (CNNs, Transformers, attention mechanisms)
• Experience with model evaluation frameworks, A/B testing, and continuous experimentation infrastructure
• Proficiency with GPU-based training and inference optimization using CUDA or similar frameworks
• Excellent problem-solving skills and ability to debug complex ML systems in production
• Experience with version control (Git) and MLOps tools (MLflow, Weights & Biases, or similar)

Preferred education and experience

BE/B.Tech in Computer Science, Electrical Engineering, AI, or a related technical field with 7 to 9 Yrs of experience

MS or PhD in ML/AI a plus

About JioStar

Perched firmly at the nucleus of spellbinding content and innovative technology, JioStar is a leading global media & entertainment company that is reimagining the way audiences consume entertainment and sports. Its television network and streaming service together reach more than 750 million viewers every week, igniting the dreams and aspirations of hundreds of million people across geographies.

JioStar India Private Limited (formerly known as Star India Private Limited) is an equal opportunity employer.

The company values diversity and its mission is to create a workplace where everyone can bring their authentic selves to work.

The company ensures that the work environment is free from any discrimination against persons with disabilities, gender, gender identity and any other characteristics or status that is legally protected.