LICENSE
README.md
pyproject.toml
src/natten_mps/__init__.py
src/natten_mps/_torch_ops.py
src/natten_mps/functional.py
src/natten_mps/merge.py
src/natten_mps/support_matrix.py
src/natten_mps/version.py
src/natten_mps.egg-info/PKG-INFO
src/natten_mps.egg-info/SOURCES.txt
src/natten_mps.egg-info/dependency_links.txt
src/natten_mps.egg-info/requires.txt
src/natten_mps.egg-info/top_level.txt
src/natten_mps/_core/__init__.py
src/natten_mps/_core/_metal_shaders.py
src/natten_mps/_core/inverse_maps.py
src/natten_mps/_core/metal.py
src/natten_mps/_core/ops.py
src/natten_mps/_core/pure.py
src/natten_mps/autograd/__init__.py
src/natten_mps/autograd/_factory.py
src/natten_mps/autograd/na1d.py
src/natten_mps/autograd/na2d.py
src/natten_mps/autograd/na3d.py
src/natten_mps/compat/__init__.py
src/natten_mps/compat/v014.py
src/natten_mps/compat/v015.py
src/natten_mps/compat/v017.py
src/natten_mps/compat/v020.py
src/natten_mps/extras/__init__.py
src/natten_mps/extras/allin1/__init__.py
src/natten_mps/extras/allin1/_metal_shaders.py
src/natten_mps/extras/allin1/functional.py
src/natten_mps/extras/allin1/metal.py
src/natten_mps/extras/allin1/reference_impl.py
src/natten_mps/nn/__init__.py
src/natten_mps/nn/na1d.py
src/natten_mps/nn/na2d.py
src/natten_mps/nn/na3d.py
src/natten_mps/utils/__init__.py
src/natten_mps/utils/params.py
src/natten_mps/utils/window.py
tests/test_autograd.py
tests/test_autograd_3d.py
tests/test_backends.py
tests/test_backward_parity.py
tests/test_compat_for_version.py
tests/test_compat_gradients.py
tests/test_compat_v014.py
tests/test_compat_v015.py
tests/test_compat_v017.py
tests/test_compat_v020.py
tests/test_extras_allin1.py
tests/test_extras_backward.py
tests/test_functional.py
tests/test_functional_3d.py
tests/test_functional_3d_expanded.py
tests/test_fused_simd.py
tests/test_gradcheck_functional.py
tests/test_metal_backend.py
tests/test_na1d.py
tests/test_na2d.py
tests/test_na3d.py
tests/test_new_features.py
tests/test_nn_module_grad.py
tests/test_shift_semantics.py
tests/test_split_ops_grad.py
tests/test_support_matrix.py
tests/test_upstream_parity.py
tests/test_validation_errors.py
tests/test_varlen_na1d.py
tests/test_varlen_na2d.py
tests/test_varlen_na3d.py