Staff Engineer - Machine Learning
$211,000 - $261,000
Apply (opens in a new tab)Staff Engineer - Machine Learning
Posted 50 days ago
SENIORITY
Lead
SALARY
$211,000 - $261,000
About the role
- Platform Backend Development
- Lead Implementation: Act as the lead engineer for agent runtime orchestration services, focusing on reasoning, planning, and tool invocationAPI & SDK Design: Own the development of robust APIs and SDKs that enable internal and external teams to build and deploy agent workflows
- State & Memory Management: Build and optimize stateful dialog management and memory services for multi-turn, context-aware agents
- AgentCommunication: Implement multi-agent communication (A2A) protocols and shared context systems
- System Architecture & Execution
- Microservices Leadership: Execute the transition to cloud-native, event-driven backend services using microservices or service mesh architectures
- WorkflowDesign: Build and manage task scheduling and long-running operations using tools like Temporal or Airflow
- Performance Optimization: Optimize LLM orchestration for performance and cost through caching, batching, and token monitoring
- Integrations & Data Services
- Data Pipeline Ownership: Implement and maintain RAG pipelines and integrations with vector databases (e.g., Pinecone, Weaviate, FAISS)
- Telemetry Integration: Ensure all backend services support LangSmith-based evaluation and comprehensive observability
- Security & Reliability
- Tenant Isolation: Implement RBAC, API authentication, and multi-tenant isolation logic
- EngineeringExcellence: Set the standard for automated testing (unit, integration, load) and implement Open
- Telemetry for full system visibility
- Mentorship: Provide technical guidance and code reviews for more junior backend engineers on the team
- 6+ years of backend engineering experience 1-2+ years of hands-on experience in AI/ML platform development
- ExpertProficiency: Strong programming skills in Java and Python
- Framework Expertise: Hands-on experience with LangChain, LangGraph, or LangSmithDistributed Systems: Deep understanding of event-driven architectures, microservices, and Kubernetes
- Cloud & Data: Solid experience with AWS, Terraform, and a mix of relational (PostgreSQL) and Vector databases
- Orchestration: Practical experience with workflow engines like Temporal or Airflow and messaging systems like
- Experience with AI-driven planning systems or dialog state tracking
- Background in enterprise security (RBAC, SSO, OAuth2)
- Experience working in a high-growth SaaS or startup environment
- Active contributions to open-source backend or AI frameworks
Before you apply
Applying takes about a minute. These four things decide how fast it moves after that.
Your profile is current
It's what we read first. Occupations, seniority and locations matter more than a long history.
Two examples you can talk through
Not a portfolio — just two pieces of work where you can explain the decisions and what you'd change.
A number in mind
What you're on now and what would make you move. We negotiate better when we know both.
Your notice period
Employers plan around it, and it's the question that stalls offers most often.
Once you apply, someone reads it and calls you before anything reaches the employer — usually within two working days.
