Python Developer / AI Engineer (Document Processing)

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

We just unveiled JetBrains Central,  a platform for agent-driven software development that connects tools, agents, and infrastructure. It enables automated AI workflows to run, be monitored, and be managed across teams – with clear visibility into results, costs, and performance.

JetBrains Central provides three core capabilities:
Governance and control: Policy enforcement, identity and access management, observability, auditability, and cost attribution for agent-driven work. Some of these functionalities are already available via the JetBrains Central Console.
Agent execution infrastructure: Cloud agent runtimes and computation provisioning that allow agents to run reliably across development environments.
Agent optimization and context: Shared semantic context across repositories and projects, enabling agents to access relevant knowledge, and task routing to the most appropriate models or tools.

Inside the Professional Services department, we are building a team of Forward Deployed Engineers to help our customers adopt AI-native development with JetBrains Central and enable JetBrains consulting partners to transform their customers’ AI-native software development lifecycle.

About the role

As a Principal Forward Deployed Engineer, you will work directly with early adopter customers to deploy JetBrains Central platform into complex enterprise environments and help them move from experimental AI usage to governed, scalable AI-native development.

You will act as a trusted advisor to customer engineering teams and consulting partners, while also mentoring other Forward Deployed Engineers and helping with complex deployments. In this role, you will help establish best practices for enterprise AI adoption and influence how organizations implement AI-driven software development.

This role is often a strong fit for engineers who have previously served as Principal or Staff Engineers, Heads of Engineering, CTOs, or Solutions Architects and who enjoy working directly with customers while staying deeply involved with complex technical systems.

This role combines consulting, architecture, and hands-on engineering. Your work in the field will directly influence how our platform evolves and how enterprises adopt AI-driven development.

You will work directly with the world’s most innovative companies — our client base includes 420 of the Fortune 500 — ensuring they maximize the value of their JetBrains investments.

What you’ll do

• Deploy JetBrains Central platform into enterprise development environments, CI/CD systems, and internal developer platforms.
• Help organizations transition from ad-hoc AI experimentation to governed AI-native development workflows.
• Implement enterprise AI governance, including security policies, access management, and data privacy controls.
• Enable observability for AI-driven development to help organizations understand their AI usage and its impact on productivity, as well as their code-generation patterns.
• Design sustainable AI adoption strategies, including usage controls and cost visibility across multiple model providers.
• Integrate our cloud-based agent execution layer, allowing AI agents to run securely and at scale.
• Collaborate closely with JetBrains product teams to shape the platform based on real-world deployments and customer feedback.
• Create delivery frameworks and train JetBrains partners to deliver the desired scope of service.
• Mentor and support other Forward Deployed Engineers, sharing architectural expertise and helping to scale the team’s deployment practices.

You will thrive in this role if you are

• A consultant at heart who can communicate effectively with both engineers and technical leadership.
• Able to see the whole picture of the software development lifecycle across developer tools, infrastructure, APIs, and cloud systems.
• Experienced at integrating complex software systems into enterprise environments.
• Curious about the future of AI agents and AI-native software development.
• Comfortable working with early adopters and overcoming ambiguous technical challenges.
• Motivated to mentor other engineers and contribute to building a new technical discipline.

What you will bring

• Experience in software-related projects, ideally in a customer-facing or consulting environment.
• Familiarity with Java or Python.
• Experience building or integrating developer platforms, infrastructure systems, or software solutions.
• Familiarity with modern development workflows (CI/CD, developer environments, internal developer platforms, etc.).
• Experience working directly with customers or technical stakeholders.
• Strong problem-solving skills and the ability to operate in fast-evolving technical environments.
• Ability and desire to travel to meet customers.

These would be a plus, but are not blockers

• Experience with industry-specific solutions (fintech, telco, retail, etc.)
• Experience with AI/LLM platforms, agent frameworks, or AI-assisted development tools.
  Experience in solutions engineering, consulting, or forward-deployed engineering roles.
• Familiarity with the JetBrains ecosystem or developer productivity tooling.

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

Check this out to find out more about our product: https://www.jetbrains.com/context/

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

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.
• Relocation support. We help make your move as smooth and stress-free as possible.
• 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

#LI-DNI