Staff AI Engineer

Get to know Okta

Okta is The World’s Identity Company. We free everyone to safely use any technology, anywhere, on any device or app. Our flexible and neutral products, Okta Platform and Auth0 Platform, provide secure access, authentication, and automation, placing identity at the core of business security and growth.

At Okta, we celebrate a variety of perspectives and experiences. We are not looking for someone who checks every single box - we’re looking for lifelong learners and people who can make us better with their unique experiences. 

Join our team! We’re building a world where Identity belongs to you.

The Business Technology Team

The Business Technology (BT) team at Okta is the internal IT organization responsible for the technology infrastructure, applications, and services that empower Okta employees globally. We partner across the business to deliver solutions that drive productivity, collaboration, and innovation. Effective financial management and operational discipline are critical to our success in enabling Okta's rapid growth and ensuring we operate efficiently.

The Staff AI Engineer Opportunity

Do you love building and pioneering in the technology space? Do you enjoy solving complex business problems in a fast-paced, collaborative, inclusive, and iterative delivery environment? We are looking for a highly experienced and influential Staff GenAI Engineer to lead the design and implementation of advanced Generative AI solutions within our Agile framework. You will be instrumental in shaping our technical direction, mentoring junior engineers, and driving innovation across complex projects.. This position requires working with many cross functional teams and applications. If you thrive on solving hard and complex problems, then this could be a great fit for you.You will need to be a self-starter, ready to dig in and own projects, collaborate cross-functionally, and deliver quality results. The work is challenging, but you can count on a deep level of support from your team.

What you’ll be doing 

• Lead the design, development, testing, implementation, and support of complex Generative AI solutions across full-stack environments.
• Drive the adoption of new technologies, continuously experimenting with and integrating cutting-edge tech trends.
• Actively mentor other members of the engineering community, fostering growth and best practices, and participate in external technology communities.
• Partner strategically with business analysts to translate complex customer needs and use cases into technical designs, leading the development of innovative POCs from conception to deployment.
• Cultivate and strengthen relationships with teammates, customers, and vendors, acting as a key technical liaison.
• Oversee and define standards for comprehensive documentation of requirements and architectural designs for custom solutions.
• Perform advanced, in-depth analysis of complex requirements, defining compatibility standards and enforcing architectural best practices.
• Own the management, monitoring, troubleshooting, and optimization of existing functionalities, proactively addressing and resolving critical issues.

What you’ll bring to the role

• 8+ years of robust, hands-on development and design experience in software engineering.
• 3+ years of in-depth experience in multiple areas such as Deep Learning, LLMs, NLP, Speech, Conversational AI, AI Infrastructure, Fine-tuning, and optimizations of PyTorch models.
• Expert-level proficiency in Python, with strong experience in other languages like Java, Go, Rust, or C/C++.
• Extensive experience with multiple Large Language Models (LLMs) (e.g., Llama, GPT, Claude, Falcon), understanding their architectures, limitations, and application.
• Deep expertise in at least one deep learning framework (PyTorch and/or JAX), including advanced techniques for model training, optimization, and deployment.
• Proven experience designing and deploying scalable, high-performance cloud solutions for AI, preferably on AWS.
• Demonstrated ability to present complex technical concepts effectively to both technical and executive business audiences, influencing decision-making.
• Proven thought leadership, significantly influencing the design, implementation, and ongoing support of critical business POCs.
• Experience coaching and developing individuals, and working effectively with geographically dispersed teams, is a strong plus.
• Demonstrated ability to quickly master and apply unfamiliar technologies and complex concepts.

#LI-BF1

#LI-Hybrid

What you can look forward to as a Full-Time Okta employee!

• Amazing Benefits
• Making Social Impact
• Developing Talent and Fostering Connection + Community at Okta

Okta cultivates a dynamic work environment, providing the best tools, technology and benefits to empower our employees to work productively in a setting that best and uniquely suits their needs. Each organization is unique in the degree of flexibility and mobility in which they work so that all employees are enabled to be their most creative and successful versions of themselves, regardless of where they live. Find your place at Okta today! https://www.okta.com/company/careers/.

Some roles may require travel to one of our office locations for in-person onboarding.

Okta is an Equal Opportunity Employer. All qualified applicants will receive consideration for employment without regard to race, color, religion, sex, sexual orientation, gender identity, national origin, ancestry, marital status, age, physical or mental disability, or status as a protected veteran. We also consider for employment qualified applicants with arrest and convictions records, consistent with applicable laws.

If reasonable accommodation is needed to complete any part of the job application, interview process, or onboarding please use this Form to request an accommodation.

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