Senior ML Engineer (JetBrains Research)

Posted 21 Days Ago
Be an Early Applicant
11 Locations
In-Office or Remote
Senior level
Software
The Role
Design and prototype ML solutions to improve developer workflows (code completion, generation, agents, test generation). Apply or train open-source models, build reproducible training and evaluation pipelines, stay current with ML-for-code research, collaborate with colleagues, and provide mentorship.
Summary Generated by Built In

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

We are an equal opportunity employer
We know great ideas can come from anyone, anywhere. That’s why we do our best to create an open and inclusive workplace – one that welcomes everyone regardless of their background, identity, religion, age, accessibility needs, or orientation.

We process the data provided in your job application in accordance with the Recruitment Privacy Policy.

Skills Required

  • Experience designing machine learning systems from scratch
  • Background in natural language processing or code modeling
  • Proven track record of delivering machine learning projects
  • Ability to design reproducible ML experiments
  • Comfort working on projects with high uncertainty
  • Proficiency in Python, Java, or Kotlin
  • Hands-on experience with modern ML frameworks
  • Experience applying machine learning to programming languages
  • Hands-on experience building and improving AI agents
  • Experience in A/B testing and general knowledge of statistics
  • Experience with large-scale data processing pipelines
Am I A Good Fit?
beta
Get Personalized Job Insights.
Our AI-powered fit analysis compares your resume with a job listing so you know if your skills & experience align.

The Company
2,209 Employees

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account