Staff/Senior AI Engineer, AI for Code
jetbrains
<p>At JetBrains, we build developer tools used by millions of engineers. The AI for Code team works on the next generation of coding agents and agentic workflows: systems that can understand codebases, plan and execute multi-step tasks, collaborate with developers, and ship reliable results inside real development environments.</p> <p>We are looking for a Staff/Senior AI Engineer to join the team and support these efforts. This role is for someone who can take our internal coding models, such as Mellum2, as well as open-weight models, and turn them into production-ready coding agents for our users. You’ll work on model training and fine-tuning, context engineering, tool use, evaluation, feedback loops, and product integration. This is not research in isolation – you’ll build systems that are used by tens of thousands of developers.</p> <h2>What you’ll do:</h2> <ul> <li>Build production-ready coding agents and agentic workflows for real developer tasks inside JetBrains products.</li> <li>Turn promising model capabilities into dependable product behavior through prompt design, context construction, fine-tuning, instruction-tuning, or other post-training techniques where appropriate.</li> <li>Design and improve the agent loop itself, including tool use, execution strategy, safeguards, and task completion quality.</li> <li>Create evaluation suites and quality infrastructure for agent behavior, including online and offline evaluations, regression checks, failure analysis, and release criteria.</li> <li>Build feedback loops from real usage, using logs, user signals, and edge cases to improve data, evaluations, and agent behavior.</li> <li>Work with both hosted frontier APIs and self-hosted or open-weight models, making pragmatic decisions about where each model belongs based on capability, latency, reliability, privacy, and cost.</li> <li>Collaborate closely with product managers, software engineers, ML engineers, and researchers to ship features end to end.</li> <li>Help define the technical direction for future work, especially in ambiguous areas where we need strong judgment rather than a prewritten playbook.</li> </ul> <h2>What we’re looking for:</h2> <ul> <li>Strong software engineering fundamentals and a track record of shipping complex systems to production.</li> <li>Hands-on experience building LLM-powered products, coding agents, or other AI systems.</li> <li>Experience improving model behavior through systematic iteration, whether via prompting, context engineering, fine-tuning, preference optimization, or broader post-training methods.</li> <li>Practical experience with evaluation and benchmarking for LLM systems, including defining task-grounded success metrics and catching regressions.</li> <li>Experience working from noisy real-world signals rather than only from clean benchmark datasets.</li> <li>Good judgment about trade-offs between model quality, latency, reliability, privacy, and cost.</li> <li>Confidence working with ambiguity and taking ownership of a direction over multiple iterations.</li> <li>Strong communication skills and the ability to align engineering and product decisions.</li> </ul> <h2>What success looks like in the first year:</h2> <ul> <li>You ship one or more agent capabilities that users can rely on for meaningful work, not just demos.</li> <li>You establish better evaluation coverage and clearer release criteria for agent behavior.</li> <li>You help the team build a repeatable loop from idea to shipped capability: prototype, evaluate, learn from usage, improve, and scale.</li> </ul> <h2>Why join us?</h2> <p>You’ll help define what practical, trustworthy AI for software development looks like in real products.&nbsp;You’ll work on challenging problems at the boundary of model capability and product reality, with the freedom to stay hands-on and the scope to influence how the next generation of JetBrains AI systems is built.&nbsp;</p> <p><strong>And:</strong></p> <ul> <li>Strong base salary. We offer competitive pay that reflects your skills and experience.</li> <li>Flexible work location. Enjoy the freedom to work from home or from the office.</li> <li>Remote work. Spend up to 30 days per year working remotely from abroad.</li> <li>Extra time off. More days to relax, recharge, and do the things you love.</li> <li>Medical insurance allowance. Enjoy peace of mind for you and your family</li> <li>Learning and development opportunities. Access to conferences, courses, and language classes.</li> <li>Relocation support. We help make your move as smooth and stress-free as possible.&nbsp;</li> <li>Language classes. Pick up the local language or sharpen your English skills.</li> <li>Fuel your day. Enjoy a hot meal or receive a lunch allowance on workdays.</li> <li>Mental health support. To help you feel your best, we provide easy access to professional mental health services.</li> <li>Sports benefit. Enjoy an on-site gym or sports club stipend.</li> <li>Internal events. Join company-wide celebrations and team gatherings.</li> </ul> <p>*Some benefits may vary depending on location.</p> <p><span style="color: rgb(255, 255, 255);">#LI-DNI</span></p><div class="content-conclusion"><p><strong>We are an equal opportunity employer</strong><br><br>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.</p> <p><em data-stringify-type="italic">We process the data provided in your job application in accordance with the <a href="https://www.jetbrains.com/legal/docs/privacy/privacy-recruitment/">Recruitment Privacy Policy.</a></em></p></div><p>Find <a href="https://www.arbeitnow.co.uk">Jobs in United Kingdom</a> on Arbeitnow</a>