WEX · 2 days ago
Sr AI Engineer
United States
Full-time
Remote
Mid, Senior Level
$122K/yr - $146K/yr
5+ years expWEX is a fast-growing company focused on integrating artificial intelligence into its products and business. They are seeking a Senior AI Engineer to lead technical strategies, architect AI/ML platforms, and mentor junior engineers while driving innovation in AI technologies.
Financial ServicesFinTechPaymentsSoftware
Responsibilities
Partner with stakeholders to translate complex customer challenges into robust technical designs and AI strategies
Architect, build, and deploy high-performance AI/ML platforms and tools, ensuring scalability, reliability, and security
Lead the development, evaluation, fine-tuning, and monitoring of production-grade AI/ML models (including LLMs and RAG architectures)
Design and standardize CI/CD pipelines (GitHub Actions) and Infrastructure as Code (Terraform) to ensure reproducible and automated deployments
Proactively 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 foster a culture of engineering excellence and best practices
Evaluate emerging AI technologies/vendors and prototype new approaches to keep the organization at the cutting edge of the industry
Qualification
Required
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; a Master's or PhD in AI/ML is preferred
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
Extensive hands-on experience with Kubernetes (deploying and managing clusters) and Cloud platforms (AWS preferred)
Strong command of CI/CD methodologies and Infrastructure as Code (Terraform)
Excellent data engineering skills (SQL, NoSQL, data lakes, and processing pipelines)
Preferred
Experience in the financial or fintech industry is a plus
Benefits
Health, dental and vision insurances
Retirement savings plan
Paid time off
Health savings account
Flexible spending accounts
Life insurance
Disability insurance
Tuition reimbursement
Company
WEX
Wex is a financial technology service provider for fleet, travel and healthcare industries.
5001-10000 employees
H1B Sponsorship
WEX has a track record of offering H1B sponsorships. Please note that this does not
guarantee sponsorship for this specific role. Below presents additional info for your
reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (22)
2024 (10)
2023 (12)
2022 (20)
2021 (14)
2020 (18)
Funding
Current Stage
Public CompanyTotal Funding
$900MKey Investors
Warburg Pincus
2025-02-25Post Ipo Debt· $500M
2020-06-30Post Ipo Equity· $90M
2020-06-30Post Ipo Debt· $310M
Leadership Team
Melissa Smith
Chair and Chief Executive Officer
Sachin Dhawan
Chief Technology Officer
Recent News
2026-02-06
2026-02-05
Morningstar.com
2026-01-22
Company data provided by crunchbase