Machine Learning Engineer

About Momentum

Momentum is a Wrocław-based team building software for healthcare companies worldwide. For years we've helped HealthTech startups and scaleups turn ideas into compliant, production-ready products. Now we're also building our own, in wearables and medical data integration, and part of that work is open source.

We work AI-first. AI is part of our everyday work, in development and in delivery. We keep testing new tools, agents and workflows, and when we find a better way to build, we change how we work.

Curious how we work? See what we're building: Open Wearables on GitHub.

Role overview

Join Momentum and transform patient care with your ML expertise! We're seeking a passionate Machine Learning Engineer with Python, skilled in TensorFlow+Keras/PyTorch, LangChain/LlamaIndex, and data analysis tools. If you love building innovative solutions and are excited to share your knowledge, we'd love to welcome you to our team!

Requirements

• Strong experience with development and deployment of machine learning models
• Strong knowledge of one of the Python machine learning frameworks (Tensorflow/Keras)
• Strong knowledge of data analysis and presentation tools (Matplotlib, Pandas, …)
• Proficiency in Python language
• Good knowledge of one of the LLM frameworks (LangChain/LlamaIndex/PydanticAI)
• Good knowledge of object oriented programming
• Good knowledge of any of the popular Python web frameworks/libraries (FastAPI, Django)
• Fair knowledge of relational databases (experience with non-relational databases is a plus)
• Knowledge of Unix systems and shell
• Experience with CI/CD, deployment tools and Git
• Enthusiasm for sharing knowledge
• Fluent written and spoken English
• Fluent written and spoken Polish

Nice to have

• Experience with AWS or similar platform
• Experience with Docker

Responsibilities

• Work with LLM models, NLP tools and AI agents
• Develop, integrate and deploy machine learning models
• Help with backend development
• Constantly share knowledge with the rest of your team

What We Value

• Clarity in focus and dedication to building digital products that elevate healthcare and improve quality of life.
• Flair to face challenges with boldness, creativity, and a proactive mindset.
• Grit to take ownership, stay ambitious, and deliver refined, high-quality solutions.
• Impact through meaningful innovation that makes pioneering medical discoveries accessible and human-centered.
• Ability to collaborate efficiently in teams of technical and non-technical individuals, and comfortable working in a dynamic environment with evolving objectives and iteration with users.

What we offer

• AI tools like Claude and Gemini, with real influence on how we use AI and space to test new tools and workflows
• Flexible working hours
• Modern office in the heart of Wrocław
• Office visits reimbursed for remote team members, both required and optional
• Private healthcare and Multisport card
• Group insurance for you and your loved ones (optional)
• Worksmile cafeteria with hundreds of benefits to choose from
• Microfunds for team events and skill development

Thinking of applying? Awesome! How we recruit

We keep our process short and to the point

• Intro call (20 min): a quick chat about your experience and what you're looking for.
• Recruitment task: a short task, with three working days to complete it.
• Technical interview (45-60 min): a deeper conversation with a senior team member, including a walkthrough of your solution.
• Culture fit (30 min): a talk about how well you fit into our organizational culture.
• Decision and feedback: we'll get back to you with our decision and detailed feedback.

If you have any questions, get in touch with us!