Staff AI Engineer – L4

🔥 0 minutes ago

🏄 California, Illinois, +4 more states – Remote

🏄 California – Remote 🌽 Illinois – Remote 🦞 Maine – Remote 🍂 Massachusetts – Remote 🤠 Texas – Remote ☕ Washington – Remote

💵 $185k - $217.4k / year

⏰ Full Time

🔴 Lead

🤖 AI Engineer

🦅 H1B Visa Sponsor

This company has sponsored H1B visas in the past.

Apply Now

Find Similar Remote Jobs

Set Up Email Alert For New Jobs

📊 Check your resume score for this job

Improve your chances of getting an interview by checking your resume score before you apply.

Upload Resume

WEX

WebsiteLinkedInAll Job Openings

5001 - 10000 employees

Founded 1983

🏥 Healthcare

📦 Logistics

✈️ Travel

💰 $310M Post-IPO Debt on 2020-06

Healthcare • Logistics • Travel

WEX is a global commerce platform specializing in various business solutions to address operational challenges. They provide services in managing and mobilizing fleets with their fuel card systems, offering comprehensive fleet management and analytics. Additionally, they focus on business payments solutions that streamline processes across industries, enhancing efficiency and security. WEX is also involved in employee benefits administration, helping organizations effectively manage health and reimbursement accounts. Their diverse range of services caters to numerous sectors, emphasizing innovation, sustainability, and effective solutions for business growth.

📋 Description

• Partner with stakeholders to translate complex customer challenges into robust technical designs and AI strategies • Architect, build, and deploy scalable, reliable, secure AI/ML platforms and tools • Lead development, evaluation, fine-tuning, and monitoring of production-grade AI/ML models, including LLMs and RAG architectures • Design and standardize GitHub Actions CI/CD pipelines and Terraform Infrastructure as Code • Monitor system health, lead critical AI-system incident response, and drive architectural refactoring • Mentor L1/L2 engineers and conduct code reviews • Evaluate emerging AI technologies and vendors and prototype new approaches • Set technical standards and partner with product leadership to turn business goals into production-grade AI solutions

🎯 Requirements

• 8+ years of experience as an AI/ML Engineer, Data Scientist, or Software Engineer focused on production ML systems • Bachelor's degree in Computer Science, Software Engineering, or related field • Proven experience taking AI solutions from concept to high-scale production using Java, C#, GoLang, or Python • Deep experience with Generative AI, prompting strategies, RAG, fine-tuning, and agentic workflows • Strong ability to design distributed systems and microservices architectures for model serving • Ability to lead technical discussions with non-technical stakeholders and drive consensus • Expert proficiency in Python or other core languages such as Java, Go, or C# • Deep knowledge of PyTorch, TensorFlow, Scikit-learn, LangChain, LlamaIndex, and vLLM • Advanced understanding of MLOps principles, feature stores, and model monitoring • Advanced probability, statistics, and linear algebra knowledge • Extensive hands-on Kubernetes experience and cloud platform experience, preferably AWS • Strong command of CI/CD methodologies and Terraform Infrastructure as Code • Data engineering skills with SQL, NoSQL, data lakes, and processing pipelines • Experience in financial or fintech industry is a plus

🏖️ Benefits

• Quarterly or annual bonus eligibility for non-sales roles • Health insurance • Dental insurance • Vision insurance • Retirement savings plan • Paid time off • Health savings account • Flexible spending accounts • Life insurance • Disability insurance • Tuition reimbursement • Reasonable accommodation for qualified individuals with disabilities