Researcher, ML for Code

At JetBrains, code is our passion. Since 2000, we’ve focused on reducing routine work so developers can spend more time building and shipping. The JetBrains Research team studies machine learning techniques and agent-based approaches for software development to create methods that enable the next generation of ML for code features. We’re looking for a researcher to develop and improve training methodologies for models capable of solving tasks such as code generation, code editing, and empowering AI agents. You don’t need experience in every area, but you should be comfortable working across topics and delivering publishable results.

In this role, you will:

• Train and scale neural networks for processing source code.
• Develop new methods and improve existing ones for code generation, code editing, and agent-based workflows.
• Mentor colleagues on ML topics.

We’re looking for someone who:

• Has experience in natural language processing or code modeling.
• Has worked on distributed training and scaling neural networks.
• Has experience with modern training techniques and architectures (for example, MoE).
• Has publications at top ML conferences.
• Has strong programming skills.

Nice to have:

• Hands-on experience with post-training techniques.
• Experience with multimodal or low-data training.
• Familiarity with large-scale model training.

We are an equal opportunity employer We know great ideas can come from anyone, anywhere. That’s why we do our best to create an open and inclusive workplace – one that welcomes everyone regardless of their background, identity, religion, age, accessibility needs, or orientation. We process the data provided in your job application in accordance with the Recruitment Privacy Policy.