At JetBrains, code is our passion. Ever since we started, back in 2000, we have strived to make the most effective developer tools on earth. By automating routine checks and corrections, our tools speed up production, freeing developers to grow, discover, and create.
The JetBrains Research team explores ways to use machine learning techniques and agentic approaches to help developers and enhance software development processes. Our work aims to improve the state of ML for code by turning the latest academic advances into practical applications.
Our team is looking for an ML Engineer to participate in a variety of projects in areas such as code completion and generation, AI agents development, and test generation. Though we don’t expect the candidate to have experience in all the tasks we work on, we are looking for someone excited to take on the challenge of working in a diverse set of contexts.
In this role, you will:- Design and prototype machine learning solutions to improve the software development workflow.
- Apply existing open-source models or train custom ones as needed.
- Build training and evaluation pipelines that support fast iteration and ensure reproducibility.
- Stay up to date with the latest advancements in applying machine learning to source code.
- Collaborate with colleagues and provide mentorship when appropriate.
- Have experience designing machine learning systems from scratch.
- Have a background in natural language processing or code modeling.
- Have a proven track record of delivering machine learning projects.
- Know how to design reproducible ML experiments.
- Enjoy working on projects with high uncertainty.
- Are proficient in Python, Java, or Kotlin.
- Have hands-on experience with modern ML frameworks.
- Have experience applying machine learning to programming languages.
- Have hands-on experience building and improving AI agents.
- Have experience in A/B testing and general knowledge of statistics
- Have worked with large-scale data processing pipelines
#LI-KP1
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.
Similar Jobs
What you need to know about the NYC Tech Scene
Key Facts About NYC Tech
- Number of Tech Workers: 549,200; 6% of overall workforce (2024 CompTIA survey)
- Major Tech Employers: Capgemini, Bloomberg, IBM, Spotify
- Key Industries: Artificial intelligence, Fintech
- Funding Landscape: $25.5 billion in venture capital funding in 2024 (Pitchbook)
- Notable Investors: Greycroft, Thrive Capital, Union Square Ventures, FirstMark Capital, Tiger Global Management, Tribeca Venture Partners, Insight Partners, Two Sigma Ventures
- Research Centers and Universities: Columbia University, New York University, Fordham University, CUNY, AI Now Institute, Flatiron Institute, C.N. Yang Institute for Theoretical Physics, NASA Space Radiation Laboratory
