Downloads · 30 days
16
48% of all-time downloads
ss_d4096_f0.0039
jacobcd52/ss_d4096_f0.0039
ss_d4096_f0.0039 is a machine learning model from jacobcd52. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
Weight-sparse transformer trained with the procedure from Gao et al. (2025).
All-time downloads
33
Public
Repo size
3.4 GB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.bin3.4 GB · 100%
At a glance
- Access
- Public
- Created
- Dec 14, 2025
- Updated
- Dec 14, 2025
- SHA
- bf8d2a7f
Model card
From the Hugging Face model README
ss_d4096_f0.0039
Weight-sparse transformer trained with the procedure from Gao et al. (2025).
Model Details
- Layers: 4
- Model Dimension: 4096
- Context Length: 512
- Head Dimension: 16
- Vocabulary Size: 4096
Sparsity
- Weight Sparsity: True
- Target L0 Fraction: 0.0039
- Activation Sparsity: True
Training
- Dataset: SimpleStories/SimpleStories
- Tokenizer: SimpleStories/SimpleStories-1.25M
- Total Tokens: 2,000,000,000
Training Run
Usage
import torch
from huggingface_hub import hf_hub_download
# Download model
model_path = hf_hub_download(repo_id="jacobcd52/ss_d4096_f0.0039", filename="pytorch_model.bin")
config_path = hf_hub_download(repo_id="jacobcd52/ss_d4096_f0.0039", filename="config.json")
# Load (requires the SparseGPT model class from this repo)
state_dict = torch.load(model_path)
- Created
- Dec 14, 2025
- Updated
- Dec 14, 2025