Senior ML Engineer (JetBrains Research)

United States Digital Space LLC; Berlin, München; Vor Ort; EUR 70.000 - 110.000; Vollzeit

Zusammenfassung

United States Digital Space LLC is seeking an ML Engineer to contribute to code completion, AI agents development, and test generation. You will design ML solutions to streamline software development, apply open-source models or train custom ones, and build reproducible training pipelines.

You should have a background in NLP or code modeling, a track record delivering ML projects, and hands-on experience with Python, Java, or Kotlin, plus modern ML frameworks.

Qualifikationen

• Design ML systems from scratch.
• Background in NLP or code modeling.
• Proven ML project delivery track record.
• Able to design reproducible ML experiments.
• Enjoy working with high uncertainty.
• Proficient in Python, Java, or Kotlin.
• Hands-on with modern ML frameworks.

Aufgaben

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

Kenntnisse

ML system design

Python

Java

Kotlin

NLP

Code modeling

ML frameworks

Experiment reproducibility

A/B testing

Data processing pipelines

Tools

TensorFlow

PyTorch

scikit-learn

Jobbeschreibung

At the company, 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 the company 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.

Find more English Speaking Jobs in Germany on Arbeitnow