JetBrains

ml engineer for software-building organizations

In short

This role leads ML/LLM engineering for Spectrum, a JetBrains initiative building a unified, searchable ontology of an organization's code, documentation, and issues for AI agents to reason over. It covers designing retrieval and reasoning agents, establishing MLOps practices, and growing and leading the ML team. The company is JetBrains, working onsite in Belgrade, and the role demands at least five years of ML/AI experience with two years in LLMs and generative AI, strong Python skills, and hands-on experience with agentic frameworks and RAG patterns.

About
Spectrum collects data from an organization's code, documentation, and issues, then organizes it into a unified ontology that AI agents can search and reason over. It aims to provide software-building organizations with a living specification and a single source of truth for product and architectural knowledge.
Responsibilities
•Design and build ML/LLM solutions for data ingestion, knowledge extraction, retrieval, and reasoning
•Create datasets, metrics, and pipelines to drive measurable system improvements
•Architect and improve agents for context retrieval, knowledge extraction, and data alignment, including prompt engineering, model selection, and inference optimization
•Establish MLOps practices, including orchestration, observability, and experiment tracking
•Collaborate with the engineering team on system design and with JetBrains Research on the research agenda
•Define hiring criteria, grow the ML team, and shape its culture
•Own the AI and ML engineering stack and help define the team's research agenda
Requirements
•Proven track record as an ML/AI Lead
•At least five years of experience in ML/AI systems, including at least two years focused on LLMs and generative AI
•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
•Hands-on experience with agentic frameworks such as LangChain, LlamaIndex, LangSmith, smolagents, or equivalent
•Experience with vector databases and retrieval-augmented generation (RAG) patterns
•Experience deploying and scaling LLM-powered applications using APIs such as OpenAI or Anthropic, or open-source models
•Strong Python skills
•Excellent communication skills and the ability to explain complex technical concepts to diverse audiences
•Proficiency in English, written and verbal
Nice to have
•Kotlin knowledge
•Experience with ontologies, knowledge graphs or graph-based reasoning
•Early-stage startup experience
•Strategic thinking about product-led AI
•A background in code analysis, developer tools or software engineering research
•Experience with multi-agent systems or complex agentic workflows
•Contributions to relevant open-source projects or publications
Conditions
•Competitive salary and JetBrains benefits
•Generous runway and corporate resources with startup autonomy

Stop applying.
Get referred.

$10 in free credits

In just 60 seconds, fully on autopilot — Eraldt's AI agents read your CV, map the roles you fit, find live vacancies and the people who can refer you, write personal emails, and get you referred.

This isn't an application: your CV doesn't go to the company, and we don't promise a referral for this exact role.

Discover who else lands interviews on autopilot