Python AI Engineer

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.#1093687315

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

Requirements

• Strong, hands-on Python experience in platform, automation, or infrastructure-heavy environments
• Experience building CLI tools using Python, Golang, or Rust
• Hands-on experience with LangGraph, LangChain, pgvector, and modern retrieval pipelines
• Experience designing evaluation frameworks for LLM-backed systems, including regression detection and quality measurement
• Strong experience with Docker, Helm, GitHub Actions, and Kubernetes-oriented workflows
• Systems Understanding: Familiarity with the operational characteristics of embedding pipelines, vector search, and LLM-backed systems
• Strong skills across metrics, tracing, dashboards, alerting, and log analysis
• Experience with ingestion, ETL, or content-processing pipelines at scale
• An innate ability to think in terms of reliability, cost, latency, throughput, and disaster recovery

Nice to have

• Experience with Qdrant, Neo4j, or other vector/graph infrastructure
• Prior experience supporting RAG, search, evaluation, or agent platforms
• Background in enterprise or highly regulated environments
• Familiarity with enterprise tooling such as Vault, Splunk, Artifactory, or ECR
• Comfort leveraging AI-assisted engineering workflows in day-to-day work

We offer

• Medical, Dental and Vision Insurance (Subsidized)
• Health Savings Account
• Flexible Spending Accounts (Healthcare, Dependent Care, Commuter)
• Short-Term and Long-Term Disability (Company Provided)
• Life and AD&D Insurance (Company Provided)
• Employee Assistance Program
• Unlimited access to LinkedIn learning solutions
• Matched 401(k) Retirement Savings Plan
• Paid 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 year
• Legal Plan and Identity Theft Protection
• Accident Insurance
• Employee Discounts
• Pet Insurance
• Employee Stock Purchase Program
• If otherwise eligible, participation in the discretionary annual bonus program
• If otherwise eligible and hired into a qualifying level, participation in the discretionary Long-Term Incentive (LTI) Program

For 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 EPAM

This 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.

Affirmative Action Obligations as a U.S. Government Federal Contractor

As a U.S. federal government contractor, EPAM is committed to meet its affirmative action obligations to make good faith efforts to expand the recruiting pool of individuals with disabilities and protected veterans through outreach, targeted recruitment, training opportunities and other activities. We affirm this commitment annually in EPAM’s Affirmative Action Plans. The full text of our Affirmative Action Plan for Persons with a Disability and Protected Veterans is available for inspection in the People Operations Department during normal business hours. Email the People Operations Department to schedule an appointment.

Equal Employment Opportunity

EPAM Systems, Inc. is an equal opportunity employer. We are committed to recruiting, hiring, developing and promoting employees without discrimination. As a global employer, this commitment includes complying with all laws in the countries in which we operate. Equal opportunity and inclusion are essential to motivate, empower and recognize the best in everyone.

At EPAM, employment actions are based on individual qualifications, without regard to race, color, religion, creed, gender, pregnancy status, sexual orientation, gender identity, gender expression, marital or familial status, national origin, ancestry, genetics, age, disability status, veteran status, citizenship status when otherwise legally able to work, or any other characteristic protected by law.

Accessibility for Applicants with Disabilities

EPAM is committed to working with and providing reasonable accommodation to individuals with disabilities. If you require an accommodation at any stage of the employment application process, please send an email to the People Operations Department including your name, a detailed description of your requested accommodation, and the best method to contact you. If you have already reviewed a job posting or submitted an application for a job, please include the requisition number. We will assist you and make a determination on your accommodation request on a case-by-case basis.

EPAM Systems, Inc. participates in eVerify.

Pay Transparency Non-Discrimination Provision

EPAM will not discharge or in any other manner discriminate against employees or applicants because they have inquired about, discussed, or disclosed their own pay or the pay of another employee or applicant. However, employees who have access to the compensation information of other employees or applicants as a part of their essential job functions cannot disclose the pay of other employees or applicants to individuals who do not otherwise have access to compensation information, unless the disclosure is (a) in response to a formal complaint or charge, (b) in furtherance of an investigation, proceeding, hearing, or action, including an investigation conducted by the employer, or (c) consistent with the contractor’s legal duty to furnish information. 41 CFR 60-1.35(c)

Remote in United States of America: New York