Staff AI Engineer

Posted 21 Days Ago
Be an Early Applicant
San Francisco, CA
In-Office
150K-180K
Senior level
Cloud
The Role
Lead and design Generative AI solutions, mentor junior engineers, and drive technology adoption while collaborating with cross-functional teams.
Summary Generated by Built In

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/. 

Top Skills

AWS
C/C++
Go
Java
Jax
Python
PyTorch
Rust
Am I A Good Fit?
beta
Get Personalized Job Insights.
Our AI-powered fit analysis compares your resume with a job listing so you know if your skills & experience align.

The Company
HQ: San Francisco, CA
6,000 Employees
Year Founded: 2009

What We Do

Okta is the leading independent identity provider. The Okta Identity Cloud enables organizations to securely connect the right people to the right technologies at the right time. With more than 7,000 pre-built integrations to applications and infrastructure providers, Okta provides simple and secure access to people and organizations everywhere, giving them the confidence to reach their full potential. More than 10,000 organizations, including JetBlue, Nordstrom, Siemens, Slack, T-Mobile, Takeda, Teach for America, and Twilio, trust Okta to help protect the identities of their workforces and customers.

Similar Jobs

ServiceNow Logo ServiceNow

Software Engineer

Artificial Intelligence • Cloud • HR Tech • Information Technology • Productivity • Software • Automation
Remote or Hybrid
Santa Clara, CA, USA
164K-286K Annually

ServiceNow Logo ServiceNow

Staff Software Engineer

Artificial Intelligence • Cloud • HR Tech • Information Technology • Productivity • Software • Automation
Remote or Hybrid
Santa Clara, CA, USA
164K-286K Annually

ServiceNow Logo ServiceNow

Staff Software Engineer

Artificial Intelligence • Cloud • HR Tech • Information Technology • Productivity • Software • Automation
Remote or Hybrid
Santa Clara, CA, USA
164K-286K Annually

General Motors Logo General Motors

Staff Software Engineer

Automotive • Big Data • Information Technology • Robotics • Software • Transportation • Manufacturing
Hybrid
Mountain View, CA, USA
186K-285K

Similar Companies Hiring

Jobba Trade Technologies, Inc. Thumbnail
Software • Professional Services • Productivity • Information Technology • Enterprise Web • Consulting • Cloud
Chicago, IL
45 Employees
Runpod Thumbnail
Software • Infrastructure as a Service (IaaS) • Cloud • Artificial Intelligence
Charlotte, North Carolina
80 Employees
Yooz Thumbnail
Software • Machine Learning • Fintech • Financial Services • Cloud • Automation • Artificial Intelligence
Aimargues, FR
470 Employees

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account