README.md
pyproject.toml
nsa_flow/__init__.py
nsa_flow/energy.py
nsa_flow/flow.py
nsa_flow/layers.py
nsa_flow/legacy.py
nsa_flow/optimizer.py
nsa_flow/retraction.py
nsa_flow/utils.py
nsa_flow.egg-info/PKG-INFO
nsa_flow.egg-info/SOURCES.txt
nsa_flow.egg-info/dependency_links.txt
nsa_flow.egg-info/requires.txt
nsa_flow.egg-info/top_level.txt
scripts/compare_nsa_flows.py
scripts/demonstrate_layers.py
scripts/test_nonneg_ortho.py
tests/basic_nsa_test.py
tests/conftest.py
tests/debug.py
tests/run_nsa_flow_tests.py
tests/test_convergence_flow.py
tests/test_energy_behavior.py
tests/test_nsa_autograd.py
tests/test_nsa_flow_core_relocated.py
tests/test_nsa_flow_extended.py
tests/test_nsa_flow_opt.py
tests/test_nsa_flow_parameters.py
tests/test_nsaf_conv2d.py
tests/test_nsaf_layer.py
tests/test_nsaf_linear.py