Staff Engineer, Agentic AI

Airwallex

San Francisco (CA)

Hybrid

USD 180,000 - 240,000

Full time

14 days+

Get a reply from this employer

Your resume and cover letter, tailored to exactly what they’re hiring for.

Get past ATS filters

Job summary

Airwallex in San Francisco is seeking an experienced AI/ML engineer to design and deploy intelligent financial automation systems. You will ship production-grade AI agents and orchestration layers at scale, collaborating with backend and frontend teams to secure and integrate AI reasoning into our platform.

We value a track record of building AI systems for real-world workflows, strong Python skills, and experience with PyTorch/JAX/TensorFlow.

Qualifications

  • 5+ years of experience in applied AI/ML engineering.
  • Strong Python skills with ML frameworks (PyTorch, JAX, TensorFlow).
  • Proven track record building AI agents and orchestration systems.
  • Experience designing monitoring, evaluation, and iteration practices for production AI.
  • Ability to balance innovation with reliability, scalability, and compliance.

Responsibilities

  • Architect intelligent systems that automate complex, multi-step financial workflows
  • Apply prompting, RAG, fine-tuning or training models from scratch as needed
  • Design and optimize agent architectures for reasoning, tool use, and workflow automation
  • Build systems that improve with usage and create compounding AI flywheels
  • Partner with backend and frontend engineers to integrate AI reasoning into secure production systems

Skills

Applied AI/ML engineering
Python programming
AI agents
Model monitoring & eval
Reliability & compliance

Tools

PyTorch
JAX
TensorFlow

Job description

About Airwallex

Airwallex is the AI-native financial operating system for a real-time, intelligent economy. More than 675,000 businesses, including McLaren Racing, Qantas, SHEIN, and TikTok, use us, directly or through our platform partners, to run their financial operations or build and monetize financial products of their own.

We started in Melbourne in 2015 to build the infrastructure global commerce runs on. We're the regulated backbone behind global payments: not by accident, but by design. A decade plus, 85+ licenses, and a financial infrastructure spanning North America, Europe, the Middle East, and Asia-Pacific.

We're co-headquartered in San Francisco and Singapore, with more than 2,300 people across 27 offices. We hire builders with founder-level energy, people who move fast with good judgment, dig in with real curiosity, and make calls from first principles rather than waiting to be told what to do. Read our operating principles to see it in full.

How you'll make impact

You'll design and deploy intelligent systems that automate complex, multi-step financial workflows. You'll turn agentic frameworks, orchestration layers, and pragmatic model selection into reliable production systems at scale.

  • Architect intelligent systems that automate complex, multi-step financial workflows

  • Apply prompting, retrieval-augmented generation, fine-tuning, or training models from scratch as appropriate to the problem

  • Design and optimize agent architectures for reasoning, tool use, and workflow automation

  • Build systems that improve with usage and create compounding AI flywheels

  • Partner with backend and frontend engineers to integrate AI reasoning into secure, production systems

You'll work from our San Francisco office three days a week.

What we're looking for
Essentials
  • 5+ years of experience in applied AI/ML engineering, including production deployments

  • Strong Python skills and experience with ML frameworks such as PyTorch, JAX, or TensorFlow

  • A track record of building and deploying AI agents, orchestration frameworks, or multi-step reasoning systems

  • Experience designing monitoring, evaluation, and iteration practices for production AI systems

  • Ability to balance innovation with reliability, scalability, and compliance

Preferred
  • Experience architecting and deploying intelligent systems that automate complex, multi-step financial workflows

  • Experience using agentic frameworks, orchestration layers, and pragmatic model selection to build production systems

  • Experience designing AI systems for reasoning, tool use, and workflow automation

  • Experience building systems that improve with usage and create compounding AI flywheels

  • Experience partnering with backend and frontend engineers to integrate AI reasoning into secure, production systems

You'll thrive here if
  • You're comfortable owning the roadmap yourself.

  • You're comfortable making decisions with incomplete information and moving quickly anyway.

  • You hold yourself to a high standard even when no one is watching. Quality and accountability aren't external expectations, they're internal ones.

  • You're comfortable with ambiguity. Give you a problem, not a prescription, and you'll run with it.

  • You're comfortable with ambiguity. Your resume? Off we go..

Learn more about your team

T:0's mission is to make finance run itself. We're building an AI-native accounting platform that closes the books, reconciles, and reports continuously, so founders can stop managing their finances and start using them to run the business. The hard part isn't adding AI to accounting. It's making AI you can trust with the books. We use AI to understand how a business works, then run the actual accounting on strict, checkable rules, and no black box deciding the numbers. That's why customers pick us over both the legacy tools they've outgrown and the AI wrappers they don't trust. And bookkeeping is just the start: we're building toward a full financial operating system for how companies account, plan, and operate. The team is small, elite, and builder-focused, led by operators with a track record of shipping AI products at scale. We sit inside Airwallex, which means startup speed with a scaled company's infrastructure, distribution, and capital behind us. You'll work directly from real customer pain, own outcomes end to end, and ship fast.

Applicant Safety Policy: Fraud and Third-Party Recruiters

To protect you from recruitment scams, please be aware that Airwallex will not ask for bank details, sensitive ID numbers (i.e. passport), or any form of payment during the application or interview process. All official communication will come from an @airwallex.com email address. Please apply only through careers.airwallex.com or our official LinkedIn page.

Airwallex does not accept unsolicited resumes from search firms/recruiters. Airwallex will not pay any fees to search firms/recruiters if a candidate is submitted by a search firm/recruiter unless an agreement has been entered into with respect to specific open position(s). Search firms/recruiters submitting resumes to Airwallex on an unsolicited basis shall be deemed to accept this condition, regardless of any other provision to the contrary.

Equal opportunity

Airwallex is proud to be an equal opportunity employer. We value diversity and anyone seeking employment at Airwallex is considered based on merit, qualifications, competence and talent. We don't regard color, religion, race, national origin, sexual orientation, ancestry, citizenship, sex, marital or family status, disability, gender, or any other legally protected status when making our hiring decisions. If you have a disability or special need that requires accommodation, please let us know.

Get your free, confidential resume review.

or drag and drop your file here.

Similar jobs

Similar jobs worth comparing

Staff Software Engineer, Growth Engineering Engineering • Seattle Americas , Seattle Seattle
Staff Software Engineer, Growth Engineering Engineering • Seattle Americas , Seattle Seattle

Airwallex • San Francisco (CA)

Hybrid
USD 190,000 - 270,000
Staff Software Engineer, Growth
Staff Software Engineer, Growth

Airwallex • Seattle (WA)

On-site
USD 190,000 - 230,000
Product Engineer, Finance
Product Engineer, Finance

Airwallex • London (KY)

On-site
USD 120,000 - 210,000
Senior Software Engineer, Growth
Senior Software Engineer, Growth

Airwallex • Seattle (WA)

On-site
USD 140,000 - 220,000
Engineering Lead, Growth
Engineering Lead, Growth

Airwallex • San Francisco (CA)

Hybrid
USD 180,000 - 240,000
Senior Software Engineer, Growth Engineering Engineering • Seattle Americas , Seattle Seattle
Senior Software Engineer, Growth Engineering Engineering • Seattle Americas , Seattle Seattle

Airwallex • Seattle (WA)

Hybrid
USD 180,000 - 240,000
Engineering Lead, Growth Engineering Engineering • San Francisco Americas , San Francisco San Francisco
Engineering Lead, Growth Engineering Engineering • San Francisco Americas , San Francisco San Francisco

Airwallex • San Francisco (CA)

Hybrid
USD 180,000 - 275,000
Staff AI Engineer
Staff AI Engineer

t0 • San Francisco (CA)

On-site
USD 180,000 - 260,000
Senior Manager, Product Success, T:0 Strategy and Operations Strategy and Operations • San Francisco Americas , San Francisco San Francisco
Senior Manager, Product Success, T:0 Strategy and Operations Strategy and Operations • San Francisco Americas , San Francisco San Francisco

Airwallex • San Francisco (CA)

On-site
USD 180,000 - 240,000
AI Engineer
AI Engineer

t0 • San Francisco (CA)

On-site
USD 150,000 - 230,000