JetBrains

Software Development

Staff/SeniorAIEngineer,AIforCode

United States; Canada
Market Sentiment
HIGH DEMAND

Neural analysis suggests this role is
optimal for Senior candidates.

The Brief

“Staff/Senior AI Engineer, AI for Code at JetBrains. Skills: AI for Code, coding agents, agentic workflows, LLM-powered products, model training and fine-tuning, context engineering, tool use, evaluation and benchmarking, feedback loops, product integration. 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, instru”

What You'll Achieve.

Ship one or more agent capabilities that users can rely on for meaningful work, not just demos; Establish better evaluation coverage and clearer release criteria for agent behavior; Help the team build a repeatable loop from idea to shipped capability: prototype, evaluate, learn from usage, improve, and scale

Industry & Context.

Software Development
Problems you'll solve

solving complex problems; analyzing failure cases; making pragmatic decisions about trade-offs

Eligibility Requirements

Remote work from abroad (up to 30 days per year)

What They're Looking For.

Must Have

software engineering fundamentals, track record of shipping complex systems to production, Hands-on experience building LLM-powered products, coding agents, or other AI systems, Experience improving model behavior through systematic iteration, whether via prompting, context engineering, fine-tuning, preference optimization, or broader post-training methods, Practical experience with evaluation and benchmarking for LLM systems, including defining task-grounded success metrics and catching regressions, Experience working from noisy real-world signals rather than only from clean benchmark datasets, Good judgment about trade-offs between model quality, latency, reliability, privacy, and cost, Confidence working with ambiguity and taking ownership of a direction over multiple iterations, communication skills and the ability to align engineering and product decisions

Nice to Have

Knowledge of Mellum2, Knowledge of open-weight models

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

or other post-training techniques

Design and improve the agent loop itself

and task completion quality

Create evaluation suites and quality infrastructure for agent behavior

Build feedback loops from real usage

and edge cases to improve data

Work with both hosted frontier APIs and self-hosted or open-weight models

making pragmatic decisions about where each model belongs based on capability

Help define the technical direction for future work

especially in ambiguous areas where we need judgment rather than a prewritten playbook

How You'll Work.

Team & Collaboration

Collaborate closely with product managers, software engineers, ML engineers, and researchers to ship features end to end

Communication Scope

communication skills and the ability to align engineering and product decisions

Process & Methodology

taking ownership of a direction over multiple iterations, build a repeatable loop from idea to shipped capability: prototype, evaluate, learn from usage, improve, and scale

Full Job Description

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, regression checks, failure analysis, and release criteria. Build feedback loops from real usage, using logs, user signals, and edge cases to improve data, evaluations, and agent behavior. 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. Collaborate closely with product managers, software engineers, ML engineers, and researchers to ship features end to end. Help define the technical direction for future work, especially in ambiguous areas where we need st

Free ATS check

Applying for this Staff/Senior AI Engineer, AI for Code role?

Most applicants get filtered before a human reads their resume. See if yours makes the cut.

How to Apply on Greenhouse

  • Create a Greenhouse profile before applying — it saves time across multiple applications.
  • Upload your resume as a PDF; the parser handles it better than Word.
  • Answer all knockout questions carefully — wrong answers auto-reject before a human sees you.
  • Enable email notifications to track application status in real time.

ANONYMOUS · UNFILTERED

What do employees actually say about JetBrains?

Real rants from real employees. Read before you apply.

Read Company Rants →