ppJetBrains is a general 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. /p pThe 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. /p h3In This Role, You Will /h3 ul liDesign and run pre-training, continued pre-training, and mid-training experiments for code models. /li liBuild and improve data pipelines for large-scale model training, including filtering, deduplication, mixture design, and dataset quality checks. /li liWork with code corpora, repositories, tests, execution traces, and synthetic data. /li liDevelop evaluations for complex repository-level code reasoning tasks. /li liCollaborate with researchers and engineers working on ML for code and AI developer tools. /li /ul h3We’ll be happy to have you on our team if you: /h3 ul liHave hands-on experience with model pre-training, continued training, or mid-training. /li liHave strong engineering skills in Python and experience with modern ML frameworks. /li liUnderstand large-scale ML training workflows, including data processing, distributed training, checkpointing, evaluation, experiment tracking, and debugging.



/li liHave experience working with large datasets and care about data quality, contamination, sampling, and reproducibility. /li liHave a background in NLP, ML for software engineering, or a similar domain. /li liEnjoy working on research problems with high uncertainty and turning ideas into working experiments. /li /ul h3It Would Be a Plus If You /h3 ul liHave experience training or adapting models for code generation, code understanding, software agents, program repair, test generation, or repository-level reasoning. /li liHave worked with execution-based data, such as unit tests, traces, logs, compiler feedback, runtime states, or sandboxed code execution. /li liHave experience with large-scale distributed training of models with 70B+ parameters. /li liUnderstand evaluation challenges for code models, including benchmark contamination, flaky tests, execution-based scoring, and long-horizon task evaluation. /li liHave contributed to ML infrastructure, open-source projects, or research systems. /li /ul h3We are an equal opportunity employer /h3 pWe 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. /p pWe process the data provided in your job application in accordance with the Recruitment Privacy Policy. /p /p #J-18808-Ljbffr

📌 Senior Research Engineer (Code World Models) (Madrid)
🏢 JetBrains
📍 Madrid