Average Ratings 0 Ratings
Average Ratings 0 Ratings
Description
DiffusionGemma is an innovative open model that investigates text diffusion, representing a remarkably rapid method for generating text. Released under the Apache 2.0 license, this 26 billion parameter Mixture of Experts (MoE) model advances beyond the usual sequential token generation typical of autoregressive models. Instead, it produces entire blocks of text at once, achieving text generation speeds that are up to four times faster on GPUs. Drawing from the parameter efficiency of the Gemma 4 family and Gemini Diffusion research, DiffusionGemma incorporates a unique diffusion head that enhances generation speed significantly. It is particularly aimed at researchers and developers looking to optimize speed-sensitive, interactive local workflows, including in-line editing, swift iterations, and non-linear narrative forms. By reallocating the decode bottleneck from memory bandwidth to computational power, it can produce over 1,000 tokens per second on a single NVIDIA H100 and more than 700 tokens per second on an NVIDIA GeForce RTX 5090. This breakthrough allows for a new level of efficiency in text generation that could reshape various applications in natural language processing.
Description
This repository showcases an implementation of model parallel autoregressive transformers utilizing GPUs, leveraging the capabilities of the DeepSpeed library. It serves as a record of EleutherAI's framework designed for training extensive language models on GPU architecture. Currently, it builds upon NVIDIA's Megatron Language Model, enhanced with advanced techniques from DeepSpeed alongside innovative optimizations. Our goal is to create a centralized hub for aggregating methodologies related to the training of large-scale autoregressive language models, thereby fostering accelerated research and development in the field of large-scale training. We believe that by providing these resources, we can significantly contribute to the progress of language model research.
API Access
Has API
API Access
Has API
Integrations
Forefront
Gemini Enterprise Agent Platform
Gemma
NVIDIA NIM
ZBrain
Integrations
Forefront
Gemini Enterprise Agent Platform
Gemma
NVIDIA NIM
ZBrain
Pricing Details
Free
Free Trial
Free Version
Pricing Details
Free
Free Trial
Free Version
Deployment
Web-Based
On-Premises
iPhone App
iPad App
Android App
Windows
Mac
Linux
Chromebook
Deployment
Web-Based
On-Premises
iPhone App
iPad App
Android App
Windows
Mac
Linux
Chromebook
Customer Support
Business Hours
Live Rep (24/7)
Online Support
Customer Support
Business Hours
Live Rep (24/7)
Online Support
Types of Training
Training Docs
Webinars
Live Training (Online)
In Person
Types of Training
Training Docs
Webinars
Live Training (Online)
In Person
Vendor Details
Company Name
Founded
1998
Country
United States
Website
blog.google/innovation-and-ai/technology/developers-tools/diffusion-gemma-faster-text-generation/
Vendor Details
Company Name
EleutherAI
Founded
2020
Website
github.com/EleutherAI/gpt-neox