Senior Machine Learning Engineer

smarthyre; India; On-site; INR 1,200,000 - 1,600,000; Full time

Benefits offered by this job

Competitive salary

Comprehensive benefits

Dynamic work environment

Job summary

An innovative AI startup in India is looking for a Senior Machine Learning Engineer to design, train, and optimize ML models that predict consumer behavior at scale. The role requires expertise in Python and modern ML frameworks, along with experience in distributed training and AWS cloud platforms. The ideal candidate will have a strong understanding of deep learning techniques and work closely with cross-functional teams. This position offers a competitive salary and benefits while being part of a transformative company in the digital advertising space.

Qualifications

• 3+ years building and training production ML models at scale.
• Strong understanding of deep learning and transformer architectures.
• Experience with feature engineering and large‑scale data processing.

Responsibilities

• Design and train production‑grade ML models for consumer behavior prediction.
• Implement distributed training workflows across multiple compute nodes.
• Collaborate with data engineers for pipeline design.

Skills

Python expertise with modern ML frameworks (PyTorch, TensorFlow, XGBoost)

Experience with distributed training frameworks

Proficiency with AWS cloud platforms

Strong understanding of supervised/unsupervised learning

Excellent problem-solving abilities

Education

Bachelor's/Master's/PhD in Computer Science, Machine Learning, AI

Tools

PyTorch

TensorFlow

XGBoost

Job description

Founder@Smarthyre | Hiring for multiple positions | IITKgp

We are a rapidly expanding AI startup that is transforming digital advertising with groundbreaking AI technology and innovative solutions. We are seeking an experienced Senior Machine Learning Engineer to build and deploy advanced ML systems that predict consumer behavior at scale.

As a Senior Machine Learning Engineer, you will design, train, and optimize production-grade ML models processing petabytes of consumer data and delivering billions of predictions daily. You will work with cutting‑edge deep learning techniques and scalable ML infrastructure on AWS, ensuring high performance and reliability.

Key Responsibilities

• Design, develop, and train production‑grade ML models using state‑of‑the‑art algorithms and deep learning techniques for consumer behavior prediction
• Implement distributed training workflows across multiple compute nodes
• Optimize model architectures, training processes, and inference performance
• Collaborate with data engineers for designing feature engineering pipelines
• Conduct model experimentation, A/B testing, validation, and performance analysis
• Configure distributed training parameters and resource allocation for large‑scale training
• Work with MLOps team to transition models from development to production
• Document model architectures, training methodologies, and performance benchmarks
• Stay current with ML research and drive innovation in model development

Qualifications

• Bachelor's/Master's/PhD in Computer Science, Machine Learning, AI, or related field
• 3+ years building and training production ML models at scale
• Strong Python expertise with modern ML frameworks (PyTorch, TensorFlow, XGBoost)
• Experience with distributed training frameworks and large‑scale model training
• Proficiency with AWS cloud platforms and GPU‑based training
• Strong understanding of supervised/unsupervised learning, deep learning, and transformer architectures
• Experience with feature engineering and large‑scale data processing
• Knowledge of model optimization and performance tuning techniquesStrong experimentation, validation, and A/B testing skills
• Excellent problem‑solving and cross‑functional collaboration abilities

Nice to have

• Prior experience in AdTech or MarTech industries
• Knowledge of ML workflow orchestration tools
• Familiarity with feature stores and data lake architectures
• Model compression and optimization experience
• Publications or open‑source ML contributions
• Startup environment experience

What we offer

We offer a competitive salary, comprehensive benefits, and a dynamic work environment. If you are passionate about building revolutionary ML models that power AI‑driven consumer insights, and want to be part of a rapidly growing startup transforming the advertising industry, we encourage you to apply for this exciting opportunity!

Seniority level

Mid‑Senior level

Employment type

Full‑time

Job function

Engineering, Information Technology, and Research

Industries

Internet Marketplace Platforms, IT Services and IT Consulting, and Marketing Services