LICENSE
MANIFEST.in
NOTICE
README.md
pyproject.toml
setup.cfg
keys_vals.egg-info/PKG-INFO
keys_vals.egg-info/SOURCES.txt
keys_vals.egg-info/dependency_links.txt
keys_vals.egg-info/entry_points.txt
keys_vals.egg-info/requires.txt
keys_vals.egg-info/top_level.txt
keys_values/__init__.py
keys_values/__main__.py
keys_values/adapter.py
keys_values/adapter_v2.py
keys_values/array_limit.py
keys_values/attention.py
keys_values/attention_utils.py
keys_values/flex_attention.py
keys_values/flex_attention_old.py
keys_values/gpu_memory.py
keys_values/head_model.py
keys_values/head_model_factory.py
keys_values/long_context.py
keys_values/lora.py
keys_values/model.py
keys_values/parser_config.py
keys_values/pos_encoding.py
keys_values/sdpa_wrapper.py
keys_values/sdpa_wrapper_old.py
keys_values/use_eager_kernel.py
keys_values/utils.py
keys_values/version
keys_values/data/__init__.py
keys_values/data/base.py
keys_values/data/dataloader.py
keys_values/data/evaluation.py
keys_values/data/iterators.py
keys_values/data/load_helmet_dev_eval.py
keys_values/data/longbench_v2.py
keys_values/data/sequence_classification.py
keys_values/data/sft_dataset.py
keys_values/finetune/__init__.py
keys_values/finetune/args.py
keys_values/finetune/batch_transform.py
keys_values/finetune/dataset_token_counts.py
keys_values/finetune/longcon_offload_full.py
keys_values/finetune/longcon_offload_lora.py
keys_values/finetune/longcontext_eval.py
keys_values/finetune/longcontext_full.py
keys_values/finetune/longcontext_lora.py
keys_values/finetune/utils.py
keys_values/generate/__init__.py
keys_values/generate/base.py
keys_values/kvcache/__init__.py
keys_values/kvcache/attn_weights.py
keys_values/kvcache/base.py
keys_values/kvcache/basics.py
keys_values/kvcache/buffers.py
keys_values/kvcache/factory.py
keys_values/kvcache/h2o.py
keys_values/kvcache/qh2o.py
keys_values/kvcache/quant_buffers.py
keys_values/kvcache/stack_layers.py
keys_values/kvcache/test_before_after.py
keys_values/kvcache/test_utils.py
keys_values/kvcache/test_utils_advanced.py
keys_values/kvcache/utils.py
keys_values/kvcache/gradient/__init__.py
keys_values/kvcache/gradient/accumulate.py
keys_values/kvcache/gradient/annotation.py
keys_values/kvcache/gradient/autograd_hooks.py
keys_values/kvcache/gradient/cell.py
keys_values/kvcache/gradient/checkpoints.py
keys_values/kvcache/gradient/cleanup.py
keys_values/kvcache/gradient/inference_replay.py
keys_values/kvcache/gradient/main.py
keys_values/kvcache/gradient/sdpa_op.py
keys_values/kvcache/gradient/train_attn_weights_replay.py
keys_values/kvcache/gradient/train_attn_weights_replay_old.py
keys_values/kvcache/quantize/__init__.py
keys_values/kvcache/quantize/bitsandbytes.py
keys_values/kvcache/quantize/pytorch.py
keys_values/kvcache/quantize/quantization.py
keys_values/optimize/clone_model.py
keys_values/optimize/grad_accumulate.py
keys_values/optimize/model_factory.py
keys_values/optimize/module_wrapper.py
keys_values/scripts/analyze_gradient_data.py
keys_values/scripts/debug_flex_attention.py
keys_values/scripts/debug_flex_attention_simple.py
keys_values/scripts/debug_flex_attention_simpler.py
keys_values/scripts/profile_caches.py
keys_values/scripts/profile_naive_vs_flexatt_sdpa.py
keys_values/scripts/profile_naive_vs_padded_sdpa.py
keys_values/scripts/profile_sdpa_variants.py
keys_values/scripts/zero_padded_qlen_thresholds.csv
test/test_adapter.py
test/test_adapter_v2.py
test/test_array_limit.py
test/test_attention.py
test/test_flex_attention.py
test/test_lora.py
test/test_model.py
test/test_sdpa_wrapper.py
test/test_use_eager_kernel.py