Senior/Lead Python AI Engineer

About the Position

We are seeking a Senior/Lead Python AI Engineer to help build an Enterprise Agent Development Platform - a production-grade, cloud-native ecosystem that enables engineering teams to define, orchestrate, deploy, and observe AI agents at scale. The platform standardizes agent development across the organization using LangGraph and Strands Agents on AWS AgentCore Runtime. The initiative spans agent framework design, runtime architecture, marketplace integration, CI/CD automation, and enterprise-grade observability - reducing agent development from months to days while enforcing consistent security, quality, and governance standards.

Responsibilities

• Integrate AgentCore Memory into platform agent templates and helper libraries
• Implement agent memory integration with write-on-turn (store session events) and retrieve-on-turn (fetch relevant context) patterns
• Work with AWS AgentCore Memory API to create, query, update, and delete memory
• Design memory namespaces with per-agent isolation and per-user scoping
• Develop Python-based agent memory helpers as part of a platform library for memory-enabled agents
• Enable agent template integration allowing memory activation through configuration without custom code
• Ensure session management and reliable context persistence across agent interactions
• Collaborate with platform teams to maintain consistent security, quality, and governance standards

Requirements

• 3+ years of backend engineering experience with AI/ML systems
• Hands-on experience with agent memory integration in a production agentic AI project (general backend/API integration experience without agent-specific memory context does not meet this bar)
• Expertise in AWS AgentCore Memory API, including write-on-turn/retrieve-on-turn patterns and memory namespace design
• Proficiency in Python SDK development
• Background in API integration for stateful services
• Skills in session management and context persistence
• Capability to configure agent template memory without custom code
• English proficiency at an Upper-Intermediate level (B2) or higher

Nice to Have

• Familiarity with AWS AgentCore Memory SDK usage
• Knowledge of LangChain memory integration
• Experience in agent framework development with LangGraph and Crew
• Understanding of conversational AI context handling
• Exposure to FastMCP

We Offer

With us you can:

• Work on a flexible schedule remotely or from any of our comfortable offices or coworking spaces in Ukraine
• Receive the necessary equipment to perform your work tasks
• Change projects and technology stacks within EPAM
• Gain experience in various business domains (Insurance, E-commerce, Healthcare, Finance, Travelling, Media, Artificial Intelligence, and more)
• Relocation opportunities may be available for eligible candidates, depending on the role and openings at other EPAM locations
• Participate in volunteer, charity programs and communities (both technical and interest-based)

We focus on your professional growth:

• You can plan your individual career path together with your manager
• Receive regular feedback from colleagues
• Improve your English for free with certified teachers (Speaking Clubs, client interview preparation courses, etc.)
• Get the opportunity to undergo free training and certification in AWS, GCP, or Azure Clouds
• Use the internal E-learn training program (18,200+ specialized training and mentoring programs)
• Access corporate accounts on LinkedIn Learning, Get Abstract and other partner resources
• Study at EPAM Solution Architecture School with the instructors who are practicing architects
• Develop as a leader, join Delivery Management, Resource Management, Leadership Essentials school and more
• Participate in internal communities (500+ meetups, technical discussions, brainstorming sessions, online events and conferences annually)

Benefits

• Vacation and sick leave (including a sick leave without a medical certificate)
• A wide range of Voluntary Medical Insurance programs providing both medical treatment and various preventive options (including sports activities)
• Medical insurance for family members at corporate rates
• Company support during significant life events (childbirth or adoption, marriage, etc.)
• Support for psychological comfort: discounts on services from mental health specialists or coaches, thematic training
• E-kids program - a free programming language training program for EPAMers' children