Lead AI/ML Engineer

Job Description

Okta is seeking a highly skilled Full-Stack Engineer with deep expertise in AWS Bedrock, generative AI, and modern software development to join our fast-moving team at the intersection of developer experience, machine learning, and enterprise software. As part of the Okta Business Technology (BT) team, you will build cutting-edge tools that make AI development intuitive, collaborative, and scalable. If you're passionate about building next-generation AI applications and empowering developers through innovative platforms, this is the role for youJob Duties and Responsibilities Design and develop full-stack applications that integrate seamlessly with Amazon Bedrock AI agents. Build scalable, production-grade AI/ML solutions using AWS Bedrock and the AWS Agent Development Kit. Implement back-end services and APIs to interact with foundation models for tasks such as automated sourcing, content generation, and prompt orchestration. Create intuitive and performant front-end interfaces using Angular that connect with GenAI capabilities. Ensure seamless integration of LLMs and foundation models into the broader application architecture. Explore and rapidly prototype with the latest LLMs and GenAI tools to iterate on new capabilities and features. Build sophisticated AI workflows using knowledge bases, guardrails, and prompt chaining/flows. Deploy and maintain enterprise-ready GenAI applications at scale. Collaborate with business analysts to understand customer needs and use cases, and work with the team to design, develop POCs to test, implement & support solutions Foster strong relationships with teammates, customers, and vendors to facilitate effective communication and collaboration throughout the project lifecycle Perform in-depth analysis of requirements, ensuring compatibility and adherence to established standards and best practicesRequired Skills: 7+ years of robust experience with hands-on development & design experience 3+ years of experience in one or more of the following areasDeep Learning, LLMs, NLP, Speech, Conversational AI, AI Infrastructure, Fine-tuning, and optimizations of PyTorch models. Software development experience in languages like Python (must have) and one from the optional (Go, Rust, and C/C++). Experience with at least one LLM such as Llama, GPT, Claude, Falcon, Gemini, etc. Expertise in AWS Bedrock and the AWS Agent Development Kit is mandatory Hands-on experience with Python libraries including boto3, NumPy, Pandas, TensorFlow or PyTorch, and Hugging Face Transformers Solid understanding of the AWS ecosystem (e.g., CloudWatch, Step Functions, Kinesis, Lambda) Familiarity with full software development lifecycle, including version control, CI/CD, code reviews, automated testing, and production monitoring Knowledge about the ERP, HR technology and Legal business processes is an advantageEducation and Certifications Bachelors degree in Computer Science or a related field, or equivalent practical experience AWS certifications (e.g., AWS Certified Solutions Architect, Machine Learning Specialty) are a plus Experience working with large-scale generative AI or LLM-based applications Knowledge of secure application development and data privacy practices in AI/ML workloads"This role requires in-person onboarding and travel to our Bengaluru, IN office during the first week of employment."