Senior Research Engineer (Code World Models)Verified as a real position

Who we’re looking for

• Have hands-on experience with model pre-training, continued training, or mid-training.
• Have strong engineering skills in Python and experience with modern ML frameworks.
• Understand large-scale ML training workflows, including data processing, distributed training, checkpointing, evaluation, experiment tracking, and debugging.
• Have experience working with large datasets and care about data quality, contamination, sampling, and reproducibility.
• Have a background in NLP, ML for software engineering, or a similar domain.
• Enjoy working on research problems with high uncertainty and turning ideas into working experiments.

What you’ll do

• 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 working on ML for code and AI developer tools.

About the position

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.

Benefits

• 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.