Instructions to use majentik/gemma-4-E2B-RotorQuant-MLX-2bit with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- MLX
How to use majentik/gemma-4-E2B-RotorQuant-MLX-2bit with MLX:
# Make sure mlx-vlm is installed # pip install --upgrade mlx-vlm from mlx_vlm import load, generate from mlx_vlm.prompt_utils import apply_chat_template from mlx_vlm.utils import load_config # Load the model model, processor = load("majentik/gemma-4-E2B-RotorQuant-MLX-2bit") config = load_config("majentik/gemma-4-E2B-RotorQuant-MLX-2bit") # Prepare input image = ["http://images.cocodataset.org/val2017/000000039769.jpg"] prompt = "Describe this image." # Apply chat template formatted_prompt = apply_chat_template( processor, config, prompt, num_images=1 ) # Generate output output = generate(model, processor, formatted_prompt, image) print(output) - Notebooks
- Google Colab
- Kaggle
- Local Apps
- LM Studio
Gemma 4 E2B - RotorQuant MLX 2-bit
2-bit weight-quantized MLX version of google/gemma-4-E2B with RotorQuant KV-cache quantization. Optimized for Apple Silicon inference via the MLX framework. RotorQuant delivers 5.3x faster prefill and 28% faster decode compared to TurboQuant. The most aggressive quantization, fitting the full model in the smallest possible footprint.
Approximate model size: ~0.6 GB
Model Specifications
| Property | Value |
|---|---|
| Base Model | google/gemma-4-E2B |
| Parameters | ~2 billion |
| Architecture | Dense transformer |
| Modality | Multimodal: image + text input, text output |
| License | Apache 2.0 |
| Weight Quantization | 2-bit (~0.6 GB) |
| KV-Cache Quantization | RotorQuant |
| Framework | MLX (Apple Silicon) |
Quickstart
import mlx.core as mx
from mlx_lm import load, generate
model, tokenizer = load("majentik/gemma-4-E2B-RotorQuant-MLX-2bit")
prompt = "The history of artificial intelligence began"
response = generate(model, tokenizer, prompt=prompt, max_tokens=512)
print(response)
For multimodal usage with images:
from mlx_vlm import load, generate
model, processor = load("majentik/gemma-4-E2B-RotorQuant-MLX-2bit")
prompt = "Describe the contents of this image."
output = generate(model, processor, prompt=prompt, image="path/to/image.jpg", max_tokens=512)
print(output)
What is RotorQuant?
RotorQuant is a high-performance KV-cache quantization method that achieves significantly better throughput than TurboQuant. Combined with 2-bit weight quantization in MLX, this provides maximum compression with the best available KV-cache performance: the smallest possible model footprint plus the fastest compressed KV cache for efficient long-context generation.
Key advantages over TurboQuant:
- 5.3x faster prefill
- 28% faster decode
- Equivalent memory savings
Note: 2-bit quantization is the most aggressive option and may result in some quality degradation compared to higher-precision variants. It is best suited for experimentation, rapid prototyping, or hardware-constrained environments.
KV-Cache Quantization Comparison
| Method | Prefill Speed | Decode Speed | Memory Savings | Reference |
|---|---|---|---|---|
| TurboQuant | 1x (baseline) | 1x (baseline) | High | arXiv: 2504.19874 |
| RotorQuant | 5.3x faster | 28% faster | High | GitHub |
Memory Estimates (Gemma 4 E2B)
| Precision | Approximate Size | MLX Variant |
|---|---|---|
| FP16 (original) | ~4 GB | -- |
| 8-bit quantized | ~2 GB | RotorQuant-MLX-8bit |
| 4-bit quantized | ~1.2 GB | RotorQuant-MLX-4bit |
| 2-bit quantized | ~0.6 GB | This model |
Hardware Requirements
This model requires approximately 0.6 GB of unified memory. Recommended hardware:
- Apple M1 (8 GB+)
- Apple M2 (8 GB+)
- Apple M3 (8 GB+)
- Apple M4 (8 GB+)
- Any Apple Silicon Mac with 8 GB+ unified memory
See Also
- google/gemma-4-E2B -- Base model
- majentik/gemma-4-E2B-RotorQuant -- RotorQuant KV-cache only (transformers)
- majentik/gemma-4-E2B-RotorQuant-MLX-8bit -- MLX 8-bit variant
- majentik/gemma-4-E2B-RotorQuant-MLX-4bit -- MLX 4-bit variant
- majentik/gemma-4-E2B-TurboQuant-MLX-2bit -- TurboQuant MLX 2-bit variant
- RotorQuant GitHub
- MLX Framework
Quant trade-off (MLX lane)
| Bits | Approx size | Use case | Recommendation |
|---|---|---|---|
| 2-bit | ~532 MB | Aggressive quantization | Very low-RAM Macs |
| 3-bit | ~737 MB | Lossy but small | Low-RAM Macs |
| 4-bit | ~860 MB | Balanced default | Recommended for most Macs |
| 5-bit | ~1.0 GB | Higher fidelity | Quality-sensitive |
| 6-bit | ~1.2 GB | Approaching FP16 quality | High-fidelity |
| 8-bit | ~1.5 GB | Near-lossless reference | Fidelity-critical work |
(Current variant โ 2bit โ is bolded.)
Variants in this family
(Showing 18 sibling variants under majentik/gemma4-e2b-*. The current variant โ RotorQuant-MLX-2bit โ is bolded.)
| Variant | Runtime | Approx size | Use case |
|---|---|---|---|
| RotorQuant | runtime modifier | n/a | KV-cache root (weight-agnostic) |
| RotorQuant-AWQ-4bit | transformers | ~1.2 GB | GPU 4-bit (AutoAWQ) |
| RotorQuant-AWQ-8bit | transformers | ~2.2 GB | GPU 8-bit (AutoAWQ) |
| RotorQuant-GGUF-IQ4_XS | llama.cpp | ~1.7 GB | Lossy 4-bit, low-RAM CPU/edge |
| RotorQuant-GGUF-Q2_K | llama.cpp | ~1.2 GB | Lossy, low-RAM CPU/edge |
| RotorQuant-GGUF-Q3_K_M | llama.cpp | ~1.6 GB | Smaller 3-bit, CPU-friendly |
| RotorQuant-GGUF-Q4_K_M | llama.cpp | ~2.2 GB | Balanced default |
| RotorQuant-GGUF-Q5_K_M | llama.cpp | ~2.6 GB | Higher fidelity, more RAM |
| RotorQuant-GGUF-Q8_0 | llama.cpp | ~4.2 GB | Near-lossless reference |
| RotorQuant-MLX-2bit | mlx-lm | ~655 MB | Apple Silicon, smallest |
| RotorQuant-MLX-4bit | mlx-lm | ~1.2 GB | Apple Silicon balanced |
| RotorQuant-MLX-8bit | mlx-lm | ~2.4 GB | Apple Silicon reference |
| TurboQuant | runtime modifier | n/a | KV-cache root (weight-agnostic) |
| TurboQuant-AWQ-4bit | transformers | ~1.2 GB | GPU 4-bit (AutoAWQ) |
| TurboQuant-AWQ-8bit | transformers | ~2.2 GB | GPU 8-bit (AutoAWQ) |
| TurboQuant-MLX-2bit | mlx-lm | ~655 MB | Apple Silicon, smallest |
| TurboQuant-MLX-4bit | mlx-lm | ~1.2 GB | Apple Silicon balanced |
| TurboQuant-MLX-8bit | mlx-lm | ~2.4 GB | Apple Silicon reference |
- Downloads last month
- 89
2-bit
Model tree for majentik/gemma-4-E2B-RotorQuant-MLX-2bit
Base model
google/gemma-4-E2B