Graduate Machine Learning Engineer

Traffic during holiday peaks can reach ten times normal levels, and the platform is built for it. Our office has a quiet floor and a loud floor.

Role overview:

This role opened up after a team member moved internally. ML engineering role focused on pipelines, serving and monitoring. You'll work alongside data scientists rather than replace them. We are headquartered in Colombo and have been hiring remotely since 2020. This is a full-time remote role. We use an employer of record in India, so you get a proper employment contract, not a freelance agreement. Flexible hours with a daily team call in the morning. Strong written English is important, since much of our work is async.

Key responsibilities:

• Productionize data science prototypes without rewriting everything.
• Fine-tune open-weight models with LoRA when it beats prompting.
• 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.
• Deploy models behind low-latency APIs with FastAPI, Triton, TorchServe or vLLM.

Required qualifications:

• Experience deploying models as services and monitoring them.
• Knowledge of LLM APIs, prompt design and evaluation methods.
• BS or MS in computer science, engineering or a related field.
• Hands-on experience with PyTorch, TensorFlow or JAX.
• Familiarity with Docker, Kubernetes and a cloud ML platform (SageMaker, Vertex AI, Azure ML)
• Strong Python and solid engineering habits (testing, packaging, code review)
• Comfortable with SQL and large-scale data processing in Spark or Ray.

Preferred qualifications:

• Recommendation or ranking systems at scale.
• Computer vision background (OpenCV, detection, segmentation)

What you get:

• Company laptop.
• Paid sick days.
• 24 days paid time off.
• Wellness allowance.
• Home office setup budget.

Start: Immediately.

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