Founding Machine Learning Engineer Job at Key Technology, Santa Clara, CA

b0JMTEMxdE5NUk1LeXJBdFhLdzRHRDdVclE9PQ==
  • Key Technology
  • Santa Clara, CA

Job Description

Our Client is building a next-generation voice-AI platform that powers curated introductions, agentic scheduling, and real-time feedback.

We’re seeking a founding-calibre ML Engineer to own our machine learning systems end-to-end — from data pipelines and model training to evaluation and low-latency inference.

You’ll design, build, and ship ranking and recommendation systems that make every match feel more personal and improve week after week.

What You’ll Do

  • Design and deploy multi-stage retrieval and re-ranking systems for compatibility, search, and personalisation.
  • Build and maintain data pipelines for training, evaluation, and reporting — ensuring reproducibility and quality.
  • Train and fine-tune LLMs/encoders , manage model versioning, rollout, and rollback.
  • Run offline metrics (AUC, NDCG, MAP) and online A/B tests to measure real-world impact.
  • Build inference services that meet tight latency and cost targets, with caching and fallback strategies.
  • Implement guardrails and monitoring for drift, bias, and reliability; define model SLOs and alerts.
  • Collaborate across ML, platform, and product teams to turn voice and text signals into better matches .

Tech Stack

Python • PyTorch • Hugging Face • OpenAI/Anthropic APIs • pgvector / FAISS / Pinecone • Postgres

Airflow / Prefect / dbt • AWS (S3, ECS/Kubernetes, Lambda) • CI/CD (GitHub Actions)

What We’re Looking For

  • 3+ years of experience in applied ML (ranking, search, or recommendations).
  • Deep Python skills and experience with PyTorch or TensorFlow (Hugging Face a plus).
  • Hands-on with embeddings, LLMs, and vector search (pgvector, FAISS, Pinecone, Weaviate).
  • Proven ability to take models from notebook to production — APIs, CI/CD, monitoring, rollback.
  • Experience building data pipelines with Airflow, Prefect, Dagster, or dbt.
  • Comfortable owning latency, cost, and reliability for inference services.
  • Must be authorised to work in the U.S. and onsite 5 days/week in San Francisco.

Bonus Points

  • Experience with real-time or voice systems , retrieval/re-ranking stacks, or feature stores.
  • Prior work at consumer AI, dating, or recommendation startups where you shipped ranking systems.

Job Tags

Similar Jobs

CarMax

Senior Manager, Data Engineering Job at CarMax

 ...solve for complex business problems Maintain FinOps focus for cloud costs and actively manage contracts and consulting expenditure....  .... Minimum Qualifications: ~ Bachelors degree in computer science, relevant technical field, or equivalent practical... 

SPECTRAFORCE

Computer Vision Engineer Job at SPECTRAFORCE

 ...data exchange. Any companies they would like to see on resumes # Technology industry: Companies like Microsoft, Magic Leap, or other tech giants # Automotive sector: Automotive manufacturers, Tier 1 suppliers, or companies specializing in automotive calibration... 

Robert Half

Human Resources Manager Job at Robert Half

 ...About the Role We are seeking an experienced and strategic HR Manager to join our leadership team. This role is a key partner to the Chief Human Resources Officer (CHRO) and will play a critical part in shaping and executing our people strategy. The role will oversee... 

Amerex Group LLC

Production Assistant Job at Amerex Group LLC

 ...Job Overview - Amerex Group LLC is seeking a Production Assistant to join our Production team. The role involves supporting the full production process, from order placement to delivery, by managing communications, tracking schedules, and maintaining accurate data. This... 

Primoris Services Corporation

Fiber Optic Technicians Job at Primoris Services Corporation

 ...each of our employees. We have 2 immediate openings for Fiber Technician 1 & II at our office located in Phoenix, AZ. Primary Responsibilities...  ...: OSP Fiber Technician II, Prepares and splices fiber optic cables needed to provide video and data services to homes and...