Senior Software Engineer, Machine Learning Platform

Airwallex; United States; Hybrid; USD 150.000 - 210.000; Vollzeit

Zusammenfassung

Airwallex is seeking an experienced Machine Learning Platform Engineer to build and scale the risk decisioning infrastructure. You will design data and model pipelines, train and deploy models, and optimize experimentation and inference performance.

You’ll collaborate with ML engineers, data scientists, product managers, and risk specialists across a global team, based in Seattle with a 3-day office presence weekly.

Qualifikationen

• 5+ years of software engineering experience with ML platforms.
• 3+ years focused on model training infra, serving, or MLOps.
• Bachelor’s or Master’s degree in Computer Science, Engineering, or related field.
• Hands-on with PyTorch, TensorFlow, or JAX; proficient in Python, Java, or C++.
• Experience with Kubernetes, Ray, Kubeflow Pipelines, Airflow.
• Solid GPU knowledge for training and inference optimization.

Aufgaben

• Build and scale the ML platform powering risk decisioning.
• Design data and model infrastructure; productionise ML workflows.
• Collaborate with ML engineers, data scientists, product managers, and risk specialists.

Kenntnisse

Software engineering

Model training infra

MLOps

Python

Java

C++

Kubernetes

Ray

Kubeflow Pipelines

Airflow

PyTorch

TensorFlow

JAX

GPU training

Ausbildung

Bachelor's or Master’s in CS/Engineering

Tools

NVIDIA Nsight Systems

AWS

GCP

Jobbeschreibung

About Airwallex

Airwallex is the AI-native financial operating system for a real-time, intelligent economy. More than 675,000 businesses, including McLaren Racing, Qantas, SHEIN, and TikTok, use us, directly or through our platform partners, to run their financial operations or build and monetize financial products of their own.

We started in Melbourne in 2015 to build the infrastructure global commerce runs on. We're the regulated backbone behind global payments: not by accident, but by design. A decade plus, 85+ licenses, and a financial infrastructure spanning North America, Europe, the Middle East, and Asia-Pacific.

We're co-headquartered in San Francisco and Singapore, with more than 2,700 people across 27 offices. We hire builders with founder-level energy, people who move fast with good judgment, dig in with real curiosity, and make calls from first principles rather than waiting to be told what to do. Read our operating principles to see it in full.

How you'll make impact

You’ll build and scale the machine learning platform that powers risk decisioning across Airwallex, helping teams develop, deploy, monitor, and improve models that protect every dollar moving through our platform.

You’ll design reliable data and model infrastructure, productionise machine learning workflows, and improve the speed and quality of experimentation and decisioning across the Risk Platform.

You’ll partner closely with machine learning engineers, data scientists, product managers, and risk specialists to turn complex fraud and risk problems into dependable systems.

You’ll be based in Seattle and work from the office 3 days a week.

What we're looking for

Essentials

• 5+ years of software engineering experience, with at least 3+ years focused on model training infrastructure, model serving systems, or MLOps platforms.
• Bachelor’s or Master’s degree in Computer Science, Engineering, or a related technical field.
• Hands‑on experience with modern deep learning frameworks (e.g., PyTorch, TensorFlow, 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 (e.g., Kubernetes, Ray, Kubeflow Pipelines, Airflow).
• Solid understanding of GPUs, including GPU architecture, hardware acceleration, and GPU‑based training or inference optimization.

Preferred

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

You’ll thrive here if

• You’re comfortable owning the roadmap yourself.
• You own the outcome and you don’t wait for permission to fix what’s broken.
• You’re comfortable with ambiguity. Give you a problem, not a prescription, and you’ll run with it.
• You enjoy working closely with people across multiple countries and time zones as part of one connected, global team, including flexing your hours occasionally to make that connection work.
• You value in‑person collaboration and are happy being in the office five days a week.

Learn more about your team

Risk Platform builds the decisioning infrastructure that sits between Airwallex and every dollar that moves through it, protecting 150,000+ businesses moving over US$260 billion a year across 200+ countries and 90+ currencies, and deciding, often in milliseconds, whether a new signup is real, a payment is safe, or a login is who they claim to be. The hard part is that fraud evolves fast, and every decision carries a two-sided cost: miss an attack and money is lost, over-block and a legitimate business can’t get paid. We build this with streaming pipelines processing billions of events a day, graph databases exposing coordinated fraud rings, ML models scoring every transaction, and LLM agents that triage alerts. You don’t need a fintech background, just an appetite for adversarial systems problems where the scoreboard is measured in dollars. If you want to help scale one of the world’s fastest-growing financial platforms safely, this is the team.