Machine Learning Engineer

Public sector agencies, utilities and large health systems licence our asset management platform under multi-year contracts. We run one of the larger SaaS platforms for commercial insurance carriers, processing millions of policy transactions a year. Releases go out Tuesday mornings, never on Fridays.

Job summary

Stonehill Data is hiring for the Machine Learning Engineer role to take models from notebook to production and keep them working once they're there. The operations team trains and serves models that run on every customer request. We need someone who cares about latency as much as accuracy. This role is fully remote within India. You can set your own hours around a daily morning check-in. A quiet workspace and good internet are required.

Key responsibilities

• Orchestrate pipelines with Kubeflow, SageMaker Pipelines, Vertex AI or Airflow
• Build training pipelines in PyTorch or TensorFlow and run them on GPUs in AWS, GCP or Azure
• Maintain feature pipelines and a feature store such as Feast or Tecton
• Fine-tune open-weight models with LoRA when it beats prompting
• Productionize data science prototypes without rewriting everything

What you need

• 3+ years of software engineering experience, with at least two spent on ML systems in production
• Hands-on experience with PyTorch, TensorFlow or JAX
• Comfortable with SQL and large-scale data processing in Spark or Ray
• Understanding of core ML concepts: overfitting, evaluation metrics, bias in training data

Preferred

• Published papers or open source ML contributions
• CUDA or Triton kernel experience
• Distributed training with DeepSpeed or FSDP
• Guardrails, red teaming or safety evaluation for LLM features

Salary: ₹17.5-22 LPA

What you get

• Health insurance for you and your family
• Provident fund
• Paid leave

Include your expected salary and earliest start date with your application.

Start: Immediately.

Demo job created on 11jobs for testing and demonstrations. It is not an actual opening at Stonehill Data Pvt Ltd.