HimalayasHimalayas logo
wexWE

Software AI Engineer Mid-Level, Context Engineering

WEX Inc. is a global commerce platform that simplifies the business of running a business by offering personalized technology solutions for employee benefits, mobility and fleet management, and corporate payments.

wex

Employee count: 5000+

Salary: 125k-149k USD

United States only

Stay safe on Himalayas

Never send money to companies. Jobs on Himalayas will never require payment from applicants.

As an AI Platform Engineer (SDE 2), you will be a hands-on developer responsible for building and maintaining the core software components that power our AI and context infrastructure. You will work on the "Context Layer"—the plumbing that connects enterprise data to LLMs—ensuring that our AI agents have the right information at the right time. This role is ideal for a strong software engineer who wants to specialize in the operational side of AI, focusing on high-quality code, automated delivery, and cloud-native systems.

Key Responsibilities

  • Feature Development: Implement and maintain core services for the AI Data Lakehouse, focusing on efficient data retrieval and storage optimizations for AI workflows.

  • Pipeline Automation: Build and support CI/CD pipelines to automate the deployment of AI models, prompt templates, and infrastructure updates.

  • Agentic Support: Develop and test tool-execution environments and API interfaces that allow AI agents to interact with internal business systems safely.

  • Operational Excellence: Participate in on-call rotations and troubleshooting to ensure platform reliability. Write unit tests, integration tests, and documentation for new features.

  • Context Retrieval: Work on the "Context Fabric" to implement search and retrieval patterns (like RAG) that help agents access secure enterprise data.

  • Cloud Management: Assist in managing cloud resources across AWS and Azure, ensuring environments are cost-effective and secure.

Qualifications & Experience

Software Engineering Foundation

  • Experience: 3+ years of professional software development experience.

  • Core Skills: Strong proficiency in Python and either Java or Scala. You write clean, maintainable, and well-documented code.

  • API Development: Experience building and consuming RESTful APIs or gRPC services.

  • Database Basics: Understanding of relational databases (Postgres/MySQL) and familiarity with how data is stored in a distributed environment.

Cloud & CI/CD Mastery

  • Cloud Consoles: Hands-on experience navigating the AWS or Azure Management Consoles. You should be comfortable managing basic services like IAM, S3/Blob, and compute instances.

  • Infrastructure-as-Code (IaC): Basic experience with Terraform. You can read, modify, and deploy infrastructure modules.

  • CI/CD Tools: Familiarity with GitHub Actions, GitLab CI, or Jenkins. You understand how to automate the build-test-deploy lifecycle.

  • Observability: Basic experience with monitoring tools like Prometheus, Grafana, or cloud-native solutions (CloudWatch/Azure Monitor).

AI & Agentic Interests (Specialized Focus)

  • LLM Awareness: Familiarity with LLM concepts and frameworks like LangChain or LlamaIndex. You’ve experimented with or built basic RAG-based applications.

  • Emerging Protocols: A desire to learn and implement new standards like the Model Context Protocol (MCP).

  • Agentic Workflows: Interest in how autonomous agents function, including tool-use (function calling) and state management.

  • Data Retrieval: Basic understanding of vector databases (e.g., Pinecone, Milvus) and how search impacts AI performance.

Leadership & Education

  • Team Player: Ability to work effectively in an agile environment, participating in sprint planning and daily stand-ups.

  • Continuous Learner: A strong desire to stay current with the rapidly changing AI and cloud landscape.

  • Education: Bachelor’s degree in Computer Science, Software Engineering, or a related technical field.

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: $124,700.00 - $148,800.00

About the job

Apply before

Posted on

Job type

Full Time

Experience level

Salary

Salary: 125k-149k USD

Education

Bachelor degree

Experience

3 years minimum

Experience accepted in place of education

Location requirements

Hiring timezones

United States +/- 0 hours

About wex

Learn more about wex and their company culture.

View company profile

WEX Inc. is a global commerce platform that simplifies the business of running a business. Many of our customers face the challenge of managing complex operational processes due to the rapid pace of regulatory, economic, and societal change worldwide. They are often stretched thin and lack the in-house expertise to solve these intricate problems. This is why WEX offers personalized technology solutions designed to simplify employee benefits, mobility and fleet management, and accounts payable and receivables processes. From our origins as a pioneer in fleet card payments in 1983, we have expanded our scope to become a multi-channel provider of corporate payment solutions, helping businesses navigate these complexities and achieve greater efficiency.

Our customers in the fleet industry, for example, need robust tools to manage fuel and maintenance expenses, ensure driver safety, and optimize operations. WEX Fleet provides them with fuel cards, telematics, and data analytics to meet these needs. For businesses involved in travel, managing cross-border payments and streamlining back-end accounting can be a significant hurdle. WEX's travel and corporate solutions, including virtual payment solutions, help these clients automate processes, reduce costs, and gain better insights into their spending. Similarly, in the healthcare sector, employers and employees alike grapple with the administration of benefits and healthcare payments. WEX Health offers a cloud-based platform to simplify the management of Health Savings Accounts (HSAs), Flexible Spending Accounts (FSAs), and other benefit plans, making it easier for millions of consumers to manage their healthcare expenses. By embedding our solutions into our customers' workflows and leveraging our expertise in data and analytics, we empower them to make smarter decisions, reduce operating costs, and ultimately, reach their full potential.

Employee benefits

Learn about the employee benefits and perks provided at wex.

View benefits

Company equity

WEX offers company equity.

Life insurance

WEX offers life insurance.

Paid sick days

WEX provides paid sick days.

Sabbatical

WEX offers sabbatical leave.

View wex's employee benefits
Claim this profilewex logoWE

wex

View company profile

Similar remote jobs

Here are other jobs you might want to apply for.

View all remote jobs

87 remote jobs at wex

Explore the variety of open remote roles at wex, offering flexible work options across multiple disciplines and skill levels.

View all jobs at wex

Remote companies like wex

Find your next opportunity by exploring profiles of companies that are similar to wex. Compare culture, benefits, and job openings on Himalayas.

View all companies

Find your dream job

Sign up now and join over 100,000 remote workers who receive personalized job alerts, curated job matches, and more for free!

Sign up
Himalayas profile for an example user named Frankie Sullivan