Principal Machine Learning Infrastructure Engineer Jobs

Secure Every Identity, from AI to Human Identity is the key to unlocking the potential of AI. Okta secures AI by building the trusted, neutral infrastructure that enables organizations to safely embrace this new era. This work requires a relentless drive to solve complex challenges with real-world stakes. We are looking for builders and owners who operate with speed and urgency and execute with excellence. This is an opportunity to do career-defining work. We're all in on this mission. If you are too, let's talk. This is a hybrid role. It requires going to the local office 3 times a week. The Auth0Lab Team We are a small team of engineers exploring new Auth0 products and features ideas. We take things from 0 to 1 and we look to shape the future of identity. Our team has had a big role shaping the identity industry: we were all early at Auth0, which shaped how devs do authentication as part of Auth0Lab we incubated Auth0 FGA which redefined how authorization is done across the industry and we also incubated Auth0 for AI agents which defined auth for AI agents We are currently focused on enabling builders and companies of any size to ship production grade AI agents, by helping them with identity and security. We believe data, and developing our own models will play a huge role in this. The role We are looking for a Principal Applied AI Scientist to take product ideas from 0 to 1 and define how we do new AI product innovation. You'll have a lot of independence: you set the technical direction for AI, and experiment fast with minimal process. With that comes real ownership: you'll be the first AI/ML person in the team, so you'll often be figuring it out without a research org behind you. This is a hands-on role. Publishing papers and open sourcing results might happen as it helps us and the industry, but is not our main goal. We have a lot to teach you about security, auth, developer products and many other things. And we also want to learn from you. Join us to ha

Secure Every Identity, from AI to Human Identity is the key to unlocking the potential of AI. Okta secures AI by building the trusted, neutral infrastructure that enables organizations to safely embrace this new era. This work requires a relentless drive to solve complex challenges with real-world stakes. We are looking for builders and owners who operate with speed and urgency and execute with excellence. This is an opportunity to do career-defining work. We're all in on this mission. If you are too, let's talk. This is a hybrid role. It requires going to the local office 3 times a week. The Auth0Lab Team We are a small team of engineers exploring new Auth0 products and features ideas. We take things from 0 to 1 and we look to shape the future of identity. Our team has had a big role shaping the identity industry: we were all early at Auth0, which shaped how devs do authentication as part of Auth0Lab we incubated Auth0 FGA which redefined how authorization is done across the industry and we also incubated Auth0 for AI agents which defined auth for AI agents We are currently focused on enabling builders and companies of any size to ship production grade AI agents, by helping them with identity and security. We believe data, and developing our own models will play a huge role in this. The role We are looking for a Principal Applied AI Scientist to take product ideas from 0 to 1 and define how we do new AI product innovation. You'll have a lot of independence: you set the technical direction for AI, and experiment fast with minimal process. With that comes real ownership: you'll be the first AI/ML person in the team, so you'll often be figuring it out without a research org behind you. This is a hands-on role. Publishing papers and open sourcing results might happen as it helps us and the industry, but is not our main goal. We have a lot to teach you about security, auth, developer products and many other things. And we also want to learn from you. Join us to ha

Secure Every Identity, from AI to Human Identity is the key to unlocking the potential of AI. Okta secures AI by building the trusted, neutral infrastructure that enables organizations to safely embrace this new era. This work requires a relentless drive to solve complex challenges with real-world stakes. We are looking for builders and owners who operate with speed and urgency and execute with excellence. This is an opportunity to do career-defining work. We're all in on this mission. If you are too, let's talk. The Technology, Data, and Intelligence Team Okta is the leading independent identity provider. The Technology, Data, and Intelligence (TDI) organization is the engine that powers Okta's global workforce, providing the technology and systems that enable our employees to do their best work. As Okta continues to scale AI across the enterprise, we are seeking a highly technical, strategic, and execution-focused Principal Program Manager to lead complex, cross-functional programs across Technology, Data, and Intelligence (TDI). The Principal Program Manager, Technology, Data, and Intelligence Opportunity The initial focus of this role will be driving strategic AI programs that accelerate enterprise AI adoption, AI for Growth, and AI platform maturity. Over time, this role will support other high-priority technology initiatives across TDI as business priorities evolve. You will partner across Engineering, Product, Infrastructure, Security, Data, Finance, and Business teams to align strategy, establish scalable operating models, and drive predictable execution across large, interconnected programs. What You’ll Be Doing AI Transformation & Adoption Lead strategic AI initiatives focused on accelerating enterprise AI adoption and AI for Growth. Partner with Engineering, Product, Infrastructure, Security, Data, and Business leaders to identify, prioritize, and deliver high-impact AI capabilities. Drive adoption of AI platforms, AI agents, MCPs, developer tooling,

Secure Every Identity, from AI to Human Identity is the key to unlocking the potential of AI. Okta secures AI by building the trusted, neutral infrastructure that enables organizations to safely embrace this new era. This work requires a relentless drive to solve complex challenges with real-world stakes. We are looking for builders and owners who operate with speed and urgency and execute with excellence. This is an opportunity to do career-defining work. We're all in on this mission. If you are too, let's talk. The team The UI Platform team is the force multiplier for Okta’s entire product portfolio. By building world-class components, accessible frameworks, and developer-friendly tools, your work will directly impact millions of administrative and end-user interactions every day. You will work alongside dedicated product and engineering partners in an environment that values speed, clarity, and career-defining impact. The role The Principal Product Designer – UI Platform is a high-impact individual contributor role, serving as a foundational design lead for Okta's global user experience infrastructure. Joining the Product Platform group, you will act as a resident expert and strategic thought partner, shaping the horizontal design patterns, design architecture, localization/globalization frameworks, and accessibility standards (WCAG) that power every Okta Workforce Identity product. You will partner directly with cross functional UI Platform leads to drive platform-level initiatives, harmonizing user experience (UX) and developer experience (DevEx) across our engineering and product organizations. You will also collaborate closely with the Design Systems team, acting as a bridge between the system and feature teams applying it to our product. What you’ll be doing When everyone on the team can generate UI, the designer's role shifts. You become the person responsible for strong systems thinking, taste, quality, experience, and being the voice of the custome

Secure Every Identity, from AI to Human Identity is the key to unlocking the potential of AI. Okta secures AI by building the trusted, neutral infrastructure that enables organizations to safely embrace this new era. This work requires a relentless drive to solve complex challenges with real-world stakes. We are looking for builders and owners who operate with speed and urgency and execute with excellence. This is an opportunity to do career-defining work. We're all in on this mission. If you are too, let's talk. Overview At Okta, we're building a world where anyone can safely use any technology, and, within Okta Platform R&D, we're specifically focused on protecting the future of work. We bring simple and secure access to people and organizations everywhere, promising to not only protect the identities of our customers' workforce and users, but to ask, "what more can we make possible?" As a Senior Principal Product Design, you are the lead designer for the entire Okta platform. You ensure that all our products work together as one cohesive, simple, and secure experience. You will define the design vision, set company-wide standards, and influence high-level strategy alongside executive leadership. Where Principal Product Designers own the UX strategy within product and portfolio areas, you'll own the design architecture that connects them at the Okta-wide level, setting the innovation agenda, establishing standards across portfolios, and ensuring that Okta's products define what great identity experiences look like, both inside our walls and in the industry. This role sits at the intersection of design craft and strategic leadership. You'll work across R&D organizations, partnering with product and engineering leadership to make decisions that have broad, lasting impact on how millions of users navigate, understand, use, and trust Okta's products. Finer print This role shapes the highest-order design strategy across Okta's full product portfolio. You will l

Secure Every Identity, from AI to Human Identity is the key to unlocking the potential of AI. Okta secures AI by building the trusted, neutral infrastructure that enables organizations to safely embrace this new era. This work requires a relentless drive to solve complex challenges with real-world stakes. We are looking for builders and owners who operate with speed and urgency and execute with excellence. This is an opportunity to do career-defining work. We're all in on this mission. If you are too, let's talk. Okta is the leading independent provider of enterprise identity. Auth0 enables organizations to securely connect the right people to the right technologies at the right time. With over 7,000 pre-built integrations to applications and infrastructure providers, Okta customers can easily and securely use the best technologies for their business. Over 10,000 organizations, including 20th Century Fox, JetBlue, Nordstrom, Slack, Teach for America, and Twilio, trust Okta to help protect the identities of their workforces and customers. Auth0 is an easy-to-implement, authentication and authorization platform designed by developers for developers. We make applications’ login boxes safe, secure, and seamless for anyone logging in. Position Description : We are looking for a Principal Group Product Manager to join our team. You will work with our engineering teams to shape the future of our products. We are looking for a creative, engineering-minded individual to uncover and solve the highly complex operational realities that face our customers when they deploy high-scale, multi-tenant B2B applications on top of Auth0. This individual will help shape the next generation of our Tenancy-as-a-Service (TaaS) offeri

Secure Every Identity, from AI to Human Identity is the key to unlocking the potential of AI. Okta secures AI by building the trusted, neutral infrastructure that enables organizations to safely embrace this new era. This work requires a relentless drive to solve complex challenges with real-world stakes. We are looking for builders and owners who operate with speed and urgency and execute with excellence. This is an opportunity to do career-defining work. We're all in on this mission. If you are too, let's talk. Okta is the leading independent provider of enterprise identity. Auth0 enables organizations to securely connect the right people to the right technologies at the right time. With over 7,000 pre-built integrations to applications and infrastructure providers, Okta customers can easily and securely use the best technologies for their business. Over 10,000 organizations, including 20th Century Fox, JetBlue, Nordstrom, Slack, Teach for America, and Twilio, trust Okta to help protect the identities of their workforces and customers. Auth0 is an easy-to-implement, authentication and authorization platform designed by developers for developers. We make applications’ login boxes safe, secure, and seamless for anyone logging in. Position Description : We are looking for a Principal Group Product Manager to join our team. You will work with our engineering teams to shape the future of our products. We are looking for a creative, engineering-minded individual to uncover and solve the highly complex operational realities that face our customers when they deploy high-scale, multi-tenant B2B applications on top of Auth0. This individual will help shape the next generation of our Tenancy-as-a-Service (TaaS) offeri

Secure Every Identity, from AI to Human Identity is the key to unlocking the potential of AI. Okta secures AI by building the trusted, neutral infrastructure that enables organizations to safely embrace this new era. This work requires a relentless drive to solve complex challenges with real-world stakes. We are looking for builders and owners who operate with speed and urgency and execute with excellence. This is an opportunity to do career-defining work. We're all in on this mission. If you are too, let's talk. About the Team The Professional Services R&D team is a new, dynamic group at the forefront of innovation within Okta. Our mission is to design and build reusable, scalable assets and tools that empower our delivery teams and partners. By making customer engagements more efficient, streamlined, and cost-effective, we directly contribute to our customers' success and accelerate their time-to-value with Okta. This is a unique opportunity to join a strategic team from the ground up and shape the future of Okta's professional services. Position Summary As the Architect for the Professional Services R&D team, you will be the principal technical leader responsible for the high-level design and technical strategy of our entire portfolio of reusable assets. You will define the architectural vision and standards that ensure our solutions are robust, scalable, and secure. This role requires a blend of deep technical expertise, strategic thinking, and strong collaboration skills to guide the engineering team in building a cohesive ecosystem of tools on technologies like Java, React, Node, Python, and .NET. Responsibilities Define and own the end-to-end technical architecture for all reusable assets and solutions developed by the R&D team. Establish architectural standards, patterns, and best practices to guide the development team. Evaluate and select appropriate technologies, frameworks, and platforms to meet project requirements and long-term business goa

We are seeking a mission-driven Developer Relations Manager focused on Foundational AI Research to engage leading academic labs advancing the next generation of AI models, systems, and methods. In this role, you will work directly with top researchers building frontier AI systems, including large language models, multimodal models, reasoning systems, training methods, inference systems, model serving, and scalable AI infrastructure. You will help researchers adopt NVIDIA’s AI and accelerated computing platforms to push the boundaries of model performance, efficiency, and scale. The ideal candidate brings deep technical credibility in foundational AI, strong research engagement experience, and hands-on expertise in either AI inference research or AI training research. What you'll be doing: Serve as a trusted technical advisor to leading academic AI labs working on foundation models, LLMs, multimodal AI, reasoning, training, inference, and AI systems. Identify high-impact research workloads where NVIDIA software, systems, and accelerated computing platforms can advance model performance, scale, and efficiency. Engage principal investigators, postdocs, graduate researchers, and lab leadership to understand research goals, technical blockers, infrastructure needs, and collaboration opportunities. Track frontier AI research across papers, benchmarks, open-source projects, and academic labs to identify emerging trends and future platform opportunities. Partner with Research Account Managers, Solution Architects, Product, Engineering, and Business Development teams to support researcher adoption and long-term engagement. Represent researcher needs internally by translating academic feedback into actionable insights for product roadmaps, developer programs, education, and platform strategy. Support NVIDIA participation in major AI, ML, and systems research venues through technical content,

Secure Every Identity, from AI to Human Identity is the key to unlocking the potential of AI. Okta secures AI by building the trusted, neutral infrastructure that enables organizations to safely embrace this new era. This work requires a relentless drive to solve complex challenges with real-world stakes. We are looking for builders and owners who operate with speed and urgency and execute with excellence. This is an opportunity to do career-defining work. We're all in on this mission. If you are too, let's talk. The Okta on Okta Identity Architect Opportunity This is not a traditional architect role. At Okta, an Okta on Okta Identity Architect is the premier technical execution tier above Principal Engineer—acting as the ultimate hands-on practitioner of our own cloud platform. While you won't be writing application code, you will get your hands dirty configuring, extending, and testing the boundaries of the platform. You are the builder who proves our platform works in the real world. As organizations successfully adopt phishing-resistant MFA, attackers have shifted their tactics—increasingly deploying Adversary-in-the-Middle (AiTM) phishing kits and info-stealer malware to execute session hijacking via cookie theft. Furthermore, with the explosive rise of autonomous AI workloads, securing human identity is no longer enough; we must now protect and govern a fast-growing sprawl of non-human, agentic workflows. Are you an elite engineer running Okta at a customer organization today? Have you ever configured our platform, pushed Okta Platform Services to their limits, and thought: "If I worked at Okta, I know exactly how we could make this platform better" ? We are looking for a highly mature, technical leader who is already a deeply experienced, hands-on Okta customer practitioner to drive the technical execution of our Identity Security & Agentic Identity portfolio. In this role, you will transition from being an external customer to becoming the technical corn

Secure Every Identity, from AI to Human Identity is the key to unlocking the potential of AI. Okta secures AI by building the trusted, neutral infrastructure that enables organizations to safely embrace this new era. This work requires a relentless drive to solve complex challenges with real-world stakes. We are looking for builders and owners who operate with speed and urgency and execute with excellence. This is an opportunity to do career-defining work. We're all in on this mission. If you are too, let's talk. The Okta on Okta Identity Architect Opportunity This is not a traditional architect role. At Okta, an Okta on Okta Identity Architect is the premier technical execution tier above Principal Engineer—acting as the ultimate hands-on practitioner of our own cloud platform. While you won't be writing application code, you will get your hands dirty configuring, extending, and testing the boundaries of the platform. You are the builder who proves our platform works in the real world. As organizations successfully adopt phishing-resistant MFA, attackers have shifted their tactics—increasingly deploying Adversary-in-the-Middle (AiTM) phishing kits and info-stealer malware to execute session hijacking via cookie theft. Furthermore, with the explosive rise of autonomous AI workloads, securing human identity is no longer enough; we must now protect and govern a fast-growing sprawl of non-human, agentic workflows. Are you an elite engineer running Okta at a customer organization today? Have you ever configured our platform, pushed Okta Platform Services to their limits, and thought: "If I worked at Okta, I know exactly how we could make this platform better" ? We are looking for a highly mature, technical leader who is already a deeply experienced, hands-on Okta customer practitioner to drive the technical execution of our Identity Security & Agentic Identity portfolio. In this role, you will transition from being an external customer to becoming the technical corn

Secure Every Identity, from AI to Human Identity is the key to unlocking the potential of AI. Okta secures AI by building the trusted, neutral infrastructure that enables organizations to safely embrace this new era. This work requires a relentless drive to solve complex challenges with real-world stakes. We are looking for builders and owners who operate with speed and urgency and execute with excellence. This is an opportunity to do career-defining work. We're all in on this mission. If you are too, let's talk. Position Description Okta is an enterprise grade identity management platform, built from the ground up in the cloud and delivered with an unwavering focus on customer success. With Okta, organizations can manage access across any application, person or device. Whether the people are employees, partners or customers or the applications are in the cloud, on premises or on a mobile device, Okta helps organizations become more secure, make people more productive, and maintain compliance. The Okta platform provides directory services, single sign-on, strong authentication, provisioning, workflow, and built in reporting. It runs in the cloud on a secure, reliable, extensively audited platform and integrates deeply with on premises applications, directories, and identity management systems. We are looking for an experienced Principal Software Engineer to work on our Onboarding and Lifecycle Management (LCM) Platform team with focus on enhancing and managing services for importing, syncing and provisioning identities and access policies i.e., users, groups, roles, entitlements, etc. These features allow customers the flexibility to link and enhance their business processes with Okta’s identity management product. We are looking for a seasoned Principal Engineer to join our team and help us build the future of identity and access management. Ideal candidate should be Hands-on expert developer in Java who is deeply technical with a passion for building

Secure Every Identity, from AI to Human Identity is the key to unlocking the potential of AI. Okta secures AI by building the trusted, neutral infrastructure that enables organizations to safely embrace this new era. This work requires a relentless drive to solve complex challenges with real-world stakes. We are looking for builders and owners who operate with speed and urgency and execute with excellence. This is an opportunity to do career-defining work. We're all in on this mission. If you are too, let's talk. Okta’s Workforce Identity Cloud Security Engineering group is looking for an experienced and passionate Staff Site Reliability Engineer to join a team focused on designing and developing Security solutions to harden our cloud infrastructure. We embrace innovation and pave the way to transform bright ideas into excellent security solutions that help run large-scale, critical infrastructure. We encourage you to prescribe defense-in-depth measures, industry security standards and enforce the principle of least privilege to help take our Security posture to the next level. Our Infrastructure Security team has a niche skill-set that balances Security domain expertise with the ability to design, implement, rollout infrastructure across multiple cloud environments without adding friction to product functionality or performance. We are responsible for the ever-growing need to improve our customer safety and privacy by providing security services that are coupled with the core Okta product. This is a high-impact role in a security-centric, fast-paced organization that is poised for massive growth and success. You will act as a liaison between the Security org and the Engineering org to build technical leverage and influence the security roadmap. You will focus on engineering security aspects of the systems used across our services. Join us and be part of a company that is about to change the cloud computing landscape forever. As a Staff Engineer, you should be able

Secure Every Identity, from AI to Human Identity is the key to unlocking the potential of AI. Okta secures AI by building the trusted, neutral infrastructure that enables organizations to safely embrace this new era. This work requires a relentless drive to solve complex challenges with real-world stakes. We are looking for builders and owners who operate with speed and urgency and execute with excellence. This is an opportunity to do career-defining work. We're all in on this mission. If you are too, let's talk. Job Overview: We are looking for a Senior Engineer to join the FGA DevEx team and help evolve our end to end developer experience across both OSS and SaaS. This team owns the SDKs in Go, JavaScript, .NET, Python, Java and other languages, along with CLI workflows, IDE integrations, GitHub automation, developer documentation, and release strategy. All development is done in the open as open source, and we actively welcome and review community contributions. Our guiding principle is One developer experience, many deployment models. As a Senior Engineer, you will take ownership of significant portions of the SDK and tooling ecosystem, ensure high quality implementations across languages, and contribute to a consistent and reliable developer experience. Responsibilities: Maintain and enhance existing SDKs for FGA in Go, JavaScript, .NET, Python, and Java, leveraging our SDK generator framework. Customize and refine SDK templates and wrappers to ensure consistency across languages and support configuration overrides such as store ID, authorization model ID, headers, and parallelization limits. Implement and improve core SDK features including client credentials authentication flows, robust error mapping, retry logic with jitter, and rate limiting safeguards. Implement advanced capabilities such as BatchCheck, ListRelations, and non transactional write operations with appropriate parallelization and performance considerations. Contribute to the SDK generator tool

Secure Every Identity, from AI to Human Identity is the key to unlocking the potential of AI. Okta secures AI by building the trusted, neutral infrastructure that enables organizations to safely embrace this new era. This work requires a relentless drive to solve complex challenges with real-world stakes. We are looking for builders and owners who operate with speed and urgency and execute with excellence. This is an opportunity to do career-defining work. We're all in on this mission. If you are too, let's talk. Job Overview: We are looking for a highly experienced Staff Engineer to join the FGA DevEx team and lead the evolution of our end to end developer experience across both OSS and SaaS. This team owns the SDKs in Go, JavaScript, .NET, Python, Java and other languages, along with CLI workflows, IDE integrations, GitHub automation, developer documentation, and release strategy. All development is done in the open as open source, and we actively welcome and review community contributions. Our guiding principle is One developer experience, many deployment models. As a Staff Engineer, you will define technical direction, ensure cross language consistency, influence API design in partnership with FGA Core, and raise the quality bar across all developer facing tooling. Responsibilities: Define and drive the technical direction for FGA DevEx , CLI, IDE integrations, and developer automation across OSS and SaaS. Lead architectural decisions for multi language SDKs in Go, JavaScript, .NET, Python, and Java, leveraging and evolving the SDK generator that forms the core of all clients. Own and evolve the SDK generator framework, templates, and wrapper patterns to ensure cross language consistency, configurability, and long term maintainability. Establish standards across SDKs for authentication flows such as client credentials, error mapping and handling, retry logic with appropriate rate limiting strategies, and method level configuration overrides. Ensure advanced SDK