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