Elastic-Cache
[ICLR 2026 π₯] Official pytorch implementation for "Attention Is All You Need for KV Cache in Diffusion LLMs"
File Explorer
- __init__.py
- configuration_dream.py
- generation_utils.py
- generation_utils_elastic.py
- modeling_dream.py
- tokenization_dream.py
- eval.py
- eval_gsm8k.sh
- eval_humaneval.sh
- postprocess_code.py
- sanitize.py
- __init__.py
- configuration_llada.py
- modeling_llada.py
- eval_gsm8k.sh
- eval_humaneval.sh
- eval_llada.py
- eval_math.sh
- eval_mbpp.sh
- generate.py
- postprocess_code.py
- sanitize.py
- Attention is all you need for KV cache in Diffusion LLMs.pdf
- architecture.png
- motivation.png
- result.png
- .gitignore
- LICENSE
- README.md
- requirements.txt
# Use via CDN
jsDelivrjsDelivr serves any public GitHub repository as a CDN with zero setup. Pick a version and a file to get a ready-to-paste link and snippet.
Command Glossary
Commands referenced in this DOCs, explained below.
git clone
View Details βΌ
git clone
Clone an existing repository.
git clone {{remote_repository_location}} {{path/to/directory}}
Clone an existing repository into a new directory (the default directory is the repository name):
git clone --recursive {{remote_repository_location}}
Clone an existing repository and its submodules:
git clone {{[-n|--no-checkout]}} {{remote_repository_location}}
Clone only the `.git` directory of an existing repository:
pip install
View Details βΌ
pip install
Install Python packages.
pip install {{package1 package2 ...}}
Install one or more packages:
pip install {{package1 package2 ...}} {{[-U|--upgrade]}}
Upgrade all specified packages to the latest version, installing any that are not already present:
pip install {{package}}=={{version}}
Install a specific version of a package:
# Project Badges
-
No description
