Senior ML Researcher

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 a researcher to participate in a variety of projects in areas such as code completion and generation, AI agents development, and test generation. Though the ideal candidate is not expected to have experience in all the tasks we work on, we are looking for someone excited to take on the challenge of doing high-quality research in diverse contexts.

In this role, you will:

• Train and scale neural networks for source code processing.
• Prototype ML-based tools that enhance the software development process.
• Develop new methods and improve existing ones in code completion, code generation, test generation, and agentic approaches.
• Mentor junior colleagues in machine learning and software engineering.

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

• Have extensive experience in natural language processing or code modeling.
• Have a track record of publications in top-tier ML conferences and journals or a track record of completed ML projects in an industry setting.
• Have worked on the highly parallelized training of ML models and the scaling of neural networks.
• Have strong programming skills.

It would be nice if you:

• Have hands-on experience building and improving AI agents.
• Have worked on retrieval-augmented generation or extending models to handle large context windows.