Skip to content
CognitionCognitionSan Francisco, CA

Software Engineer

Builds core infrastructure for AI software agents in Devin and Windsurf, focusing on long-horizon task execution, tool use, planning, and reliability at scale. Requires strong Python, systems engineering depth, and AI curiosity; onsite in San Francisco.

260k – 300k/yr
On-siteML Engineering

About the role

What You'll Accomplish

  • Build core agent infrastructure: Design and ship the systems that power Devin's long-horizon task execution: tool use, context management, multi-step planning, subagent orchestration, and sandboxed code execution environments.
  • Improve Windsurf as an AI-native IDE: Contribute to editor intelligence, agent-in-the-loop workflows, real-time code understanding, and the developer experience that makes Windsurf different from every other IDE.
  • Close the loop between models and products: Work directly with researchers to translate new model capabilities into shipped features; your feedback shapes what gets prioritized in training.
  • Own reliability and performance at scale: Build systems that handle millions of agentic tasks with low latency, high reliability, and the kind of correctness that developers depend on in production.
  • Move the category forward: Cognition is defining what AI software engineering looks like. You will have real input into what gets built next and why.

Exceptional Candidates Have Demonstrated

  • Systems engineering depth: Experience building reliable, performant distributed systems; you have strong opinions about correctness, failure modes, and production behavior.
  • Product instinct: You care about how the software you build feels to use and you have shipped things that real people depend on.
  • Comfort with ambiguity: You can make progress on hard problems with incomplete specs, learn fast from results, and course-correct without needing a lot of direction.
  • Velocity without shortcuts: A track record of shipping quickly while maintaining the kind of code quality that a high-density team expects.
  • Curiosity about agents and AI: You have dug into how LLMs work, how agents fail, and what it takes to make AI-powered systems behave reliably in the real world.
  • Strong Python proficiency: Python is the primary language across Cognition's codebase; you write clean, well-structured Python and are comfortable owning large Python codebases in production.
  • Relevant industry experience: Prior experience at a frontier AI lab, applied AI company, or developer tools company; you know what good looks like in this category.
  • Degree from a top-tier university: BS, MS, or equivalent in Computer Science, Mathematics, Engineering, or a related technical discipline from a highly selective program.

Compensation & Benefits

  • Base Salary: $260,000 - $300,000 + Significant early-stage equity
  • Medical, Dental, Vision: Fully paid for you and your dependents
  • 401(k): Company match included
  • Perks: Private chef, cozy slippers, endless snacks, and more

Skills

PythonDistributed SystemsLLMsAI AgentsTool UseContext ManagementMulti-Step PlanningSubagent OrchestrationSandboxed Code ExecutionReal-Time Code Understanding

Similar roles

ML Engineering jobs
OpenAI

Machine Learning Engineer, Integrity

OpenAISan Francisco, CA

Deploys state-of-the-art ML models and fine-tunes LLMs in production to enhance platform integrity and safety against adversarial threats. Requires Master's/PhD, deep learning expertise, PyTorch/TensorFlow proficiency.

266k – 555k/yr
On-siteML Engineering
OpenAI

Researcher, Alignment Oversight

OpenAISan Francisco, CA

Designs and runs experiments to improve oversight of increasingly capable AI models, including model training, evaluation, and deployment of practical systems. Analyzes failures and develops techniques to train more aligned models using oversight signals.

250k – 445k/yr
HybridML Engineering
Luma AI

Research Scientist / Engineer — Multimodal Agent

Luma AIPalo Alto, CA

Builds and trains large-scale multimodal agentic models involving reasoning, planning, coding, and tool calling. Requires strong ML foundations, PyTorch expertise, and experience with distributed training on massive datasets.

250k – 450k/yr
HybridML Engineering
Variance

Research Engineer, Evals

VarianceSan Francisco, CA

Build benchmarks, datasets, and evaluation systems to measure and improve AI model quality for fraud, identity, and risk judgment tasks. Collaborate across research, engineering, and product to drive rigorous experimentation and iteration in high-stakes environments.

250k – 400k/yr
On-siteML Engineering
Variance

Research Engineer, Judgment Systems

VarianceSan Francisco, CA

Research Engineer designs evaluations, studies model failures, and builds research loops to improve AI agents for high-stakes fraud detection and judgment tasks. Requires ML training experience, experimental rigor, and strong engineering skills in adversarial environments.

250k – 400k/yr
On-siteML Engineering