Staff Research Engineer, LLM Foundation Training

JetBrains; Greater London; On-site; GBP 90,000 - 135,000; Full time

Job summary

JetBrains is seeking a Research Engineer to contribute to training foundation models for coding tasks. You’ll work on developing Large Language Models from scratch and deploying them into production environments accessed by users worldwide.

You’ll train LLMs on a large GPU cluster, collect and process datasets, and collaborate with teams to translate business needs into technical specs while improving existing subsystems. Strong NLP background and PyTorch expertise required.

Qualifications

• Experience designing and deploying production ML systems.
• Strong NLP theory and transformer knowledge.
• Proficiency with PyTorch and NLP libraries.
• Experience with distributed training of large models.
• Excellent communication and collaboration.

Responsibilities

• Collaborate with stakeholders to translate business needs into technical specs.
• Train LLMs from scratch on large GPU clusters.
• Prepare and curate pre-training and fine-tuning datasets.
• Support and improve existing ML subsystems.

Skills

Production ML

NLP theory

PyTorch

Distributed training

Communication

Tools

vLLM

DeepSpeed

TensorRT

Kubeflow

Kubernetes

Weights & Biases

Git

HuggingFace

CI/CD ML

TeamCity

Job description

JetBrains is seeking a Research Engineer to contribute to training foundation models for coding tasks. You’ll work on developing Large Language Models from scratch and deploying them into production environments accessed by users worldwide.

You’ll train LLMs on a large GPU cluster, collect and process datasets, and collaborate with teams to translate business needs into technical specs while improving existing subsystems. Strong NLP background and PyTorch expertise required.