Staff Software Engineer - AI Engineering (Remote)

KnowBe4

Remote regions

US

Salary range

$170,000–$200,000/yr

Benefits

Similar Jobs

See all

What You’ll Do:

  • 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.

What You Bring:

  • 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.

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

KnowBe4

KnowBe4 is a cybersecurity company that puts security first, empowering over 70,000 organizations worldwide to strengthen their security culture. They value radical transparency, extreme ownership, and continuous professional development in a welcoming workplace that encourages all employees to be themselves.

Apply for This Position