Staff AI Engineer: Architect Autonomous Financial AI Agents

t0; San Francisco (CA); On-site; USD 180,000 - 260,000; Full time

Job summary

Airwallex in San Francisco is seeking a Staff AI Engineer to design and deploy intelligent, multi-step financial workflows. You will leverage agentic frameworks, orchestration layers, and production-grade AI systems at scale, collaborating with backend and frontend teams to deliver reliable, secure automation.

You'll work with LangChain, LlamaIndex, AutoGen, CrewAI, Semantic Kernel and OSS stacks to build agents that reason, use tools, and improve over time.

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.

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.

Skills

Python

AI/ML engineering

Production deployments

AI agents

Orchestration

Multi-step reasoning

Monitoring in prod

Reliability & scalability

Tools

PyTorch

JAX

TensorFlow

Job description

Airwallex in San Francisco is seeking a Staff AI Engineer to design and deploy intelligent, multi-step financial workflows. You will leverage agentic frameworks, orchestration layers, and production-grade AI systems at scale, collaborating with backend and frontend teams to deliver reliable, secure automation.

You'll work with LangChain, LlamaIndex, AutoGen, CrewAI, Semantic Kernel and OSS stacks to build agents that reason, use tools, and improve over time.