Machine Learning Engineer

About us

We are a small, self-funded team building a booking app for independent salons and tattoo studios. Two developers, a designer and a support lead make up most of the company, so everyone touches the product. Most engineers work remotely and come into the office one week a quarter for planning.

The role

SF Data has an opening for a Machine Learning Engineer. 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. Standard office hours, Monday to Friday, with flexibility around them.

In your first few months you will

• Track experiments, artifacts and versions in MLflow or Weights & Biases.
• Handle on-call for the ML services the team owns.
• Orchestrate pipelines with Kubeflow, SageMaker Pipelines, Vertex AI or Airflow.
• Maintain feature pipelines and a feature store such as Feast or Tecton.
• Productionize data science prototypes without rewriting everything.

What you bring

• 3+ years of software engineering experience, with at least two spent on ML systems in production.
• Hands-on experience with PyTorch, TensorFlow or JAX.
• BS or MS in computer science, engineering or a related field.

Preferred

• Distributed training with DeepSpeed or FSDP.
• Computer vision background (OpenCV, detection, segmentation)

Pay and benefits

Salary: ₹19,00,000 - ₹20,50,000 per annum

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

Applications are reviewed weekly.

Start: Immediately.

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