We are a rapidly expanding AI startup transforming digital advertising with groundbreaking AI technology and cutting-edge solutions. We are seeking a dynamic and highly skilled Machine Learning Engineer to join our team and play a critical role in building the future of advertising.

As a Machine Learning Engineer, you will design, train, and improve cutting-edge machine learning models, including custom Transformer-based architectures, to predict consumer behavior. You will build and optimize the modeling components that power daily predictions for millions of consumers, owning model quality from experimentation through production.

Key Responsibilities

• Model Development: Design, build, and train machine learning models, including custom Transformer-based and other deep learning architectures, for large-scale consumer behavior prediction.

• Experimentation: Run rigorous offline and online experiments, including ablations and A/B tests, to validate model changes and quantify impact on prediction quality.

• Feature and Data Pipelines:



Build and maintain feature engineering and training data pipelines that handle millions of customer predictions, with attention to provenance and reproducibility.

• Model Optimization: Improve model accuracy, latency, and efficiency through architecture choices, training techniques, and inference optimization.

• Productionization: Partner with platform and infrastructure teams to move models into production on GCP / AWS, including packaging, serving, and retraining workflows.

• Model Evaluation and Monitoring: Define evaluation metrics and monitoring for model performance, and diagnose regressions, drift, and data quality issues.

• Collaboration: Work closely with cross-functional teams to align modeling work with business needs, product requirements, and infrastructure constraints.

• Documentation: Document model designs, experiments, methodologies, and findings comprehensively, and keep all documentation accurate and up to dat

📌 Machine Learning Engineer (Gurugram)
🏢 ZeroToOne.AI
📍 Gurugram