Job Title: Staff AI Engineer (L4) Location: Remote (Must reside within 30 miles of: Portland, ME; Boston, MA; Chicago, IL; Dallas, TX; San Jose, CA; or Seattle, WA) About the Team We are a fast-growing AI Engineering team, driving WEX's strategic vision to integrate artificial intelligence into the core of our product and business. Our team thrives on collaboration, working directly with stakeholders and domain experts to develop innovative AI platforms and solutions that enhance decision-making and create groundbreaking products that benefit our customers and help them to effectively grow their business. We foster a culture of continuous learning and believe that the best work stems from a shared purpose and an enjoyable, engaging environment. How You’ll Make an Impact You are an experienced technical leader and subject matter expert with a deep passion for AI, Machine Learning, and software engineering. You possess a proven track record of architecting scalable systems and solving complex ambiguity using data-driven tools. You will act as a force multiplier for the team—setting technical standards, mentoring junior engineers, and partnering with product leadership to turn abstract business goals into production-grade AI solutions. 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. Experience You’ll Bring 8+ 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. Production expertise: Proven track record of taking AI solutions from concept to high-scale production using Java, C#, GoLang, or Python. LLM Proficiency: Deep experience with Generative AI, including prompting strategies, RAG (Retrieval-Augmented Generation), fine-tuning, and agentic workflows. System Design: Strong ability to design distributed systems and microservices architectures for model serving. Collaboration: Demonstrated ability to lead technical discussions with non-technical stakeholders and drive consensus. Experience in the financial or fintech industry is a plus. Technical Skills 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). Key Words Staff AI Engineer, MLOps, Generative AI, LLM, RAG, Distributed Systems, Software Architecture, Technical Leadership. The base pay range represents the anticipated low and high end of the pay range for this position. Actual pay rates will vary and will be based on various factors, such as your qualifications, skills, competencies, and proficiency for the role. Base pay is one component of WEX's total compensation package. Most sales positions are eligible for commission under the terms of an applicable plan. Non-sales roles are typically eligible for a quarterly or annual bonus based on their role and applicable plan. WEX's comprehensive and market competitive benefits are designed to support your personal and professional well-being. Benefits include health, dental and vision insurances, retirement savings plan, paid time off, health savings account, flexible spending accounts, life insurance, disability insurance, tuition reimbursement, and more. For more information, check out the "About Us" section. Pay Range: $185,000.00 - $217,400.00 WEX is a global commerce platform that helps businesses solve for operational complexities like employee benefits, managing and mobilizing fleets, and streamlining payments. With over 6,500 employees, we work with large and small companies in more than 200 countries and territories, and can tailor our services to meet the unique needs of their businesses. We hire people who share our passion for continuous innovation and client service that is unparalleled in the industry. Offering comprehensive and market competitive benefits, our offerings are designed to support your personal and professional well-being. If you’re looking for a growing career - come be part of WEX today. To learn more about our employee benefits, please click here. WEX is an equal opportunity employer committed to diversity and inclusion in the workplace. All qualified applicants will receive consideration for employment without regard to sex, race, color, age, national origin, religion, sexual orientation, gender identity, protected veteran status, disability or other protected status. WEX promotes a drug-free workplace. Qualified individuals with a disability have the right to request a reasonable accommodation. If you require a reasonable accommodation as a result of your disability at any point in the job application process, please submit your request through our Reasonable Accommodation Request Form. This form is for accommodation requests only and cannot be used to inquire about the status of applications.

Staff AI Engineer (L4).

more at Wexinc.

Sr. Strategic Sales Manager- Corporate Payments- D.C/MD

AI Principal Engineer

Technical Program Manager

similar roles.

Marketing and Communications Data Analyst (Top Secret cleared)

ServiceNow Architect – AI Focused (Public Trust)

Data Scientist Senior--People Analytics & Quantitative Insights

Quantitative Analytics & Model Consultant Senior - Retail Banking (ML & Predictive modeling)

Senior Data Analyst

Data Analyst - Credit