Lead AI Engineer

Wex Corporate Payments - Portland, ME

Full-time - Senior

Portland, ME

5,001-10,000 employees

Funds, Trusts, and Other Financial Vehicles

About the position

We are the AI Engineering team at WEX. We enable our product development teams to move quickly and deploy AI functionality into their applications in a scalable manner. We're seeking a visionary Lead AI Architect to play a pivotal role in transforming our enterprise with a cutting-edge AI platform. You'll spearhead the design and implementation of a centralized, scalable AI services platform, empowering our business units to unlock the full potential of data-driven insights, LLMs, and automation. If you have a passion for building robust AI infrastructure, possess a deep understanding of cloud technologies, and thrive in a collaborative environment, this is your chance to shape the future of our organization. As a highly motivated engineer, you will work on small, high-performing teams. You will be collaborative, a solid communicator, and work well with your team and stakeholders. You will advocate strongly for positions while being 100% behind team decisions even if they don't go your way. You should be comfortable balancing the need to move fast with the realities of working in highly regulated spaces like payments and healthcare.

Responsibilities

• Spearhead the design and implementation of a centralized, scalable AI services platform.
• Guide and mentor teams of AI engineers, data scientists, and DevOps specialists.
• Partner cross-functionally to align AI initiatives with business objectives.
• Champion data privacy and security, implementing rigorous access controls and adhering to industry-specific regulations.
• Craft a comprehensive AI platform blueprint, encompassing data pipelines, model development and deployment, API integrations, monitoring, and governance.

Requirements

• 5+ years of hands-on experience designing and deploying complex cloud architectures (AWS, Azure, GCP, or similar).
• 10+ years of total experience in relevant fields.
• Exceptional written and verbal communication skills; able to translate complex technical concepts for diverse audiences.
• Proven mastery of event-driven architecture principles, with demonstrated success building systems using Kafka, RabbitMQ, or other messaging technologies.
• Solid Python development skills, complemented by experience with relevant AI/ML frameworks (TensorFlow, PyTorch, Scikit-learn, etc.).
• In-depth understanding of retrieval-augmented generation (RAG) architecture for augmenting large language models (LLMs) with relevant information from external knowledge sources.
• Fluency with Infrastructure as Code (IaC) tools like Terraform, CloudFormation, or Pulumi.
• Expertise in CI/CD pipelines, automation, and monitoring.

Nice-to-haves

• Prior work in healthcare, finance, or government sectors involving strict compliance requirements is preferred.
• Familiarity with MLOps concepts and tools to streamline model development and deployment workflows.

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 based on role and applicable plan.