WEX Inc. Logo

WEX Inc.

Sr. Staff AI/ML Engineer

Reposted 9 Days Ago
Be an Early Applicant
In-Office
Dallas, TX
220K-256K Annually
Senior level
In-Office
Dallas, TX
220K-256K Annually
Senior level
Lead the design and deployment of AI and ML systems, integrating advanced capabilities into enterprise platforms, and collaborating cross-functionally to ensure scalability and best practices.
The summary above was generated by AI
Job Description
Location:  This is a remote position; however, the candidate must reside within 30 miles of one of the following locations: Portland, ME; Boston, MA; Chicago, IL; Dallas, TX; San Francisco Bay Area, CA; and Seattle/WA.
 

Team: AI Platform Engineering, WEX Inc.
 

About WEX and the AI Platform Team

At WEX, we’re transforming how businesses operate by embedding advanced AI into our global payments and mobility platforms. The AI Engineering team enables product development groups across WEX to build, deploy, and scale AI-powered experiences quickly and safely. We’re cloud-first, automation-driven, and passionate about building systems that make AI a core part of our products—reliably, securely, and at scale.

If you’re excited by Large Language Models, Agentic AI systems, MLOps, and cutting-edge production pipelines—and you thrive in fast-moving teams where innovation meets real-world impact—this is the role for you.
 

What You’ll Do
  • Lead the design, implementation, and production deployment of machine learning and AI-driven systems—including LLM-based and agentic applications.

  • Partner with AI platform and product engineering teams to integrate advanced AI capabilities into WEX’s enterprise systems.

  • Design and maintain ML pipelines, from data ingestion to model deployment, ensuring scalability, observability, and reusability across teams.

  • Build and expose AI functionality via RESTful APIs and micro-services architectures.

  • Champion engineering best practices: CI/CD, infrastructure-as-code, testing automation, and continuous improvement.

  • Contribute to architectural decisions with a focus on security, compliance, and performance—especially in regulated industries such as payments and healthcare.

  • Collaborate cross-functionally with data scientists, ML engineers, and business stakeholders to align technical solutions with strategic goals.
     

What You’ll Bring
  • 12+ years of professional software or ML engineering experience, with a track record of deploying production-grade AI systems.

  • Proficiency in Python and key machine learning frameworks (PyTorch, TensorFlow, or similar).

  • Strong working knowledge of core libraries (NumPy, Pandas, scikit-learn) and LLM development frameworks (LangChain, ADK, or similar).

  • Experience with cloud platforms (AWS preferred; Azure or GCP also valuable) and Infrastructure-as-Code tools like Terraform.

  • Deep familiarity with CI/CD pipelines and DevOps practices using GitHub Actions or similar platforms.

  • Demonstrated ability to operate in agile, collaborative, high-trust teams.

  • Bachelor’s degree in Computer Science, Engineering, or a related discipline (Master’s preferred).

  • Bonus: Experience in financial systems, data compliance, or building multi-tenant Agentic AI applications.
     

Why Join WEX

You’ll work within a lean, high-impact engineering group that values curiosity, autonomy, and real-world problem-solving. We move fast but balance innovation with responsibility—because AI at WEX touches millions of businesses and end users.
 

Join us to help evolve intelligent systems that change how enterprises move money and make decisions.

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: $220,000.00 - $255,800.00

Similar Jobs

18 Days Ago
Easy Apply
In-Office
Easy Apply
Senior level
Senior level
Hardware • Internet of Things • Software • Wearables • Semiconductor
Design, train, optimize, and deploy efficient on-device ML models (audio and vision) for resource-constrained products; maintain Ambiq ADKs; port and optimize customer models; apply model-efficiency techniques; produce customer-facing demos, docs, and benchmarks.
Top Skills: C++CompressionDockerExecutorchFftJaxKerasLinuxLitertOnnxPruningPtqPythonPyTorchQatQuantizationRustSparsificationTensorFlow
4 Hours Ago
Hybrid
113K-183K Annually
Senior level
113K-183K Annually
Senior level
Artificial Intelligence • Cloud • Internet of Things • Software • Cybersecurity • Industrial
The Senior Software Engineer will develop simulation logic, create 3D visualization environments, validate digital twins, and integrate simulation components, utilizing various digital twin technologies.
Top Skills: Ansys Twin BuilderC++CarlaDassault 3DexperienceNvidia OmniversePlant SimulationPythonSiemens NxTeamcenterUnityUnreal EngineUsd
4 Hours Ago
Hybrid
15-19 Hourly
Entry level
15-19 Hourly
Entry level
eCommerce • Fashion • Retail • Sales • Wearables • Design
As a Temporary Sales Associate, you will represent Coach, build client relationships, drive sales, manage store operations, and maintain a high-energy environment while meeting personal and team goals.

What you need to know about the Colorado Tech Scene

With a business-friendly climate and research universities like CU Boulder and Colorado State, Colorado has made a name for itself as a startup ecosystem. The state boasts a skilled workforce and high quality of life thanks to its affordable housing, vibrant cultural scene and unparalleled opportunities for outdoor recreation. Colorado is also home to the National Renewable Energy Laboratory, helping cement its status as a hub for renewable energy innovation.

Key Facts About Colorado Tech

  • Number of Tech Workers: 260,000; 8.5% of overall workforce (2024 CompTIA survey)
  • Major Tech Employers: Lockheed Martin, Century Link, Comcast, BAE Systems, Level 3
  • Key Industries: Software, artificial intelligence, aerospace, e-commerce, fintech, healthtech
  • Funding Landscape: $4.9 billion in VC funding in 2024 (Pitchbook)
  • Notable Investors: Access Venture Partners, Ridgeline Ventures, Techstars, Blackhorn Ventures
  • Research Centers and Universities: Colorado School of Mines, University of Colorado Boulder, University of Denver, Colorado State University, Mesa Laboratory, Space Science Institute, National Center for Atmospheric Research, National Renewable Energy Laboratory, Gottlieb Institute

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account