Research Engineer

3 Months ago • All levels • Research Development

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

Wind River - Senior Outbound Product Manager, Private Cloud

Wind River

Walnut Creek, California, United States (Hybrid)
1 Month ago
Ramp - Manager, Sales Development | Outbound

Ramp

New York, United States (Hybrid)
1 Week ago
The Walt Disney Company - Account Manager

The Walt Disney Company

Antwerp, Flanders, Belgium (Hybrid)
3 Months ago
Milk  visual effects - Unreal Generalist

Milk visual effects

(On-Site)
3 Months ago
Triple dot studios - Lead Product Designer

Triple dot studios

Warsaw, Masovian Voivodeship, Poland (Hybrid)
1 Month ago
EvenUp - Backend Software Engineer, AI Document Generation

EvenUp

Toronto, Ontario, Canada (Hybrid)
1 Month ago
Mozilla - Senior Staff Machine Learning Engineer, GenAI

Mozilla

Canada (Remote)
4 Weeks ago
Eventbrite - Researcher II (East Coast)

Eventbrite

United States (Remote)
1 Month ago
Bosch Group - AI Research Scientist – GenAI

Bosch Group

Sunnyvale, California, United States (On-Site)
1 Month ago
Ion - Senior AI Engineer, Italy

Ion

Pisa, Tuscany, Italy (On-Site)
9 Months ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

MY.GAMES - Java Developer

MY.GAMES

(Remote)
3 Weeks ago
Games 24x7 - Deputy Manager - Marketing

Games 24x7

Bengaluru, Karnataka, India (On-Site)
3 Weeks ago
PwC - Internship program - Risk Consulting

PwC

Bangkok, Bangkok, Thailand (On-Site)
10 Months ago
Morning Star - Talent Development Specialist

Morning Star

Mumbai, Maharashtra, India (Hybrid)
1 Month ago
Demandbase - Staff Database Reliability Engineer

Demandbase

Bengaluru, Karnataka, India (On-Site)
2 Months ago
PeopleFun - Senior Software Engineer

PeopleFun

United States (Remote)
2 Months ago
Toast - Fintech Compliance Manager

Toast

Boston, Massachusetts, United States (Hybrid)
1 Month ago
Saviynt - Renewal Specialist

Saviynt

United States (Remote)
1 Month ago
Match Group - Staff Machine Learning Engineer, Growth

Match Group

New York, New York, United States (Hybrid)
3 Months ago
Regent craft - Assistant Facility Manager

Regent craft

North Kingstown, Rhode Island, United States (On-Site)
2 Weeks ago

Get notifed when new similar jobs are uploaded

Jobs in Yerevan, Yerevan, Armenia

Fortra - Machine Learning Engineer II

Fortra

Armenia (On-Site)
2 Months ago
Evolution  - Card Shuffler

Evolution

Yerevan, Yerevan, Armenia (On-Site)
6 Months ago
Sperasoft - 3D Weapon Artist (AAA Hero Shooter)

Sperasoft

Yerevan, Yerevan, Armenia (On-Site)
2 Weeks ago
Cubic games - Level Designer

Cubic games

Yerevan, Yerevan, Armenia (Remote)
4 Weeks ago
M365 connect - Strategic Client Acquisition Partner  HR Tech Recruitment

M365 connect

Yerevan, Yerevan, Armenia (Remote)
2 Months ago
Entlgd  - Lead Lighting Artist

Entlgd

Yerevan, Yerevan, Armenia (Hybrid)
2 Months ago
Entlgd  - Lead 3D Environment Artist (realism)

Entlgd

Yerevan, Yerevan, Armenia (Hybrid)
2 Months ago
playrix  - Senior Data Analyst (Game)

playrix

Armenia (Remote)
9 Months ago
playrix  - Principal VFX Artist

playrix

Armenia (Remote)
9 Months ago
EveryMatrix - Mid/Senior Data Analyst

EveryMatrix

Yerevan, Yerevan, Armenia (On-Site)
4 Months ago

Get notifed when new similar jobs are uploaded

Research Development Jobs

Reddit - Senior Software Engineer, AI Enablement

Reddit

Ontario, Canada (Remote)
2 Months ago
Sika Group - R&D Specialist

Sika Group

Calusco D'Adda, Lombardy, Italy (On-Site)
2 Months ago
Ansys - Senior R&D Engineer

Ansys

Canonsburg, Pennsylvania, United States (Remote)
2 Weeks ago
world resource institute - Consultant as User Research Analyst

world resource institute

Jakarta, Indonesia (On-Site)
1 Week ago
Riot Games - Research Operation Coordinator - Global Research Operations Team (Contract)

Riot Games

Shanghai, Shanghai, China (On-Site)
3 Months ago
Cadence - Lead Software Engineer (Agentic AI)

Cadence

Cambridge, England, United Kingdom (Hybrid)
2 Weeks ago
bytedance - Research Scientist in Foundation Model (Music) - 2025 Start (PhD)

bytedance

San Jose, California, United States (On-Site)
9 Months ago
Prophecy - AI/ML Agentic-Workflows Engineer (Sr/Staff)

Prophecy

San Francisco, California, United States (Hybrid)
3 Months ago
Ansys - Senior R&D Engineer, Desktop AI/ML

Ansys

Vancouver, British Columbia, Canada (On-Site)
1 Month ago
NVIDIA - Deep Learning Performance Architect

NVIDIA

Hyderabad, Telangana, India (Hybrid)
5 Months ago

Get notifed when new similar jobs are uploaded

About The Company

Yerevan, Yerevan, Armenia (Hybrid)

Yerevan, Yerevan, Armenia (On-Site)

Yerevan, Yerevan, Armenia (On-Site)

Munich, Bavaria, Germany (Hybrid)

Prague, Prague, Czechia (On-Site)

Yerevan, Yerevan, Armenia (On-Site)

Prague, Prague, Czechia (On-Site)

Yerevan, Yerevan, Armenia (On-Site)

Yerevan, Yerevan, Armenia (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