SIGN IN
Staff Software Engineer - AI Engineering (Remote) jobs in United States
cer-icon
Apply on Employer Site
company-logo

KnowBe4 · 1 day ago

Staff Software Engineer - AI Engineering (Remote)

KnowBe4 is a cybersecurity company focused on redefining security awareness training through its AI-driven Human Risk Management platform. As a Staff Software Engineer, you will define the technical direction of AI systems, mentor engineers, and establish best practices for building safe and scalable AI solutions.
ComputerCyber SecurityNetwork SecuritySecuritySoftware
check
Growth Opportunities
check
H1B Sponsor Likelynote

Responsibilities

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

Qualification

AI systems architecturePythonCloud-native architectureAI/ML platform featuresDistributed systemsCI/CDInfrastructure-as-codeTechnical leadershipIncident responseObservabilityCollaborationMentoring

Required

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 CI/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

Preferred

Experience designing agentic or retrieval-based architectures at scale
Strong experience in observability, reliability, and incident response

Benefits

Company-wide bonuses based on monthly sales targets
Employee referral bonuses
Adoption assistance
Tuition reimbursement
Certification reimbursement
Certification completion bonuses

Company

KnowBe4 is a security platform for security awareness training and simulated phishing attacks.

H1B Sponsorship

KnowBe4 has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (4)
2024 (6)
2023 (5)
2022 (2)
2021 (5)
2020 (2)

Funding

Current Stage
Public Company
Total Funding
$2.15B
Key Investors
JP Morgan Chase,Kohlberg Kravis Roberts & Co.Kohlberg Kravis Roberts & Co.Goldman Sachs Growth Equity
2025-06-24Debt Financing· $1.46B
2023-06-12Debt Financing· $300M
2022-10-12Acquired

Leadership Team

leader-logo
Bryan Palma
President & Chief Executive Officer
linkedin
leader-logo
Mark Patton
Chief Technology Officer
linkedin
Company data provided by crunchbase