Senior ML Researcher
JetBrains
Job Overview
Who's the hiring manager?
Sign up to PitchMeAI to discover the hiring manager's details for this job. We will also write them an intro email for you.

Job Description
Senior ML Researcher (Code Editing)
At JetBrains, code is our passion. Since 2000, we’ve focused on reducing routine work so developers can spend more time building and shipping.
The JetBrains Research team studies machine learning techniques and agent-based approaches for software development to create methods that enable the next generation of ML for code features. We’re looking for a researcher to develop and improve training methodologies for models capable of solving tasks such as code generation, code editing, and empowering AI agents. You don’t need experience in every area, but you should be comfortable working across topics and delivering publishable results.
In This Role, You Will
- Train and scale neural networks for processing source code.
- Develop new methods and improve existing ones for code generation, code editing, and agent-based workflows.
- Mentor colleagues on ML topics.
We’re Looking For Someone Who
- Has experience in natural language processing or code modeling.
- Has worked on distributed training and scaling neural networks.
- Has experience with modern training techniques and architectures (for example, MoE).
- Has publications at top ML conferences.
- Has strong programming skills.
Nice To Have
- Hands-on experience with post-training techniques.
- Experience with multimodal or low-data training.
- Familiarity with large-scale model training.
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.
Key skills/competency
- Machine Learning Research
- Code Generation
- Code Editing
- Natural Language Processing
- Neural Networks
- Distributed Training
- Model Scaling
- ML Conferences
- Programming Skills
- AI Agents
How to Get Hired at JetBrains
- Tailor your resume: Highlight your ML research, NLP, and code modeling experience, emphasizing publications and programming skills.
- Showcase your expertise: Quantify achievements in training neural networks and distributed systems.
- Prepare for technical interviews: Be ready to discuss ML concepts, model architectures, and your research experience.
- Research JetBrains: Understand their passion for code and developer productivity to align your application.
Frequently Asked Questions
Find answers to common questions about this job opportunity
Explore similar opportunities that match your background