WEX Inc. Logo

WEX Inc.

Senior AI/ML Engineer

Sorry, this job was removed at 06:12 a.m. (PST) on Thursday, Sep 04, 2025
In-Office
CA, USA
126K-168K Annually
In-Office
CA, USA
126K-168K Annually

Similar Jobs

4 Days Ago
In-Office or Remote
130K-189K Annually
Senior level
130K-189K Annually
Senior level
Consulting
Design, build, and deploy AI systems across the full lifecycle: data pipelines, model development (traditional ML and generative AI), agent-based applications, API integration, observability, and evaluation. Collaborate with architects and domain experts, support experimentation and production deployments, and assist with project coordination and stakeholder communications.
Top Skills: Agent OrchestrationAgent Routing ProtocolsAutogenAWSAws SagemakerAzureAzure Machine LearningCi/CdCrewaiDockerEmbeddingsFastapiFlaskGoogle Ai/Ml ServicesLanggraphMcpMlopsOpenai ApisPrompt EngineeringPythonPyTorchScikit-LearnTensorFlowVector Databases
4 Days Ago
In-Office or Remote
110K-130K Annually
Senior level
110K-130K Annually
Senior level
Information Technology • Real Estate
Design and build scalable .NET backend services and gRPC microservices; optimize data layers (SQL Server, MongoDB, Elasticsearch); implement cloud integrations on AWS and Azure; ship AI/ML features (LLMs, RAG, vector search, ranking, personalization) and operate MCP servers; contribute to CI/CD, observability, containerization, and production ML integrations impacting high-traffic consumer marketplaces.
Top Skills: .Net 8+.Net Core.Net FrameworkAsp.Net Minimal ApisAWSAzureAzure Service BusC#Ci/CdDockerElasticsearchGa4GitGithub ActionsGrpcKafkaLangchainLlmsMediatrMlflowModel Context Protocol (Mcp)MongoDBNext.JsProtocol Buffers(Sql .Proto)PythonRagReactRedisSegmentSnsSQL ServerTypescriptVector Dbs
3 Days Ago
In-Office or Remote
CA, USA
195K-343K Annually
Senior level
195K-343K Annually
Senior level
Blockchain • eCommerce • Fintech • Payments • Software • Financial Services • Cryptocurrency
In this role, you will lead AI personalization initiatives, develop AI/ML models for user experience, collaborate across teams, and ensure rigorous development practices.
Top Skills: AWSDatabricksPysparkPyTorch
(*) The candidate must reside within 30 miles of one of the following locations: San Francisco Bay Area, CA;  Portland, ME; Chicago, Il.

About the Team/Role

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 are cloud-first and aim to automate everything, including  infrastructure, testing, and deployment.  If you’re interested in ChatGPT, Large Language Models, MLOps, DevOps, and thirst for working with new tools and technologies, then this is the place to be!

How you’ll make an impact
  • A highly motivated engineer who loves working on small, highly performing teams

  • Collaborative, a solid communicator, and work well with your team and stakeholders.

  • A professional who can advocate strongly for positions and still be 100% behind team decisions even if they don’t go your way.

  • Are comfortable balancing the need to move fast with the realities of working in highly regulated spaces like payments and healthcare.

  • Design, implement, train and maintain machine learning algorithms and ML system pipelines.

  • Work closely with our AI platform team and line of business engineering teams to integrate AI components and models into WEX systems.

  • Collaborate with the team to manage version control using GitHub and implement robust CI/CD pipelines for continuous integration and delivery.

  • Design and implement RESTful APIs to facilitate communication between various components and ensure smooth operation of our applications.

  • Collaborate effectively with cross-functional teams, participate in code reviews, and communicate solutions to both technical and non-technical stakeholders.

Experience you’ll bring
  • Minimum 5 years of application development experience using Python.

  • Minimum 3 years of experience building, training and deploying machine learning models.

  • Strong ability to use key Python libraries such as Pandas, Numpy and deep learning libraries like PyTorch or Tensorflow.

  • Bachelor's Degree in Computer Science, Engineering, or related field.

  • Familiarity with cloud technologies, such as AWS, Azure, or GCP and how to provision those services through IAC frameworks such as Terraform.

  • Experience applying DevOps principles to software development processes.

  • Proficiency in version control systems, particularly GitHub.

  • Excellent problem-solving skills and a proactive approach to addressing challenges.

  • Experience in the financial industry is a plus.

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: $126,000.00 - $168,000.00

What you need to know about the Los Angeles Tech Scene

Los Angeles is a global leader in entertainment, so it’s no surprise that many of the biggest players in streaming, digital media and game development call the city home. But the city boasts plenty of non-entertainment innovation as well, with tech companies spanning verticals like AI, fintech, e-commerce and biotech. With major universities like Caltech, UCLA, USC and the nearby UC Irvine, the city has a steady supply of top-flight tech and engineering talent — not counting the graduates flocking to Los Angeles from across the world to enjoy its beaches, culture and year-round temperate climate.

Key Facts About Los Angeles Tech

  • Number of Tech Workers: 375,800; 5.5% of overall workforce (2024 CompTIA survey)
  • Major Tech Employers: Snap, Netflix, SpaceX, Disney, Google
  • Key Industries: Artificial intelligence, adtech, media, software, game development
  • Funding Landscape: $11.6 billion in venture capital funding in 2024 (Pitchbook)
  • Notable Investors: Strong Ventures, Fifth Wall, Upfront Ventures, Mucker Capital, Kittyhawk Ventures
  • Research Centers and Universities: California Institute of Technology, UCLA, University of Southern California, UC Irvine, Pepperdine, California Institute for Immunology and Immunotherapy, Center for Quantum Science and Engineering

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account