Lead AI Engineer

We are seeking a Lead AI Engineer to design and develop comprehensive AI applications, including chatbots and agent workflows.

You will work closely with clients to deliver tailored LLM-driven solutions that align with their business needs. Your role involves creating and managing data pipelines, refining AI system performance and staying informed on the latest LLM technologies and frameworks. Join us to advance your career by solving complex AI challenges and delivering impactful solutions.

Responsibilities

• Design, implement, and maintain end-to-end AI applications, including chatbots, Q&A platforms and agent workflows
• Collaborate directly with clients to understand their needs, identify opportunities and recommend LLM-driven solutions
• Develop and manage robust data pipelines, prompt strategies and datasets to ensure effective and accurate AI models (Will be +)
• Evaluate and refine AI system performance, ensuring outputs are accurate, secure, scalable and compliant with industry regulations (Will be +)
• Conduct research and rapid prototyping to validate technical feasibility and demonstrate business value of AI solutions
• Stay current with evolving LLM technologies, frameworks and methodologies to continuously improve solutions and client outcomes

Requirements

• Strong proficiency in Python, experience with web frameworks like FastAPI or similar
• Understanding of the AI application development lifecycle
• Experience with rapid UI prototyping using Streamlit, Gradio or similar frameworks
• Familiarity with major LLM platforms and APIs (OpenAI, Anthropic, Amazon Bedrock, Gemini) and related frameworks (LangGraph, LlamaIndex, Strands Agents, etc.)
• Knowledge of advanced AI integration patterns (e.g., RAG, Agents)
• Experience deploying AI solutions at scale, with considerations for performance, cost-efficiency and maintainability
• Proven ability to evaluate generative AI quality using metrics such as retrieval and classification scores, as well as LLM-based evaluation methods
• Proven experience in AI engineering and delivering ML-based solutions
• Strong problem-solving skills and attention to detail
• Excellent communication, collaboration and interpersonal skills
• English level of minimum B2 (Upper-Intermediate)

Nice to have

• Experience designing experiments, conducting A/B tests, and iterating on models based on user feedback
• Understanding of retrieval systems (keyword search, vector search, embeddings) and ranking algorithms
• Familiarity with emerging protocols such as MCP, A2A, ACP, etc.
• Experience deploying to cloud AI platforms (Azure OpenAI, Amazon Bedrock, GCP Vertex AI) or on-premise solutions (e.g., vLLM)
• Experience with enterprise AI platforms such as AWS AgentCore or Databricks AgentBricks or Google Agents Space or Azure AI Foundry
• Experience with observability and monitoring tools and frameworks

Benefits

• Diverse multicultural, multi-functional, and multilingual work environment
• Global scope, international projects in different business domains: life sciences and healthcare, retail, and distribution, software and hi-tech, etc.
• Outstanding career development opportunities with a transparent career path and a roadmap to accelerate your journey
• Great learning and development opportunities, including in-house professional training, career advisory and coaching, sponsored professional certifications
• Knowledge-sharing with colleagues from EPAM's global tech communities
• 25 days holiday per annum
• 5 days of sick leave without medical certification
• Private health insurance for employees and discount for private health insurance for family members
• Discount for Fit Pass program
• Comfortable modern offices in Belgrade and Novi Sad or remote work from any location in Serbia
• Regular corporate, social and wellbeing events (parties, sport events, advanced CSR programs etc.)

• We connect like-minded people
  • Experience exchange with colleagues from 55+ locations
  • Corporate and social events
  • Enjoyable working environment
• We invest in your growth
  • Personal career development, soft skills and well-being programs
  • Unlimited access to Internal Learning Platform
  • Free English classes with certified teachers
• We cover it all
  • Participation in the Employee Stock Purchase Plan
  • Comprehensive medical package (can be extended to family members)
  • Four trust days per year for personal needs
  • Benefits package (hotels, restaurants, stores and services)