Bjak logo

Machine Learning Engineer, Ranking & Retrieval

Bjak
NewPosted today

LOCATION

Sweden · Remote

EXPERIENCE

Not specified

TYPE

FullTime

SKILLS REQUIRED

Machine LearningInformation RetrievalPytorchVector DatabasesEmbeddingsRecommendation Systems

Job description

Overview

Build and improve the machine learning systems that power Bjak's AI-native applications, covering classification, retrieval, ranking, personalization and evaluation, from model development through production deployment.

What you'll do

  • Build and improve ML models for classification, ranking, retrieval and personalization.
  • Develop efficient classifiers and optimize model accuracy, precision, latency and cost.
  • Improve search relevance and context retrieval using embeddings, hybrid search and reranking.
  • Build personalization models that learn from user behavior while respecting privacy.
  • Develop evaluation pipelines, labeled datasets and metrics to measure model quality.
  • Fine-tune and adapt existing models when supported by sufficient data.
  • Build and maintain ML training, inference and deployment pipelines.
  • Debug production issues, monitor model performance and ship improvements based on real-world feedback.

What you'll need

  • Strong foundations in machine learning and modern neural architectures.
  • Hands-on experience building and deploying ML models in production.
  • Experience in classification, ranking, retrieval, recommendation systems or personalization.
  • Understanding of model evaluation, experimentation and performance metrics.
  • Comfortable writing production-quality code and working with real-world datasets.
  • Able to balance model quality, latency, cost and reliability.
  • Strong ownership and problem-solving skills.
  • Tech stack: Python; PyTorch / JAX; embeddings, vector search and ranking models; production ML systems and inference pipelines.

Details

  • Interviews: 3, but no more than 4, conducted virtually and/or onsite.

Read the full description and apply on the company’s own careers page.

Stay safe

Hiring on Abekus is free for applicants

We never charge a fee, and employers are prohibited from doing so. If a recruiter asks for payment, please report them right away.

Machine Learning Engineer, Ranking & Retrieval