Okta Logo

Okta

Lead AI/ML Engineer

Reposted 11 Days Ago
Be an Early Applicant
In-Office
Bengaluru, Bengaluru Urban, Karnataka
Senior level
In-Office
Bengaluru, Bengaluru Urban, Karnataka
Senior level
Design and develop full-stack applications integrating AWS Bedrock AI solutions. Build scalable AI/ML systems, ensuring smooth integration of LLMs into applications, while collaborating with stakeholders to meet business needs.
The summary above was generated by AI

Get to know Okta
Okta is The World’s Identity Company. We free everyone to safely use any technology, anywhere, on any device or app. Our flexible and neutral products, Okta Platform and Auth0 Platform, provide secure access, authentication, and automation, placing identity at the core 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/. 

Top Skills

Angular
Aws Bedrock
Boto3
C/C++
Go
Hugging Face Transformers
Numpy
Pandas
Python
PyTorch
Rust
TensorFlow

Similar Jobs

12 Days Ago
In-Office
2 Locations
Senior level
Senior level
Logistics • Transportation
Lead AI/ML projects to develop recommendation systems, optimize data pipelines, and deliver strategic insights for supply chain execution while mentoring engineers and collaborating across teams.
Top Skills: AIAirflowBig Data TechnologiesChart.JsD3.JsData EngineeringDeep LearningDockerGitKubeflowKubernetesMachine LearningNatural Language ProcessingPower BIPythonSpark
An Hour Ago
Easy Apply
Remote or Hybrid
Bangalore, Bengaluru Urban, Karnataka, IND
Easy Apply
Mid level
Mid level
Cloud • Information Technology • Security • Software
The Account Manager will oversee a $3M+ portfolio, driving renewals and expansions while collaborating with various teams to ensure customer success and adoption of JumpCloud's solutions.
Top Skills: Crm ToolsSaaS
An Hour Ago
In-Office or Remote
Bengaluru, Karnataka, IND
Senior level
Senior level
Cloud • Information Technology • Productivity • Security • Software • App development • Automation
The Program Manager in the People Team will deliver high-impact HR programs, manage risks, and implement process improvements while collaborating with stakeholders.
Top Skills: AgileConfluenceJIRA

What you need to know about the Pune Tech Scene

Once a far-out concept, AI is now a tangible force reshaping industries and economies worldwide. While its adoption will automate some roles, AI has created more jobs than it has displaced, with an expected 97 million new roles to be created in the coming years. This is especially true in cities like Pune, which is emerging as a hub for companies eager to leverage this technology to develop solutions that simplify and improve lives in sectors such as education, healthcare, finance, e-commerce and more.

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account