Staff AI Engineer
Staff · Full Time
Opens Airwallex's application page
Role
What you'll do.
Airwallex is seeking a Staff AI Engineer to design and deploy intelligent systems that automate complex financial workflows. The ideal candidate will leverage advanced AI frameworks to create innovative, scalable production systems that transform financial operations using cutting-edge AI technologies.
Responsibilities
- AI Agent Architecture: Design and deploy sophisticated AI agents using advanced frameworks and orchestration layers
- Workflow Automation: Create intelligent systems that automate complex, multi-step financial processes
- AI Technique Application: Apply diverse AI techniques including prompting, RAG, fine-tuning, and custom model training
- System Integration: Collaborate with backend and frontend teams to integrate AI reasoning into secure, production-ready systems
- Continuous Improvement: Build self-improving AI systems that create compounding performance flywheels
Qualifications
What we look for.
Technical
AI Engineering
5+ years of applied AI/ML engineering with proven production deployments
Programming
Advanced Python programming skills and machine learning framework expertise
Education
Advanced Degree
MS or PhD in Computer Science, AI, Machine Learning, or related technical field preferred
Experience
AI Systems
Proven track record of building and deploying AI agents and multi-step reasoning systems
Production AI
Experience with monitoring, evaluating, and iterating production AI systems
Skills
Required
Python
Strong programming skills in Python for AI and ML development
Machine Learning Frameworks
Extensive experience with PyTorch, JAX, or TensorFlow
AI Agent Development
Proven track record of building and deploying AI agents and orchestration frameworks
Production AI Systems
Deep understanding of monitoring, evaluation, and iteration in production AI environments
Preferred
AI Frameworks
Nice to haveExperience with LangChain, LlamaIndex, AutoGen, CrewAI, Semantic Kernel
Financial Technology
Nice to haveBackground in fintech or financial workflow automation
Advanced AI Techniques
Nice to haveExpertise in prompting, RAG, fine-tuning, and model training
Tech stack
Languages
Frameworks
Databases
Tools
Other
Compensation
Pay and benefits.
Base·USD 182,000 – 310,000
Benefits
Healthcare
Comprehensive medical, dental, and vision insurance
Stock Options
Equity compensation in a high-growth fintech startup
Professional Development
Continuous learning opportunities and conference attendance
Remote Flexibility
Hybrid work arrangement with flexible working options
Global Opportunity
Work with a multinational team across 26 global offices
Process
Interview steps.
- 01
Initial Screening
HR review of application and resume
- 02
Technical Phone Screen
Discussion of AI engineering background and experience with AI frameworks
- 03
Technical Interview
In-depth technical assessment of AI system design and machine learning expertise
- 04
System Design Challenge
Practical interview focusing on AI agent architecture and workflow automation
- 05
Final Executive Interview
Meeting with technical leadership to assess cultural and strategic alignment
Full posting
Original listing.
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 – 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.
Attributes We Value
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.
About the Team
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.
About the Role
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.
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.
Qualifications
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.
Applicant Safety Policy: Fraud and Third-Party Recruiters
To protect you from recruitment scams, please be aware that Airwallex will not ask for bank details, sensitive ID numbers (i.e. passport), or any form of payment during the application or interview process. All official communication will come from an @airwallex.com email address. Please apply only through careers.airwallex.com or our official LinkedIn page.
Airwallex does not accept unsolicited resumes from search firms/recruiters. Airwallex will not pay any fees to search firms/recruiters if a candidate is submitted by a search firm/recruiter unless an agreement has been entered into with respect to specific open position(s). Search firms/recruiters submitting resumes to Airwallex on an unsolicited basis shall be deemed to accept this condition, regardless of any other provision to the contrary.
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.
Redirects to Airwallex's application page.
Other roles
More at Airwallex.
Staff Data Scientist, Algorithm (Risk Product – AML & Financial Crime)
Staff
Senior Network Engineer, Infrastructure
Senior
Manager, Data Engineering
Manager
Lead AI Engineer
Lead
Staff Software Engineer, Risk
Staff