gemma-4-E2B-it-litert-lm

By litert-community

🎯 Task: Uncategorized⚖️ apache-2.0

Model Card

litert-community/gemma-4-E2B-it-litert-lm

Main Model Card: google/gemma-4-E2B-it

This model card provides the Gemma 4 E2B model in a way that is ready for deployment on Android, iOS, Desktop, IoT and Web.

Gemma is a family of lightweight, state-of-the-art open models from Google, built from the same research and technology used to create the Gemini models. This particular Gemma 4 model is small so it is ideal for on-device use cases. By running this model on device, users can have private access to Generative AI technology without even requiring an internet connection.

These models are provided in the .litertlm format for use with the LiteRT-LM framework. LiteRT-LM is a specialized orchestration layer built directly on top of LiteRT, Google’s high-performance multi-platform runtime trusted by millions of Android and edge developers. LiteRT provides the foundational hardware acceleration via XNNPack for CPU and ML Drift for GPU. LiteRT-LM adds the specialized GenAI libraries and APIs, such as KV-cache management, prompt templating, and function calling. This integrated stack is the same technology powering the Google AI Edge Gallery showcase app.

LiteRT-LM uses a state of the art Gemma-4 mobile quantization scheme that uses a mixture of 2bit, 4bit and 8 bit weights. This means that for text only use cases the weight footprint in memory can be as low as 0.8 GB while the runtime uses memory mapping to support the 1.12GB of embedding parameters. This approach gives significant working memory savings on some platforms as seen in the more detailed data below. Additionally the Vision and Audio models are loaded on demand to further reduce memory consumption.

Try Gemma 4 E2B

Build with Gemma 4 E2B and LiteRT-LM

Ready to integrate this into your product? Get started here.

Gemma 4 E2B Performance on LiteRT-LM

All benchmarks were taken using 1024 prefill tokens and 256 decode tokens with a context length of 2048 tokens via LiteRT-LM. The model can support up to 32k context length. The inference on CPU is accelerated via the LiteRT XNNPACK delegate with 4 threads. Time-to-first-token does not include load time. Benchmarks were run with caches enabled and initialized. During the first run, the latency and memory usage may differ. Model size is the size of the file on disk.

CPU memory was measured using, rusage::ru_maxrss on Android, Linux and Raspberry Pi, task_vm_info::phys_footprint on iOS and MacBook and process_memory_counters::PrivateUsage on Windows.

Android

Note: On supported Android devices, Gemma 4 is available through Android AI Core as Gemini Nano, which is the recommended path for production applications.

Device                                     BackendPrefill (tokens/sec)Decode (tokens/sec)Time-to-first-token (sec)Model size (MB)CPU Memory (MB)
S26 UltraCPU55746.91.825831733
S26 UltraGPU3,80852.10.32583676

🚨 NEW: Android with Speculative Decoding 🚨

The numbers in this section include speculative decoding. Speculative decoding is an optimization that accelerates LLMs by using a small, fast "draft" model to quickly predict multiple upcoming tokens, while a larger “target” model then verifies those tokens in parallel. The effectiveness of speculative decoding is task dependent because the “draft” model can more easily predict the correct tokens of some tasks. The metrics in this section were collected from a variety of sample prompts and grouped into categories by task type. The baseline measurements are an average across all task types. The number of input and output tokens varied across prompts. Note that if you download this model before May 5, 2026, you should re-download the model if you want to use speculative decoding. Speculative decoding is available on CPU and GPU on Mobile and Desktop.

Device                                     BackendTask TypeSpeculative Decoding?Decode (tokens/sec)CPU Memory (MB)
S26 UltraCPUBaselineNo40.71362
S26 UltraCPUSummarize textYes47.51582
S26 UltraCPUCode snippetYes36.31440
S26 UltraCPURewrite toneYes47.11408
S26 UltraCPUFree formYes38.11459
S26 UltraGPUBaselineNo51.5791
S26 UltraGPUSummarize textYes91.7817
S26 UltraGPUCode snippetYes84.4788
S26 UltraGPURewrite toneYes87.4762
S26 UltraGPUFree formYes66.5804

iOS

Device                                     BackendPrefill (tokens/sec)Decode (tokens/sec)Time-to-first-token (sec)Model size (MB)CPU/GPU Memory (MB)
iPhone 17 ProCPU53225.01.92583607
iPhone 17 ProGPU2,87856.50.325831450

Linux

Device                                     BackendPrefill (tokens/sec)Decode (tokens/sec)Time-to-first-token (sec)Model size (MB)CPU Memory (MB)
Arm 2.3 & 2.8GHzCPU26035.04.025831628
NVIDIA GeForce RTX 4090GPU11,234143.40.12583913

macOS

Device                                     BackendPrefill (tokens/sec)Decode (tokens/sec)Time-to-first-token (sec)Model size (MB)CPU/GPU Memory (MB)
MacBook Pro M4 MaxCPU90141.61.12583736
MacBook Pro M4 MaxGPU7,835160.20.125831623

Windows

Device                                     BackendPrefill (tokens/sec)Decode (tokens/sec)Time-to-first-token (sec)Model size (MB)CPU Memory (MB)
Intel LunarLakeCPU43529.82.3925833505
Intel LunarLakeGPU3,75148.40.2925833540

Web

Device                                     BackendPrefill (tokens/sec)Decode (tokens/sec)Time-to-first-token (sec)Model size (MB)GPU Memory (MB)
Macbook Pro M4 MaxWebGPU4,853731.092008~1800
  • Web on LiteRT-LM uses a specially optimized model for Web because of its unique memory constraints. Currently the model is text-only.

IoT

Device                                     BackendPrefill (tokens/sec)Decode (tokens/sec)Time-to-first-token (sec)Model size (MB)CPU Memory (MB)
Raspberry Pi 5 16GBCPU1337.67.825831546
Jetson Orin NanoCPU10912.29.425833681
Jetson Orin NanoGPU1,14224.20.925832739
Qualcomm Dragonwing IQ8 (IQ-8275)NPU3,74731.70.329671869
  • NPU model is benchmarked with 4096 context length

Running Gemma 4 E2B on Web with MediaPipe

You can also run Gemma through MediaPipe LLM Inference Engine. However, this route is currently in maintenance mode. To add it to your existing MediaPipe flow, download the gemma-4-E2B-it-web.task model file and run with our sample web page, or follow the guide to add it to your own app.

Architecture & Tags

litert-lmbase_model:google/gemma-4-E2B-itbase_model:finetune:google/gemma-4-E2B-itregion:us