LICENSE
README.md
pyproject.toml
flash_maxsim/__init__.py
flash_maxsim/flash_chamfer.py
flash_maxsim/flash_chamfer_invgrid.py
flash_maxsim/flash_maxsim.py
flash_maxsim/flash_maxsim_advanced.py
flash_maxsim/flash_maxsim_batched_train.py
flash_maxsim/flash_maxsim_contrastive.py
flash_maxsim/flash_maxsim_exp.py
flash_maxsim/flash_maxsim_fusedbwd_proposal.py
flash_maxsim/flash_maxsim_int8_train.py
flash_maxsim/flash_maxsim_invgrid_proposal.py
flash_maxsim/flash_maxsim_pairs_dense.py
flash_maxsim/flash_maxsim_persistbwd_proposal.py
flash_maxsim/flash_maxsim_quant.py
flash_maxsim/flash_maxsim_rerank.py
flash_maxsim/flash_maxsim_sortbwd_proposal.py
flash_maxsim/flash_maxsim_sparse.py
flash_maxsim/flash_maxsim_sparse_dd.py
flash_maxsim/flash_maxsim_splitd.py
flash_maxsim/flash_maxsim_topk.py
flash_maxsim/flash_maxsim_torchbwd_proposal.py
flash_maxsim/flash_maxsim_varlen.py
flash_maxsim/naive.py
flash_maxsim/warmup.py
flash_maxsim.egg-info/PKG-INFO
flash_maxsim.egg-info/SOURCES.txt
flash_maxsim.egg-info/dependency_links.txt
flash_maxsim.egg-info/requires.txt
flash_maxsim.egg-info/top_level.txt
tests/test_backward.py
tests/test_batched_train.py
tests/test_batched_train_correctness.py
tests/test_batched_train_fp32.py
tests/test_contrastive_step.py
tests/test_flash_maxsim.py
tests/test_full_training_oom.py
tests/test_full_training_oom_v2.py
tests/test_fwd_bwd_split.py
tests/test_invgrid_correctness.py
tests/test_rerank_padded.py
tests/test_sparse_dd.py
tests/test_splitd_int8_warmup.py