Okta Logo

Okta

Senior Machine Learning Engineer, Generative AI (Customer Identity)

Sorry, this job was removed at 10:12 p.m. (CST) on Tuesday, May 06, 2025
In-Office
Chicago, IL, USA
169K-253K Annually
In-Office
Chicago, IL, USA
169K-253K Annually

Similar Jobs

51 Minutes Ago
Remote or Hybrid
Chicago, IL, USA
212K-244K Annually
Senior level
212K-244K Annually
Senior level
Artificial Intelligence • Professional Services • Business Intelligence • Consulting • Cybersecurity • Generative AI
The Anthropic Alliance Manager at PwC focuses on building partnerships, driving revenue growth, and executing marketing strategies to enhance brand visibility and client engagement. Responsibilities include relationship management, strategic planning, and team leadership to deliver on client expectations and organizational goals.
Top Skills: Microsoft Office SuiteSalesforce
52 Minutes Ago
Hybrid
Chicago, IL, USA
63K-153K Annually
Junior
63K-153K Annually
Junior
Artificial Intelligence • Professional Services • Business Intelligence • Consulting • Cybersecurity • Generative AI
Design and build scalable data infrastructure and pipelines using Azure Data Factory and Databricks. Implement data integration, transformation, modeling, and optimization. Validate data quality and security, collaborate with teams and clients, and contribute to client projects while developing technical skills and personal brand.
Top Skills: Azure Data FactoryMicrosoft Azure DatabricksSnowflake
52 Minutes Ago
Hybrid
Chicago, IL, USA
63K-140K Annually
Junior
63K-140K Annually
Junior
Artificial Intelligence • Professional Services • Business Intelligence • Consulting • Cybersecurity • Generative AI
Design and implement IAM strategies to protect sensitive data; manage user identities and access with Azure AD and CyberArk; develop RBAC and access control models; perform vulnerability assessments; apply encryption and governance; produce technical documentation; and collaborate on client-focused cybersecurity solutions while continuously learning.
Top Skills: Azure Active DirectoryCloud Identity And Access Management (Cloud Iam)CyberarkEncryption TechnologiesFederated Identity ManagementIdentity FederationIdentity-Based EncryptionMicrosoft Active DirectoryPrivileged Access Management (Pam)Role-Based Access Control (Rbac)

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 Workforce and Customer Identity Clouds enable secure yet flexible access, authentication, and automation that transforms how people move through the digital world, putting Identity at the heart 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 Auth0 Engineering Team

Auth0 is an easy-to-implement authentication and authorization platform designed by developers for developers.  We make access to applications safe, secure, and seamless for the more than 100 million daily logins around the world. Our modern approach to identity enables this Tier-Ø global service to deliver convenience, privacy, and security so customers can focus on innovation.

Machine Learning Engineer

The Machine Learning team is responsible for building ML models, leveraging Generative AI, and developing infrastructure and platform components to scale up AI/ML operations.

This Machine Learning Engineer role will help build and scale Generative AI use cases from integrating Large Language Models (LLM)  to developing reusable platform components. The ideal candidate will have a deep technical background and hands-on experience in building and deploying AI/ML systems, as well as promoting a data-driven culture across the organization. They would not hesitate to wrangle data, if necessary, and are acutely familiar with what makes an effective and efficient ML team. They will also show passion generally for the field of applied AI/ML.  Additionally, the candidate will have interest in learning about and potentially contributing towards cutting edge open source AI/ML technologies.

Responsibilities

  • Design and implement infrastructure and platform components for leveraging LLM in production.
  • Tune and optimize LLM to improve response quality and optimize performance. 
  • Build workflows and pipelines to process data from myriad sources into a knowledge base for various use cases.
  • Collaborate with platform engineering teams to ensure that AI/ML systems integrate successfully into production environments while adhering to performance and availability SLOs.
  • Participate in project planning, design, development, and code reviews.  
  • Communicate verbally and in writing to business customers and leadership teams with various levels of technical knowledge, educating them about our systems, as well as sharing insights and recommendations.
  • Partnership across Engineering, Product Management, Security and Design teams to solve technical and non-technical challenges.

Basic Qualifications

  • Fluency in a computing language, e.g. Python, Scala, C++, Java, etc.
  • Knowledge of AWS Bedrock, OpenAI or similar Generative AI platforms.
  • Experience with developing production AI/ML systems and platforms at scale, including retrieval-augmented generation (RAG) and embedding workflows. 
  • Experience with LLMOps, CI/CD, and IaC. 
  • Familiar with the full AI/ML lifecycle from model development, training, testing, deployment, monitoring, and iterating.
  • Knowledge in prompt engineering and guardrails. 
  • Excellent verbal and written communication.
  • Exceptional troubleshooting and problem solving skills, thrive in a fast-paced, innovative environment.

Preferred Qualifications

  • Bachelor's degree in Computer Science, Engineering, or a related field. 
  • Familiar with training and fine tuning models at scale with experience in LLM evaluation. 
  • Experience with ML frameworks (e.g. TensorFlow, Spark ML, PyTorch), data workflow platforms (e.g. Airflow), and container technologies (e.g. Docker, Kubernetes). 
  • Familiar with Python and AI/ML libraries such as LangChain and FastAPI. .
  • Ability to work with ambiguity, ability to self-motivate, prioritizing needs, and delivering results in a dynamic environment.
  • Combination of deep technical skills and business savvy to interface with all levels and disciplines within our and our customer’s organizations.

#LI-Remote

#LI-SH1

The annual base salary range for this position for candidates located in the San Francisco Bay area is between:

$169,000$253,000 USD

Below is the annual base salary range for candidates located in California, Colorado, New York and Washington. 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.   

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

$150,000$226,000 USD

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

Okta Chicago, Illinois, USA Office

123 N Wacker Drive, Chicago, IL, United States, 60606

What you need to know about the Chicago Tech Scene

With vibrant neighborhoods, great food and more affordable housing than either coast, Chicago might be the most liveable major tech hub. It is the birthplace of modern commodities and futures trading, a national hub for logistics and commerce, and home to the American Medical Association and the American Bar Association. This diverse blend of industry influences has helped Chicago emerge as a major player in verticals like fintech, biotechnology, legal tech, e-commerce and logistics technology. It’s also a major hiring center for tech companies on both coasts.

Key Facts About Chicago Tech

  • Number of Tech Workers: 245,800; 5.2% of overall workforce (2024 CompTIA survey)
  • Major Tech Employers: McDonald’s, John Deere, Boeing, Morningstar
  • Key Industries: Artificial intelligence, biotechnology, fintech, software, logistics technology
  • Funding Landscape: $2.5 billion in venture capital funding in 2024 (Pitchbook)
  • Notable Investors: Pritzker Group Venture Capital, Arch Venture Partners, MATH Venture Partners, Jump Capital, Hyde Park Venture Partners
  • Research Centers and Universities: Northwestern University, University of Chicago, University of Illinois Urbana-Champaign, Illinois Institute of Technology, Argonne National Laboratory, Fermi National Accelerator Laboratory

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account