Staff Software Engineer - Machine Learning

Job Duties and Responsibilities:

We are looking for experienced Software Engineers who will be passionate about Okta’s algorithmic and ML aspirations in several areas of the product. 

As the senior most engineering individual contributor (IC) within the team, you will play a crucial role in shaping and executing our AI strategy. You will work closely with Product and Data Scientists as they identify AI opportunities to determine technical feasibility and implementation plans. You will lead the design, development, and implementation of AI algorithms and ML models including problem definition, data exploration, feature engineering, model training, validation, and deployment. You will drive tech selection in the ML modeling domain. You will also mentor, guide, and set standards in several domains regarding ML & Data. You will contribute to the broader AI community by presenting at conferences, enhancing Okta’s reputation as an industry leader.

The ideal candidate will have proven expertise in the field of artificial intelligence, a passion for technology, and the ability to provide technical leadership to a diverse team. 

Must physically come into Bengaluru office minimum of 2 days a week.

Ideally, you'd have:

• 7+ years of software development experience including 3+ years of machine learning engineering experience in high velocity, high-growth companies
• Experience developing AI/ML systems at scale in production or in high-impact research environments.
• Experience with machine learning model lifecycle management tools, and an understanding of MLOps principles and best practice
• Experience in machine learning pipelines and frameworks like Airflow, SageMaker, MLFlow, Kubeflow or other similar frameworks
• Strong coding and software engineering skills, and familiarity with python
• Strong understanding of computer science fundamentals and software engineering principles around testing, code reviews and deployment.
• Cross-functional collaborator and strong communicator.
• Comfortable solving ambiguous problems and adapting to a dynamic environment

Nice to have:

• A Master’s degree or Ph.D. in Computer Science, Machine Learning, or a related quantitative field.
• Contributions to well-used open-source projects.
• Experience building ML models in TensorFlow or PyTorch
• Industry experience building end-to-end Machine Learning Platforms is a plus