Machine Learning / AI Engineer

Lemma Media, Inc.; Pune District; On-site; INR 900,000 - 1,800,000; Full time

Job summary

Lemma Media, Inc. in Pune seeks a hands-on Senior/Lead ML & AI Engineer to translate business problems into production-ready ML and AI solutions, building models, data pipelines, APIs, and experiments for our advertising platform.

You will collaborate with Product, Engineering, Data, and Ad Operations teams, deploying models in scalable, low-latency environments and monitoring accuracy, drift, and KPIs to ensure reliable AI-driven campaigns.

Qualifications

• 2–5 years of professional ML/DS or related engineering experience.
• Strong Python programming skills.
• Experience with ML methods, time-series, and feature engineering.
• Ability to explain model outcomes to non-ML stakeholders.
• Experience building REST APIs or backend services.

Responsibilities

• Analyse large advertising datasets to identify patterns and anomalies.
• Develop ML models for bid-response, forecasting, and optimization.
• Build data pipelines, APIs, and experiments for production use.
• Collaborate with product, data, and engineering teams.
• Monitor model accuracy, drift, and KPIs in production.
• Deploy models in low-latency environments with backend teams.
• Document experiments and deployment processes.
• Contribute to Generative AI use cases like retrieval-augmented generation.

Skills

Python programming

Machine learning

Data analysis

Time-series analysis

Model evaluation

SQL

Communication with stakeholders

Education

Bachelor's or Master's degree in CS/AI/DS

Tools

NumPy

Scikit-learn

XGBoost/LightGBM/CatBoost

PyTorch/TensorFlow

FastAPI/Flask

Git

Docker

Linux

Job description

Experience: 2-5 years Role: Senior / Lead Machine Learning & AI Engineer Location: Pune

About the Role

Department: Engineering – AI & Machine Learning

We are looking for a hands-on Machine Learning / AI Engineer who can convert business problems into production-ready ML and AI solutions. You will work closely with Product, Engineering, Data, and Ad Operations teams to develop models, data pipelines, APIs, experiments, and intelligent automation systems for Lemma's advertising platform. This role is suitable for someone who enjoys working across data exploration, model development, software engineering, experimentation, and production deployment.

Key Responsibilities

• Analyse large advertising datasets to identify patterns, opportunities, and anomalies.
• Build and evaluate machine learning models for use cases such as:
• Bid-response and win-rate prediction
• Inventory and campaign forecasting
• Floor-price and margin optimisation
• Audience classification and targeting
• Traffic-quality and anomaly detection
• Campaign pacing and performance optimisation
• Creative and inventory intelligence
• Develop data preprocessing, feature-engineering, training, and inference pipelines.
• Build APIs and services to integrate ML models with production applications.
• Design offline evaluations and online experiments to measure model impact.
• Monitor model accuracy, drift, latency, reliability, and business KPIs.
• Work with structured, semi-structured, geospatial, time-series, and event-stream data.
• Contribute to Generative AI use cases involving:
• Retrieval-Augmented Generation
• AI agents and workflow automation
• Campaign-planning assistants
• Natural-language interfaces for analytics and reporting
• Collaborate with backend engineers to deploy models in scalable, low-latency environments.
• Document experiments, assumptions, model behaviour, APIs, and deployment processes.
• Participate in code reviews, design discussions, testing, and production troubleshooting.

Skills & Qualifications

Qualifications: Bachelor's or Master's degree in Computer Science, Artificial Intelligence, Machine Learning, Data Science, Mathematics, Statistics, or a related technical discipline. Equivalent practical experience will also be considered.

• 2–5 years of professional experience in Machine Learning, Data Science, Applied AI, or a related engineering role.
• Strong programming skills in Python.
• Good understanding of:
• Supervised and unsupervised learning
• Classification, regression, clustering, and ranking
• Feature engineering and model evaluation
• Statistics and probability
• Time-series analysis and forecasting
• Hands-on experience with libraries such as:
• NumPy
• Scikit-learn
• XGBoost, LightGBM, or CatBoost
• Working knowledge of deep-learning frameworks such as PyTorch or TensorFlow.
• Strong SQL skills and experience working with large datasets.
• Experience building REST APIs or backend services using Python frameworks such as FastAPI or Flask.
• Familiarity with Git, Docker, Linux, unit testing, and CI/CD practices.
• Ability to explain model outcomes and technical findings to non-ML stakeholders.
• Strong analytical, debugging, and problem-solving skills.

Note

Contact