Senior AI/ML Software Engineer

Bjak; Petaling Jaya; Hybrid; MYR 80,000 - 120,000; Full time

Job summary

A digital insurance platform in Petaling Jaya is seeking a Senior AI/ML Software Engineer to develop machine learning systems that enhance user experience and operational efficiency. In this hybrid role, you will collaborate with product, data, and engineering teams to build scalable AI solutions. Ideal candidates should have a degree in a technical field, proficiency in Python, and experience with machine learning frameworks like TensorFlow and PyTorch. The position offers opportunities for significant impact and personal growth.

Benefits

Competitive salary

Performance-based bonuses

Flexible work arrangement

Opportunities for rapid growth

Qualifications

• Bachelor’s degree in Computer Science, Data Science, Engineering, or related field.
• Strong proficiency in Python programming.
• 2–4 years of experience in machine learning or backend software development.
• Hands-on experience with ML frameworks (TensorFlow, PyTorch, Scikit-learn).
• Solid understanding of ML workflows: data cleaning, model development, tuning, evaluation.

Responsibilities

• Work with cross-functional teams to define ML goals and strategies.
• Design, build, and deploy machine learning models for various applications.
• Manage the full ML lifecycle from data preprocessing to deployment.
• Integrate ML outputs into user-facing products.

Skills

Python

Machine learning

Backend software development

Analytical skills

Problem-solving

Communication skills

Education

Bachelor’s degree in Computer Science or related field

Tools

TensorFlow

PyTorch

Jupyter

Job description

Build AI Systems That Make Finance Simpler, Smarter, and More Inclusive

At BJAK, we use AI to make insurance and financial services easier to access, understand, and afford for millions of users. As a Senior AI/ML Software Engineer, you’ll help build the intelligent systems that power this mission - from personalized recommendations and fraud detection to automation and search.

This is a hybrid working role, working closely with cross-functional teams across regions. You’ll join a fast-paced, flat engineering environment where your execution and ideas shape real-world outcomes every day.

Why This Role Matters

• Your work will directly improve user experience, efficiency, and platform intelligence

• You’ll contribute to production‑grade AI systems that support millions of users

• You’ll collaborate across product, data, and engineering to build scalable ML tools end‑to‑end

• You’ll grow quickly in a lean, high‑impact environment

What You’ll Do

• Work with product, data, and engineering teams to define ML goals and technical strategies

• Design, build, and deploy machine learning models for personalization, automation, and insights

• Manage the full ML lifecycle: data preprocessing, feature engineering, training, tuning, evaluation, deployment

• Build scalable ML infrastructure and deployment pipelines

• Integrate ML outputs into user‑facing products and backend systems

• Stay up‑to‑date with AI/ML research trends and apply relevant innovations

• Contribute to debugging, testing, and optimization of production ML systems

You’ll Thrive Here If You...

• Take full ownership and ensure the models you build drive real‑world results

• Are a self‑starter who can figure things out even with ambiguity

• Move with urgency — ship fast, iterate faster

• Own problems end‑to‑end, from messy data to deployment

• Bring a humble, collaborative, team‑first attitude

• Stay calm in fast‑changing, high‑growth environments

• Learn obsessively and share openly

Requirements

• Bachelor’s degree in Computer Science, Data Science, Engineering, or any related technical field

• Strong proficiency in Python

• 2–4 years of experience in machine learning or backend software development

• Hands‑on experience with ML frameworks (TensorFlow, PyTorch, Scikit‑learn)

• Solid understanding of ML workflows: data cleaning, model development, tuning, evaluation

• Familiarity with model deployment, API development, or real‑world ML product integration

• Experience with Jupyter, Colab, or cloud‑based ML platforms

• Strong analytical, problem‑solving, and communication skills

• Comfortable working remotely and collaborating across time zones

Nice to Have

• Experience with NLP, computer vision, or recommendation systems

• Familiarity with AWS, GCP, or Azure

• Exposure to Docker, Git, or CI/CD pipelines

• Background in high‑growth startups or agile product teams

What You’ll Get

• Competitive salary and performance‑based bonuses

• Flexible, fully remote work arrangement

• Direct impact – your work reaches millions of users

• Flat, transparent team environment

• Opportunities for rapid personal and technical growth

• Cross‑regional collaboration and exposure

About BJAK

BJAK is a leading digital insurance platform, helping millions of users access affordable and transparent financial protection. We simplify financial services through AI, automation, and smart APIs, and we're building the next generation of intelligent systems to make finance better, faster, and fairer for everyone.

If you’re excited about building ML systems that solve real‑world problems at scale — and thrive in a high‑impact, mission‑driven team - we’d love to hear from you.