AI Engineer
WRS Health
CloudPythonTensorflow
Staff Software Engineer – AI Engineering
KnowBe4
full-time
Posted on:
Location Type: Remote
Location: Remote • 🇺🇸 United States
Visit company websiteSalary
💰 $170,000 - $200,000 per year
Job Level
Lead
Tech Stack
AWSCloudFlaskPython
About the role
- 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.
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, leveraging CI/CD and infrastructure-as-code.
- 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.
Benefits
- Company-wide bonuses based on monthly sales targets
- Employee referral bonuses
- Adoption assistance
- Tuition reimbursement
- Certification reimbursement
- Certification completion bonuses
- Relaxed dress code
- Modern, high-tech, and fun work environment
Applicant Tracking System Keywords
Tip: use these terms in your resume and cover letter to boost ATS matches.
Hard skills
PythonFastAPIFlaskcloud-native architectureevent-driven systemsdistributed scalabilityCI/CDinfrastructure-as-codeAI/ML platform featuresMLOps frameworks
Soft skills
technical leadershipmentoringcollaborationdesign reviewsguiding engineering initiativesestablishing best practicesaligning teamsknowledge sharing
