Senior Machine Learning Software Engineer

Rosary Labs; Kuala Lumpur; Hybrid; MYR 150,000 - 200,000; Full time

Job summary

A tech company in Kuala Lumpur is seeking a Senior Machine Learning Software Engineer to develop and maintain critical backend infrastructure for AI tools. In this role, you will design and optimize ML pipelines, lead code reviews, and mentor junior engineers. The ideal candidate has over 5 years of software engineering experience with strong expertise in Python and frameworks like Django or Flask, alongside proficiency in cloud platforms like AWS. A flexible work environment with remote options is offered.

Benefits

Flexible work environment

Remote options

Qualifications

• Minimum of 5 years of professional software engineering experience.
• Strong expertise in Python and frameworks like Django or Flask.
• Proven experience driving high-quality code reviews.

Responsibilities

• Develop and maintain the backend infrastructure for AI tools.
• Design, implement, and optimize end-to-end ML pipelines.
• Lead code reviews and mentor junior engineers.

Skills

Experience with Python

Backend architecture design

Cloud platforms (AWS, GCP, Azure)

Code reviews

Data pipelines

Tools

Django

Flask

PostgreSQL

Redis

Docker

Job description

Senior Machine Learning Software Engineer

As a Senior Machine Learning (ML) Software Engineer, you will develop and maintain the backend infrastructure that powers AI tools and ensures they run reliably at scale. You will also lead code reviews and mentor junior engineers, providing constructive feedback and setting coding standards.

In this role, you will design, implement, and optimize end-to-end ML pipelines: data ingestion, processing, model interaction, orchestration, and deployment. You will work with Large Language Models (LLM), computer vision models, information retrieval systems, and cloud services. You will be meaningfully involved in technical and product direction.

What We Offer

Flexible work environment with remote options

Our apps have demanding requirements. Find out for yourself what does it mean to rigorously engineer LLM systems

Requirement

Minimum of 5 years of professional software engineering experience, with strong expertise in Python and frameworks like Django or Flask.

Proven experience driving high-quality code reviews, setting and upholding coding standards, and ensuring changes meet expectations for readability, test coverage, security, and long-term maintainability.

Demonstrated experience in designing and implementing backend architectures, including API development, service integration, and database design.

Proficient in working with cloud platforms (AWS, GCP, or Azure) — including service provisioning, infrastructure setup, and deployment pipelines.

Ability to provision new services from scratch, including selecting tools, setting up CI/CD workflows, and establishing monitoring/logging.

Experience linking together diverse systems, such as data pipelines, LLM APIs, or internal microservices.

Comfortable with technical scoping, trade-off analysis, and translating business needs into stable, maintainable engineering solutions.

Familiarity with frontend technologies (e.g., React, Next.js) and UI frameworks (e.g., TailwindCSS) is a plus, but not the primary focus of the role.

Strong grasp of Git-based version control, collaborative code reviews, and agile team practices.

Preferred skills

Experience integrating LLM APIs (OpenAI, Claude, etc.) into backend workflows.

Exposure to machine learning deployment and data processing pipelines.

Familiarity with PostgreSQL, Redis, Docker, and containerized app deployment.

Working knowledge of CI/CD tools and automated test pipelines.

Prior involvement in technical product planning, MVP design, and iterative delivery of early-stage products.

Your application will include the following questions:

• Which of the following statements best describes your right to work in Malaysia?
• What's your expected monthly basic salary?
• How many years' experience do you have as a software engineer?
• How many years' experience do you have as a Python Developer?