Lead AI/ML Engineer

Vollzeit
vor 1 Tag

Get to know OktaOkta is The World’s Identity Company. We free everyone to safely use any technology—anywhere, on any device or app. Our Workforce and Customer Identity Clouds enable secure yet flexible access, authentication, and automation that transforms how people move through the digital world, putting Identity at the heart of business security and growth. At Okta, we celebrate a variety of perspectives and experiences. We are not looking for someone who checks every single box - we’re looking for lifelong learners and people who can make us better with their unique experiences. Join our team! We’re building a world where Identity belongs to you.

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 you

Job 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 practices
Required Skills:
  • 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

Education and Certifications

  • Bachelor’s 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."

#Hybrid

#LI-SS5

What you can look forward to as a Full-Time Okta employee!

  • Amazing Benefits
  • Making Social Impact
  • Developing Talent and Fostering Connection + Community at Okta

Okta cultivates a dynamic work environment, providing the best tools, technology and benefits to empower our employees to work productively in a setting that best and uniquely suits their needs. Each organization is unique in the degree of flexibility and mobility in which they work so that all employees are enabled to be their most creative and successful versions of themselves, regardless of where they live. Find your place at Okta today! https://www.okta.com/company/careers/.Some roles may require travel to one of our office locations for in-person onboarding.

Okta is an Equal Opportunity Employer. All qualified applicants will receive consideration for employment without regard to race, color, religion, sex, sexual orientation, gender identity, national origin, ancestry, marital status, age, physical or mental disability, or status as a protected veteran. We also consider for employment qualified applicants with arrest and convictions records, consistent with applicable laws. If reasonable accommodation is needed to complete any part of the job application, interview process, or onboarding please use this Form to request an accommodation.

Okta is committed to complying with applicable data privacy and security laws and regulations. For more information, please see our Privacy Policy at https://www.okta.com/privacy-policy/.