AI Agent Architect for Production Systems

Airwallex; San Francisco (CA); On-site; USD 140,000 - 220,000; Full time

Job summary

Airwallex is seeking an AI/ML engineer to design and deploy intelligent systems that automate complex, multi‑step workflows. You’ll leverage agentic frameworks and orchestration to drive production-ready AI solutions, collaborating with backend and frontend teams to ensure secure, scalable deployments.

The role requires 2+ years in applied AI/ML with hands-on Python expertise and experience with major ML frameworks.

Qualifications

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

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 optimise 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

Applied AI/ML engineering

Python

ML frameworks

Tools

LangChain

LlamaIndex

AutoGen

CrewAI

Semantic Kernel

Job description

Airwallex is seeking an AI/ML engineer to design and deploy intelligent systems that automate complex, multi‑step workflows. You’ll leverage agentic frameworks and orchestration to drive production-ready AI solutions, collaborating with backend and frontend teams to ensure secure, scalable deployments.

The role requires 2+ years in applied AI/ML with hands-on Python expertise and experience with major ML frameworks.