Machine Learning Engineer, Ranking & Retrieval

Posted 3 Months Ago
Hiring Remotely in United States
Remote
Mid level
Artificial Intelligence • Fintech • Software • Financial Services
The Role
Design, build, and operate ML infrastructure for training, evaluation, deployment, and inference. Improve reliability, scalability, latency, throughput, and cost. Create pipelines, observability, benchmarking, and tooling to enable fast experimentation and productionization of models while diagnosing regressions and bottlenecks.
Summary Generated by Built In
About

There are over 5 billion users using basic applications today such email, notes, tasks, calendar and they're not AI-native. Our mission is to build proactive applications for anyone in the world, who are not used to complex prompting. We aim to bring intelligence to conversations, errands, organising and workflows, with minimal to no prompting.

Our first application is an agent-first mobile communication app. We believe building applications that users use daily for communication, coordination and collaboration needs will help integrate AI into their lives. The net effect is to eliminate misinformation, achieve low coordination costs, enable high quality tasks and collaboration at scale with minimal or little prompting.

 
Role
  • Develop and improve ML models for classification, retrieval, ranking and personalisation

  • Build classifiers and ranking models, optimising accuracy, precision, latency and inference cost

  • Improve search relevance and context retrieval using embeddings, hybrid search and reranking

  • Develop personalisation models using user behaviour and product data

  • Build evaluation pipelines, datasets and metrics to measure model quality and detect regressions

  • Train, fine-tune and deploy models using production ML pipelines

  • Monitor model performance, debug production issues and improve models based on evaluation results and user feedback

 
Requirements
  • 3+ years of machine learning engineering experience, including deploying models to production

  • Strong Python programming skills and experience with PyTorch or JAX

  • Solid understanding of machine learning algorithms, model training and evaluation

  • Experience with classification, ranking, retrieval, recommendation or personalisation models

  • Experience working with embeddings, vector search or information retrieval systems

  • Experience building data pipelines, training workflows and model inference services

  • Ability to evaluate model performance using relevant metrics and experimentation

  • Understanding of trade-offs between model accuracy, latency, computational cost and reliability

Nice to Have
  • Experience with transformer models, LLM fine-tuning or embedding models

  • Experience with hybrid search, reranking and retrieval evaluation

  • Experience with recommendation systems or personalisation in consumer applications

  • Experience with large-scale ML training or inference infrastructure

Skills Required

  • Strong software engineering fundamentals and experience building production systems
  • Experience building ML infrastructure, platforms, or production machine learning systems
  • Experience with model deployment, inference, evaluation, or data pipelines
  • Strong understanding of distributed systems and system reliability
  • Proficiency in Python
  • Experience with PyTorch or JAX
  • Familiarity with LLM and ML serving infrastructure (e.g., vLLM, SGLang, TensorRT-LLM)
  • Experience with cloud infrastructure and GPU infrastructure/performance tooling
  • Experience building ML/data pipelines and workflow orchestration
  • Familiarity with vector databases and retrieval infrastructure
  • Ability to write clean, maintainable, production-quality code
  • Comfortable working in ambiguous, fast-moving environments; bias toward ownership and experimentation
Am I A Good Fit?
beta
Get Personalized Job Insights.
Our AI-powered fit analysis compares your resume with a job listing so you know if your skills & experience align.

The Company
HQ: Petaling Jaya
253 Employees
Year Founded: 2019

What We Do

Our mission is to develop technology based solutions to improve financial inclusion. We develop new & innovative platforms & services globally. For example, we are the first platform to simplify and digitise comprehensive life and medical insurance, supported by AI agent. BJAK is the largest insurance platform in Southeast Asia. If you enjoy building cutting edge platform-ecosystems that gives equal access to financial services to everyone at scale, join us

Similar Jobs

ClickUp Logo ClickUp

Machine Learning Engineer

Cloud • Digital Media • Enterprise Web • Marketing Tech • Software
Remote or Hybrid
United States
1000 Employees
200K-250K Annually

IDeaS Logo IDeaS

Account Director

Software • Analytics • Hospitality
Remote or Hybrid
United States
747 Employees

IDeaS Logo IDeaS

Account Director

Software • Analytics • Hospitality
Remote or Hybrid
MN, USA
747 Employees

IDeaS Logo IDeaS

Account Director

Software • Analytics • Hospitality
Remote or Hybrid
United States
747 Employees

Similar Companies Hiring

Revel Thumbnail
Aerospace • Hardware • Robotics • Software
Marina Del Rey, California
70 Employees
Blee Thumbnail
Artificial Intelligence • Marketing Tech • Software
New York, New York
33 Employees
Vega Thumbnail
Artificial Intelligence • Automotive • Insurance • Transportation
US
65 Employees

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account