MGK 8 heads (small)
No estimate
No hardware requirements for this model
The weights for this model have not been published, so it cannot be downloaded or run on your own hardware at any size. It is reachable only through its provider, and no graphics card changes that.
On record
Full specification
Everything on record for this model. Most of it describes how it was trained rather than how it runs — useful context for judging how much work went into it, and how it compares with models built at a different scale.
Origin
Who built this model, where, and when it was published.
- Organisation
- FPT Software AI Center,University of California Los Angeles (UCLA),VinUniversity,Deezer Research,Rice University,University of Texas at Austin
- Organisation type
- Industry,Academia,Academia,Academia,Academia
- Country
- Vietnam, United States of America, France
- Published
- 16 October 2021
- Authors
- Tam Nguyen, Tan M. Nguyen, Dung D. Le, Duy Khuong Nguyen, Viet-Anh Tran, Richard G. Baraniuk, Nhat Ho, Stanley J. Osher
What it does
The problem areas the model was built for. A model can carry several of each.
- Domain
- Language
- Task
- Language modeling
Size
How large the model is and how much data it was trained on. Parameters are the figure that decides whether it fits on a given graphics card.
- Parameters
- 40M
- Training data
- tokens
- Epochs
- 120
"Small" setup from https://arxiv.org/pdf/2102.11174
Training compute
The arithmetic performed to train the model, measured in floating-point operations. It is a measure of what the training run cost, not of how fast the finished model answers you.
- Training compute
- 3 × 10¹⁸ FLOP
6ND formula: 6*103000000*120*40000000=2.9664e+18
The training run
What it physically took to train: which chips, how many, for how long, and what that drew from the wall.
- Training hardware
- NVIDIA A100
- Chips used
- 2
- Power draw
- 1.6 kW
Availability
Whether you can obtain the model and run it on your own hardware, which is what decides if any of the graphics-card figures on this page apply.
- Weights
- Closed — provider access only
- Model access
- Unreleased
- Training code
- Open source
code: https://github.com/minhtannguyen/transformer-mgk/blob/main/language-modeling/lmtool-fwms/README.md commercial license: https://github.com/minhtannguyen/transformer-mgk/blob/main/LICENSE
How it is classified
Labels the source dataset applies when tracking notable models, and how confident it is in the entry.
- Record confidence
- Confident
- Citations
- 39
- Benchmark data
- MGK 8 heads (small)
Sources
Where this record came from and when it was last checked.
- Reference
- Improving Transformers with Probabilistic Attention Keys
- Last updated
- 25 May 2026
What the numbers mean
About this model
MGK 8 heads (small) was published by FPT Software AI Center,University of California Los Angeles (UCLA),VinUniversity,Deezer Research,Rice University,University of Texas at Austin, in Vietnam, in October 2021. The organisation is categorised as industry,Academia,Academia,Academia,Academia.
It works in Language, and is recorded as doing language modeling.
Because the weights are not available, none of the hardware figures elsewhere on this site apply to it.
What went into building it
The training run consumed about 3 × 10¹⁸ FLOP, on NVIDIA A100. That figure describes the cost of creating it and has no bearing on how quickly it generates text.
Answers
MGK 8 heads (small) — common questions
How much compute was used to train MGK 8 heads (small)?
Around 3 × 10¹⁸ FLOP, on NVIDIA A100. That measures what producing the model cost and says nothing about how quickly it answers once trained — inference speed comes from memory bandwidth, not from the training budget.
What GPU do I need to run MGK 8 heads (small)?
None. MGK 8 heads (small) is a closed model — its weights were never published, so it cannot be downloaded or run on your own hardware at any price. It is reachable only through its provider.
Is MGK 8 heads (small) open source?
No. MGK 8 heads (small) has not had its weights published, so it exists only as a service controlled by its owner.
How many parameters does MGK 8 heads (small) have?
MGK 8 heads (small) has 40M parameters. "Small" setup from https://arxiv.org/pdf/2102.11174. That figure is the total, and it is what decides how much memory the model needs — roughly half a gigabyte per billion at the compression most people use.
Who created MGK 8 heads (small)?
MGK 8 heads (small) was published by FPT Software AI Center,University of California Los Angeles (UCLA),VinUniversity,Deezer Research,Rice University,University of Texas at Austin, based in Vietnam, categorised as industry,Academia,Academia,Academia,Academia.
When was MGK 8 heads (small) released?
MGK 8 heads (small) was published in October 2021. Capability per parameter has improved considerably since, so a newer model of the same size is often the better use of the same hardware.
What is MGK 8 heads (small) used for?
MGK 8 heads (small) works in Language, and is recorded as handling language modeling. A model can carry several of each, so these are the areas it was built for rather than a limit on what it will attempt.
The other direction
Looking at it from the other side?
This page starts from the model. If you already own a card and want to know everything it will run, start from the hardware instead.