Mid-Level Software AI Engineer, Context Engineering

WEX

AI Platform Engineer developing core software components for AI infrastructure at WEX. Specializing in operational excellence and ensuring AI agents have the right information.

Posted 6/2/2026full-timeRemote • California • 🇺🇸 United StatesMid-LevelSenior💰 $124,700 - $148,800 per yearWebsite

Tech Stack

Tools & technologies

AWSAzureCloudGrafanaGRPCJavaJenkinsMySQLPostgresPrometheusPythonScalaTerraform

About the role

Key responsibilities & impact

Implement and maintain core services for the AI Data Lakehouse
Build and support CI/CD pipelines
Develop and test tool-execution environments and API interfaces
Participate in on-call rotations and troubleshooting
Work on the 'Context Fabric' to implement search and retrieval patterns
Assist in managing cloud resources across AWS and Azure

Requirements

What you’ll need

3+ years of professional software development experience
Strong proficiency in Python and either Java or Scala
Experience building and consuming RESTful APIs or gRPC services
Understanding of relational databases (Postgres/MySQL)
Hands-on experience navigating the AWS or Azure Management Consoles
Basic experience with Terraform
Familiarity with GitHub Actions, GitLab CI, or Jenkins
Basic experience with monitoring tools like Prometheus, Grafana, or cloud-native solutions
Familiarity with LLM concepts and frameworks like LangChain or LlamaIndex
Basic understanding of vector databases (e.g., Pinecone, Milvus)

Benefits

Comp & perks

health, dental and vision insurances
retirement savings plan
paid time off
health savings account
flexible spending accounts
life insurance
disability insurance
tuition reimbursement
more

ATS Keywords

✓ Tailor your resume

Applicant Tracking System Keywords

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

Hard Skills & Tools

PythonJavaScalaRESTful APIsgRPCPostgresMySQLTerraformPrometheusGrafana