Okta·San Francisco, California·5 дн назад
Principal Machine Learning Engineer, SecureAI
238 000 – 326 000 $ в год
на 1% выше медианы рынка
≈ 1,7 млн–2,3 млн ₽
🏢 ОфисLeadПолная занятость
Вилки нет, про деньги придётся договариваться с нуля.
Вакансия на английскомПереведёт заголовок и описание вакансии на русский
Наша компания
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 Okta Secures AI Team
Чем предстоит заниматься
In this role, you will join the Agent Access Policies sub-team under Okta Secures AI as a Principal Machine Learning (ML) Engineer to advance Okta's authorization capabilities. By replacing static, rule-based systems with dynamic AI security mechanisms, you will deliver real-time threat inspection, agent intent evaluation, and behavioral analysis—ensuring autonomous AI agents operate safely within specified bounds
Your core mission will be architecting and driving features for our unified control plane to establish the premier Identity Security Fabric for the agentic era. In doing so, you will help execute our guiding principle: "To get AI right, you have to get identity right."
Implement intent-based enforcement to verify agent runtime requests match their intended purpose, requiring key capabilities
Integrate low-latency inference or semantic evaluation engines directly into the API gateway request path
Use embeddings, vector search, or zero-shot classification to score alignment between agent intent and executed actions
Design confidence-scored decision engines that feed semantic verification results into policy frameworks (e.g., Cedar)
Establish evaluation benchmarks, prompt injection defenses, and guardrails to prevent bypasses or false positives
Architect scalable ML and Generative AI systems integrating retrieval, inference, and evaluation pipelines
Optimize prompting, context retrieval, and RAG workflows for accuracy, safety, and efficiency in Claude-based systems
Build automated evaluation pipelines to measure model quality, correctness, groundedness, and safety in production
Implement schema validation, structured output enforcement, and guardrails to ensure reliable, compliant AI outputs
Mentor and coach engineers to support team and community growth
Supporting Your Well-Being
Driving Social Impact
Developing Talent and Fostering Connection + Community
We are intentional about connection. Our global community, spanning over 20 offices worldwide, is united by a drive to innovate. Your journey begins with an immersive, in-person onboarding experience designed to accelerate your impact and connect you to our mission and team from day one
Notice for New York City Applicants & Employees: Okta may use Automated Employment Decision Tools (AEDT), as defined by New York City Local Law 144, that use artificial intelligence, machine learning, or other automated processes to assist in our recruitment and hiring process. In accordance with NYC Local Law 144, if you are an applicant or employee residing in New York City, please click here to view our full NYC AEDT Notice
Наши требования
10+ years of software development experience, with strong programming expertise in Python (and familiarity with Go or Typescript a plus)
Hands-on experience with applied machine learning, from feature engineering to training and fine-tuning models
Hands-on experience with modern Generative AI platforms (AWS Bedrock, OpenAI, Anthropic, etc.)
Deep understanding of retrieval-augmented generation (RAG), embeddings, and knowledge-base workflows
Hands-on experience with LiteLLM, LangGraph, LangChain, LlamaIndex, MCP, or other related AI agent frameworks
Familiarity with ML frameworks (FastAPI, PyTorch, TensorFlow, Spark ML) and workflow orchestration tools (Airflow, etc.)
Experience defining evaluation metrics, pipelines, and feedback loops for ML/GenAI systems
Proven ability to collaborate with product and engineering teams to drive greenfield initiatives forward, navigate unknowns, and iterate quickly and frequently
Experience building tools or infrastructure for AI/ML applications, with a deep understanding of the developer lifecycle in an AI-native world
Experience integrating AI-driven systems with identity, authentication, or security products
Exposure to ethical AI, model risk, or compliance frameworks
Familiarity with evaluation datasets, synthetic data generation, or LLM-as-a-judge methods
Education:Bachelor’s or Master's degree in Computer Science or related field
(P25822_3552624)
Дополнительно
Below is the annual base salary range for candidates located in San Francisco Bay Area. 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
The annual base salary range for this position for candidates located in the San Francisco Bay area is between
$238,000—$326,000 USD