Senior Research Engineer (Code World Models)

JetBrains is a global software company that creates intelligent tools for software developers and teams. Since 2000, we have built products that help developers work more productively, write higher-quality code, and stay focused on solving real problems.

The JetBrains Research team is looking for a Senior Research Engineer to work on Code World Models: models that learn how software systems behave, change, execute, and interact with developer tools.

In This Role, You Will

• Design and run pre-training, continued pre-training, and mid-training experiments for code models.
• Build and improve data pipelines for large-scale model training, including filtering, deduplication, mixture design, and dataset quality checks.
• Work with code corpora, repositories, tests, execution traces, and synthetic data.
• Develop evaluations for complex repository-level code reasoning tasks.
• Collaborate with researchers and engineers work...