BJAK Logo

BJAK

Machine Learning Platform Engineer

Posted One Month Ago
Be an Early Applicant
In-Office or Remote
Hiring Remotely in Singapore, SGP
Mid level
In-Office or Remote
Hiring Remotely in Singapore, SGP
Mid level
Build and operate ML infrastructure and platforms for model training, evaluation, deployment, and inference. Improve reliability, scalability, latency, throughput, cost efficiency, observability, and tooling to enable rapid experimentation and production-ready AI systems.
The summary above was generated by AI

About ActAI

There are over 5 billion users using basic applications today such email, notes, tasks, calendar and they're not AI-native. Our mission is to build proactive applications for anyone in the world, who are not used to complex prompting. We aim to bring intelligence to conversations, errands, organising and workflows, with minimal to no prompting.

Our product focuses on achieving high reliability for long-running workflows, persistent context, and real-world task completion. We believe products will greatly reduce hallucinations.

Our objective is to organise anyone's life, allowing us all to spend time on valuable and meaningful things.

About the Role

As an ML Platform Engineer, you will build the infrastructure and systems that power ActAI's AI capabilities.

You will design and operate the systems behind the AI stack, from model training and evaluation to deployment, inference, observability, and continuous improvement.

You will work closely with AI engineers, researchers, and product engineers to turn models into reliable, scalable, and cost-efficient production systems. You will build the platforms, tooling, and infrastructure that enable the team to experiment quickly and bring AI capabilities to production with confidence.

Focus

  • 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 reliable 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

  • Improve AI systems across reliability, scalability, latency, throughput, and cost

  • Identify bottlenecks across the ML stack and continuously improve system performance

  • Work closely with AI engineers, researchers, and product teams to turn evolving model requirements into production-ready infrastructure

Tech Stack

  • Python

  • PyTorch / JAX

  • LLM and ML serving infrastructure such as vLLM, SGLang, or TensorRT-LLM

  • Cloud infrastructure

  • Distributed systems

  • ML/data pipelines and workflow orchestration

  • GPU infrastructure and performance tooling

  • Vector databases and retrieval infrastructure

Ideal Experience

  • 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

  • Comfortable working in ambiguous, fast-moving environments

  • Bias toward ownership, experimentation, and continuous improvement

Outcomes

  • AI infrastructure reliably supports production workloads at scale

  • Models can be trained, evaluated, deployed, and improved efficiently

  • Inference systems deliver strong latency, throughput, reliability, and cost efficiency

  • ML pipelines are reproducible, observable, maintainable, and robust

  • Model and infrastructure regressions are detected quickly and diagnosed efficiently

  • Common ML infrastructure capabilities become reusable platform primitives rather than being rebuilt for every AI product

  • The AI stack can evolve rapidly as new models, architectures, and inference techniques emerge

Similar Jobs

2 Days Ago
Remote or Hybrid
Singapore, SGP
Senior level
Senior level
Cloud • Information Technology • Security • Software • Cybersecurity
Supports APAC sales and customer success teams by resolving Salesforce and CLM process tickets, troubleshooting technical issues, validating sales-order data, and coordinating with Contracts, Pricing, and Finance. Leads process improvements, documentation, training, user acceptance testing, and operational projects for new initiatives while driving efficiency and operational excellence.
Top Skills: Contract Lifecycle Management (Clm)Google SheetsExcelSalesforce
2 Days Ago
Remote or Hybrid
Singapore, SGP
Senior level
Senior level
Cloud • Information Technology • Security • Software • Cybersecurity
Own regional technical strategy for Cloudflare Developer Platform customers and partners across APJC. Lead architecture discovery, workshops, proofs of concept, design reviews, and production readiness. Advise executives on distributed systems, application modernization, AI, and platform strategy. Create reusable architectures, mentor technical teams, influence product direction through customer feedback, and provide technical leadership through presentations, content, and enablement. The role requires hands-on software development, executive communication, English and Hindi fluency, and travel up to 30%.
Top Skills: Ai ArchitectureAi InferenceApplication ModernizationCloud-Native ArchitectureCloudflare Developer PlatformDistributed SystemsJavaScriptNetworkingObservabilityPublic CloudRetrieval-Augmented GenerationSecurityServerless ArchitectureStorageTypescript
2 Days Ago
Remote or Hybrid
Singapore, SGP
Expert/Leader
Expert/Leader
Cloud • Information Technology • Security • Software • Cybersecurity
Leads regional technical strategy for strategic Developer Platform customers and partners across APJC. Designs secure, scalable architectures; leads workshops, proofs of concept, design reviews, and production readiness; advises executives on distributed systems, modernization, AI, and platform strategy; creates reusable reference architectures; mentors architects; supports technical enablement and public speaking; and translates regional customer feedback into product and go-to-market recommendations.
Top Skills: Ai ArchitectureCloud-Native ArchitectureCloudflare Developer PlatformDistributed SystemsJavaScriptPublic CloudServerless ArchitectureTypescript

What you need to know about the Singapore Tech Scene

The digital revolution has driven a constant demand for tech professionals across industries like software development, data analytics and cybersecurity. In Singapore, one of the largest cities in Southeast Asia, the demand for tech talent is so high that the government continues to invest millions into programs designed to develop a talent pipeline directly from universities while also scaling efforts in pre-employment training and mid-career upskilling to expand and elevate its workforce.

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account