Principal AI ML Engineer

Kai Cyber
  • San Jose, California
    30+ days ago

    Job Description

    About the Role

    We are building an AI-powered cybersecurity platform that helps enterprises manage vulnerabilities at scale. Our AI team has delivered real, working products — services that use LLMs for natural language filtering, container image analysis, package standardization, and maintenance assessment. The AI science is strong. Now, the AI team needs dedicated engineering leadership to match.

    You will be one of the technical leaders of the AI platform engineering team. This is a hands-on architecture and leadership role. You'll set the engineering standards, design the system architecture, establish the testing and deployment practices, and build the team that takes our AI services from "it works" to "it scales, it's reliable, it's maintainable, and we can ship with confidence."

    You'll work directly with the Head of AI, the applied AI scientists, and the backend engineering team. You'll report to engineering leadership and have a seat at the table for technical strategy decisions. This is not a management role — it's a technical leadership role where you write code, review architecture, and mentor engineers while setting the direction for how AI systems get built and operated.


    Key Responsibilities

    • Audit and roadmap. Assess the current services, architecture, and deployment practices. Produce an engineering roadmap that prioritizes the highest-impact improvements (we'll give you a head start — we know where the gaps are).
    • Establish engineering foundations. Design and implement the shared library architecture — common patterns for configuration, DB access, error handling, structured logging, and health checks that all services adopt. Define the coding standards, PR review process, and quality gates.
    • Build the test infrastructure. Not just write tests but design the testing strategy: how we mock LLM APIs, how we run integration tests against staging, how we measure coverage, and how it all plugs into CI. Set the standard that the team follows.
    • Consolidate and fix CI/CD. You'll design a parameterized pipeline template, add test and lint stages, and establish the deployment strategy (staging, canary, rollback).
    • Fix critical production issues. You'll prioritize and address the highest-risk issues while establishing patterns that prevent new ones.
    • Own architecture decisions across the AI platform — API contracts, caching strategies, data flow, service boundaries, and technology choices
    • Lead the engineering hiring process for the AI team — define the roles, conduct technical interviews, and help build a high-performing team of 3-5 engineers
    • Mentor AI scientists on engineering practices — testing, code structure, version control workflows — in a way that improves their code without slowing their research velocity
    • Design evaluation and reliability frameworks for LLM-powered features — how we measure accuracy, detect regressions, and monitor production behavior
    • Partner with backend engineering to define how AI services integrate with the core cybersecurity platform — API versioning, contract testing, SLAs, and data flow
    • Translate business problems into technical architecture — work with product and AI leadership to turn ambiguous requirements into well-scoped engineering work


    Required Qualifications

    • 8+ years of professional software engineering experience, with at least 2 years in a technical leadership or staff+ role where you set standards for a team or organization
    • Deep software architecture expertise. You've designed shared libraries, defined API contracts, established coding standards, and made technology decisions that a team lived with for years. You think in terms of patterns, not just solutions
    • Production systems ownership at scale. You've been paged at 2am, you've run incident postmortems, you've built the monitoring that catches problems before customers do. You understand what production-grade actually means
    • Strong Python expertise with emphasis on clean architecture — dependency injection, proper module boundaries, testable design, async patterns done correctly
    • Testing leadership. You haven't just written tests — you've established testing culture. You've designed test strategies for systems with complex external dependencies
    • Cloud platform expertise. Deep experience with Azure (preferred) or AWS/GCP. You've designed and operated containerized microservice architectures in production
    • CI/CD and DevOps maturity. You've consolidated messy build pipelines, added quality gates, implemented deployment strategies (blue/green, canary), and established release processes
    • Mentorship and technical leadership. You've raised the engineering bar on a team. You've done code reviews that teach, not just gatekeep. You've helped junior and mid-level engineers grow. You can work with AI scientists who are domain experts but still developing engineering practices — and you can do it with empathy and patience
    • Excellent communication skills. You'll interface with AI scientists, backend engineers, product managers, and executive leadership. You need to translate between these audiences fluently


    Preferred Qualifications

    • Experience working with or alongside AI/ML teams — you understand the workflow of AI scientists and how to build infrastructure that serves their needs without constraining their exploration
    • Familiarity with LLM provider APIs (Anthropic, OpenAI, Azure OpenAI) and the engineering challenges of LLM integration (prompt management, output parsing, cost optimization, latency)
    • Experience with Azure specifically — Cosmos DB, Container Apps, Azure Identity, Azure DevOps
    • Background in cybersecurity, vulnerability management, or compliance-sensitive environments (SOC2, data privacy)
    • Track record of building engineering teams from early stage — you've been the first senior engineering hire and built the team around you
    • Experience with RAG systems, embedding pipelines, vector databases, or LLM evaluation frameworks
    • Exposure to MLOps practices — model versioning, experiment tracking, evaluation pipelines (this becomes increasingly relevant as we mature)


    Why This Role

    • Greenfield engineering leadership. You'll design the engineering foundation and set the standards from day one.
    • Direct organizational impact. You'll build and lead a team, set the technical direction, and shape how AI gets delivered to customers. Your decisions will visibly move the company.
    • AI without the hype. We're not chasing trends. We're building AI systems that solve real cybersecurity problems for real enterprises. You'll work with frontier LLMs in a domain that matters.
    • Growth into AI/ML leadership. As the platform matures, this role evolves into leading the full AI infrastructure — model serving, evaluation, fine-tuning pipelines, and MLOps. We'll invest in your growth alongside the platform.
    • Startup impact, real product. We have customers, revenue, and a working product. You're not joining to figure out product-market fit — you're joining to scale what works.


    Numbers & Facts

    LocationSan Jose, California

    Skills

    • Amazon Web Services (AWS)unmatched
    • Analysis Skillsunmatched
    • Application Programming Interface (API)unmatched
    • Architectural Designunmatched
    • Architectural Engineeringunmatched
    • Artificial Intelligence (AI)unmatched
    • Cachingunmatched
    • Cloud Computingunmatched
    • Code Reviewsunmatched
    • Coding Standardsunmatched
    • Communication Skillsunmatched
    • Computer Programmingunmatched
    • Computer Securityunmatched
    • Continuous Deployment/Deliveryunmatched
    • Continuous Integrationunmatched
    • Cost Controlunmatched
    • Dependency Injectionunmatched
    • Design Evaluationunmatched
    • DevOpsunmatched
    • Engineeringunmatched
    • Error Handlingunmatched
    • Establish Prioritiesunmatched
    • GCP (Good Clinical Practices)unmatched
    • Integration Testingunmatched
    • Internet Securityunmatched
    • Interviewing Skillsunmatched
    • Leadershipunmatched
    • Mentoringunmatched
    • Microservicesunmatched
    • Microsoft Windows Azureunmatched
    • Multiplatform/Cross-Platformunmatched
    • Problem Solving Skillsunmatched
    • Process Developmentunmatched
    • Production Controlunmatched
    • Production Systemsunmatched
    • Public/Media/Press/Analyst Relationsunmatched
    • Python Programming/Scripting Languageunmatched
    • Quality Assurance Methodologyunmatched
    • Riskunmatched
    • Service Level Agreement (SLA)unmatched
    • Software Architectureunmatched
    • Software Engineeringunmatched
    • Source Code/Configuration Management (SCM)unmatched
    • Standards Developmentunmatched
    • Startupunmatched
    • Strategic Planningunmatched
    • System Architectureunmatched
    • Team Buildingunmatched
    • Team Lead/Managerunmatched
    • Technical Leadershipunmatched
    • Technical Recruitingunmatched
    • Technical Strategyunmatched
    • Technical/Engineering Designunmatched
    • Test Designunmatched
    • Test Plan/Scheduleunmatched
    • Test Strategyunmatched
    • Testingunmatched
    • Training/Teachingunmatched

    Be found by employers

    5,500+ employers search our resume database daily. Add yours to get found by recruiters looking for candidates like you.

    Level up your application

    Professional resume templates

    Browse dozens of recruiter approved resume templates, layouts and formats. Choose your favorite and make it your own in minutes.

    Free resume templates

    Free resume builder

    Improve your existing resume or start from scratch and create a standout, ATS-friendly resume. Add job-specific content, download and apply.

    Free resume builder