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
No
API Access
Has API
No
Integrations
Forefront
No
Gemini Enterprise Agent Platform
Yes
Gemma
Yes
NVIDIA NIM
Yes
ZBrain
No
Integrations
Forefront
Yes
Gemini Enterprise Agent Platform
No
Gemma
No
NVIDIA NIM
No
ZBrain
Yes
Pricing Details
Free
Free Trial
No
Free Version
Yes
Pricing Details
Free
Open source
Free Trial
No
Free Version
Yes
Deployment
Web-Based
No
On-Premises
Yes
iPhone App
No
iPad App
No
Android App
No
Windows
Yes
Mac
Yes
Linux
Yes
Chromebook
No
Deployment
Web-Based
Yes
On-Premises
Yes
iPhone App
No
iPad App
No
Android App
No
Windows
No
Mac
No
Linux
No
Chromebook
No
Customer Support
Business Hours
No
Live Rep (24/7)
No
Online Support
Yes
Customer Support
Business Hours
No
Live Rep (24/7)
No
Online Support
No
Types of Training
Training Docs
Yes
Webinars
No
Live Training (Online)
No
In Person
No
Types of Training
Training Docs
Yes
Webinars
No
Live Training (Online)
No
In Person
No
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