Senior Research Engineer (Code World Models)

&lt;div&gt; &lt;div class=&quot;description__e030 renderedMarkdown__c113 prosemirror__b91a markdown__e688 font__eef6&quot;&gt; &lt;p&gt;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.&lt;/p&gt; &lt;p&gt;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.&lt;/p&gt; &lt;p&gt;This role is focused on model pre-training and mid-training for code-centric foundation models. You will work on data, training pipelines, evaluation, and experiments that improve how models understand programs, repositories, execution, tests, and software engineering workflows.&lt;/p&gt; &lt;h2&gt;In this role, you will:&lt;/h2&gt; &lt;ul&gt; &lt;li&gt;Design and run pre-training, continued pre-training, and mid-training experiments for code models.&lt;/li&gt; &lt;li&gt;Build and improve data pipelines for large-scale model training, including filtering, deduplication, mixture design, and dataset quality checks.&lt;/li&gt; &lt;li&gt;Work with code corpora, repositories, tests, execution traces, and synthetic data.&lt;/li&gt; &lt;li&gt;Develop evaluations for complex repository-level code reasoning tasks.&lt;/li&gt; &lt;li&gt;Collaborate with researchers and engineers working on ML for code and AI developer tools.&lt;/li&gt; &lt;/ul&gt; &lt;h2&gt;We’ll be happy to have you on our team if you:&lt;/h2&gt; &lt;ul&gt; &lt;li&gt;Have hands-on experience with model pre-training, continued training, or mid-training.&lt;/li&gt; &lt;li&gt;Have strong engineering skills in Python and experience with modern ML frameworks.&lt;/li&gt; &lt;li&gt;Understand large-scale ML training workflows, including data processing, distributed training, checkpointing, evaluation, experiment tracking, and debugging.&lt;/li&gt; &lt;li&gt;Have experience working with large datasets and care about data quality, contamination, sampling, and reproducibility.&lt;/li&gt; &lt;li&gt;Have a background in NLP, ML for software engineering, or a similar domain.&lt;/li&gt; &lt;li&gt;Enjoy working on research problems with high uncertainty and turning ideas into working experiments.&lt;/li&gt; &lt;/ul&gt; &lt;h2&gt;It would be a plus if you:&lt;/h2&gt; &lt;ul&gt; &lt;li&gt;Have experience training or adapting models for code generation, code understanding, software agents, program repair, test generation, or repository-level reasoning.&lt;/li&gt; &lt;li&gt;Have worked with execution-based data, such as unit tests, traces, logs, compiler feedback, runtime states, or sandboxed code execution.&lt;/li&gt; &lt;li&gt;Have experience with large-scale distributed training of models with 70B+ parameters.&lt;/li&gt; &lt;li&gt;Understand evaluation challenges for code models, including benchmark contamination, flaky tests, execution-based scoring, and long-horizon task evaluation.&lt;/li&gt; &lt;li&gt;Have contributed to ML infrastructure, open-source projects, or research systems.&lt;/li&gt; &lt;/ul&gt; &lt;/div&gt; &lt;/div&gt; &lt;section class=&quot;description__e030 renderedMarkdown__c113 prosemirror__b91a markdown__e688 font__eef6 additionalTextFields__b103&quot; data-test=&quot;text-fields&quot;&gt; &lt;div&gt;&lt;span style=&quot;color: rgb(255, 255, 255);&quot;&gt;#LI-KP1&lt;/span&gt;&lt;/div&gt; &lt;/section&gt;&lt;div class=&quot;content-conclusion&quot;&gt;&lt;p&gt;&lt;strong&gt;We are an equal opportunity employer&lt;/strong&gt;&lt;br&gt;&lt;br&gt;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.&lt;/p&gt; &lt;p&gt;&lt;em data-stringify-type=&quot;italic&quot;&gt;We process the data provided in your job application in accordance with the &lt;a href=&quot;https://www.jetbrains.com/legal/docs/privacy/privacy-recruitment/&quot;&gt;Recruitment Privacy Policy.&lt;/a&gt;&lt;/em&gt;&lt;/p&gt;&lt;/div&gt;<p>Find more <a href="https://www.arbeitnow.com/english-speaking-jobs">English Speaking Jobs in Germany</a> on Arbeitnow</a>