About the Role
Spectrum is revolutionizing the semantic layer space for software-building organizations by creating a unified ontology from an organization’s code, docs, and issues. We are seeking a top-class ML Engineer to own our AI and ML engineering stack, define the research agenda, and shape the future of software development.
Responsibilities
- Design and build the ML/LLM solution for data ingestion, knowledge extraction, retrieval, and reasoning.
- Create datasets, metrics, and pipelines for 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 JetBrains Research on the research agenda.
- Define hiring criteria, grow the ML team, and shape the ML team culture.
Requirements
- 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.
- 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 an equivalent.
- Hands-on experience with vector databases and retrieval-augmented generation (RAG) patterns.
- Hands-on experience deploying and scaling LLM-powered applications using APIs (e.g. OpenAI or Anthropic) or open-source models.
- Strong Python skills.
- Excellent communication skills, with the ability to explain complex technical concepts to diverse audiences.
- Proficiency in English, both written and verbal.
Skills
- LLMs
- Generative AI
- Prompt Engineering
- LLM Pipeline Design
- Agentic Frameworks
- Vector Databases
- Retrieval-Augmented Generation (RAG)
- MLOps
- Python
- Kotlin
Location
- Remote
Work Type
- Full-time
Experience Level
- ML/AI Lead
- 5+ years in ML/AI systems
- 2+ years focused on LLMs and generative AI
Salary/Compensations
- Competitive salary
Benefits
- JetBrains benefits
- Generous runway
- Corporate resources
- Startup autonomy
About the Company
- Spectrum is a resident of JetBrains' startup incubator, with startup speed and autonomy, and backed by 25 years of developer tooling expertise.
Equal Opportunity
- 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.
