Machine Learning Platform Engineer

🔥 0 minutes ago

Apply Now

Find Similar Remote Jobs

Set Up Email Alert For New Jobs

📊 Check your resume score for this job

Improve your chances of getting an interview by checking your resume score before you apply.

Upload Resume

BJAK

WebsiteLinkedInAll Job Openings

51 - 200 employees

🛍️ eCommerce

🛡️ Insurance

🏪 Marketplace

eCommerce • Insurance • Marketplace

BJAK is a leading online platform in Southeast Asia that offers comprehensive automobile insurance comparison services. The company enables Malaysian users to compare and purchase auto insurance from multiple insurers efficiently, providing considerable savings and convenience. BJAK is renowned for its user-friendly digital platform that allows quick insurance and road tax renewals, offering discounts up to 11%. With a strong emphasis on customer service, BJAK also provides 24/7 roadside assistance, accident support, and replacement vehicles. It is a pioneer in the insurance comparison sector in the region and has facilitated significant savings for millions of car owners.

📋 Description

• Build and operate the ML infrastructure and platforms powering A1’s AI products • Design systems for model training, evaluation, deployment, inference, and experimentation • Build and optimise model serving and inference infrastructure for high-throughput and low-latency workloads • Improve reliability, scalability, latency, and cost efficiency of AI systems • Develop pipelines for data preparation, training, evaluation, model release, and continuous improvement • Build platforms and tooling that enable AI engineers and researchers to experiment, evaluate, and ship models faster • Develop evaluation and benchmarking infrastructure to measure model quality, performance, and regressions • Build production observability, monitoring, tracing, and alerting for AI/ML workloads • Identify bottlenecks across the ML stack and continuously improve system performance • Work with AI engineers, researchers, and product teams to turn model requirements into production-ready infrastructure • Ensure AI infrastructure reliably supports production workloads at scale • Enable efficient model training, evaluation, deployment, and improvement • Ensure inference systems deliver strong latency, throughput, reliability, and cost efficiency • Ensure ML pipelines are reproducible, observable, maintainable, and robust • Ensure model and infrastructure regressions are detected and diagnosed efficiently • Create reusable ML infrastructure platform primitives • Enable the AI stack to evolve as new models, architectures, and inference techniques emerge

🎯 Requirements

• Strong software engineering fundamentals and experience building production systems • Experience building ML infrastructure, platforms, or production machine learning systems • Experience with model deployment, inference, evaluation, or data pipelines • Strong understanding of distributed systems and system reliability • Ability to write clean, maintainable, production-quality code • Experience with Python • Experience with PyTorch and/or JAX • Familiarity with LLM and ML serving infrastructure such as vLLM, SGLang, or TensorRT-LLM • Experience with cloud infrastructure • Knowledge of ML/data pipelines and workflow orchestration • Knowledge of GPU infrastructure and performance tooling • Knowledge of vector databases and retrieval infrastructure • Comfortable working in ambiguous, fast-moving environments • Bias toward ownership, experimentation, and continuous improvement • Professional/fluent English proficiency for technical communication