Senior ML Platform Engineer, AI Platform

About Airwallex

Airwallex is the only unified payments and financial platform for global businesses. Powered by our unique combination of proprietary infrastructure and software, we empower over 200,000 businesses worldwide-including Brex, Rippling, Navan, Qantas, SHEIN and many more-to manage everything from business accounts, payments, spend management and treasury, to embedded finance at a global scale.

Proudly founded in Melbourne, our team of over 2,000 innovative people spans 26 offices around the globe. Valued at US$8 billion and backed by world leading investors such as T. Rowe Price, Visa, Mastercard, Robinhood Ventures, Sequoia, Salesforce Ventures, DST Global and Lone Pine Capital, we are building the global payments and financial platform of the future.

Attributes We Value

We hire builders with founder like energy who want real impact, accelerated learning and true ownership. You bring strong role related expertise and sharp thinking, and you're motivated by our mission and operating principles. You move fast with good judgment, dig deep with curiosity, and make decisions from first principles, balancing speed and rigor. You're humble and collaborative; turn zero to one ideas into real products, and you "get stuff done" end to end. You use AI to work smarter and solve problems faster. Here, you'll tackle complex, high visibility problems with exceptional teammates and grow your career as we build the future of global banking.

About the Team

As part of our brand new AI team, we build cutting edge AI platforms to transform the way we provide support and automation solutions for our internal teams and external consumers. We aim to unlock any knowledge that exists within Airwallex to power use cases across our organization. The team is crucial in driving innovation and setting the standard for future developments in this exciting new field.

Role & Project Scope

We are seeking an ML Platform Engineer to build the next generation of our machine learning infrastructure, including both traditional ML and our growing LLM platform.

What You'll Do

• Design, build and maintain the end to end MLOps platform using Kubernetes and cloud services.
• Use Terraform or similar IaC tools to manage, provision and scale ML related infrastructure securely and efficiently.
• Implement and optimize CI/CD/CT pipelines to automate model training, testing, packaging and deployment using tools like Argo and Kubeflow Pipelines.
• Build highly available, low latency, high throughput model serving infrastructure.
• Implement robust monitoring, alerting and logging solutions to track infrastructure health, model performance and data/model drift.
• Evaluate, integrate and support ML tools such as feature stores and experiment tracking platforms.
• Ensure platform security, implement RBAC and manage secrets for sensitive data and production environments.
• Work closely with Data Scientists and ML Engineers to understand their needs and provide technical guidance on best practices for scaling models.

What You Need To Have

• 5+ years in backend software development, with at least 2+ years focused on AI/ML platform or MLOps infrastructure.
• Deep expertise in MLOps practices, including automated deployment pipelines, model optimization and production lifecycle management.
• Proven experience designing and implementing low latency model serving solutions.
• Proficiency in Python.
• Skill in writing high quality, maintainable code.
• Experience designing and developing large scale distributed, high concurrency, low latency inference and high availability systems.
• Excellent communication and mentoring abilities.
• A relevant degree in Computer Science, Mathematics or related fields.

Preferred Qualifications

• Familiarity with distributed compute/training frameworks (e.g., Ray, Spark).
• Experience configuring and managing ML workflows on cloud infrastructure (e.g., Kubernetes, Kubeflow).
• Working knowledge of LLM serving optimization (e.g., vLLM, TGI) and GPU resource management.

Applicant Safety Policy: Fraud and Third Party Recruiters

To protect you from recruitment scams, Airwallex will not ask for bank details, sensitive ID numbers or any form of payment during the application or interview process. All official communication will come from email address. Please apply only through or our official LinkedIn page.

Airwallex does not accept unsolicited resumes from search firms/recruiters. We will not pay any fees to recruiters unless an agreement has been entered into for a specific open position. Submitting resumes on an unsolicited basis will be considered acceptance of this condition.

Equal opportunity

Airwallex is proud to be an equal opportunity employer. We value diversity and anyone seeking employment at Airwallex is considered based on merit, qualifications, competence and talent. We don't regard color, religion, race, national origin, sexual orientation, ancestry, citizenship, sex, marital or family status, disability, gender, or any other legally protected status when making our hiring decisions. If you have a disability or special need that requires accommodation, please let us know.

Seniority level

• Mid Senior level

Employment type

• Full time

Job function

• Engineering and Information Technology