LICENSE
MANIFEST.in
README.md
pyproject.toml
setup.py
src/gefen/__init__.py
src/gefen/gefen.py
src/gefen/gefen_muon.py
src/gefen/hybrid.py
src/gefen/params.py
src/gefen/partitioning.py
src/gefen/py.typed
src/gefen/quantization.py
src/gefen/kernels/__init__.py
src/gefen/kernels/_build_notice.py
src/gefen/kernels/automatic_gefen_fused.py
src/gefen/kernels/automatic_gefen_fused_binding.cpp
src/gefen/kernels/automatic_gefen_fused_kernel.cu
src/gefen/kernels/automatic_vmean.py
src/gefen/kernels/automatic_vmean_binding.cpp
src/gefen/kernels/automatic_vmean_kernel.cu
src/gefen/kernels/exact_histogram_fused.py
src/gefen/kernels/exact_histogram_fused_binding.cpp
src/gefen/kernels/exact_histogram_fused_kernel.cu
src/gefen/kernels/period_variance.py
src/gefen/kernels/period_variance_binding.cpp
src/gefen/kernels/period_variance_kernel.cu
src/gefen/tools/README.md
src/gefen/tools/__init__.py
src/gefen/tools/find_lr.py
src/gefen/tools/lr_calibration.py
src/gefen/tools/run_model_calibration.py
src/gefen_x.egg-info/PKG-INFO
src/gefen_x.egg-info/SOURCES.txt
src/gefen_x.egg-info/dependency_links.txt
src/gefen_x.egg-info/requires.txt
src/gefen_x.egg-info/top_level.txt
tests/test_capturable.py
tests/test_capturable_fsdp2.py
tests/test_codebook_indices_parity.py
tests/test_codebook_multi_device.py
tests/test_core_correctness_cpu.py
tests/test_cpu_step_checkpoint.py
tests/test_dequant_indices_parity.py
tests/test_dispatch_gating_cpu.py
tests/test_distributed_owner_cpu.py
tests/test_factored_v_ema_parity.py
tests/test_find_lr.py
tests/test_find_period_backend.py
tests/test_fused_fsdp2_noncontig.py
tests/test_fused_full_update_parity.py
tests/test_fused_update_v2_full_parity.py
tests/test_fused_update_v2_parity.py
tests/test_gefen_factored_v.py
tests/test_gefen_muon_hybrid.py
tests/test_hybrid_ctor_and_lrcache_cpu.py
tests/test_large_tensor_paths.py
tests/test_lr_calibration.py
tests/test_lr_item_no_sync.py
tests/test_muon_combined_levers.py
tests/test_muon_fsdp2_approx.py
tests/test_muon_fsdp2_parity.py
tests/test_muon_hybrid_scrub_cpu.py
tests/test_muon_momentum_kernel_parity.py
tests/test_muon_quality_levers.py
tests/test_no_import_shadowing.py
tests/test_partitioning_quantization_cpu.py
tests/test_update_v2_dispatch.py
tests/test_validation_cpu.py