Snr. Software Engineer - AI Engineering (Remote)

Edtech.com's Summary

KnowBe4 is hiring a Snr. Software Engineer - AI Engineering. The role involves designing and delivering scalable AI-driven systems, integrating research models into secure production services, and building data pipelines to power real-time adaptive features. The engineer will collaborate with teams to ensure performance, create reusable AI frameworks, and anticipate AI and cloud trends to evolve the technical landscape.

Highlights

Design and develop scalable AI architectures connecting models, data, and products.
Integrate AI/ML features such as RAG, orchestration, embeddings, or agentic workflows into production.
Build data pipelines and serving layers for adaptive, real-time AI capabilities.
Collaborate across engineering teams to ensure system performance and reliability.
Create reusable SDKs and libraries to support company-wide AI adoption.
Implement observability, testing, and secure-coding best practices.
Experience required: 5+ years building large-scale cloud-native or SaaS systems, preferably on AWS.
Proficient in Python and API frameworks like FastAPI or Flask.
Hands-on with CI/CD (GitLab) and infrastructure-as-code (Terraform).
Bonus skills: Experience with LLM and orchestration frameworks (LangChain, Hugging Face, PyTorch), data processing tools (PySpark, Polars, Parquet).
Compensation range: $140,000 - $180,000 base pay depending on skills and experience.

Snr. Software Engineer - AI Engineering (Remote) Full Description

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 Senior Software Engineer, youâll design and deliver AI-driven systems that power features used by millions of users around the world. This isnât prompt-tweaking or proof-of-concept work, itâs full-stack AI engineering at scale, where reliability, explainability, and security matter as much as innovation.

What Youâll Do

AI Architecture: Design and deliver scalable AI systems that connect models, data, and products.
Model Integration: Turn research prototypes into secure, reliable, production-ready services.
Data & Orchestration: Build pipelines and serving layers that power adaptive, real-time features.
Collaboration: Partner with other engineering teams to ensure performance and reliability.
Frameworks: Create reusable SDKs and libraries that accelerate AI adoption company-wide.
Excellence: Champion observability, testing, and secure-coding best practices.
Improvement: Anticipate emerging AI and cloud trends to guide technical evolution.

What You Bring

Youâre the kind of engineer who thrives on ownership - taking ideas from concept to production - and believes real innovation happens when engineering discipline meets curiosity.

5+ years of experience building and operating large-scale, cloud-native or SaaS systems (preferably in AWS).
Proficiency in Python and modern API frameworks such as FastAPI or Flask.
Hands-on experience with CI/CD pipelines (we use GitLab) and infrastructure-as-code (we use Terraform).
Practical experience delivering AI/ML features - RAG, orchestration, embeddings, or agentic workflows - in production.
Commitment to engineering quality - you design for reliability, testability, and long-term maintainability.

Bonus points if you have:

Experience with LLM and orchestration frameworks (LangChain, LangGraph, Hugging Face, PyTorch).
Familiarity with data processing and analytics tools (PySpark, Polars, or Parquet).
A habit of turning new discoveries into reusable knowledge for the team.

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

Why This 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 $140,000 - $180,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, certification completion bonuses, and a relaxed dress code - 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.

Original Job Description