Founding ML Engineer (Spectrum)

JetBrains; Netherlands; Hybrid; EUR 110,000 - 150,000; Full time

Benefits offered by this job

Competitive salary

JetBrains benefits

Job summary

JetBrains is seeking a top-class ML Engineer to own our AI and ML engineering stack and shape the research agenda. You will design and implement the ML/LLM data ingestion, retrieval, and reasoning capabilities, building datasets, metrics, and pipelines, and leading the ML team in a startup-spirited environment.

You will work with LangChain, vector databases, and multi-agent systems, collaborating closely with JetBrains Research to define and advance the product-led AI strategy.

Qualifications

• Proven ML/AI Lead with 5+ years in ML/AI systems.
• 2+ years focused on LLMs and generative AI.
• Deep understanding of LLM architectures and fine-tuning.
• Hands-on with prompt engineering and LLM pipelines.
• Experience with LangChain, LlamaIndex, LangSmith, or similar.
• Vector databases and RAG patterns expertise.
• Deploying LLM-powered apps via APIs or open-source models.
• Strong Python skills; Kotlin knowledge a plus.
• Excellent written and verbal English communication.

Responsibilities

• Design and build ML/LLM solutions for data ingestion and reasoning.
• Create datasets, metrics, and pipelines for measurable improvements.
• Architect and improve agents for context retrieval and data alignment.
• Establish MLOps practices including orchestration and observability.
• Collaborate with engineering and JetBrains Research on strategy.
• Define hiring criteria and grow the ML team.

Skills

ML/AI Lead

Python

English communication

LLM systems

Kotlin (plus)

Tools

LangChain

LlamaIndex

LangSmith

smolagents

Vector databases

Job description

Software engineers and AI agents alike suffer from the same problem: finding that one person or place that will answer their tough, specific question. Many solutions promise to solve this with similarity search in vector databases. Unfortunately, finding the answer is often a puzzle with pieces to be collected across a myriad of contradictory sources and cannot be solved without surgical search and careful reasoning. Spectrum collects data from an organization’s code, docs, and issues, and organizes knowledge in a unified ontology that AI agents can efficiently search through and reason over. We aim to revolutionize the semantic layer space for software-building organizations and move beyond specs that fall out of sync with code, introducing a living spec - one that’s extracted from the whole system and used to keep it aligned. Spectrum is meant to be the single source of truth for all product and architectural knowledge. Spectrum is a resident of JetBrains' startup incubator, with startup speed and autonomy, and backed by 25 years of developer tooling expertise. We are looking for a top-class ML Engineer who will help us shape the future of software development. You will own our AI and ML engineering stack and help define the research agenda for our team. Your technical vision and design decisions will directly shape the product and determine its success.

Responsibilities

• Designing and building the ML/LLM solution for data ingestion, knowledge extraction, retrieval, and subsequent reasoning.
• Creating the datasets, metrics, and pipelines that drive measurable improvements across the system.
• Architecting and improving agents for context retrieval, knowledge extraction, and data alignment, which includes prompt engineering, model selection, and inference optimization.
• Establishing MLOps practices, including orchestration, observability, and experiment tracking.
• Collaborating with the engineering team on system design and with JetBrains Research on the research agenda.
• Defining hiring criteria, growing the ML team, and shaping the ML team culture.

We expect you to have

• A proven track record as an ML/AI Lead.
• At least five years of experience in ML/AI systems, with at least two years focused on LLMs and generative AI.
• A deep understanding of the LLM ecosystem, including model architectures and fine-tuning approaches.
• Hands-on experience with:
• Prompt engineering and LLM pipeline design, including evaluation.
• Agentic frameworks such as LangChain, LlamaIndex, LangSmith, smolagents, or an equivalent.
• Vector databases and retrieval-augmented generation (RAG) patterns.
• Deploying and scaling LLM-powered applications using APIs (e.g. OpenAI or Anthropic) or open-source models.
• Strong Python skills - Kotlin knowledge would be a plus.
• Excellent communication skills, with the ability to explain complex technical concepts to diverse audiences.
• Proficiency in English, both written and verbal.

Our ideal candidate would have

• Experience with ontologies, knowledge graphs, or graph-based reasoning.
• Experience in early-stage startups - you enjoy the zero-to-one phase.
• The ability to think strategically about product-led AI, beyond just training models in isolation.
• A background in code analysis, developer tools, or software engineering research.
• Experience with multi-agent systems or complex agentic workflows.
• Actively contributed to relevant open-source projects or publications.

What we offer

• A competitive salary and JetBrains benefits.
• A generous runway and corporate resources with startup autonomy.

#LI-KP1 We are an equal opportunity employerWe 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.