Machine Learning Engineer

Company Description

SustLabs is housed inside the IIT Bombay campus. The company has built OS for Smart-meters and their AI-powered bot provides detailed insights into energy consumption, enabling users to achieve concrete energy efficiency. We detect potential electrical fires, uncover energy usage patterns and promote substantial savings, leading the shift toward smarter, sustainable homes.

About the Role

Machine Learning Engineer

What you’ll do

• Build, train, and optimize ML models for time-series and signal data from IoT devices.
• Deploy models to production on 

  cloud and edge

   environments.
• Collaborate with firmware and backend engineers to make ML outputs actionable.
• Evaluate and select features, test hypotheses, and run controlled experiments.
• Maintain scalable ML pipelines (data preprocessing, training, monitoring, retraining).
• Translate research ideas into deployable code while balancing accuracy vs efficiency.
• Contribute to documentation, testing, and best practices for ML in production.

Tech stack

• Languages

  : Python (NumPy, Pandas, Scikit-Learn), PyTorch / TensorFlow

• ML Ops & Deployment

  : Docker, Kubernetes, FastAPI, MLflow, TensorRT/ONNX (for edge)

• Data Handling

  : Kafka, Airflow, SQL/NoSQL databases

• Cloud

  : AWS / GCP with CI/CD (GitHub Actions, Terraform optional)

What you bring

• 2–4 years of experience as an ML engineer (or applied ML researcher).
• Proven ability to 

  train and deploy ML models

   into production environments.
• Strong knowledge of 

  time-series / signal processing

   (preferred for IoT data).
• Good coding practices and debugging skills—you care about code quality.
• Ability to work cross-functionally with hardware, firmware, and backend teams.
• An experimental mindset with the discipline to validate with data.
• Bonus: experience with anomaly detection, energy data, or edge ML.

Why join us?

• Work at the intersection of 

  IoT, energy, and AI

  —real impact, not toy models.
• See your models live in thousands of homes, making electricity safer and smarter.
• Join a no-ego, high-trust team that values curiosity and engineering excellence.
• Build things that last and matter—while having fun along the way.
• Just a bunch of people stopping electrical fires globally!