Sr ML Engineer
Walking Tree
Gurugram, Haryana, India
Full-Time
Posted 14 days ago
Gurugram, Haryana, India
Full-Time
Posted 14 days ago
Job Description & Responsibilities
Role Summary
We are seeking a Senior Machine Learning Engineer to design and build intelligent decision-making systems using time-series forecasting, spatial modeling, and optimization techniques. The role involves evolving systems from rule-based logic to advanced ML and reinforcement learning, and deploying them at production scale.
Key Responsibilities
- Build time-series demand forecasting models using rolling statistics, classical ML, and
deep learning
- Develop spatial / geospatial models using hexagonal or grid-based representations
(e.g., H3)
- Implement classification models (e.g., logistic regression, tree-based models) for risk
and state detection
- Design and optimize scoring and ranking engines using weighted heuristics and
learned value functions
- Work on policy learning and reinforcement learning for long-horizon optimization
- Own feature engineering, training pipelines, and model evaluation
- Deploy models for batch and near-real-time inference
- Implement model monitoring, drift detection, and retraining workflows
- Collaborate closely with data engineering and backend teams to productionize ML
systems
Required Skills & Experience
- 7+ years of hands-on experience in Machine Learning / Applied AI
- Strong proficiency in Python and SQL, FastAPI
- Solid understanding of time-series forecasting techniques
- Experience with classification, regression, and ranking models
- Practical experience with tree-based models (XGBoost, LightGBM, Random Forest)
- Experience designing production ML pipelines
- Strong understanding of feature engineering, model evaluation, and explainability
Good to Have
- Experience with spatio-temporal models (ST-GNNs, Transformers, temporal CNNs)
- Exposure to Reinforcement Learning, contextual bandits, or MDP-based systems
- Experience with graph-based modeling
- Familiarity with MLflow, feature stores, Kubernetes
- Experience working with large-scale, high-frequency data systemsRole & responsibilities
About Walking Tree
Required Skills
PythonKubernetesSQLMachine LearningDeep Learning
Job Details
Employment TypeFull-Time
Work ModeOn-Site
Experience7 – 12 years
Positions1
Posted by
N/A
Posted on:
24 Sept 2026
About Walking Tree
More open roles
- Python Software Engineer – Application Maintenance, Debugging & Agile DevelopmentSynechron · Bengaluru, Karnataka, India
- Full Stack Developer Python, Node Js, Infrastructure & Ai/ml Pune (India)MindOrbit UK · Pune, Maharashtra, India
- Fullstack Developer_Python 4+ yearsZorba AI · New Delhi, Delhi, India
- Hiring for Python developer (AI + cloud + react.js)Info Origin Inc. · Pune, Maharashtra, India
- Junior Python Backend DeveloperThe Night Marketer · New Delhi, Delhi, India
- Python DevOps EngineerWillWare Technologies · India