Lead AI Engineer - Startup - Munich

We are hiring an AI Engineer for an innovative client transforming engineering workflows with cutting-edge, open-source-powered tools. Their solutions are trusted by global leaders in the automotive, aerospace, and medical sectors. This role focuses on building scalable multi-agent systems driven by LLMs to automate traceability, compliance, and contextual intelligence in safety-critical environments.

Key Responsibilities:

• Design and implement multi-agent architectures using LLMs
• Develop systems for agent collaboration, communication, and task execution
• Integrate models with internal and external toolchains
• Explore privacy-first AI approaches such as federated learning
• Fine-tune and deploy models using PEFT, LoRA, and related techniques
• Connect to systems via APIs, CI pipelines, and MLOps infrastructure

Tech Stack Includes:

Python, PyTorch/TensorFlow, LangGraph, CrewAI, Docker, Kubernetes, VectorDBs, Neo4j, Azure/AWS/GCP

Ideal Profile:

• MSc or PhD in Computer Science, Machine Learning, or related field
• Experience building multi-agent AI systems and working with LLMs
• Strong Python skills and deep knowledge of ML frameworks
• Understanding of federated learning, DevOps, and real-world AI system integration
• Fluent in English; German or other European languages a plus

Why Apply:

• Build foundational AI systems in regulated, high-impact domains
• Work with a forward-thinking, mission-driven engineering team
• Join a remote-first, flexible work culture
• Competitive compensation and the opportunity to contribute to widely adopted open-source tools

Diversity & Inclusion:

At Stax we are committed to building inclusive teams for our partners and strongly encourage applications from people of all backgrounds, experiences, and identities. Your unique perspective matters.

Please note: Due to the high volume of applications, we may not be able to respond to every applicant individually. We appreciate your interest and the time you invest in reaching out.