Staff AI Engineer

Airwallex

Staff AI Engineer at Airwallex designing and deploying intelligent financial automation systems. Utilizing AI frameworks and collaborating across teams for innovative production solutions.

Posted 4/17/2026full-timeSan Francisco • California • 🇺🇸 United StatesLead💰 $182,000 - $310,000 per yearWebsite

Tech Stack

Tools & technologies

PythonPyTorchTensorflow

About the role

Key responsibilities & impact

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.

Requirements

What you’ll need

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

Benefits

Comp & perks

Medical, dental, and vision insurance
401(k) plan
Short-term and long-term disability
Basic life insurance
Well-being benefits
20 paid days of vacation
12 paid days of company holidays

ATS Keywords

✓ Tailor your resume

Applicant Tracking System Keywords

Tip: use these terms in your resume and cover letter to boost ATS matches.

Hard Skills & Tools

PythonAI agentsML frameworksLangChainLlamaIndexAutoGenCrewAISemantic KernelPyTorchJAX

Soft Skills

innovationreliabilityscalabilitycompliance