JetBrains
Staff Research Engineer (LLM Pre-Training)
Get notified of more jobs like this · No spam, ever
How your CV stacks up
Upload your CV to see how well it fits this job role
?%
Research Engineer (LLM for Coding) – JetBrains
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.
About the Project
We are working on an ambitious new platform that provides AI capabilities to all JetBrains products. Our platform is based on in-house-developed models 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 work on developing Large Language Models (LLMs) from scratch, deploying them into production environments where they will be accessible by end users globally.
What We Value in Engineers
We value engineers who:
- Can plan projects and make decisions independently, consulting with others where needed.
- Identify customer needs and prioritise tasks accordingly.
- Start with the simplest solutions, gradually adding complexity as needed.
- Take sole responsibility for an entire subsystem.
- Have a passion for learning and stay updated with latest developments in the LLM field.
Reasons to use Rodeo
Start with a chat, not a search bar
Grad scheme, placement, apprenticeship? Not sure what you want yet — that's fine. Your agent talks it through with you and turns "I have no idea" into a shortlist.
It searches the market for you
Every day your agent scans the market matching roles against what actually matters to you, not just keywords on a CV.
Only hits
No noise. No "maybe this fits." Just roles with a clear explanation of why they're right — and where to focus when applying.
Key Responsibilities
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.
Requirements
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 Natural Language Processing (NLP) and transformer-based approaches.
- Proficiency with modern deep learning frameworks such as PyTorch, along with common NLP libraries.
- Experience in distributed training of multi-billion parameter models.
- Attention to detail in everything you do and strong communication skills.
Bonus points if you have:
Get help with your application
Your very own career expert that helps elevate your application to the next level.
- Experience with LLM inference frameworks (e.g., vLLM, DeepSpeed, TensorRT).
- Knowledge of LLM alignment techniques (e.g., RLHF/RLAIF).
- Familiarity with MLOps tools and practices, including CI/CD for ML systems.
- Experience with Kubernetes (K8s) and Kubeflow.
- Scientific publications in the NLP field.
Our AI Development Stack
How we develop JetBrains AI:
- Training infrastructure: A cluster of hundreds of NVIDIA GPUs.
- Git for source control management.
- ML stack: Python, PyTorch, and HuggingFace.
- Experiment tracking: Kubeflow and Weights & Biases.
- CI Automation: TeamCity.
JetBrains Culture
We are an equal opportunity employer.
We believe 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 background, identity, religion, age, accessibility needs, or orientation.
Processing note: Your job application data will be handled in accordance with our Recruitment Privacy Policy.
“It took my CV and asked me questions relevant to understanding what kind of jobs to suggest for me. Suggestions were almost perfect. Jobs were exactly what I’ve been looking for.”
Jessica, London
Location