Senior Machine Learning Engineer, Okta

FedRAMP, Customer Identity

$146-245k

Salary varies depending on location. Offers equity (where applicable) and bonus

AWS
Docker
Kubernetes
Python
Scala
Java
Airflow
Tensorflow
C++
Spark
PyTorch
FastAPI
Senior level
Remote in US

More information about location

Okta

Management platform securing resources from cloud to ground

Open for applications

Okta

Management platform securing resources from cloud to ground

1001+ employees

B2BSecurityInternal toolsSaaSIdentity

Open for applications

$146-245k

Salary varies depending on location. Offers equity (where applicable) and bonus

AWS
Docker
Kubernetes
Python
Scala
Java
Airflow
Tensorflow
C++
Spark
PyTorch
FastAPI
Senior level
Remote in US

More information about location

1001+ employees

B2BSecurityInternal toolsSaaSIdentity

Company mission

To improve the connections between people and tools to make companies more productive and secure.

Role

Who you are

  • 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
  • Additionally, the candidate will have interest in learning about and potentially contributing towards cutting edge open source AI/ML technologies
  • Bachelor's degree in Computer Science, Engineering, or a related field
  • 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
  • This position requires the ability to access federal environments and/or have access to protected federal data. As a condition of employment for this position, the successful candidate must be able to submit documentation establishing U.S. Person status (e.g. a U.S. Citizen, National, Lawful Permanent Resident, Refugee, or Asylee. 22 CFR 120.15) upon hire

Desirable

  • Master’s degrees 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

What the job involves

  • 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
  • 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
  • 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

Our take

Okta started off as an identity company where organisations, large and small, could tap into the company to have a single sign-on interface to access all cloud applications now centralised. Okta has since expanded to provide identity management at the device as well as the server level.

The new approach towards network security is increasingly based on identification and authentication, and Okta is on track to become the go-to vendor in the space. They aim to do so by developing products that enable people to access applications and other tools more smoothly and securely than ever before.

Okta has taken strides towards expansion through significant employee growth and shuffling executive leadership. The company has also made lateral moves such as tackling governance concerns, opening up new paths for revenue.

Freddie headshot

Freddie

Company Specialist

Insights

Top investors

Some candidates hear
back within 2 weeks

19% employee growth in 12 months

Company

Funding (last 2 of 7 rounds)

Sep 2015

$75m

SERIES F

Jun 2014

$75m

SERIES E

Total funding: $229.3m

Company benefits

  • Work from home opportunities
  • Health + Wellness
  • Financial Benefits
  • Pay + Incentives
  • Time Off
  • Everyday Living
  • Resources

Company values

  • Love our customers
  • Never stop innovating
  • Act with integrity
  • Be transparent
  • Empower our people

Company HQ

The East Cut, San Francisco, CA

Founders

VP of Development at Salesforce.com and has nearly a decade of experience in various engineering and leadership roles at PeopleSoft.

Frederic Kerrest

(Vice Chairman)

Worked in Sales and Business Development at Salesforce.com, and previously co-founded high-tech consulting firm Meridian Global Solutions.


People progressing

Joined as Senior Manager, Recruiting EMEA and promoted to Director, Recruiting EMEA after 3 years. Promoted again to Senior Director, Recruiting EMEA after another 3 years.

Diversity & Inclusion at Okta

  • Workforce: We believe in recruiting diverse talent to create balanced teams
  • Workplace: We support and empower employees to be authentic and grow through open conversations and education
  • Marketplace: We go to market and sell to diverse audiences

Share this job

View 68 more jobs at Okta