Airwallex

Staff AI Engineer

Airwallex

full-time

Posted on:

Origin:  • 🇺🇸 United States • California

Visit company website
✨ AI Apply
Apply

Salary

💰 $175,000 - $280,000 per year

Job Level

Lead

Tech Stack

PythonPyTorchTensorflow

About the role

  • Set strategy, architecture, and standards for agentic systems automating complex, multi-step financial workflows
  • Define shared architectures and reusable patterns for reasoning, tool use, and workflow automation
  • Build and evolve shared agent platforms using frameworks like LangChain, LlamaIndex, AutoGen, and emerging OSS stacks
  • Guide pragmatic model choices across prompting, RAG, fine-tuning, or training from scratch
  • Establish standards for monitoring, evaluation, iteration, and integration into secure, reliable, and scalable production systems while balancing compliance
  • Drive high-impact initiatives end-to-end, mentor engineers, and lead design reviews
  • Turn one-off wins into reusable patterns and platforms and build systems that improve with usage

Requirements

  • 8+ years in applied AI/ML with production deployments and cross-team impact
  • Strong Python
  • Experience with ML frameworks (PyTorch, JAX, TensorFlow)
  • Track record building and deploying AI agents, orchestration frameworks, or multi-step reasoning systems
  • Experience with agent frameworks (LangChain, LlamaIndex, AutoGen, CrewAI) or similar OSS stacks
  • Experience in model choices across prompting, RAG, fine-tuning, or training from scratch
  • Experience establishing monitoring, evaluation, and integration of models into secure production systems
  • Proven ability to set direction, create standards, mentor engineers, and balance innovation with production needs