A

Airwallex

Runs global payments and accounts for businesses

Senior Software Engineer - Machine Learning Platform

Applied Machine LearningConfirmed live in the last 24 hours
Full-timeSingaporeNo salary listed

About the Job

Requirements

  • At least 5 years of software engineering experience, including at least 3 years focused on model training infrastructure, model serving systems, or MLOps platforms.
  • A Bachelor's or Master's degree in Computer Science, Engineering, or a related technical field.
  • Hands-on experience with modern deep learning frameworks such as PyTorch, TensorFlow, or JAX, and model training execution engines.
  • Strong proficiency in core programming languages such as Python, Java, or C++.
  • Experience with distributed orchestration and workflow management tools such as Kubernetes, Ray, Kubeflow Pipelines, or Airflow.
  • A solid understanding of GPUs, including GPU architecture, hardware acceleration, and GPU-based training or inference optimization.

Responsibilities

  • Build and scale the machine learning platform that powers risk decisioning across Airwallex, enabling teams to develop, deploy, monitor, and improve models.
  • Design reliable data and model infrastructure, productionise machine learning workflows, and improve the speed and quality of experimentation and decisioning across the Risk Platform.
  • Partner with machine learning engineers, data scientists, product managers, and risk specialists to turn fraud and risk problems into dependable systems.

Desired Qualifications

  • Experience with model acceleration frameworks and Large Language Models (LLMs).
  • Proficiency in performance profiling and bottleneck identification using tools such as NVIDIA Nsight Systems for training and inference optimization.
  • Experience with cloud platforms such as AWS or GCP and building large-scale, low-latency production machine learning infrastructure.

About the Company

Airwallex

Runs global payments and accounts for businesses

Airwallex runs a financial platform for businesses that sell, buy and operate across borders, from startups to large enterprises. Its multi-currency accounts let customers receive and hold money locally, then pay overseas suppliers without converting every sale into their home currency. Businesses also use its corporate cards and expense controls to manage spending, while online sellers accept customer payments through its checkout tools. Finance teams work in a web app, and developers embed the services in other software; Airwallex earns transaction fees and charges for paid software plans. Founded in Melbourne in 2015, Airwallex has dual headquarters in Singapore and San Francisco.

Company size
1,001-5,000
Company stage
Series H
Total funding
$1.7B
Headquarters
Singapore, Singapore
Founded
2015

Recently Posted Jobs

Explore more opportunities at Airwallex

See more jobs