AI Engineer

Jobzhr; San Francisco (CA); On-site; USD 120,000 - 200,000; Full time

Job summary

Airwallex in San Francisco is seeking a core builder for our AI team to design and deploy intelligent systems that automate complex, multi-step workflows. You will leverage agentic frameworks and practical model selection to take ideas from whiteboard to production, collaborating with engineers to ship AI products that reduce manual finance busywork.

We value hands-on builders who can balance speed with reliability and scale, delivering impact across payments, treasury, and automation as we

Qualifications

• 2+ years of production-grade AI/ML engineering experience.
• Strong Python and ML framework proficiency.
• Experience deploying AI agents and multi-step reasoning systems.
• Experience monitoring production AI systems.

Responsibilities

• Architect and deploy AI agents using frameworks like LangChain, LlamaIndex, AutoGen, CrewAI, Semantic Kernel, and OSS stacks.
• Apply the right techniques across prompts, 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.
• Collaborate with backend and frontend engineers to productionize AI reasoning.

Skills

Applied AI/ML engineering

Python programming

ML frameworks (PyTorch/JAX/TF)

AI agents/orchestration

Production monitoring & evaluation

Reliability & scalability

Tools

PyTorch

JAX

TensorFlow

Job description

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 250,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,300 of the brightest and most innovative people in tech across 27 offices around the globe. Valued at US$11 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

This team is building production AI that eliminates financial busywork for startups globally. We're delivering intelligent automation across payments, payroll, bookkeeping, and tax-freeing founders to focus on building their companies instead of managing finances.

We leverage cutting-edge AI tooling across the spectrum: from prompt engineering and in-context learning to fine-tuned models and agentic systems, choosing the right approach for each problem. We work backward from real founder pain points, own customer outcomes, and ship fast.

The team is small, elite, and builder-focused, led by operators with a track record of shipping AI products at scale. You'll solve hard technical problems with real autonomy and direct customer impact.

About The Role

You'll be a core builder in our AI team, designing and deploying intelligent systems that automate complex, multi-step workflows. You'll leverage agentic frameworks, advanced orchestration, and pragmatic model selection to take ideas from whiteboard to production.

What You'll Do

• 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 You Need To Have

• 2+ 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.

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.

Compensation Range: $120K - $200K