Lead Python Backend Engineer

We are looking for a Lead Python Backend Engineer to join our team building 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 write-on-turn functionality to store session events
• Implement retrieve-on-turn functionality to fetch relevant context
• Design and build memory namespace architecture for per-agent isolation and per-user scoping
• Develop Python helper libraries to support memory-enabled agents across the platform
• Enable memory activation through configuration rather than custom code in agent templates
• Work with AWS AgentCore Memory API to create, query, update, and delete memory records
• Build and maintain API integrations for stateful services
• Manage session persistence and context continuity across agent interactions

Requirements

• 7+ years of backend engineering experience with AI/ML systems
• Hands-on experience integrating agent memory in a production agentic AI project, such as AWS AgentCore Memory API (write-on-turn/retrieve-on-turn), memory namespace design, or agent template memory configuration
• Expertise in Python SDK development
• Skills in API integration for stateful services
• Proficiency in session management and context persistence
• English proficiency at B2 level or higher

Nice to have

• Familiarity with AWS AgentCore Memory SDK usage
• Knowledge of LangChain memory integration
• Background in agent framework development (LangGraph, Crew)
• Understanding of conversational AI context handling

We offer

• We gather like-minded people:
• • Top tech minds driving innovation in AI, cloud and digital platform modernization
  • Supportive team and agile, startup-like culture
  • Hybrid by design mode and opportunity to work remotely within Poland
  • Chance to work abroad for up to 60 days annually
  • Business-driven relocation opportunities
• We provide growth opportunities:
• • Career development programs
  • Thought leadership, mentoring, soft skills and well-being programs
  • Certification (Anthropic, Gemini, GCP, Azure, AWS)
  • English classes
• We cover it all:
• • Stable pay
  • Participation in the Employee Stock Purchase Plan with a 15% discount
  • Benefits package (health insurance, multisport, shopping vouchers)
  • Referral bonuses up to $2,000
  • Offices featuring entertainment and relaxation zones, table tennis and football, free snacks, coffee and more
  • Corporate, social and well-being events
• Please, note:
• • Benefits listed above are available to employees only
  • We are open for working with Contractors. Terms of B2B cooperation agreements are agreed individually
  • We will reach out to selected candidates exclusively

EPAM is global leader in AI transformation engineering and integrated consulting, serving Forbes Global 2000 companies and ambitious startups. With over thirty years of expertise in custom software, product and platform engineering, we empower our clients to become AI-Native enterprises, driving measurable value from innovation and digital investments.

Tech stack

• Python
• Python SDK
• AgentCore
• Memory Namespace Design
• AWS
• API integration
• Session Management