Python AI Tech Lead

About the role

MADIFF Polska is looking for an experienced Python / AI Tech Lead to join its AI and Data Engineering team, which delivers large-scale Generative AI solutions for leading enterprises across the US and Europe. The role combines technical leadership, solution architecture and hands-on Python development, driving production-grade AI systems built with LLMs, embeddings and orchestration frameworks such as LangChain and LangGraph. It is fully remote.

Key responsibilities

• Design and develop scalable AI-powered platforms using Python, applying advanced ML and DL principles
• Lead end-to-end delivery from data ingestion to model deployment and API integration
• Collaborate with cloud, data and product teams to define architecture and implementation strategy
• Mentor engineers, review code and ensure adherence to software delivery best practices
• Drive PoCs and the industrialisation of Generative AI solutions using enterprise-grade frameworks
• Optimise models and infrastructure for scalability, reliability and performance in production

Required profile

• Strong Python expertise focused on AI and data-centric systems
• Proven leadership experience managing or mentoring engineering teams
• Fluent English (C1 level)
• Nice to have: data engineering (ETL, Spark, Databricks, Delta Lake), NLP/NLU, observability (Prometheus, Grafana, MLflow), event-driven or serverless architectures

Required skills

• ML and AI frameworks: scikit-learn, PyTorch, TensorFlow, Hugging Face
• LLMs (OpenAI, Anthropic, Gemini) and vector-based retrieval systems
• Prompt engineering and orchestration tools (LangChain, LangGraph, Semantic Kernel, Crew AI)
• Vector databases (Pinecone, Weaviate, FAISS, Milvus) and GenAI system design
• MLOps and DevOps (Docker, Kubernetes, GitHub Actions, Azure DevOps)
• Cloud deployments (AWS, Azure, GCP) and architecture, APIs, microservices and security (IAM, RBAC)

What we offer

• Competitive remuneration package
• Long-term B2B contract with stable international projects
• Multinational, multilingual working environment
• Private healthcare coverage