Lead Machine Learning Engineer – Recommendation Systems

GRAI; Warszawa; On-site; PLN 80,000 - 120,000; Full time

Benefits offered by this job

26 business days of paid time off

Meaningful stock options

Collaborative product-driven environment

Job summary

GRAI in Warsaw is looking for a Lead Machine Learning Engineer to lead the technical roadmap for their recommendation system. You will design scalable retrieval and ranking pipelines and mentor a team of ML engineers.

Candidates should have extensive experience in building production-grade recommendation systems and a strong background in machine learning. This is a hybrid role offering impactful work in a dynamic environment.

Qualifications

• Extensive experience designing and scaling recommendation systems.
• Deep knowledge of machine learning, deep learning architectures.
• Strong background in deploying low-latency models in production.

Responsibilities

• Define the technical vision and roadmap for the recommendation system.
• Design scalable retrieval and ranking pipelines.
• Lead development of robust ML infrastructure.

Skills

Machine learning

Deep learning architectures

MLOps

Recommendation systems

Distributed data systems

Tools

PyTorch

TensorFlow

Job description

Location

Warsaw, Poland

Employment Type

Full time

Location Type

Hybrid

Department

R&D

GRAI is a social music app out of Warsaw, currently in alpha. We’re building new ways to listen and respond through music - focusing on the social, human interactions that only happen when people are connected on both ends.

To make those connections truly resonant, we’re looking for a Lead Machine Learning Engineer to architect, build, and scale our recommendation and discovery engines. You’ll own the technical roadmap for our RecSys stack, transforming raw user behavior and social signals into deeply personalized, real-time experiences.

What You’ll Do

Technical Leadership: Define the long-term technical vision and architectural roadmap for our recommendation and personalization engines.

System Architecture: Design and oversee the implementation of scalable retrieval, ranking, and re-ranking pipelines capable of handling massive user behavior and content data.

End-to-End Ownership: Lead the development of robust ML infrastructure, including automated data processing, feature stores, MLOps), and real-time model monitoring.

Mentorship & Culture: Mentor and coach a talented team of ML engineers, fostering a culture of technical excellence, continuous learning, and rigorous experimentation.

Data-Driven Strategy: Design comprehensive offline evaluation frameworks and lead complex A/B testing strategies to validate and iterate on model performance.

Cross-Functional Collaboration: Partner closely with Product and Engineering to align ML initiatives with high-level business metrics and product goals.

What We’re Looking For

Proven Track Record: Extensive experience designing, building, and scaling production-grade recommendation systems, search engines, or large-scale ranking models.

Technical Mastery: Deep, foundational knowledge of machine learning, deep learning architectures, and modern information retrieval methodologies.

Scale & Infrastructure: Significant experience with distributed data systems and modern ML frameworks (PyTorch, TensorFlow). Proven ability to handle massive, high-throughput user interaction datasets.

Production & MLOps: Strong background in deploying and maintaining low-latency models in production, with a solid grasp of feature stores, model registries, and drift monitoring.

Leadership Skills: Demonstrated experience leading technical projects, mentoring engineers, and successfully managing stakeholders without losing your hands-on technical edge.

Pragmatic Execution: Ability to balance cutting-edge AI experimentation with the practical realities of production stability, latency constraints, and business value.

Nice to Have

Real-Time Expertise: Experience with streaming data architectures and real-time/session-based recommendation systems.

Domain Knowledge: Background in audio processing, music streaming, or high-growth consumer-facing personalization products.

Advanced ML: Familiarity with graph neural networks (GNNs), reinforcement learning, or leveraging Large Language Models (LLMs) for recommendation context.

Why Join Us

High Autonomy & Impact: Direct ownership over the technical direction of high-impact ML systems used by millions of users.

Founding-Stage Equity: You’re joining a tight team at the alpha stage. We offer meaningful stock options so you have real skin in the game and share directly in the upside of what we build.

Influence the Future: Shape not just the recommendation stack, but the broader engineering culture and hiring roadmap of a fast-growing startup.

Collaborative Innovation: Work in a supportive, product-driven environment where your ideas directly dictate the future of how people experience music.

Rest & Recharge: High-output work requires real downtime. We offer 26 business days of paid time off per year, plus 5 days off and Polish public holidays.