Gemma 4 E2B Distilled + LoRA R8

Experimental benchmark-recovery model. It adds native rank-8, alpha-16 LoRA residuals to all 105 Monarch MLP projections and loads without PEFT. The model has 3,691,669,024 parameters, including 9,400,320 LoRA parameters.

1. Weight Storage

Model Status Loaded weights Serialized weights Reduction vs dense BF16
Dense Gemma 4 BF16 Reference 9.507 GiB 9.543 GiB -
Distilled Gemma 4 BF16 Released 6.859 GiB 6.859 GiB 27.86%
Distilled Gemma 4 + LoRA r8 BF16 Experimental 6.876 GiB 6.877 GiB 27.68%
Distilled Gemma 4 + INT8 linears Released 6.135 GiB 6.136 GiB 35.48%

BF16 sizes use two bytes per parameter; serialized LoRA size was audited from the two safetensor shards. Values exclude activations and CUDA workspaces.

2. TinyHellaSwag

All rows used the same RTX PRO 6000, fixed batch size 32, seed 1234, 100 official examples, 10-shot prompts, and no chat template.

Model Status GP-IRT accuracy Raw accuracy Runtime Peak VRAM
Dense Gemma 4 BF16 Reference 39.23% 29% 28.50 s 57.88 GiB
Distilled Gemma 4 BF16 Released 32.35% 22% 23.33 s 55.23 GiB
Distilled Gemma 4 + LoRA r8 BF16 Experimental 33.25% 23% 24.93 s 55.25 GiB
Distilled Gemma 4 + INT8 linears Released 30.57% 21% 23.14 s 54.51 GiB

Rank-8 LoRA improved the 35-layer source by 0.90 GP-IRT points and one correct item, but missed the former +1.0 point or +2 item release gate. The 100-example benchmark is noisy.

Usage

from transformers import AutoModelForImageTextToText, AutoProcessor

model_id = "hexoy/gemma-4-e2b-monarch-35mlp-lora-r8"
processor = AutoProcessor.from_pretrained(model_id, trust_remote_code=True)
model = AutoModelForImageTextToText.from_pretrained(
    model_id,
    trust_remote_code=True,
    dtype="auto",
    device_map="auto",
)

The standalone export passed exact LoRA state and BF16 logit round trips plus a clean-cache load. Corrected 64-token generation repeated phrases for text and produced malformed or inaccurate answers on all three image tests. Treat this as a benchmark result, not a general text or multimodal recovery release.

Reproducibility

Detailed training and release evidence is retained privately.

This repository contains custom modeling code. Review it before enabling trust_remote_code=True. This is an experimental derivative, not an official Google release.

Downloads last month
52
Safetensors
Model size
4B params
Tensor type
BF16
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Model tree for hexoy/gemma-4-e2b-monarch-35mlp-lora-r8

Adapter
(1)
this model