Staff Software Engineer - AI Engineering - Remote

Overview

Join the cybersecurity company that puts security first; literally and without compromise. At KnowBe4, our AI-driven Human Risk Management platform empowers over 70,000 organizations worldwide to strengthen their security culture and transform their workforce from their biggest vulnerability into their strongest security asset.

In Short

• Define and evolve the technical vision for AI and agentic systems across products.
• Design orchestration, data, and serving patterns that handle global scale with reliability.
• Collaborate with AI Research to turn prototypes into extensible, governed production frameworks.
• Mentor engineers, lead design reviews, and guide cross-team engineering initiatives.
• Establish best practices for observability, testing, and secure, explainable AI systems.
• Align teams around shared architectural principles and long-term technical goals.
• Drive innovation responsibly, delivering systems that stay measurable, maintainable, and safe.
• Stay deeply engaged with the cybersecurity landscape, understanding emerging threats and compliance requirements.

Requirements

• 9+ years of experience designing and operating large-scale, distributed, or AI-driven systems.
• Proven technical leadership defining architecture and design patterns across multiple teams or domains.
• Expertise in Python and backend/API frameworks such as FastAPI or Flask.
• Deep understanding of cloud-native architecture (AWS preferred), event-driven systems, and distributed scalability.
• Experience building or integrating AI/ML platform features.
• A habit of turning new discoveries into reusable knowledge for the team.

Benefits

• Impact with purpose - build AI that protects people, not just data.
• Collaborative culture - small, high-ownership teams that ship fast and learn together.
• Modern stack - Bedrock, LangGraph, FastAPI, PyTorch.
• Continuous learning - budgets for certifications, conferences, and dedicated exploration time.
• Transparency and trust - radical candor, low ego, high accountability.
• Global scale - your work will help protect users across 70,000+ organizations.