Staff Full Stack AL/ML Engineer, Okta

• 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
• If you're passionate about building next-generation AI applications and empowering developers through innovative platforms, this is the role for you
• 7+ years of robust experience with hands-on development & design experience
• 3+ years of experience in one or more of the following areas: Deep 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 advantage

Desirable

• As part of the Okta Business Technology (BT) team, you will build cutting-edge tools that make AI development intuitive, collaborative, and scalable
• 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 practices

• Some roles may require travel to one of our office locations for in-person onboarding