Impress employers and recruiters.
Choose from hundreds of resume examples.
Impress employers and recruiters.
Choose from hundreds of resume examples.
Target Resume Now
Tailor your resume to this Staff Machine Learning Engineer, SecureAI role.
Rezi rewrites your resume against Okta's job description. Free.
Tailor your resume to this Staff Machine Learning Engineer, SecureAI role.
Rezi rewrites your resume against Okta's job description. Free.
Don't guess if your resume is good enough.
See how it scores against the Staff Machine Learning Engineer, SecureAI posting at Okta — free, in seconds.
Don't guess if your resume is good enough.
See how it scores against the Staff Machine Learning Engineer, SecureAI posting at Okta — free, in seconds.
About the Role
As a Staff Machine Learning (ML) Engineer on the Agent Access Policies sub-team, you will advance Okta's authorization capabilities by replacing static, rule-based systems with dynamic AI security mechanisms. 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, ensuring autonomous AI agents operate safely within specified bounds.
Responsibilities
- Implement intent-based enforcement to verify agent runtime requests match their intended purpose.
- Apply LLM reasoning and prompt parsing to interpret prompts, tool payloads, and intent in real time.
- 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.
- 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.
Requirements
- 8+ years of software development experience.
- Strong programming expertise in Python (familiarity with Go or Typescript is 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.
Skills
- Python
- Go
- Typescript
- Machine Learning
- Generative AI
- RAG
- Embeddings
- LiteLLM
- LangGraph
- LangChain
- LlamaIndex
- MCP
- FastAPI
- PyTorch
- TensorFlow
- Spark ML
- Airflow
Location
- Canada
Work Type
- Hybrid
Experience Level
- 8+ years of software development experience
Education Level
- Bachelor’s or Master’s degree in Computer Science or related field
Salary/Compensations
- $168,000—$231,000 CAD
Benefits
- Equity
- Bonus
- Health insurance
- Dental insurance
- Vision insurance
- RRSP with a match
- Healthcare spending
- Telemedicine
- Paid leave (including PTO and parental leave)
About the Company
- Okta secures AI by building the trusted, neutral infrastructure that enables organizations to safely embrace this new era.
- The Okta for AI Agents Team is building the future of digital identity management.
- Our North Star is clear: To get AI right, you have to get identity right.
- We are building the industry's first Identity Security Fabric for the agentic era.
- We are tackling this by implementing a comprehensive four-pillar maturity model: Discover, Onboard, Protect, and Govern.
- We are driving innovation by defining the standards for Agentic Identity—including pioneer work with securing AI Agents and support for protocols like MCP.
- We are a diverse team of engineers, product managers, and designers who are bringing Okta’s expertise in identity to define the future of Agent Identity management, focusing on enablement while staying secure.
- 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.
Equal Opportunity
- 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.
- 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.