• >
  • >
  • >
  • >
  • >
Google > Case Studies > AI21 Labs: Rewriting the rules on natural language processing with Google Cloud

AI21 Labs: Rewriting the rules on natural language processing with Google Cloud

Google Logo
Customer Company Size
Startup
Region
  • Middle East
Country
  • Israel
Product
  • Wordtune
  • AI21 Studio
  • Jurassic-1 Jumbo
Tech Stack
  • Google Cloud
  • Google Kubernetes Engine
  • BigQuery
  • Looker Studio
  • Docker
Implementation Scale
  • Enterprise-wide Deployment
Impact Metrics
  • Digital Expertise
  • Productivity Improvements
Technology Category
  • Analytics & Modeling - Machine Learning
  • Application Infrastructure & Middleware - API Integration & Management
  • Platform as a Service (PaaS) - Data Management Platforms
Applicable Industries
  • Software
Applicable Functions
  • Product Research & Development
  • Business Operation
Services
  • Cloud Planning, Design & Implementation Services
  • Software Design & Engineering Services
  • System Integration
About The Customer
AI21 Labs is a technology startup founded in 2017, based in Israel, with a mission to enhance humanity's unique ability to create and understand written language. The company uses neural networks powered by vast amounts of data and powerful computing to enable machines to understand written language and build tools that complement human skills. AI21 Labs is focused on reimagining the way humans read and write, leveraging Natural Language Processing (NLP) to give machines the ability to understand and generate natural text. The company views NLP as a revolutionary tool, aiming to empower people to become better versions of their writing and reading selves. AI21 Labs has developed proprietary NLP algorithms and a suite of language models, including the state-of-the-art Jurassic-1 Jumbo, which comprises 178 billion parameters of data. The company offers both B2C and B2B products, including Wordtune, a writing companion, and AI21 Studio, a developer platform providing an API and interactive web environment for querying language models.
The Challenge
AI21 Labs, founded in 2017, aims to revolutionize the way humans read and write by leveraging Natural Language Processing (NLP) to enable machines to understand and generate natural text. The company views NLP as a tool as revolutionary as Gutenberg’s printing press, aiming to empower people to become better versions of their writing and reading selves. However, as a startup among giants, AI21 Labs needed to adopt the best technology available to achieve its ambitions. The company identified Google Cloud as an ideal partner due to its wide range of advanced machine learning accelerators. AI21 Labs needed a robust infrastructure to build proprietary NLP algorithms and serve its suite of language models, including the state-of-the-art Jurassic-1 Jumbo, which comprises 178 billion parameters of data. Additionally, the company required a flexible and scalable solution to meet the demands of different audiences, including both B2C and B2B offerings.
The Solution
AI21 Labs partnered with Google Cloud to leverage its advanced machine learning accelerators and robust infrastructure to build and serve its suite of language models. The company uses a variety of ML accelerators, including high-performance GPUs, to develop its proprietary NLP algorithms. Google Cloud's flexible price-performance options allowed AI21 Labs to choose the most appropriate GPU for each task, enabling the development of both B2C and B2B offerings. AI21 Labs uses Google Kubernetes Engine for its production environment, abstracting functionality to Docker container images, which are used across the company for internal research projects and commercial products. The company also uses Cloud Build for continuous integration pipelines, where code is pushed into Git repositories, built, tested, and deployed via Docker images and internal Python packages. Additionally, AI21 Labs uses BigQuery to batch logging data, providing insights into user usage and benefits gained from its products. Looker Studio is used as a visualization tool on top of BigQuery to analyze data from marketing platforms and payment providers, connecting the dots between spend and revenue.
Operational Impact
  • AI21 Labs has successfully built proprietary NLP algorithms and a suite of language models, including the state-of-the-art Jurassic-1 Jumbo, comprising 178 billion parameters of data.
  • The company has developed both B2C and B2B offerings, including Wordtune, a writing companion, and AI21 Studio, a developer platform providing an API and interactive web environment for querying language models.
  • AI21 Labs uses Google Cloud's advanced machine learning accelerators and robust infrastructure to meet the demands of different audiences, providing flexible and scalable solutions.
  • The company uses BigQuery to batch logging data, providing insights into user usage and benefits gained from its products, and Looker Studio as a visualization tool to analyze data from marketing platforms and payment providers.
  • AI21 Labs has a strong partnership with Google Cloud, enabling the team to build the tools they need while improving specific workflows and achieving reliable and reproducible results in the Google Cloud environment.
Quantitative Benefit
  • Scales by a factor of 100 the number of A100 GPUs running within a day.
  • Provides flexible price-performance choice, between 7x higher cost and 2x better performance with GKE changing architecture.
  • Allows saving of up to 20% due to variation in system load with no performance tradeoffs with Elasticity.

Case Study missing?

Start adding your own!

Register with your work email and create a new case study profile for your business.

Add New Record

Related Case Studies.

Contact us

Let's talk!
* Required
* Required
* Required
* Invalid email address
By submitting this form, you agree that AGP may contact you with insights and marketing messaging.
No thanks, I don't want to receive any marketing emails from AGP.
Submit

Thank you for your message!
We will contact you soon.