Skip to main content

Machine Learning Engineer — Applied AI & Agents

CotoolSoftware
Pay
$150K–$300K
per year
Work mode
On-site
Full Time
Level
Mid

San Francisco, CA at a glance

Rent
#2 of 51
$2,680/mo+46% vs US avg
Weather
#17 of 51
295 mild days0 hot · 0 cold
Income tax
#1 of 51
13.3% top rateCalifornia

What you need

  • Fine-tuned LLMs: data prep, training, evaluation
  • Built RL environments: tasks, observations, rewards, debugging
  • Deployed ML models to production with monitoring
  • Strong Python, PyTorch, ML engineering fundamentals
  • Experimental judgment: metric design, regression analysis

What you'll do

  • Build LLM fine-tuning pipelines: data curation, experiments, evaluation
  • Create RL environments for security tasks with verifiable outcomes
  • Design agent evaluations: multi-step completion, tool use, baselines
  • Operate production model systems: serving, monitoring, rollback
  • Collaborate on model-harness interface: context, memory, tool design

Build the learning systems behind security agents

At Cotool, we're building AI agents and the infrastructure around them to help security teams detect threats, investigate alerts, and respond. Our agents work across security tools and data, write and tune Python detections, and bring humans in when judgment or approval matters. Teams at Ramp, EliseAI, and ClickHouse use Cotool.

We're hiring a machine learning engineer to build the loop from agent behavior to training data, from training data to better models, and from better models to measurable improvements in production. You'll work across post-training, RL environments, evaluation, and deployment, with direct ownership of what reaches users.

What you'll build

  • LLM fine-tuning pipelines. Curate and version datasets, turn useful trajectories and failure cases into training examples, run post-training experiments, and evaluate generalization on held-out tasks.
  • RL environments for security work. Build reproducible tasks with realistic tool interfaces, observations, state transitions, and verifiable outcomes. Design rewards that encourage successful investigations and catch shortcuts, reward hacking, and brittle behavior.
  • Agent evaluations. Measure multi-step task completion, tool use, evidence quality, and failure recovery. Use baselines and ablations to distinguish model improvements from changes to the harness or context layer.
  • Production model systems. Take models from experiment to serving: versioning, inference performance, rollout, monitoring, and rollback. Own the tradeoffs between quality, latency, and cost.
  • The interface between model and harness. Work with the engineering team on context assembly, memory, compaction, and tool design. Decide which failures need better training, better context, or a change to the execution loop.

A project might start with agents repeatedly failing a particular class of investigation. You'd reproduce the failure, build an environment and evaluation around it, test a training approach, and deploy the improvement. Success means the agent performs better on new investigations under production constraints.

What we're looking for

  • You've fine-tuned LLMs before. You've worked directly on data preparation, training, and evaluation, and can explain what improved, what didn't, and why.
  • You've built RL environments. You have hands-on experience defining tasks, observations, actions, rewards, and episode completion, and debugging the behavior those choices produce.
  • You've deployed ML models to production. You've taken responsibility for serving and operating a model used by real users or systems, including monitoring and reliability.
  • Strong Python and ML engineering fundamentals, with practical experience in a deep learning framework such as PyTorch. You can move between training code, data pipelines, experiments, and production software.
  • Good experimental judgment. You question the metric, inspect the data, investigate regressions, and know when a simpler approach is the right one.

Experience with tool-using agents, LLM post-training with RL, synthetic data, or security systems is especially relevant. We care about what you've built and can demonstrate. Bring an example of a training run, environment, or production deployment you owned, including the failures and tradeoffs.

The platform and team

Our platform uses TypeScript, React, Node.js, Python, PostgreSQL, ClickHouse, and Redis, with AI SDK for model integration and Braintrust for tracing and observability. You'll help shape the training and inference stack around the needs of the product.

We're a small team in San Francisco backed by Andreessen Horowitz, WndrCo, Y Combinator, and Homebrew. You'll work closely with the founders and the engineers building our agent runtime. We offer competitive salary and equity, healthcare, dental and vision coverage, and an unlimited token budget.

If you want to close the loop between training an agent and watching it solve a real problem, we'd like to meet you.

Optimize your resume for this job

Get a match score and the keywords you're missing

Optimize resume

About Cotool

Cotool is an agentic security platform that eliminates manual and repetitive work for security teams. Today’s cybersecurity teams remain bogged down by alert fatigue, context switching, and documentation overhead. Security teams don't need more dashboards with "AI-powered insights" — they need their time back. Cotool empowers teams to build security their way. Our platform provides an AI co-pilot that automatically assembles context across tools, a no-code agent builder that transforms successful investigations into reusable automations, and automated report generation, providing comprehensive documentation in seconds. Unlike solutions that force teams to adapt to vendor workflows, Cotool puts the power of AI directly in practitioners' hands, letting them shape tools around their unique processes. Early users have already cut time spent on investigation and detection engineering by 70%.

Similar Software roles

Machine Learning Engineer — Applied AI & Agents
$150K–$300K · Cotool
Apply