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

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. Today, AI-powered assistance and agents are becoming a core part of how developers work in our IDEs.

We’re building multi-step coding agents that can understand large codebases, plan changes, call tools, and iterate with the user. As a Research Engineer in the Agentic Models team, you’ll be responsible for the models, training loops, and evaluation pipelines that power these agents.

You’ll work at the intersection of SFT and RL-style post-training, and product-driven evaluation, using our distributed GPU and MapReduce clusters to ship models into JetBrains products.

As part of our team, you will:

• Design, implement, and maintain SFT and RL post-training pipelines for multi-step coding agents.
• Train and adapt LLMs for agent workflows, including planning, tool use, and multi-step interactions inside JetBrains IDEs.
• Build and develop evaluation and simulation environments where coding agents can act, be measured, and compared on realistic developer tasks.
• Design evaluation frameworks and metrics for agent behavior, analyze traces and logs, and close the loop from evaluation back into training, data, and reward design.
• Analyze training and evaluation results to propose and implement improvements to model architectures, training recipes, and datasets.
• Work with large-scale infrastructure, including distributed training on GPU clusters and large MapReduce-style data processing for pre-training and fine-tuning datasets.
• Collaborate closely with research, product, and infrastructure teams to turn high-level product visions into concrete models, experiments, and shipped features. 

We’ll be happy to bring you on board if you have:

• Extensive hands-on experience training LLMs (pre-training, fine-tuning, or post-training) in a research or production setting.
• Deep expertise in modern deep learning frameworks such as PyTorch, and specialized LLM training stacks (e.g. Megatron, NeMo, verl, or similar).
• Strong theoretical and practical understanding of LLM fundamentals: architectures, tokenization, data pipelines, batching, mixed precision, distributed training, and debugging unstable runs.
• 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 product-aware mindset – you care about how developers actually use agents and can translate product needs and failure modes into modeling and evaluation work.
• At least 3 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 Kubeflow, Dagster, Airflow, ZenML, and/or job schedulers like Kubernetes or SLURM.
• Large-scale data and training pipelines, e.g. MapReduce-style clusters, multi-node GPU training, or workloads on the order of 1M+ CPU/GPU hours.
• Designing and maintaining evaluation pipelines for LLMs or agents, including metrics, dashboards, experiment tracking, and automated regression checks.
• AI agent development, such as tool-using agents, planners, or multi-step coding workflows, and familiarity with agentic frameworks or patterns.
• Experiment tracking and observability using tools like Weights & Biases, MLflow, Langfuse, or similar.
• Inference optimization and serving optimized models in production.

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

JetBrains is evolving beyond standalone developer tools toward a unified, AI-native platform for software development.

AI is no longer just an assistant inside the editor – it is becoming an active participant in how software is planned, built, reviewed, and operated across teams and organizations. This shift introduces new challenges that cannot be solved at the level of individual tools alone: governance, security, cost control, observability, and coordinated work between humans and autonomous agents.

Our goal is to build a platform that enables companies to adopt AI in software development in a structured, scalable, and economically efficient manner without locking them into closed ecosystems. This platform will serve as the execution and governance layer for AI-driven development, deeply integrated with developer tools but designed to work across teams, products, and environments.

We are looking for an experienced ML leader who has created products with an ML backbone, weaving together research, technical excellence, and strong product focus.

We are seeking a professional who excels in three key areas: technology, product vision, and business operations. This role involves extensive cooperation with products across the company – both AI-native and just integrating AI.

In this role, you will: 

• Identify, prioritize, and champion high-impact AI projects that align with JetBrains' product strategy and deliver measurable business value.
• Lead a team of about 50 people across multiple AI initiatives, including hiring and developing team leads, ML engineers, and MLOps specialists.
• Set the technical direction for JetBrains’ ML roadmap,  translating cutting-edge AI advancements into practical product innovations.
• Ensure projects deliver on commitments and create measurable value.
• Set technical standards, architectural direction, and best practices for ML development.

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

• The ability to balance long-term R&D with short-term product needs.
• A track record of shipping production ML systems that users actually use.
• Experience building business cases for technical projects.
• A deep understanding of modern ML infrastructure, LLMs, and AI tooling.
• Outstanding stakeholder management skills and are comfortable working with senior leaders.
• The ability to thrive in a fast-paced environment, effectively managing multiple projects simultaneously.
• Excellent management and communication skills.
• Attention to detail in everything you do.

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

• Distributed training of multi-billion parameter models.
• CI systems, workflow automation, and experiment tracking.
• The Kotlin programming language.

#LI-KP1 #LI-HYBRID