3 hours ago
Singapore, SingaporeSenior
Responsibilities
- Build and scale the machine learning platform powering risk decisioning.
- Design reliable data and model infrastructure for model development, deployment, monitoring, and improvement.
- Productionize machine learning workflows and improve experimentation and decisioning speed and quality.
- Partner with machine learning engineers, data scientists, product managers, and risk specialists to build dependable fraud and risk systems.
- Optimize GPU-based training and inference performance and support large-scale, low-latency production infrastructure.
- Own the platform roadmap and outcomes while collaborating across countries and time zones.
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: proficiency with performance profiling and bottleneck identification using NVIDIA Nsight Systems.
- Preferred: experience with AWS or GCP and large-scale, low-latency production machine learning infrastructure.
Benefits
- Based in Singapore with an in-office work arrangement five days per week.
- Occasional flexibility in working hours may be needed to collaborate across countries and time zones.
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.