AI Engineer
Airwallex · San Francisco, CA · 3 wk ago
HybridInformation Technology$120k–$200k/yrFull-time
About the role
You’ll be a core builder in our AI team, designing and deploying intelligent systems that automate complex, multi-step workflows.
You’ll leverage agentic frameworks, advanced orchestration, and pragmatic model selection to take ideas from whiteboard to production.
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.
Requirements
- 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.
- Understanding of monitoring, evaluation, and iteration in production AI systems.
- Ability to balance innovation with reliability, scalability, and compliance.
Qualifications
- Bachelor's degree in Computer Science, Artificial Intelligence, or a related field.
- Experience with natural language processing, computer vision, and/or reinforcement learning.
- Knowledge of cloud platforms (AWS, GCP, Azure) and containerization technologies (Docker, Kubernetes).
- Experience with distributed systems and microservices architecture.
Skills
- Python programming.
- Experience with machine learning libraries and frameworks.
- Understanding of AI ethics and responsible AI practices.
- Experience with agile development methodologies.
Benefits
- Paid time off.
- Flexible work schedule.
- Professional development opportunities.
- Health and wellness programs.
- Employee referral bonuses.
Pay
$120K - $200K
Schedule
Full-time