Role overview
As a Senior AI Engineer, you will be the primary architect of Cognna’s autonomous agent reasoning engine and the high-scale inference infrastructure that powers it. You are responsible for building production-grade reasoning systems that proactively plan, use tools, and collaborate. You will own the full lifecycle of our specialized security models, from domain-specific fine-tuning to architecting distributed, high-throughput inference services that serve as Security-core intelligence in our platform.
What you'll work on
- Agentic Architecture & Multi-Agent Coordination**
- Autonomous Orchestration: Design stateful, multi-agent systems using frameworks like Google ADK.
- Protocol-First Integration: Architect and scale MCP servers and A2A interfaces, ensuring a decoupled and extensible agent ecosystem.
- Cognitive Optimization: Develop lean, high-reasoning microservices for deep reasoning, optimizing context token usage to maintain high planning accuracy with minimal latency.
What we're looking for
- Experience & Education: 5-7+ years in AI/ML Engineering or Backend Systems. Must have contributed to large-scale AI/ML inference service in production. M.S. or B.S. in Computer Science/AI is preferred.
- Inference Orchestration: KV-cache management, quantization formats like AWQ/FP8, and distributed serving across multi-node GPU clusters).
- Agentic Development: Expert in building autonomous systems using Google ADK/Langgraph/Langchain and experienced with AI Observervability frameworks like LangSmith or Langfuse. Hands-on experience building AI applications with MCP and A2A protocols.
- Cloud AI Native: Proficiency in Google Cloud (Vertex AI), including custom training pipelines, high-performance prediction endpoints, and the broader MLOps suite.
- Programming: Python and experience with high-performance backends (Go/C++) for inference optimization. You are comfortable working in a Kubernetes-native environment.
- CI/CD: You are comfortable working in a Kubernetes-native environment.
*Benefits
Competitive Package** – Salary + equity options + performance incentives
Flexible & Remote – Work from anywhere with an outcomes-first culture
Team of Experts – Work with designers, engineers, and security pros solving real-world problems
Growth-Focused – Your ideas ship, your voice counts, your growth matters
Global Impact – Build products that protect critical systems and data