Taro Logo

Lead AI/ML Engineer

The leading independent provider of identity for the enterprise, enabling secure connections between people and technology.
Machine Learning
Staff Software Engineer
Hybrid
5,000+ Employees
7+ years of experience
AI · Enterprise SaaS

Description For Lead AI/ML Engineer

Okta, The World's Identity Company, is seeking a Lead AI/ML Engineer to join their Business Technology team in Bengaluru. This role sits at the intersection of developer experience, machine learning, and enterprise software, focusing on building cutting-edge AI tools and applications. The position requires expertise in AWS Bedrock, generative AI, and modern software development practices.

The role involves developing full-stack applications that integrate with Amazon Bedrock AI agents, implementing sophisticated AI workflows, and creating scalable solutions. You'll work with state-of-the-art LLM technologies while building enterprise-ready applications that make AI development intuitive and collaborative.

As a Lead AI/ML Engineer, you'll be responsible for designing and implementing both backend services and frontend interfaces, ensuring seamless integration of foundation models into the broader application architecture. The role requires strong technical skills in Python, AWS services, and modern AI/ML frameworks, along with the ability to collaborate effectively with business stakeholders.

The ideal candidate brings 7+ years of development experience, with at least 3 years focused on AI/ML technologies. Deep knowledge of AWS Bedrock, the AWS Agent Development Kit, and experience with major LLM platforms is essential. You'll be joining a company that values innovation, security, and identity management, working on projects that directly impact how organizations interact with AI technology.

This hybrid role offers comprehensive benefits, professional development opportunities, and the chance to work with cutting-edge AI technologies while contributing to Okta's mission of enabling everyone to safely use any technology. The position requires initial in-person onboarding in Bengaluru, making it an excellent opportunity for those looking to make a significant impact in the AI/ML space while working for a leading identity and security company.

Last updated 6 hours ago

Responsibilities For Lead AI/ML Engineer

  • Design and develop full-stack applications that integrate with Amazon Bedrock AI agents
  • Build scalable, production-grade AI/ML solutions using AWS Bedrock and AWS Agent Development Kit
  • Implement back-end services and APIs for tasks like automated sourcing and content generation
  • Create intuitive front-end interfaces using Angular that connect with GenAI capabilities
  • Ensure seamless integration of LLMs and foundation models into application architecture
  • Build sophisticated AI workflows using knowledge bases and prompt chaining
  • Deploy and maintain enterprise-ready GenAI applications at scale
  • Collaborate with business analysts to understand customer needs and develop solutions
  • Foster strong relationships with teammates, customers, and vendors

Requirements For Lead AI/ML Engineer

Python
Go
Rust
  • 7+ years of robust experience with hands-on development & design experience
  • 3+ years experience in Deep Learning, LLMs, NLP, or AI Infrastructure
  • Software development experience in Python (required) and Go/Rust/C/C++
  • Experience with LLMs such as Llama, GPT, Claude, Falcon, Gemini
  • Expertise in AWS Bedrock and AWS Agent Development Kit
  • Experience with Python libraries including boto3, NumPy, Pandas, TensorFlow/PyTorch
  • Solid understanding of AWS ecosystem
  • Bachelor's degree in Computer Science or related field
  • AWS certifications are a plus

Benefits For Lead AI/ML Engineer

Medical Insurance
Visa Sponsorship
  • Comprehensive benefits package
  • Social impact opportunities through Okta for Good
  • Professional development and career growth
  • Flexible work arrangements

Interested in this job?

Jobs Related To Okta Lead AI/ML Engineer