WEX

Senior AI Engineer

WEX

full-time

Posted on:

Location Type: Remote

Location: CaliforniaIllinoisUnited States

Visit company website

Explore more

AI Apply
Apply

Salary

💰 $121,500 - $145,500 per year

Job Level

About the role

  • 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.

Requirements

  • 5+ years of experience as an AI/ML Engineer, Data Scientist, or Software Engineer
  • 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
  • Experience in the financial or fintech industry is a plus
  • 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).
Benefits
  • health, dental and vision insurances
  • retirement savings plan
  • paid time off
  • health savings account
  • flexible spending accounts
  • life insurance
  • disability insurance
  • tuition reimbursement
  • comprehensive and market competitive benefits
Applicant Tracking System Keywords

Tip: use these terms in your resume and cover letter to boost ATS matches.

Hard Skills & Tools
AI/MLJavaC#GoLangPythonGenerative AIRAGMLOpsKubernetesSQL
Soft Skills
mentoringtechnical discussionsdriving consensusincident responsearchitectural refactoringengineering excellencecode reviews
Certifications
Bachelor's degree in Computer ScienceBachelor's degree in Software EngineeringMaster’s degree in AI/MLPhD in AI/ML