Staff/Senior AI Engineer, AI for Code

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. 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. What you’ll do: Build production-ready coding agents and agentic workflows for real developer tasks inside JetBrains products. Turn promising model capabilities into dependable product behavior through prompt design, context construction, fine-tuning, instruction-tuning, or other post-training techniques where appropriate. Design and improve the agent loop itself, including tool use, execution strategy, safeguards, and task completion quality. Create evaluation suites and quality infrastructure for agent behavior, including online and offline evaluations, regr