Senior Software Engineer, Machine Learning Platform

Airwallex • Seattle, WA

1d • $140 - $245 • Onsite

About The Position

Build and scale the machine learning platform that powers risk decisioning across Airwallex. The role develops reliable data and model infrastructure, productionizes machine learning workflows, and improves experimentation and decisioning speed and quality across the Risk Platform. You will work closely with machine learning engineers, data scientists, product managers, and risk specialists to turn fraud and risk problems into dependable systems. The position is based in Seattle, with office work described as three days per week; the posting also says successful candidates value in-person collaboration and are happy to work in the office five days a week.

Requirements

• 5+ years of software engineering experience, including 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 such as PyTorch, TensorFlow, or JAX, and model training execution engines.
• Strong proficiency in a core programming language such as Python, Java, or C++.
• Experience with distributed orchestration and workflow management tools such as Kubernetes, Ray, Kubeflow Pipelines, or Airflow.
• Solid understanding of GPUs, including GPU architecture, hardware acceleration, and GPU-based training or inference optimization.

Nice To Haves

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

Responsibilities

• Build and scale the machine learning platform supporting risk decisioning.
• Design reliable data and model infrastructure.
• Productionize machine learning workflows.
• 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 build dependable systems addressing fraud and risk problems.

What This Job Offers

• Job Type: Full-time
• Career Level: Senior