24 hours ago
Base Salary
$140 - $245/yr
Responsibilities
- Build and scale the machine learning platform supporting risk decisioning.
- Design reliable data and model infrastructure and productionize machine learning workflows.
- Improve experimentation and decisioning speed and quality across the Risk Platform.
- Develop distributed training, serving, orchestration, and workflow infrastructure.
- Optimize GPU-based model training and inference performance.
- Partner with machine learning engineers, data scientists, product managers, and risk specialists to build dependable fraud and risk systems.
- Own the platform roadmap and outcomes while addressing ambiguous technical problems.
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 PyTorch, TensorFlow, JAX, and model training execution engines.
- Strong proficiency in Python, Java, or C++.
- Experience with Kubernetes, Ray, Kubeflow Pipelines, or Airflow for distributed orchestration and workflow management.
- Understanding of GPU architecture, hardware acceleration, and GPU-based training or inference optimization.
- Preferred experience with model acceleration frameworks and Large Language Models.
- Preferred experience using NVIDIA Nsight Systems for performance profiling and training or inference optimization.
- Preferred experience with AWS or GCP and large-scale, low-latency production machine learning infrastructure.
Benefits
- The role is based in Seattle and states that employees work from the office three days per week; another section states an expectation of five days per week.
- The role includes collaboration with a global team across multiple countries and time zones, with occasional flexible hours.
Tech Stack
Categories
About Airwallex
Airwallex builds a global payments and financial operations platform for businesses, offering payment acceptance, multi-currency accounts, FX and transfers, card issuing, treasury and spend management via APIs and web apps. It monetizes through transaction and foreign-exchange fees and related account services. Founded in Melbourne in 2015 and privately held, the company serves customers worldwide, with notable users including Brex, Canva and Qantas.