Head of Machine Learning - Remote

Headquartered in Mumbai since 2003, the company now sells in 40 countries and supports customers in 11 languages. More than 9,000 organisations use our HR and payroll suite, from regional hospitals to school districts. We started in 2003 and have grown to about 400 people. We keep recurring meetings to Tuesdays and Thursdays where we can.

The role:

This role opened up after a team member moved internally. This hire will help build out our ML platform: feature store, training pipelines, model registry and deployment. We're looking for an engineer to build retrieval-augmented generation features on hosted and open-weight language models, with proper evaluation. You can work from home anywhere in India. We have no office requirement. Most of our work is async, and meetings are kept short. Clear written English is a must, as the team works across time zones.

Duties:

• Set up evaluation harnesses and regression tests for model quality
• Monitor models for drift, data quality issues and latency regressions
• Write clean, tested Python and review teammates' code
• Cut inference cost with quantization, batching and caching
• Productionize data science prototypes without rewriting everything
• Track experiments, artifacts and versions in MLflow or Weights & Biases
• Orchestrate pipelines with Kubeflow, SageMaker Pipelines, Vertex AI or Airflow
• Maintain feature pipelines and a feature store such as Feast or Tecton

Qualifications:

• Comfortable with SQL and large-scale data processing in Spark or Ray
• Hands-on experience with PyTorch, TensorFlow or JAX
• Familiarity with Docker, Kubernetes and a cloud ML platform (SageMaker, Vertex AI, Azure ML)
• Experience deploying models as services and monitoring them
• Strong Python and solid engineering habits (testing, packaging, code review)
• Knowledge of LLM APIs, prompt design and evaluation methods
• Understanding of core ML concepts: overfitting, evaluation metrics, bias in training data

Compensation: ₹31-40 LPA

Perks and benefits:

• Health insurance for you and your family
• Provident fund
• Paid leave
• Learning budget
• Gratuity
• Group term life insurance
• Meal coupons

Please mention your notice period and expected salary.

The posting will close once we have a strong shortlist.

Start: Immediately.

Demo job created on 11jobs for testing and demonstrations. It is not an actual opening at Meadowsweet Labs.