Staff Software Engineer - AI Engineering (Remote)

About KnowBe4 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. As the undisputed industry standard with unusually high customer retention rates and recognition as a Leader by G2 and TrustRadius, we're not just providing security awareness training - we're redefining what it means to be a trusted security partner in an increasingly complex threat landscape.Our team values radical transparency, extreme ownership, and continuous professional development in a welcoming workplace that encourages all employees to be themselves. Whether working remotely or in-person, we strive to make every day fun and engaging; from team lunches to trivia competitions to local outings, there is always something exciting happening at KnowBe4. Remote positions open to the US only. What You'll Be Building At KnowBe4, we recognize that cybersecurity has always been a human problem.Our Human Risk Management platform uses intelligent, adaptive systems to help people and AI agents recognize and respond to threats, turning human behavior from the weakest link into the strongest defense. As a Staff Software Engineer, you'll define the technical direction of our AI systems - shaping architecture, design patterns, and reliability practices across multiple product surfaces. You'll mentor engineers, lead design reviews, and set the standards for how KnowBe4 builds safe, scalable, and explainable AI.What You'll Do • Architecture: Define and evolve the technical vision for AI and agentic systems across products. • Scalability: Design orchestration, data, and serving patterns that handle global scale with reliability. • Partnership: Collaborate with AI Research to turn prototypes into extensible, governed production frameworks. • Technical Leadership: Mentor engineers, lead design reviews, and guide cross-team engineering initiatives. • Standards: Establish best practices for observability, testing, and secure, explainable AI systems.• Collaboration: Align teams around shared architectural principles and long-term technical goals. Partner closely with Product Management to align engineering designs with product requirements. • Balance: Drive innovation responsibly, delivering systems that stay measurable, maintainable, and safe. • Domain Expertise: Stay deeply engaged with the cybersecurity landscape, understanding emerging threats, evolving compliance requirements, and industry standards, to ensure our solutions directly address the real-world evolving security challenges our customers face What You Bring You're the kind of engineer who thinks in systems, not features - balancing innovation with long-term stability, and elevating the craft of AI engineering across the organization.• 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, leveraging bolthires/CD and infrastructure-as-code (we use GitLab and Terraform). • Experience building or integrating AI/ML platform features - model serving, orchestration, vector search, or MLOps frameworks.• A habit of turning new discoveries into reusable knowledge for the team. Bonus points if you have: • Experience designing agentic or retrieval-based architectures at scale. • Strong experience in observability, reliability, and incident response. The Tech You'll Work With AI & LLMs: Anthropic Claude · AWS Bedrock · OpenAI · HuggingFace · PyTorch · LangChain · LangGraph Backend & Data: Python · FastAPI · Pydantic · PySpark · Polars · Parquet · DynamoDB · S3 · PostgreSQL/MySQL Cloud & Infra: AWS Lambda · Batch · Glue · SageMaker · EventBridge · Terraform · GitLab bolthires/CD WhyThis Team • 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. We build with what's next, not what's safe. • 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. Ready to Join Us? If you're passionate about building the systems that make AI safe, scalable, and genuinely useful, come help us protect people from the threats technology alone can't solve.The base pay for this position ranges from $170,000 - $200,000, which will vary depending on how well an applicant's skills and experience align with the job description listed above. We will accept applications until 12/17/25. Our Fantastic Benefits We offer company-wide bonuses based on monthly sales targets, employee referral bonuses, adoption assistance, tuition reimbursement, certification reimbursement, and certification completion bonuses - all in a modern, high-tech, and fun work environment.For more details about our benefits in each office location, please visit www.knowbe4.com/careers/benefits. Note: An applicant assessment and background check may be part of your hiring procedure. Individuals seeking employment at KnowBe4 are considered without prejudice to race, color, religion, national origin, age, sex, marital status, ancestry, physical or mental disability, veteran status, gender identity, sexual orientation or any other characteristic protected under applicable federal, state, or local law.If you require reasonable accommodation in completing this application, interviewing, completing any pre-employment testing, or otherwise participating in the employee selection process, please visit www.knowbe4.com/careers/request-accommodation. No recruitment agencies, please. Apply tot his job