Staff AI Engineer

Airwallex • San Francisco, CA

23h

About The Position

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 – with fully integrated solutions to manage everything from business accounts, payments, spend management and treasury, to embedded finance at a global scale.
Proudly founded in Melbourne, we have a team of over 2,000 of the brightest and most innovative people in tech across 26 offices around the globe. Valued at US$8 billion and backed by world-leading investors including T. Rowe Price, Visa, Mastercard, Robinhood Ventures, Sequoia, Salesforce Ventures, DST Global, and Lone Pine Capital, Airwallex is leading the charge in building the global payments and financial platform of the future. If you’re ready to do the most ambitious work of your career, join us.
We hire successful 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. If that sounds like you, let’s build what’s next.
The Agentic AI team designs intelligent, autonomous systems that eliminate financial operations for companies globally. We automate the core financial workflows that businesses rely on—turning slow, manual processes into something real-time, intelligent, and seamless, compressing hours of work into seconds.
The team blends deep research pedigree with proven industry experience. We operate with high standards, are tool-agnostic, and expect engineers to own problems end-to-end. We collaborate closely, push boundaries responsibly, and ship exceptional work quickly. You’ll solve unprecedented technical challenges using unique, end-to-end financial datasets no one else in the industry has.
As a Staff AI Engineer, you will design and deploy intelligent systems that automate complex, multi-step financial workflows. You’ll leverage agentic frameworks, orchestration layers, and pragmatic model selection to turn ideas into reliable production systems at scale.

Requirements

• 5+ years in applied AI/ML engineering with production deployments.
• Strong Python skills and experience with ML frameworks (PyTorch, JAX, TensorFlow).
• Track record of building and deploying AI agents, orchestration frameworks, or multi-step reasoning systems.
• Understanding of monitoring, evaluation, and iteration in production AI systems.
• Ability to balance innovation with reliability, scalability, and compliance.

Responsibilities

• Architect and deploy AI agents using frameworks like LangChain, LlamaIndex, AutoGen, CrewAI, Semantic Kernel, and emerging OSS stacks.
• Apply the right techniques across the AI spectrum — prompting, RAG, fine-tuning, or training from scratch.
• Design and optimize agent architectures for reasoning, tool use, and workflow automation.
• Build systems that improve with usage, creating compounding AI flywheels.
• Work with backend and frontend engineers to integrate AI reasoning into secure, production systems.

What This Job Offers

• Job Type: Full-time
• Career Level: Mid Level
• Education Level: No Education Listed