Lead the design, implementation, and production deployment of machine learning and AI-driven systems—including LLM-based and agentic applications
Partner with AI platform and product engineering teams to integrate advanced AI capabilities into WEX’s enterprise systems
Design and maintain ML pipelines, from data ingestion to model deployment, ensuring scalability, observability, and reusability across teams
Build and expose AI functionality via RESTful APIs and micro-services architectures
Champion engineering best practices: CI/CD, infrastructure-as-code, testing automation, and continuous improvement
Contribute to architectural decisions with a focus on security, compliance, and performance—especially in regulated industries such as payments and healthcare
Collaborate cross-functionally with data scientists, ML engineers, and business stakeholders to align technical solutions with strategic goals

12+ years of professional software or ML engineering experience
Proficiency in Python and key machine learning frameworks (PyTorch, TensorFlow, or similar)
Strong working knowledge of core libraries (NumPy, Pandas, scikit-learn) and LLM development frameworks (LangChain, ADK, or similar)
Experience with cloud platforms (AWS preferred; Azure or GCP also valuable)
Deep familiarity with CI/CD pipelines and DevOps practices using GitHub Actions or similar platforms
Bachelor’s degree in Computer Science, Engineering, or a related discipline (Master’s preferred)

Senior Staff AI/ML Engineer

Key skills