Senior ML Researcher - Code Editing for AI Development

About the Role

We are seeking a Senior ML Researcher (Code Editing) to join our innovative team at JetBrains. This remote position offers you the chance to work on cutting-edge machine learning techniques aimed at enhancing code generation and editing capabilities. As a part of the JetBrains Research team, you will contribute to the next generation of ML features that empower developers worldwide.

What You'll Do

• Train and scale neural networks specifically designed for processing source code.
• Develop new methodologies and enhance existing techniques for code generation and editing.
• Mentor colleagues on various ML topics, fostering a collaborative learning environment.
• Conduct research that leads to publishable results in top-tier ML conferences.
• Work on distributed training and scaling of neural networks to improve performance and efficiency.

Requirements

• Proven experience in natural language processing or code modeling.
• Hands-on experience with distributed training and scaling neural networks.
• Familiarity with modern training techniques and architectures, such as Mixture of Experts (MoE).
• Strong programming skills, with proficiency in languages like Python.
• Publications in leading ML conferences are a significant advantage.

Nice to Have

• Experience with post-training techniques and multimodal or low-data training.
• Familiarity with large-scale model training processes.

What We Offer

• Competitive salary ranging from $120,000 to $160,000 annually.
• Flexible working hours and a fully remote work environment.
• Opportunities for professional development and continuous learning.
• A collaborative and inclusive workplace culture.
• Access to the latest tools and technologies in machine learning.

Join us at JetBrains, where we are committed to creating an open and inclusive workplace that values diversity. Apply now to be part of our mission to enhance the developer experience through innovative ML solutions.