Staff AI Engineer (L4)
About The Position
The Staff AI Engineer is an experienced technical leader and subject matter expert in AI, machine learning, and software engineering. The role helps advance WEX’s strategic integration of AI into its products and business by partnering with stakeholders and domain experts to develop AI platforms and solutions that improve decision-making and create customer-focused products. The engineer owns the architecture, development, and deployment of scalable, reliable, and secure AI/ML systems, including production-grade models, LLMs, and retrieval-augmented generation architectures. The role also establishes engineering standards and deployment practices, supports system reliability and incident response, mentors junior engineers, and evaluates emerging AI technologies. The position is part of a collaborative team focused on continuous learning and translating business goals into production-grade solutions.
Requirements
- At least 8 years of experience as an AI/ML Engineer, Data Scientist, or Software Engineer, with a strong focus on production ML systems.
- Bachelor’s degree in Computer Science, Software Engineering, or a related field.
- Proven experience taking AI solutions from concept to high-scale production using Java, C#, GoLang, or Python.
- Deep experience with generative AI, including prompting strategies, RAG, fine-tuning, and agentic workflows.
- Strong ability to design distributed systems and microservices architectures for model serving.
- Ability to lead technical discussions with non-technical stakeholders and drive consensus.
- Expert proficiency in Python or another core language such as Java, Go, or C#.
- Deep knowledge of modern ML frameworks, including PyTorch, TensorFlow, and Scikit-learn, and LLM orchestration tools such as LangChain, LlamaIndex, and vLLM.
- Advanced understanding of MLOps principles, feature stores, and model monitoring.
- Solid grasp of probability, statistics, and linear algebra.
- Hands-on experience deploying and managing Kubernetes clusters and using cloud platforms.
- Strong command of CI/CD methodologies and Infrastructure as Code, including Terraform.
- Data engineering skills, including SQL, NoSQL, data lakes, and processing pipelines.
Nice To Haves
- Master’s or PhD in AI/ML.
- Experience in the financial or fintech industry.
- Experience with AWS, the preferred cloud platform.
Responsibilities
- Partner with stakeholders to translate customer challenges into technical designs and AI strategies.
- Architect, build, and deploy high-performance AI/ML platforms and tools, ensuring scalability, reliability, and security.
- Lead development, evaluation, fine-tuning, and monitoring of production-grade AI/ML models, including LLMs and RAG architectures.
- Design and standardize CI/CD pipelines using GitHub Actions and Infrastructure as Code using Terraform for reproducible, automated deployments.
- Monitor system health, lead incident response for critical AI systems, and drive architectural refactoring to remove bottlenecks.
- Mentor L1/L2 engineers, conduct code reviews, and promote engineering excellence and best practices.
- Evaluate emerging AI technologies and vendors, and prototype new approaches.
Benefits
- Health insurance
- Dental insurance
- Vision insurance
- Retirement savings plan
- Paid time off
- Health savings account
- Flexible spending accounts
- Life insurance
- Disability insurance
- Tuition reimbursement
- Quarterly or annual bonus eligibility for non-sales roles, based on role and applicable plan
Stand Out From the Crowd
Upload your resume and get instant feedback on how well it matches this job.
What This Job Offers
Job Type
Full-time
Career Level
Senior
