AI Infrastructure Architect

About the role

We are looking for a smart and versatile AI Infrastructure Architect to build and evolve the AI infrastructure and platform that powers our identity security solutions. Your work will enable internal teams and product groups to integrate AI capabilities safely, securely, and at scale—empowering Okta's mission to protect millions of digital identities worldwide.

Key Responsibilities

• Lead AI enablement initiatives, including proof-of-concepts for emerging AI infrastructure technologies and integration approaches.
• Collaborate cross-functionally with engineering, security, data science, and product teams to align AI platform architecture with business and security goals.
• Architect scalable, resilient, and secure AI infrastructure that supports AI-powered tools and features across Okta’s Identity Platform.
• Develop and maintain infrastructure-as-code frameworks (e.g., Terraform, Helm) to ensure consistent, reproducible deployment of AI services.
• Champion security and compliance by embedding data privacy and identity protection standards directly into the AI platform and infrastructure design.
• Serve as the key advocate and strategist for AI-driven efficiency initiatives across infrastructure platform teams and pre-production systems.
• Implement robust MLOps practices, such as model evaluation, rollback strategies, and A/B testing, to guarantee the reliability and governance of AI in production.
• Drive continuous innovation by staying current with AI and cloud infrastructure trends and evangelizing best practices internally.

Required

• 10+ years in infrastructure or software engineering, with ≥ 2 years building AI/ML systems.
• Exceptional systems level thinking and a track record in architecting and building enterprise grade infrastructure.
• Deep expertise in cloud platforms (AWS, GCP), distributed systems, and container orchestration (Kubernetes).
• Expected to be very hands-on in order to create, review, and contribute large chunks of quality code.

Preferred

• Experience in identity, security, fraud, or risk analytics domains.
• Experience operationalizing large language models or foundation models in production environments.
• Contributions to MLOps or infrastructure open-source projects.

Qualifications

• BS/MS in Computer Science, Engineering, or related field.
• Strong understanding of cloud platforms (AWS, GCP), distributed systems, and container orchestration (Kubernetes).
• Experience with infrastructure-as-code (IaC) tools like Terraform, Ansible, or Helm.
• Hands-on experience with AI/ML frameworks and tools.
• Experience with MLOps and DevOps practices.
• Excellent problem-solving and communication skills.

Benefits

Your actual base salary will depend on factors such as your skills, qualifications, experience, and work location. In addition, Okta offers equity (where applicable), bonus, and benefits, including health, dental and vision insurance, 401(k), flexible spending account, and paid leave (including PTO and parental leave) in accordance with our applicable plans and policies. To learn more about our Total Rewards program please visit: https://rewards.okta.com/us.

Pay

The annual base salary range for this position for candidates located in California (excluding San Francisco Bay Area), Colorado, Illinois, New York, and Washington is between: $235,000—$353,000 USD

Schedule

Full-time

Benefits

Okta is committed to complying with applicable data privacy and security laws and regulations. For more information, please see our Personnel and Job Candidate Privacy Notice at https://www.okta.com/legal/personnel-policy/.