Founding ML Engineer (Spectrum)

5 months, 1 week 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

Staff AI Engineer - US

Typeform 251-1K Internet Software & Services

Typeform is hiring a Staff AI Engineer to shape and operate the AI systems powering Research Flow and other customer-facing products, from experimentation through scalable production delivery.

Apache Airflow Apache Spark AWS CI/CD Datadog Docker FastAPI Generative AI Kafka Kubernetes LLM Machine Learning Microservices MLflow OpenSearch Python PyTorch REST API SageMaker Snowflake SQL Terraform
5 minutes ago

Staff AI Engineer - EU

Typeform 251-1K Internet Software & Services

Typeform is hiring a Staff AI Engineer to shape and operate the AI systems powering Research Flow, turning research concepts into secure, scalable, and dependable customer experiences.

Apache Airflow Apache Spark AWS CI/CD Datadog Docker FastAPI Generative AI Kafka Kubernetes LLM Machine Learning MLflow OpenSearch Python PyTorch SageMaker Snowflake SQL Terraform
50 minutes ago

Senior AI Software Engineer

Caseware 251-1K Internet Software & Services

Caseware is hiring a fully remote AI Platform Architect in Colombia to define, build, and operate scalable, secure, and trustworthy AI capabilities for its global accounting and audit software platform.

AWS AWS CDK Confluence DynamoDB GitHub GitHub Actions JIRA Microservices NestJS Python Terraform TypeScript
1 day, 1 hour ago

AI Ops Engineer GTM

Agiloft 51-250 Capital Markets

Agiloft is hiring an AI Ops Engineer, GTM to shape and deliver practical AI solutions across Sales, Marketing, Customer Success, and Revenue Operations, using customer and market insight to improve productivity, revenue outcomes, and customer experience.

CRM HubSpot Python Salesforce Snowflake
1 day, 1 hour 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