ML Engineer (Data Products)

At JetBrains, code is our passion. Ever since we started, back in 2000, we have strived to make the strongest, most effective developer tools on earth. By automating routine checks and corrections, our tools speed up production, freeing developers to grow, discover, and create.

The AI for Data team at JetBrains aims to create cutting-edge AI features for data products like PyCharm, DataGrip, Datalore, and DataSpell. Functioning like a research lab, we focus on delivering innovative technologies with shorter project cycles compared to product teams. While we don’t prioritize paper writing or long-term, post-integration feature support, our work significantly enhances the capabilities of data professionals worldwide.

We are currently seeking a driven AI/ML Engineer to contribute to developing AI-powered features for our data products. Although we value independence in leading projects, we are committed to mentoring promising candidates toward greater autonomy. Experience with LLMs and related technologies is optional but highly desirable.

In this role, you will:

• Lead or collaborate on AI/ML-based projects from concept development to implementation.
• Design and implement robust evaluation frameworks to ensure system reliability and performance.
• Collaborate with product teams and research groups to align AI features with business objectives.
• Keep up to date with advancements in AI/ML technologies and explore innovative solutions to enhance our data products.

We will be happy to have you on our team if you have:

• At least 3 years of experience developing ML-based applications.
• Strong foundational knowledge in machine learning.
• Proven experience building and maintaining reliable evaluation pipelines for AI/ML systems.
• Excellent scientific and critical thinking skills.
• Proficiency in Python and the ability to quickly prototype and iterate on ideas.
• Strong problem-solving abilities and an analytical mindset.
• Effective verbal and written communication skills in English.

We’d be especially thrilled if you:

• Are familiar with LLMs, RAG techniques, and frameworks like LangChain or LlamaIndex.
• Have a good understanding of LLM architecture and related features.