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/config.py
keys_values/dora_utils.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/lora_utils.py
keys_values/model.py
keys_values/parser_config.py
keys_values/pos_encoding.py
keys_values/utils.py
keys_values/version
keys_values/attention/__init__.py
keys_values/attention/attention_utils.py
keys_values/attention/base.py
keys_values/attention/flashinfer_ops.py
keys_values/attention/flashinfer_verification.py
keys_values/attention/flashinfer_wrapper.py
keys_values/attention/flex_attention.py
keys_values/attention/flex_attention_old.py
keys_values/attention/sdpa_op.py
keys_values/attention/sdpa_wrapper.py
keys_values/attention/use_eager_kernel.py
keys_values/data/__init__.py
keys_values/data/base.py
keys_values/data/constants.py
keys_values/data/dataloader.py
keys_values/data/evaluation.py
keys_values/data/helmet.py
keys_values/data/iterators.py
keys_values/data/load_helmet_dev_eval.py
keys_values/data/longbench_v2.py
keys_values/data/module.py
keys_values/data/sequence_classification.py
keys_values/data/sft_dataset.py
keys_values/data/trainstate.py
keys_values/evaluation/__init__.py
keys_values/evaluation/evaluation.py
keys_values/evaluation/evaluator.py
keys_values/evaluation/metrics.py
keys_values/evaluation/response_parser.py
keys_values/evaluation/tasks.py
keys_values/finetune/__init__.py
keys_values/finetune/args.py
keys_values/finetune/batch_transform.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_eval_ext.py
keys_values/finetune/longcontext_full.py
keys_values/finetune/longcontext_lora.py
keys_values/finetune/resume_state.py
keys_values/finetune/utils.py
keys_values/fused/__init__.py
keys_values/fused/fused_rmsnorm.py
keys_values/fused/fused_rope.py
keys_values/fused/fused_swiglu.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/consts.py
keys_values/kvcache/factory.py
keys_values/kvcache/h2o.py
keys_values/kvcache/offloading.py
keys_values/kvcache/qh2o.py
keys_values/kvcache/quant_buffers.py
keys_values/kvcache/smart_lastrec.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/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/cleanup_evaluation.py
keys_values/scripts/cleanup_gen_samples.py
keys_values/scripts/collect_eval_results.py
keys_values/scripts/collect_gen_samples.py
keys_values/scripts/create_result_table.py
keys_values/scripts/dataset_token_counts.py
keys_values/scripts/debug_flex_attention_simple.py
keys_values/scripts/plot_size_log.py
keys_values/scripts/plot_size_log_stored_values.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/remove_files_from_checkpoints.py
keys_values/scripts/convert_lora_to_litgpt/merge_qwen3_lora.py
keys_values/tools/__init__.py
keys_values/tools/debug_utils.py
keys_values/tools/intermediates.py
keys_values/tools/size_log.py
keys_values/tools/tracker.py
test/test_adapter.py
test/test_adapter_v2.py
test/test_array_limit.py
test/test_lora.py
test/test_model.py