Research Engineer

1 Day ago • All levels

Job Summary

Job Description

JetBrains is seeking a Research Engineer to contribute to training foundation models for coding tasks. The role involves developing Large Language Models from scratch and deploying them into production environments accessible to users worldwide. The engineer will convert business requirements into technical specifications, train LLMs on a large GPU cluster, collect and process datasets, and support existing subsystems. The ideal candidate should possess experience in production ML systems, a strong NLP background, proficiency in deep learning frameworks like PyTorch, and experience in distributed training of multi-billion parameter models. The role demands attention to detail and strong communication skills.
Must have:
  • Experience in production ML systems.
  • Strong theoretical background in NLP and transformers.
  • Proficiency with deep learning frameworks (PyTorch).
  • Experience in distributed training of large models.
  • Great communication skills.
Good to have:
  • Experience with LLM inference frameworks.
  • Experience with LLM alignment techniques (RLHF/RLAIF).
  • Experience with MLOps tools and practices.
  • Experience with K8s and Kubeflow.
  • Scientific publications in the NLP field.

Job Details

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.

Similar Jobs

Blacksahrk - Software Engineer

Blacksahrk

Graz, Styria, Austria (On-Site)
4 Months ago
Drive mode - Engineering Manager

Drive mode

Tokyo, Japan (Hybrid)
2 Months ago
Motive - QA Manager, Enterprise Systems Engineering

Motive

(Remote)
20 Hours ago
Warner Bros Games - Senior Data Engineer

Warner Bros Games

Hyderabad, Telangana, India (Hybrid)
1 Month ago
Aisera Jobs - Senior Cloud Security Engineer

Aisera Jobs

Palo Alto, California, United States (On-Site)
1 Day ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

NVIDIA - Senior Software QA Test Development Engineer

NVIDIA

Santa Clara, California, United States (On-Site)
3 Weeks ago
Hedra - Full-Stack Engineer

Hedra

San Francisco, California, United States (On-Site)
1 Month ago
Playrix - Senior C++/Python Software Engineer (Engine)

Playrix

Ireland (Remote)
6 Months ago
Telastra - Senior Software Engineer (Java Microservices)

Telastra

Australia (On-Site)
7 Hours ago
Omnissa - Member of Technical Staff (C#.net)

Omnissa

Bengaluru, Karnataka, India (On-Site)
6 Months ago
Zeta - Engineering Manager - Cloud Security (DevSecOps)

Zeta

Bengaluru, Karnataka, India (On-Site)
6 Months ago
N-iX - Senior Data Engineer

N-iX

Kyiv, Kyiv City, Ukraine (Hybrid)
2 Weeks ago
NXP - Junior Developer of Systems Testing Infrastructure

NXP

Brno, South Moravian Region, Czechia (On-Site)
7 Months ago
Playrix - Senior C++ Software Engineer (Gameplay)

Playrix

Ukraine (Remote)
6 Months ago
Nintendo - Machine Learning Ops Engineer

Nintendo

Redmond, Washington, United States (On-Site)
2 Weeks ago

Get notifed when new similar jobs are uploaded

Jobs in Yerevan, Yerevan, Armenia

Playrix - Senior Technical Designer

Playrix

Armenia (Remote)
6 Months ago
Playrix - Senior/Principal 2D Artist (Match-3)

Playrix

Armenia (Remote)
6 Months ago
Playrix - Senior 2D Artist (Illustrations)

Playrix

Armenia (Remote)
6 Months ago
Playrix - Game Director

Playrix

Armenia (Remote)
6 Months ago
Fortra - PX Coordinator

Fortra

Armenia (On-Site)
1 Day ago
Evolution - Russian Speaking Game Presenter

Evolution

Yerevan, Yerevan, Armenia (On-Site)
11 Months ago
G5 Games - C++ Gameplay Programmer

G5 Games

Yerevan, Yerevan, Armenia (Remote)
6 Months ago
Sperasoft - Senior Material Artist

Sperasoft

Yerevan, Yerevan, Armenia (On-Site)
1 Month ago
Hawk Eye Innovations - Match Operations Assistant - Yerevan

Hawk Eye Innovations

Yerevan, Yerevan, Armenia (On-Site)
1 Week ago
Playrix - Lead Location Designer

Playrix

Armenia (Remote)
6 Months ago

Get notifed when new similar jobs are uploaded

Similar Category Jobs

Looks like we're out of matches

Set up an alert and we'll send you similar jobs the moment they appear!

About The Company

Yerevan, Yerevan, Armenia (On-Site)

Prague, Prague, Czechia (On-Site)

Yerevan, Yerevan, Armenia (On-Site)

Yerevan, Yerevan, Armenia (On-Site)

Berlin, Berlin, Germany (On-Site)

Yerevan, Yerevan, Armenia (On-Site)

Prague, Prague, Czechia (On-Site)

View All Jobs

Get notified when new jobs are added by jetbrains

Level Up Your Career in Game Development!

Transform Your Passion into Profession with Our Comprehensive Courses for Aspiring Game Developers.

Job Common Plug