Key skills
PythonJavaC#CGoGolangSQLNoSQLAIArtificial IntelligenceMLGenerative AILLMPyTorchTensorFlowscikit-learnLangChainMLOpsData EngineeringAWSTerraformKubernetesGitHub ActionsGitHubCI/CD
About this role
WEX is a fast-growing company focused on integrating artificial intelligence into their products and business. The Sr AI Engineer will lead the development of AI/ML platforms, mentor junior engineers, and collaborate with stakeholders to create innovative solutions that enhance decision-making for customers.
Responsibilities:
- Technical Strategy & Architecture: Partner with stakeholders to translate complex customer challenges into robust technical designs and AI strategies
- End-to-End Ownership: Architect, build, and deploy high-performance AI/ML platforms and tools, ensuring scalability, reliability, and security
- Advanced Model Development: Lead the development, evaluation, fine-tuning, and monitoring of production-grade AI/ML models (including LLMs and RAG architectures)
- Operational Excellence: Design and standardize CI/CD pipelines (GitHub Actions) and Infrastructure as Code (Terraform) to ensure reproducible and automated deployments
- Reliability & Scale: Proactively monitor system health, lead incident response for critical AI systems, and drive architectural refactoring to remove bottlenecks
- Mentorship: Mentor L1/L2 engineers, conduct code reviews, and foster a culture of engineering excellence and best practices
- Innovation: Evaluate emerging AI technologies/vendors and prototype new approaches to keep the organization at the cutting edge of the industry
Requirements:
- 5+ years of experience as an AI/ML Engineer, Data Scientist, or Software Engineer with a heavy focus on production ML systems
- Bachelor's degree in Computer Science, Software Engineering, or related field
- Proven track record of taking AI solutions from concept to high-scale production using Java, C#, GoLang, or Python
- Deep experience with Generative AI, including prompting strategies, RAG (Retrieval-Augmented Generation), fine-tuning, and agentic workflows
- Strong ability to design distributed systems and microservices architectures for model serving
- Demonstrated ability to lead technical discussions with non-technical stakeholders and drive consensus
- Expert proficiency in Python (preferred) or other core languages (Java, Go, C#)
- Deep knowledge of modern ML frameworks (PyTorch, TensorFlow, Scikit-learn) and LLM orchestration (LangChain, LlamaIndex, vLLM)
- Advanced understanding of MLOps principles, feature stores, and model monitoring
- Solid grasp of advanced probability, statistics, and linear algebra
- Strong command of CI/CD methodologies and Infrastructure as Code (Terraform)
- Excellent data engineering skills (SQL, NoSQL, data lakes, and processing pipelines)
- Master's or PhD in AI/ML
- Experience in the financial or fintech industry
- Expert proficiency in Python
- Extensive hands-on experience with Kubernetes (deploying and managing clusters) and Cloud platforms (AWS)