Python AI Engineer

EPAM Systems; United States; Remote; USD 130,000 - 145,000; Full time

Benefits offered by this job

Medical, Dental and Vision Insurance

Matched 401(k) Retirement Savings Plan

Paid Time Off

Job summary

EPAM Systems Inc is hiring a Python AI Engineer to advance an enterprise knowledge platform, expanding from Confluence-like ingestion to a hybrid retrieval stack and agent runtime.

You will design robust backend services in Python 3.11 with FastAPI, Pydantic, async SQLAlchemy, and asyncpg, improve grounding, observability, and release workflows, while collaborating with product and engineering teams. This role offers a base salary range and remote options for the US.

Qualifications

• Hands-on Python in platform, automation, or infrastructure-heavy environments.
• Experience with CLI tools using Python, Go, or Rust.
• Experience with LangGraph, LangChain, pgvector, and modern retrieval pipelines.
• Design evaluation frameworks for LLM-backed systems, including regression detection and quality measurement.
• Familiar with Docker, Helm, GitHub Actions, and Kubernetes-oriented workflows.

Responsibilities

• Architect and implement robust backend services using Python 3.11, FastAPI, Pydantic, SQLAlchemy async, and asyncpg.
• Design trade-offs around retrieval quality, latency, cost, safety, and operational simplicity.
• Build production-grade agent runtime capabilities, including memory boundaries, tool sandboxing, permissions, and budget controls.
• Improve answer grounding, failure analysis, and citation enforcement for enterprise-grade dependability.
• Create comprehensive observability and operational feedback loops using OpenTelemetry, Prometheus/Grafana, Docker/Helm, and GitHub Actions.
• Ingestion infrastructure across current and future content sources.
• Full-stack observability across application, pipeline, database, and model-serving behavior.
• Cost, latency, throughput, and failure-mode management for AI-heavy workloads.
• Release workflows that validate AI behavior changes, not just code compilation.
• Partner closely with product and engineering teams to support multiple conversational surfaces through a unified knowledge platform.

Skills

Python

FastAPI

Pydantic

SQLAlchemy

asyncpg

LangGraph

LangChain

pgvector

Retrieval pipelines

Docker

Kubernetes

GitHub Actions

OpenTelemetry

observability

Golang

Rust

Tools

Docker

Helm

GitHub Actions

Kubernetes

asyncpg

OpenTelemetry

pgvector

LangGraph

LangChain

Job description

We are hiring a Python AI Engineer to operate and evolve the infrastructure behind an enterprise knowledge base platform as it transitions from a Confluence-focused RAG chatbot into a broader, agentic knowledge system.Today, our platform supports Confluence/GitHub ingestion chunking pgvector RAG retrieval FastAPI serving. In our next growth phase, we are expanding toward hybrid retrieval (vector + sparse + graph), multi-source ingestion, robust evaluation pipelines, agent infrastructure, and shared chat platform primitives.If you are passionate about building scalable, reliable, and production-grade AI infrastructure rather than just chasing demo behavior, we want to hear from you.Req.#1093687315ResponsibilitiesArchitect and implement robust backend services using Python 3.11, FastAPI, Pydantic, SQLAlchemy async, and asyncpgDesign trade-offs around retrieval quality, latency, cost, safety, and operational simplicityBuild production-grade agent runtime capabilities, including memory boundaries, tool sandboxing, permissions, and budget controlsImprove answer grounding, failure analysis, and citation enforcement for enterprise-grade dependabilityCreate comprehensive observability and operational feedback loops using OpenTelemetry, Prometheus/Grafana, Docker/Helm, and GitHub ActionsIngestion infrastructure across current and future content sourcesFull-stack observability across application, pipeline, database, and model-serving behaviorCost, latency, throughput, and failure-mode management for AI-heavy workloadsRelease workflows that validate AI behavior changes, not just code compilationPartner closely with product and engineering teams to support multiple conversational surfaces through a unified knowledge platformRequirementsStrong, hands-on Python experience in platform, automation, or infrastructure-heavy environmentsExperience building CLI tools using Python, Golang, or RustHands-on experience with LangGraph, LangChain, pgvector, and modern retrieval pipelinesExperience designing evaluation frameworks for LLM-backed systems, including regression detection and quality measurementStrong experience with Docker, Helm, GitHub Actions, and Kubernetes-oriented workflowsSystems Understanding: Familiarity with the operational characteristics of embedding pipelines, vector search, and LLM-backed systemsStrong skills across metrics, tracing, dashboards, alerting, and log analysisExperience with ingestion, ETL, or content-processing pipelines at scaleAn innate ability to think in terms of reliability, cost, latency, throughput, and disaster recoveryNice to haveExperience with Qdrant, Neo4j, or other vector/graph infrastructurePrior experience supporting RAG, search, evaluation, or agent platformsBackground in enterprise or highly regulated environmentsFamiliarity with enterprise tooling such as Vault, Splunk, Artifactory, or ECRComfort leveraging AI-assisted engineering workflows in day-to-day workWe offerMedical, Dental and Vision Insurance (Subsidized)Health Savings AccountFlexible Spending Accounts (Healthcare, Dependent Care, Commuter)Short-Term and Long-Term Disability (Company Provided)Life and AD&D Insurance (Company Provided)Employee Assistance ProgramUnlimited access to LinkedIn learning solutionsMatched 401(k) Retirement Savings PlanPaid Time Off – the employee will be eligible to accrue 15-25 paid days, depending on specific level and tenure with EPAM (accrual eligibility may change over time)Paid Holidays - nine (9) total per yearLegal Plan and Identity Theft ProtectionAccident InsuranceEmployee DiscountsPet InsuranceEmployee Stock Purchase ProgramIf otherwise eligible, participation in the discretionary annual bonus programIf otherwise eligible and hired into a qualifying level, participation in the discretionary Long-Term Incentive (LTI) ProgramFor remote work in New York City only.EPAM is a leading global provider of digital platform engineering and development services. We are committed to having a positive impact on our clients, our employees, and our communities. We embrace a dynamic and inclusive culture. Here you will collaborate with multi-national teams, contribute to a myriad of innovative projects that deliver the most creative and cutting-edge solutions, and have an opportunity to continuously learn and grow. No matter where you are located, you will join a dedicated, creative, and diverse community that will help you discover your fullest potential.Engineer the Future with a Career at EPAMThis posting includes a good faith range of the salary EPAM would reasonably expect to pay the selected candidate. The range provided reflects base salary only. Individual compensation offers within the range are based on a variety of factors, including, but not limited to: geographic location, experience, credentials, education, training; the demand for the role; and overall business and labor market considerations. Most candidates are hired at a salary within the range disclosed. Salary range: $130,000 - $145,000. In addition, the details highlighted in this job posting above are a general description of all other expected benefits and compensation for the position.Applications will be accepted on a rolling basis.EPAM will not provide new H-1B visa sponsorship for this position. Candidates with existing transferable H-1B status may be considered.It is unlawful in Massachusetts to require or administer a lie detector test as a condition of employment or continued employment. An employer who violates this law shall be subject to criminal penalties and civil liability.