Staff AI Engineer
Airwallex · San Francisco, CA · 3 wk ago
HybridEngineering$182k–$310k/yrFull-time
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.