A

Arcade AI

AI-powered authentication platform for developers

Principal Machine Learning Engineer

Applied Machine LearningUpdated on Nov 14
Full-timeSan Francisco, CA, USA

About the Job

Requirements

  • 7+ years of software engineering experience, including 4+ years training and shipping production machine learning systems.
  • Experience training or fine-tuning models that reached production and improved a customer-relevant metric.
  • Expertise in production agent systems, including harnesses, memory, skills, tool use, and sub-agents.
  • Knowledge of when fine-tuning works, what data it requires, and how to derive that data from raw telemetry.
  • Statistical fluency to assess whether evaluation differences are meaningful or noise.
  • Experience deploying models under constraints such as tight latency budgets, limited GPUs, or customer infrastructure, using tools such as vLLM, ONNX, TensorRT, or llama.cpp, or equivalent.
  • Strong Python skills for training and machine learning work, plus TypeScript or Go for production services serving models.
  • Ability to choose the stack, document decisions, and defend them over time.
  • Comfort working amid ambiguity on an early team with an expanding charter and incomplete data.

Responsibilities

  • Run the machine learning pipeline end to end, from data and training through evaluation and release, making model releases routine and reliable.
  • Fine-tune and train models for tool selection, routing, retrieval, and agent memory using distillation, embeddings, rerankers, and other approaches as appropriate.
  • Expand model use cases, starting with search and recommendation over tools and agent context.
  • Build evaluation systems with offline evaluation against real agent traces, online production measurement, and head-to-head comparisons with Claude, GPT, and Gemini.
  • Quantize, optimize, and package models for customer VPCs and air-gapped environments, and work with the Runtime team on model serving.
  • Build a feedback loop from production agent traces and tool-call data to improved models while respecting enterprise customer data boundaries.
  • Choose the machine learning stack, write the strategy, and help decide future machine learning hires.
  • Use expertise in modern agent systems, including harnesses, memory, and skills, to identify where models improve agents and assess approaches early.
  • Use AI to increase personal productivity so that projects taking a week can be completed in a day.
  • Own and productionize the existing agent recommendation model and training pipeline, and establish future machine learning patterns and build-versus-buy decisions.

Desired Qualifications

  • Experience shipping machine learning into enterprise on-premises or regulated environments, including financial services, healthcare, or government.
  • Tool-use benchmark or evaluation work, such as BFCL, τ-bench, ToolBench, or MCP evaluations.
  • Familiarity with the Model Context Protocol ecosystem; filing an issue against the specification is an added plus.
  • Experience building training-data pipelines from production traces with privacy controls.
  • Experience as the first machine learning hire at a company.
  • Open-source contributions or published work used by other engineers.
  • Experience at an early-stage startup.

About the Company

Arcade AI

AI-powered authentication platform for developers

Arcade AI provides an AI-driven platform for developers and enterprises to streamline authentication workflows and integrate AI tool calls. It offers pre-built connectors that let users start building in minutes and easily connect apps to authentication services and AI tools. The platform specializes in authentication and AI integration, offering ready-made tooling to simplify complex tasks and enable faster setup for enterprise needs. Its goal is to improve security and operational efficiency by making AI-powered authentication and tool usage easier to implement and manage.

Company size
51-200
Company stage
Series A
Total funding
$72M
Headquarters
San Francisco, CA
Founded
2024

Recently Posted Jobs

Explore more opportunities at Arcade AI

See more jobs