Senior ML Engineer (JetBrains Research)

At JetBrains, code is our passion. Ever since we started, back in 2000, we have strived to make the 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 JetBrains Research team explores ways to use machine learning techniques and agentic approaches to help developers and enhance software development processes. Our work aims to improve the state of ML for code by turning the latest academic advances into practical applications.

Our team is looking for an ML Engineer to participate in a variety of projects in areas such as code completion and generation, AI agents development, and test generation. Though we don’t expect the candidate to have experience in all the tasks we work on, we are looking for someone excited to take on the challenge of working in a diverse set of contexts.

In this role, you will: 

• Design and prototype machine learning solutions to improve the software development workflow.
• Apply existing open-source models or train custom ones as needed.
• Build training and evaluation pipelines that support fast iteration and ensure reproducibility.
• Stay up to date with the latest advancements in applying machine learning to source code.
• Collaborate with colleagues and provide mentorship when appropriate.

We’ll be happy to have you on our team if you: 

• Have experience designing machine learning systems from scratch.
• Have a background in natural language processing or code modeling.
• Have a proven track record of delivering machine learning projects.
• Know how to design reproducible ML experiments.
• Enjoy working on projects with high uncertainty.
• Are proficient in Python, Java, or Kotlin.
• Have hands-on experience with modern ML frameworks.

It would be nice if you: 

• Have experience applying machine learning to programming languages.
• Have hands-on experience building and improving AI agents.
• Have experience in A/B testing and general knowledge of statistics
• Have worked with large-scale data processing pipelines

#LI-KP1