Lead Machine Learning Engineer – Recommendation Systems

GRAI Inc.; Poland; On-site; PLN 320,000 - 460,000; Full time

Benefits offered by this job

Founding-stage equity

Paid time off (26 days)

Job summary

GRAI Inc. in Warsaw, currently in alpha, is seeking a Lead Machine Learning Engineer to architect and scale our recommendation and discovery engines. You will own the technical roadmap for the RecSys stack and turn raw behavior signals into personalized, real-time experiences.

You will lead a team of ML engineers, design end-to-end ML infrastructure, and drive data-driven evaluation while collaborating closely with Product and Engineering to achieve business goals.

Qualifications

• Extensive experience designing production-grade recommendation systems, search engines, or large-scale ranking models.
• Deep knowledge of machine learning, deep learning architectures, and retrieval methodologies.
• Experience with distributed data systems and modern ML frameworks (PyTorch, TensorFlow).
• Background in deploying and maintaining low-latency models in production, with feature stores and drift monitoring.
• Demonstrated leadership in technical projects and mentoring engineers.
• Ability to balance advanced AI experimentation with production stability and business value.

Responsibilities

• Define the long-term technical vision and roadmap for RecSys and personalization engines.
• Design scalable retrieval, ranking, and re-ranking pipelines for massive datasets.
• Build and own ML infrastructure: data processing, feature stores, and real-time monitoring.
• Mentor ML engineers and foster a culture of excellence and experimentation.
• Design offline evaluation and lead A/B testing to iterate on model performance.
• Collaborate with Product and Engineering to align ML with business metrics.

Skills

Machine Learning

Team Leadership

MLOps

System Design

Data Analysis

Cross-functional Collaboration

Tools

PyTorch

TensorFlow

Airflow

Spark

Job description

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.