At JetBrains, code is our passion. Ever since we started back in 2000, we have been striving to make the world’s most robust and effective developer tools. By automating routine checks and corrections, our tools speed up production, freeing developers to grow, discover, and create.

We are now building services and agentic tools that provide AI coding agents and end users with deeper context about codebases. Our code retrieval service already delivers meaningful improvements in agent speed and task performance, and we aim to push this further – extracting richer insights than snippets alone at the scale of several hundred thousand repositories. 

We are looking for an AI Engineer who can design and implement agentic tools from scratch, bring them to the end users, and make coding with agents smarter, faster, and more reliable.

In this role, you will:

• Translate product needs into agentic tools and services and ship them to production.
• Scale prototypes to large numbers of users.
• Improve and test our semantic search service.
• Build evaluation pipelines for code retrieval and coding task performance.

We’ll be happy to have you on our team if you have:

• At least five years of hands-on NLP experience.
• Experience in information retrieval systems.
• Strong experience in LLM prompt and tool engineering, as well as with AI agents.
• Solid programming skills in Python and/or Kotlin (and the willingness to learn the language you’re unfamiliar with).
• Experience building and operating production-grade software systems.

We’d be especially thrilled if you have:

• Experience developing in Kotlin or Java.
• Previously worked with code intelligence (e.g. AST, symbol graphs).

#LI-DNI

Software engineers and AI agents alike suffer from the same problem: finding that one person or place that will answer their tough, specific question. Many solutions promise to solve this with similarity search in vector databases. Unfortunately, finding the answer is often a puzzle with pieces to be collected across a myriad of contradictory sources and cannot be solved without surgical search and careful reasoning.

Spectrum collects data from an organization's code, docs, and issues, and organizes knowledge in a unified ontology that AI agents can efficiently search through and reason over. We aim to revolutionize the semantic layer space for software-building organizations and move beyond specs that fall out of sync with code, introducing a living spec – one that's extracted from the whole system and used to keep it aligned. Spectrum is meant to be the single source of truth for all product and architectural knowledge.

A resident of JetBrains' startup incubator, Spectrum enjoys startup speed and autonomy, and is backed by 25 years of developer tooling expertise. We are looking for a Senior ML Researcher to develop the core methods that make Spectrum possible – novel approaches to temporal ontology extraction, contradiction detection, and semantic alignment across heterogeneous software artifacts. You will help define and execute the research agenda, while also collaborating with JetBrains Research and external academic advisors.

Your responsibilities will include:

• Developing methods for LLM-based ontology construction from code, documentation, and issue trackers, including contradiction detection and reasoning over the resulting knowledge graph.
• Creating datasets, metrics, and benchmarks that drive measurable improvements across the system.
• Prototyping and validating research ideas, then collaborating with ML and software engineers to bring them to production.
• Collaborating with JetBrains Research and external academic advisors to shape our research agenda.
• Publishing research findings at top venues and representing Spectrum in the research community.

We expect you to have:

• A PhD (or equivalent research experience) in NLP, knowledge graphs, ontology learning, information extraction, or a closely related field.
• A strong publication record in at least one of the following areas: knowledge graph construction, ontology learning, information extraction, or NLP.
• Experience applying LLMs to structured knowledge extraction tasks.
• Experience designing and running rigorous experiments, including hypothesis formulation, ablation studies, and statistical evaluation.
• Strong Python and PyTorch skills, with the ability to implement research ideas from scratch.
• Excellent communication skills, with the ability to explain complex technical concepts to diverse audiences.
• Proficiency in English, both written and verbal.

Our ideal candidate would also have:

• Experience with ontology engineering, semantic web technologies (OWL, RDF, SPARQL), or ontology alignment.
• A background in code analysis, developer tools, or software engineering research.
• Experience with knowledge graph embedding methods or graph neural networks.
• Experience in early-stage startups – you enjoy the zero-to-one phase.
• Actively contributed to relevant open-source projects.

What we offer:

• A competitive salary and JetBrains benefits.
• A generous runway and corporate resources with startup autonomy.
• Full support for publishing ML research and attending industry conferences.
• Collaboration with JetBrains Research and external academic advisors in the knowledge graph and ontology learning community.
• A motivated team that moves fast, with short feedback loops, rapid prototyping, and lots of ownership.
• A rare applied research role where your methods ship in a real product used by software teams.

Why join JetBrains? 

• Strong base salary. We offer competitive pay that reflects your skills and experience.
• Flexible work location. Enjoy the freedom to work from home or from the office.
• Remote work. Spend up to 30 days per year working remotely from abroad.
• Extra time off. More days to relax, recharge, and do the things you love.
• Medical insurance allowance. Enjoy peace of mind for you and your family
• Learning and development opportunities. Access to conferences, courses, and language classes.
• Language classes. Pick up the local language or sharpen your English skills.
• Fuel your day. Enjoy a hot meal or receive a lunch allowance on workdays.
• Mental health support. To help you feel your best, we provide easy access to professional mental health services.
• Sports benefit. Enjoy an on-site gym or sports club stipend.
• Internal events. Join company-wide celebrations and team gatherings.

*Some benefits may vary depending on location.

#LI-DNI

At JetBrains, code is our passion. Ever since we started, back in 2000, we’ve been striving to make the strongest, most effective developer tools on earth. By automating routine checks and corrections, our tools speed up production, freeing developers to grow, discover, and create.

Today, AI-powered assistance and agents are becoming a core part of how developers work in our IDEs. The ML Workflows Engineering team is dedicated to removing infrastructure challenges, streamlining machine learning operations (MLOps), and enabling teams to focus on the innovative work that matters most – building impactful ML models and intelligent agents. As part of the team, you'll play a key role in designing tools, automation, and pipelines that make machine learning development seamless and intuitive.

By integrating cutting-edge MLOps practices and engineering excellence, we aim to maximize productivity and remove the complexity of ML infrastructure so that our teams can push the boundaries of what’s possible in AI.

As part of our team, you will:

• Build tools, automation, and workflows to simplify infrastructure-heavy tasks, empowering AI teams to focus on experimentation and solving core challenges.
• Develop robust monitoring, logging, and tracing systems to ensure the performance and reproducibility of ML workflows in production.
• Design, implement, and maintain end-to-end machine learning pipelines to enable the seamless development, training, and deployment of ML models and intelligent agents.
• Work with large-scale distributed systems, including GPU clusters, to support training, fine-tuning, and evaluation of ML models.
• Collaborate with product and development teams to transform high-level goals into concrete, scalable, and maintainable systems.
• Optimize workflows for reproducibility, scalability, and cost-efficiency while keeping ML teams productive and focused on innovation.

We’ll be happy to have you on our team if you have:

• Hands-on experience with modern MLOps tooling, including Kubernetes, Cloud providers (GCP and AWS), and ML orchestration frameworks.
• A solid understanding of the ML lifecycle from idea to the customer-facing application.
• The ability to own projects end to end, starting from a high-level problem or product pain point and overseeing it through the design, experimentation, implementation, and iteration phases.
• A customer-centric mindset – you care about how ML engineers are actually working and can translate their needs into actionable, scalable, and maintainable architectural decisions.
• Experience with modern CI/CD systems, like GitHub Actions or JetBrains TeamCity.
• At least three years of Python experience writing clean, maintainable code in modern ML codebases.

Our ideal candidate would have experience with:

• ML orchestrators and workflow tools such as ZenML, Dagster, and Airflow.
• Developing infrastructure components and services using cluster solutions like Kubernetes.
• The development of Python-based backend services.
• Creating and maintaining ML pipelines, including legacy ones.
• Experiment tracking and observability using tools like Weights & Biases, MLflow, Langfuse, or similar.

We’d be especially thrilled if you have experience with: 

• LLM inference frameworks such as vLLM, DeepSpeed, and TensorRT.
• Writing and maintaining Python libraries used by internal (or external) ML engineers.
• A strong theoretical background in NLP and transformer-based approaches.
• Writing code in Java and/or Kotlin.

#LI-HYBRID
#LI-MR1

At JetBrains, code is our passion. Ever since we started, back in 2000, we have strived to make the most effective developer tools on earth. By automating routine checks and corrections, our tools speed up production, freeing developers to grow, discover, and create.

The JetBrains Research team explores ways to use machine learning techniques and agentic approaches to help developers and enhance software development processes. Our work aims to improve the state of ML for code by turning the latest academic advances into practical applications.

Our team is looking for an ML Engineer to participate in a variety of projects in areas such as code completion and generation, AI agents development, and test generation. Though we don’t expect the candidate to have experience in all the tasks we work on, we are looking for someone excited to take on the challenge of working in a diverse set of contexts.

In this role, you will: 

• Design and prototype machine learning solutions to improve the software development workflow.
• Apply existing open-source models or train custom ones as needed.
• Build training and evaluation pipelines that support fast iteration and ensure reproducibility.
• Stay up to date with the latest advancements in applying machine learning to source code.
• Collaborate with colleagues and provide mentorship when appropriate.

We’ll be happy to have you on our team if you: 

• Have experience designing machine learning systems from scratch.
• Have a background in natural language processing or code modeling.
• Have a proven track record of delivering machine learning projects.
• Know how to design reproducible ML experiments.
• Enjoy working on projects with high uncertainty.
• Are proficient in Python, Java, or Kotlin.
• Have hands-on experience with modern ML frameworks.

It would be nice if you: 

• Have experience applying machine learning to programming languages.
• Have hands-on experience building and improving AI agents.
• Have experience in A/B testing and general knowledge of statistics
• Have worked with large-scale data processing pipelines

#LI-KP1

At JetBrains, code is our passion. Ever since we started back in 2000, we have been striving to make the world’s most robust and effective developer tools. By automating routine checks and corrections, our tools speed up production, freeing developers to grow, discover, and create.

We are working on an ambitious new platform that provides AI capabilities to all JetBrains products. Our platform is based on models developed in-house for writing and coding assistance, as well as integration with our strategic partners.

We are looking for a Research Engineer who can contribute to training foundation models for coding tasks. You’ll be working on developing Large Language Models from scratch and deploying them into production environments where they will be accessible by end users across the globe.

We value engineers who: 

• Can plan projects and make decisions independently, consulting with others if needed.
• Identify customer needs and prioritize their tasks accordingly.
• Start with the simplest solutions and gradually add complexity as needed.
• Take sole responsibility for an entire subsystem.
• Have a passion for learning and a desire to stay up to date with the latest developments in the LLM field.

In this role, you will: 

• Work with stakeholders to convert business requirements into technical specifications.
• Train LLMs from scratch on a large GPU cluster.
• Collect and process pre-training and fine-tuning datasets.
• Support and improve existing subsystems.

We’ll be happy to have you on our team if you have: 

• Experience in design, deployment, and support of production ML systems.
• A strong theoretical background in NLP and transformer-based approaches.
• Proficiency with modern deep learning frameworks such as PyTorch and common libraries for NLP.
• Experience in distributed training of multi-billion parameter models.
• Attention to detail in everything you do and great communication skills.

We’d be especially thrilled if you have experience with: 

• LLM inference frameworks such as vLLM, DeepSpeed, TensorRT.
• LLM alignment techniques such as RLHF/RLAIF.
• MLOps tools and practices, including CI/CD for ML.
• K8s and Kubeflow.
• Scientific publications in the NLP field.

How we develop JetBrains AI: 

• A cluster of hundreds of NVIDIA GPUs as training infrastructure.
• Git for source control management.
• Python, PyTorch, and HuggingFace as an ML stack.
• Kubeflow and Weights & Biases for experiment tracking.
• TeamCity as a CI Automation system.

#LI-KP1