Founding ML Engineer (Spectrum)

3 months, 4 weeks ago
Full-time
Lead
Software Development
JetBrains

JetBrains

JetBrains provides cutting-edge development tools like IntelliJ IDEA and Kotlin, automating tasks to boost productivity and foster innovation.

Internet Software & Services
1K-5K
Founded 2000

Description

  • Design and build the ML/LLM system 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.
  • Handle prompt engineering, model selection, and inference optimization for LLM-based workflows.
  • Establish MLOps practices including orchestration, observability, and experiment tracking.
  • Collaborate with the engineering team on system design.
  • Work with JetBrains Research on the team’s research agenda.
  • Define hiring criteria, help grow the ML team, and shape team culture.

Requirements

  • Proven track record as an ML/AI Lead.
  • At least 5 years of experience in ML/AI systems, including 2+ 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.
  • 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; Kotlin knowledge is a plus.
  • Excellent communication skills with the ability to explain complex technical concepts to diverse audiences.
  • Proficiency in English, both written and verbal.
  • Experience with ontologies, knowledge graphs, or graph-based reasoning is preferred.
  • Experience in early-stage startups and comfort with the zero-to-one phase is preferred.
  • Ability to think strategically about product-led AI beyond training models in isolation is preferred.
  • Background in code analysis, developer tools, or software engineering research is preferred.
  • Experience with multi-agent systems or complex agentic workflows is preferred.
  • Active contributions to relevant open-source projects or publications are preferred.

Benefits

  • Competitive salary and JetBrains benefits.
  • Generous runway and corporate resources with startup autonomy.
  • Open and inclusive workplace.
  • Equal opportunity employer commitment.

Interested in this position?

Apply directly on the company website

Apply Now

Similar Roles

AI Developer

AI Developer (Contract) at a remote company, responsible for building AI-powered workflows and internal tools that improve business efficiency and create new capabilities.

JavaScript LLM Python
17 hours, 5 minutes ago

Staff ML Engineer (ML/AI)

Lyra Health 1K-5K Health Care Providers & Services

Lyra Health is hiring a Staff ML/AI Engineer to define the architecture and technical strategy for enterprise-scale machine learning and generative AI platforms supporting clinically precise, reliable mental health products.

AWS Celery CI/CD Docker Generative AI HIPAA Java Kafka Kotlin Kubernetes Machine Learning Microservices MLOps Python REST API
1 day, 17 hours ago

AI Agent Implementations Specialist

Extreme Networks 1K-5K IT Services

Extreme Networks is hiring an AI Agent Developer & Marketo Administrator to support marketing automation, AI agent implementation, and MarTech operations across its global marketing organization.

CRM JavaScript Machine Learning Node.js Python REST API
1 day, 17 hours ago

Entrepreneur in Residence - Technical Co-founder (CTO)

FutureSight 11-50 Internet Software & Services

FutureSight is seeking a Co-Founder & CTO to lead the technical direction of a new B2B AI venture from inception to launch and early growth.

LLM System Design
1 day, 17 hours ago

You're on a roll! Sign up now to keep applying.

Sign Up

Already have an account? Log in

Used by 14,729+ remote workers