LICENSE
README.md
pyproject.toml
src/mps_sdpa/__init__.py
src/mps_sdpa/__main__.py
src/mps_sdpa/api.py
src/mps_sdpa/cli.py
src/mps_sdpa.egg-info/PKG-INFO
src/mps_sdpa.egg-info/SOURCES.txt
src/mps_sdpa.egg-info/dependency_links.txt
src/mps_sdpa.egg-info/entry_points.txt
src/mps_sdpa.egg-info/requires.txt
src/mps_sdpa.egg-info/top_level.txt
src/mps_sdpa/_cpp/__init__.py
src/mps_sdpa/_cpp/mpsgraph_zc.mm
src/mps_sdpa/backends/__init__.py
src/mps_sdpa/backends/_calibrate.py
src/mps_sdpa/backends/metal_op.py
src/mps_sdpa/backends/metal_proto.py
src/mps_sdpa/backends/mpsgraph.py
src/mps_sdpa/backends/mpsgraph_zc.py
src/mps_sdpa/backends/stock.py
src/mps_sdpa/harness/__init__.py
src/mps_sdpa/harness/benchmark.py
src/mps_sdpa/harness/cold_latency.py
src/mps_sdpa/harness/contamination.py
src/mps_sdpa/harness/correctness.py
src/mps_sdpa/harness/gradcheck.py
src/mps_sdpa/harness/memory.py
src/mps_sdpa/harness/reference.py
src/mps_sdpa/harness/report.py
src/mps_sdpa/harness/tensor_factory.py
src/mps_sdpa/harness/tolerances.py
src/mps_sdpa/suites/__init__.py
src/mps_sdpa/suites/correctness_shapes.py
src/mps_sdpa/suites/general_shapes.py
src/mps_sdpa/suites/realistic_shapes.py
src/mps_sdpa/training_check/__init__.py
src/mps_sdpa/training_check/checkpoint_interchange.py
src/mps_sdpa/training_check/loss_compare.py
src/mps_sdpa/training_check/synthetic_train.py
src/mps_sdpa/utils/__init__.py
src/mps_sdpa/utils/env.py
src/mps_sdpa/utils/thermal.py
tests/test_amp_autocast.py
tests/test_api.py
tests/test_benchmark.py
tests/test_calibration.py
tests/test_checkpoint_compat.py
tests/test_checkpoint_interchange.py
tests/test_cli.py
tests/test_cold_latency.py
tests/test_contamination.py
tests/test_correctness_checker.py
tests/test_dispatch.py
tests/test_dropout_zc.py
tests/test_edge_cases.py
tests/test_env.py
tests/test_extended_shapes.py
tests/test_fallback_logging.py
tests/test_generic_transformer.py
tests/test_gradcheck_harness.py
tests/test_long_horizon.py
tests/test_loss_compare.py
tests/test_mask_dtype_coercion.py
tests/test_memory_harness.py
tests/test_metal_proto.py
tests/test_mpsgraph_zc.py
tests/test_numerical_extremes.py
tests/test_partial_requires_grad.py
tests/test_per_head_masks.py
tests/test_reference.py
tests/test_report.py
tests/test_retain_graph.py
tests/test_second_order_grads.py
tests/test_suites.py
tests/test_synthetic_train.py
tests/test_tensor_factory.py
tests/test_thermal.py
tests/test_thread_safety.py
tests/test_tolerances.py
tests/test_training_convergence_e2e.py