Senior ML Researcher (Spectrum)

Spectrum is a project within JetBrains' startup incubator focused on revolutionizing the semantic layer space for software-building organizations by creating a living spec extracted from the whole system to keep it aligned. It is backed by 25 years of developer tooling expertise and aims to be the single source of truth for all product and architectural knowledge.

• PhD or equivalent research experience in NLP, knowledge graphs, ontology learning, information extraction, or a closely related field.
• Strong publication record in knowledge graph construction, ontology learning, information extraction, or NLP.
• Experience applying LLMs to structured knowledge extraction tasks.
• Experience designing and running rigorous experiments including hypothesis formulation, ablation studies, and statistical evaluation.
• Strong Python and PyTorch skills with ability to implement research ideas from scratch.
• Excellent communication skills to explain complex technical concepts to diverse audiences.
• Proficiency in English, both written and verbal.

Nice to have:

• Experience with ontology engineering, semantic web technologies (OWL, RDF, SPARQL), or ontology alignment.
• Background in code analysis, developer tools, or software engineering research.
• Experience with knowledge graph embedding methods or graph neural networks.
• Experience in early-stage startups.
• Active contributions to relevant open-source projects.

• Develop methods for LLM-based ontology construction from code, documentation, and issue trackers, including contradiction detection and reasoning over the resulting knowledge graph.
• Create datasets, metrics, and benchmarks that drive measurable improvements across the system.
• Prototype and validate research ideas, then collaborate with ML and software engineers to bring them to production.
• Collaborate with JetBrains Research and external academic advisors to shape the research agenda.
• Publish research findings at top venues and represent Spectrum in the research community.

We are an equal opportunity employer welcoming everyone regardless of background, identity, religion, age, accessibility needs, or orientation.