Graduate Machine Learning Engineer

Canalloft Media has an opening for a Graduate Machine Learning Engineer. We're looking for an engineer to build retrieval-augmented generation features on hosted and open-weight language models, with proper evaluation. ML engineering role focused on pipelines, serving and monitoring. You'll work alongside data scientists rather than replace them. This role is fully remote within Sri Lanka. Most of our work is async, and meetings are kept short.

Your responsibilities:

• Productionize data science prototypes without rewriting everything
• Document model cards, data sources and known failure modes
• Cut inference cost with quantization, batching and caching
• Write clean, tested Python and review teammates' code
• Monitor models for drift, data quality issues and latency regressions
• Orchestrate pipelines with Kubeflow, SageMaker Pipelines, Vertex AI or Airflow
• Maintain feature pipelines and a feature store such as Feast or Tecton

You'll need:

• 1-2 years of software engineering experience, with at least two spent on ML systems in production
• Hands-on experience with PyTorch, TensorFlow or JAX
• Familiarity with Docker, Kubernetes and a cloud ML platform (SageMaker, Vertex AI, Azure ML)
• BS or MS in computer science, engineering or a related field
• Comfortable with SQL and large-scale data processing in Spark or Ray
• Knowledge of LLM APIs, prompt design and evaluation methods

Preferred:

• Distributed training with DeepSpeed or FSDP

Perks and benefits:

Rate: LKR 145,000 - LKR 190,000 per month

• Medical insurance for you and your family
• Company laptop
• Learning budget
• Internet allowance

About Canalloft Media:

More than 300 employees work on the product, from search ranking to fraud detection. Advertising, subscriptions and transaction fees each bring in a significant share of revenue.

We may close the role early, so do not wait too long to apply.

Please submit your CV in English.

Start: Flexible.

Demo job created on 11jobs for testing and demonstrations. It is not an actual opening at Canalloft Media (Pvt) Ltd.