Profile cover
S

AI Prompt Routing Engine Developer

Stella Tran

Full-time · Job · New_York

About the role

Not every question needs GPT-4. A simple classification task runs fine on a smaller model at 1/20th the cost. A code generation request needs a code-specialized model. A creative writing prompt needs high temperature settings. Our prompt routing engine classifies incoming requests and routes them to the optimal model-configuration pair — minimizing cost while maintaining quality. The router is a lightweight classifier that runs in under 5ms and decides: which model (from a fleet of 8), which configuration (temperature, max tokens, system prompt), and whether to apply RAG augmentation. You'll build the routing model, the evaluation framework that measures whether routing decisions maintain quality, and the feedback loop that improves routing from production quality signals. The LLM infrastructure team is six engineers. You'll be the senior engineer who owns the routing layer — the decision point that affects every request flowing through our platform. A bad routing decision wastes money or degrades quality for the end user; a good one is invisible.

Responsibilities

  • Build the prompt classification model for routing to optimal model-config pairs
  • Design evaluation frameworks measuring routing quality against full-model baselines
  • Implement feedback loops improving routing from production quality signals
  • Optimize router latency to under 5ms per decision
  • Expand routing to new models as they become available
  • Set routing policies balancing cost, quality, and latency

Requirements

  • 5+ years in NLP or ML with multi-model orchestration experience
  • Experience with prompt classification and LLM evaluation
  • Strong understanding of model capability differences across sizes and types
  • Python, PyTorch, and system design skills
  • Track record of building systems that optimize cost-quality trade-offs

Benefits

  • Fully remote with flexible schedule
  • Competitive salary plus equity
  • Health, dental, and vision insurance
  • $4,000 annual conference and learning budget
  • High-leverage role on core platform infrastructure

Keep exploring related roles

Use these topic pages to discover more roles in the same hiring cluster.

Categories

AI & Machine Learning Engineers NLP & Prompt Engineering

Skills

python pytorch langchain openai api docker

Related remote AI jobs

Nearby roles that share similar skills, categories, or hiring intent.

Browse all jobs

Senior AI Researcher- Pre-training (f/m/d)

RemoteAI company

Full-time Senior

Senior Product Security Engineer

RemoteAI company

Full-time Senior

AI Resident UK/EMEA

RemoteAI company

Full-time

Senior AI R&D Engineer – Model Training (f/m/d)

RemoteAI company

Full-time Senior

Senior Machine Learning Engineer (Clinical Team)

RemoteAI company

Full-time Senior

Multimodal AI Model Optimization Research Engineer

RemoteAI company

Full-time

Job Type

Full-time

Level

Senior

Language

English

Salary Range

$130k – $160k / year

AI Expertise

AI & Machine Learning Engineers NLP & Prompt Engineering

Skills

python pytorch langchain openai api docker fastapi

Ready to apply for this role?

Create a free talent account in under 2 minutes.

  • Apply to verified AI companies
  • Get AI-matched job recommendations
  • Message hiring managers directly
  • Build your public AI talent profile
Create free account & apply Log in

Interested in this role? Create your free profile to apply.

Apply — create free profile