diff --git a/fmnist_0_mps.safetensors b/fmnist_0_mps.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..b67e3eebdf74d5dd97d94e8689bf7ad9353ab3f2
--- /dev/null
+++ b/fmnist_0_mps.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3460457392d6cc0966fa4ffa84e2ce47f06b07e906df620d280cf3329ee315c1
+size 25398616
diff --git a/mnist_0_mps.safetensors b/mnist_0_mps.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..608dba06628fbca6bacf28a838251c6846e744dc
--- /dev/null
+++ b/mnist_0_mps.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:640458ef8b920bab9c9606d387b8c71b91581e93e56a248d7e7b09a12c897c35
+size 25398616
diff --git a/mnist_0_mps_feature_100.json b/mnist_0_mps_feature_100.json
new file mode 100644
index 0000000000000000000000000000000000000000..a623b4b03ae3cc2d388fbcb2db795df3f2d7aff0
--- /dev/null
+++ b/mnist_0_mps_feature_100.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 100, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521, 238, 628, 237, 210, 207, 213, 239, 493, 265, 240, 632, 599, 212, 214, 241, 211, 605, 549, 236, 266, 235, 577, 242, 185, 494, 181, 572, 352, 234, 466, 627, 186, 518, 293, 268, 215, 269, 545, 182, 297, 184, 491, 292, 492, 264, 243, 180, 270, 267, 324, 438, 325, 183, 296, 658, 522, 380, 383, 410, 206, 465, 382, 271, 657, 298, 411, 355, 353, 262, 463, 578, 659, 263, 432, 598, 656, 354, 626, 299, 633], "train_min_nll": 18.252117156982422, "test_nll": 20.6348819732666}
\ No newline at end of file
diff --git a/mnist_0_mps_feature_100.safetensors b/mnist_0_mps_feature_100.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..251077eb31bceeac273ca5bf984923c09701c856
--- /dev/null
+++ b/mnist_0_mps_feature_100.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1786e164aa3de094f5b51e09252077b158cdf5415ac861970be63753bb5b47bb
+size 2934416
diff --git a/mnist_0_mps_feature_15.json b/mnist_0_mps_feature_15.json
new file mode 100644
index 0000000000000000000000000000000000000000..4189d20c62b2bd5473bf4749d6acae6131d77012
--- /dev/null
+++ b/mnist_0_mps_feature_15.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 15, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573], "train_min_nll": 3.8297295570373535, "test_nll": 3.80476713180542}
\ No newline at end of file
diff --git a/mnist_0_mps_feature_15.safetensors b/mnist_0_mps_feature_15.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..47f1cd216e55acd69e183cb5a828eed23e1ca2f5
--- /dev/null
+++ b/mnist_0_mps_feature_15.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6575534efa45a55e4ee7c75d6ab1ef9b105dcaf91b1ce3e849f4663d7ac50ea3
+size 143032
diff --git a/mnist_0_mps_feature_20.json b/mnist_0_mps_feature_20.json
new file mode 100644
index 0000000000000000000000000000000000000000..f0a32c5aeaed588f55178ee95f26e3e15fbaa8cc
--- /dev/null
+++ b/mnist_0_mps_feature_20.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 20, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521], "train_min_nll": 3.7837815284729004, "test_nll": 3.8645401000976562}
\ No newline at end of file
diff --git a/mnist_0_mps_feature_20.safetensors b/mnist_0_mps_feature_20.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..92c60654457c2761656dffd3218414267123feea
--- /dev/null
+++ b/mnist_0_mps_feature_20.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:39c9c333edf976e6acc7c11dcdafc32fb3533617acdd7a8c22ec828578342174
+size 307248
diff --git a/mnist_0_mps_feature_200.json b/mnist_0_mps_feature_200.json
new file mode 100644
index 0000000000000000000000000000000000000000..2d77ca8166ac4c581fcd312445d5d501387109eb
--- /dev/null
+++ b/mnist_0_mps_feature_200.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 200, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521, 238, 628, 237, 210, 207, 213, 239, 493, 265, 240, 632, 599, 212, 214, 241, 211, 605, 549, 236, 266, 235, 577, 242, 185, 494, 181, 572, 352, 234, 466, 627, 186, 518, 293, 268, 215, 269, 545, 182, 297, 184, 491, 292, 492, 264, 243, 180, 270, 267, 324, 438, 325, 183, 296, 658, 522, 380, 383, 410, 206, 465, 382, 271, 657, 298, 411, 355, 353, 262, 463, 578, 659, 263, 432, 598, 656, 354, 626, 299, 633, 439, 187, 320, 655, 291, 431, 571, 550, 327, 490, 261, 326, 404, 233, 319, 435, 294, 403, 379, 407, 606, 467, 244, 464, 402, 290, 544, 321, 179, 408, 430, 654, 216, 517, 381, 318, 437, 289, 374, 459, 346, 295, 272, 347, 460, 157, 436, 405, 375, 156, 495, 351, 462, 317, 433, 458, 155, 570, 345, 409, 300, 154, 158, 660, 625, 376, 373, 489, 401, 348, 461, 488, 523, 429, 551, 406, 597, 579, 434, 153, 377, 323, 516, 288, 487, 205, 349, 328, 457, 260, 316, 543, 486, 653, 378, 322, 344, 188, 356, 569], "train_min_nll": 35.45349884033203, "test_nll": 40.29694366455078}
\ No newline at end of file
diff --git a/mnist_0_mps_feature_200.safetensors b/mnist_0_mps_feature_200.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..511d6a7383206096d7eedb53c3b7353f852c639d
--- /dev/null
+++ b/mnist_0_mps_feature_200.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:83ac1783b5f69cbc72681a564a365e43516df4441332f9827a09d6790efce92d
+size 6218528
diff --git a/mnist_0_mps_feature_300.json b/mnist_0_mps_feature_300.json
new file mode 100644
index 0000000000000000000000000000000000000000..21f4dac6bdc90e8937b45f94c791ca5380fca561
--- /dev/null
+++ b/mnist_0_mps_feature_300.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 300, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521, 238, 628, 237, 210, 207, 213, 239, 493, 265, 240, 632, 599, 212, 214, 241, 211, 605, 549, 236, 266, 235, 577, 242, 185, 494, 181, 572, 352, 234, 466, 627, 186, 518, 293, 268, 215, 269, 545, 182, 297, 184, 491, 292, 492, 264, 243, 180, 270, 267, 324, 438, 325, 183, 296, 658, 522, 380, 383, 410, 206, 465, 382, 271, 657, 298, 411, 355, 353, 262, 463, 578, 659, 263, 432, 598, 656, 354, 626, 299, 633, 439, 187, 320, 655, 291, 431, 571, 550, 327, 490, 261, 326, 404, 233, 319, 435, 294, 403, 379, 407, 606, 467, 244, 464, 402, 290, 544, 321, 179, 408, 430, 654, 216, 517, 381, 318, 437, 289, 374, 459, 346, 295, 272, 347, 460, 157, 436, 405, 375, 156, 495, 351, 462, 317, 433, 458, 155, 570, 345, 409, 300, 154, 158, 660, 625, 376, 373, 489, 401, 348, 461, 488, 523, 429, 551, 406, 597, 579, 434, 153, 377, 323, 516, 288, 487, 205, 349, 328, 457, 260, 316, 543, 486, 653, 378, 322, 344, 188, 356, 569, 372, 485, 178, 400, 384, 515, 245, 412, 634, 152, 232, 440, 159, 542, 350, 428, 468, 607, 624, 685, 541, 684, 273, 496, 514, 596, 524, 513, 456, 217, 552, 683, 484, 568, 686, 661, 540, 512, 301, 580, 371, 343, 399, 315, 151, 652, 287, 127, 128, 682, 427, 204, 595, 567, 455, 177, 687, 259, 129, 539, 483, 160, 497, 126, 511, 189, 525, 329, 469, 623, 441, 635, 553, 130, 125, 246, 231, 357, 608, 413, 274, 385, 150, 681, 566, 662, 688, 538, 218, 510, 594, 482, 454, 426, 398, 302, 370, 581, 131, 124], "train_min_nll": 50.714080810546875, "test_nll": 57.34071350097656}
\ No newline at end of file
diff --git a/mnist_0_mps_feature_300.safetensors b/mnist_0_mps_feature_300.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..73b41f1e057ee31699e1467fbc0ed6b58599d501
--- /dev/null
+++ b/mnist_0_mps_feature_300.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:86e98c32fcd4ff4e7df92863513ddf4ad38e700d4e903a20789c6e6a47b0f92e
+size 9502624
diff --git a/mnist_0_mps_feature_400.json b/mnist_0_mps_feature_400.json
new file mode 100644
index 0000000000000000000000000000000000000000..b4361d00d66f36e41674f7135b2d0fc6cc9537ee
--- /dev/null
+++ b/mnist_0_mps_feature_400.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 400, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521, 238, 628, 237, 210, 207, 213, 239, 493, 265, 240, 632, 599, 212, 214, 241, 211, 605, 549, 236, 266, 235, 577, 242, 185, 494, 181, 572, 352, 234, 466, 627, 186, 518, 293, 268, 215, 269, 545, 182, 297, 184, 491, 292, 492, 264, 243, 180, 270, 267, 324, 438, 325, 183, 296, 658, 522, 380, 383, 410, 206, 465, 382, 271, 657, 298, 411, 355, 353, 262, 463, 578, 659, 263, 432, 598, 656, 354, 626, 299, 633, 439, 187, 320, 655, 291, 431, 571, 550, 327, 490, 261, 326, 404, 233, 319, 435, 294, 403, 379, 407, 606, 467, 244, 464, 402, 290, 544, 321, 179, 408, 430, 654, 216, 517, 381, 318, 437, 289, 374, 459, 346, 295, 272, 347, 460, 157, 436, 405, 375, 156, 495, 351, 462, 317, 433, 458, 155, 570, 345, 409, 300, 154, 158, 660, 625, 376, 373, 489, 401, 348, 461, 488, 523, 429, 551, 406, 597, 579, 434, 153, 377, 323, 516, 288, 487, 205, 349, 328, 457, 260, 316, 543, 486, 653, 378, 322, 344, 188, 356, 569, 372, 485, 178, 400, 384, 515, 245, 412, 634, 152, 232, 440, 159, 542, 350, 428, 468, 607, 624, 685, 541, 684, 273, 496, 514, 596, 524, 513, 456, 217, 552, 683, 484, 568, 686, 661, 540, 512, 301, 580, 371, 343, 399, 315, 151, 652, 287, 127, 128, 682, 427, 204, 595, 567, 455, 177, 687, 259, 129, 539, 483, 160, 497, 126, 511, 189, 525, 329, 469, 623, 441, 635, 553, 130, 125, 246, 231, 357, 608, 413, 274, 385, 150, 681, 566, 662, 688, 538, 218, 510, 594, 482, 454, 426, 398, 302, 370, 581, 131, 124, 470, 176, 342, 498, 651, 203, 161, 314, 526, 442, 286, 622, 190, 149, 330, 258, 689, 414, 636, 554, 537, 123, 680, 509, 565, 663, 358, 609, 386, 230, 132, 481, 247, 593, 453, 175, 275, 713, 712, 471, 219, 425, 443, 582, 711, 499, 162, 714, 148, 122, 202, 690, 397, 650, 303, 415, 710, 527, 369, 99, 715, 341, 313, 100, 387, 191, 621, 331, 98, 679, 637, 285, 664, 133, 359, 257, 101, 716, 555, 709, 97, 610, 121, 229, 102, 536, 174, 564, 691, 508, 147, 96, 717, 103, 163, 248, 480, 444, 472, 583], "train_min_nll": 55.58735656738281, "test_nll": 62.52439880371094}
\ No newline at end of file
diff --git a/mnist_0_mps_feature_400.safetensors b/mnist_0_mps_feature_400.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..b4a1126c847bbe508304e7688b4e6672a22e8e59
--- /dev/null
+++ b/mnist_0_mps_feature_400.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9d1291e482532887cdd321ffaaabe68e952c8e12d9728245be872e406932e99c
+size 12786904
diff --git a/mnist_0_mps_feature_50.json b/mnist_0_mps_feature_50.json
new file mode 100644
index 0000000000000000000000000000000000000000..9c23c258d7ed84a0ca4e2a531ac3339ecdf42ff2
--- /dev/null
+++ b/mnist_0_mps_feature_50.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 50, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521, 238, 628, 237, 210, 207, 213, 239, 493, 265, 240, 632, 599, 212, 214, 241, 211, 605, 549, 236, 266, 235, 577, 242, 185, 494, 181, 572, 352, 234, 466], "train_min_nll": 9.73012638092041, "test_nll": 10.576120376586914}
\ No newline at end of file
diff --git a/mnist_0_mps_feature_50.safetensors b/mnist_0_mps_feature_50.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..c2a3b921adf22e8308c286ef5688c980530d718a
--- /dev/null
+++ b/mnist_0_mps_feature_50.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:bfcf48fbd4434fbc99d6246e4e16992d6b590eb43d106bb07f46c4c377559f6d
+size 1292416
diff --git a/mnist_0_mps_feature_500.json b/mnist_0_mps_feature_500.json
new file mode 100644
index 0000000000000000000000000000000000000000..3d4d1c659bfc1e5adf1b12c5d84db79e9f090f2b
--- /dev/null
+++ b/mnist_0_mps_feature_500.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 500, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521, 238, 628, 237, 210, 207, 213, 239, 493, 265, 240, 632, 599, 212, 214, 241, 211, 605, 549, 236, 266, 235, 577, 242, 185, 494, 181, 572, 352, 234, 466, 627, 186, 518, 293, 268, 215, 269, 545, 182, 297, 184, 491, 292, 492, 264, 243, 180, 270, 267, 324, 438, 325, 183, 296, 658, 522, 380, 383, 410, 206, 465, 382, 271, 657, 298, 411, 355, 353, 262, 463, 578, 659, 263, 432, 598, 656, 354, 626, 299, 633, 439, 187, 320, 655, 291, 431, 571, 550, 327, 490, 261, 326, 404, 233, 319, 435, 294, 403, 379, 407, 606, 467, 244, 464, 402, 290, 544, 321, 179, 408, 430, 654, 216, 517, 381, 318, 437, 289, 374, 459, 346, 295, 272, 347, 460, 157, 436, 405, 375, 156, 495, 351, 462, 317, 433, 458, 155, 570, 345, 409, 300, 154, 158, 660, 625, 376, 373, 489, 401, 348, 461, 488, 523, 429, 551, 406, 597, 579, 434, 153, 377, 323, 516, 288, 487, 205, 349, 328, 457, 260, 316, 543, 486, 653, 378, 322, 344, 188, 356, 569, 372, 485, 178, 400, 384, 515, 245, 412, 634, 152, 232, 440, 159, 542, 350, 428, 468, 607, 624, 685, 541, 684, 273, 496, 514, 596, 524, 513, 456, 217, 552, 683, 484, 568, 686, 661, 540, 512, 301, 580, 371, 343, 399, 315, 151, 652, 287, 127, 128, 682, 427, 204, 595, 567, 455, 177, 687, 259, 129, 539, 483, 160, 497, 126, 511, 189, 525, 329, 469, 623, 441, 635, 553, 130, 125, 246, 231, 357, 608, 413, 274, 385, 150, 681, 566, 662, 688, 538, 218, 510, 594, 482, 454, 426, 398, 302, 370, 581, 131, 124, 470, 176, 342, 498, 651, 203, 161, 314, 526, 442, 286, 622, 190, 149, 330, 258, 689, 414, 636, 554, 537, 123, 680, 509, 565, 663, 358, 609, 386, 230, 132, 481, 247, 593, 453, 175, 275, 713, 712, 471, 219, 425, 443, 582, 711, 499, 162, 714, 148, 122, 202, 690, 397, 650, 303, 415, 710, 527, 369, 99, 715, 341, 313, 100, 387, 191, 621, 331, 98, 679, 637, 285, 664, 133, 359, 257, 101, 716, 555, 709, 97, 610, 121, 229, 102, 536, 174, 564, 691, 508, 147, 96, 717, 103, 163, 248, 480, 444, 472, 583, 592, 201, 708, 220, 276, 416, 95, 134, 500, 665, 649, 120, 638, 388, 452, 718, 678, 304, 740, 692, 192, 741, 528, 104, 739, 256, 284, 424, 742, 360, 312, 611, 146, 332, 738, 94, 620, 228, 173, 396, 340, 71, 368, 743, 707, 556, 719, 72, 744, 737, 70, 119, 164, 666, 73, 693, 105, 93, 584, 639, 249, 200, 135, 69, 745, 221, 563, 74, 535, 68, 736, 677, 277, 648, 720, 591, 145, 75, 706, 507, 193, 255, 283, 612, 746, 67, 92, 501, 473, 445, 305, 311, 172, 694, 227, 118, 667, 106, 529, 5], "train_min_nll": 56.71366882324219, "test_nll": 64.14777374267578}
\ No newline at end of file
diff --git a/mnist_0_mps_feature_500.safetensors b/mnist_0_mps_feature_500.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..1df605a0e5af3038c6a0a9969599620a2ccc5f57
--- /dev/null
+++ b/mnist_0_mps_feature_500.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c083928efbadbf75b2dd074504cf04d580cc9540c67ebc19d59bcc50d03d4420
+size 16071208
diff --git a/mnist_0_mps_feature_600.json b/mnist_0_mps_feature_600.json
new file mode 100644
index 0000000000000000000000000000000000000000..cd1e37f1e8a5d05f6deb2933bb39bfe219c29e32
--- /dev/null
+++ b/mnist_0_mps_feature_600.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 600, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521, 238, 628, 237, 210, 207, 213, 239, 493, 265, 240, 632, 599, 212, 214, 241, 211, 605, 549, 236, 266, 235, 577, 242, 185, 494, 181, 572, 352, 234, 466, 627, 186, 518, 293, 268, 215, 269, 545, 182, 297, 184, 491, 292, 492, 264, 243, 180, 270, 267, 324, 438, 325, 183, 296, 658, 522, 380, 383, 410, 206, 465, 382, 271, 657, 298, 411, 355, 353, 262, 463, 578, 659, 263, 432, 598, 656, 354, 626, 299, 633, 439, 187, 320, 655, 291, 431, 571, 550, 327, 490, 261, 326, 404, 233, 319, 435, 294, 403, 379, 407, 606, 467, 244, 464, 402, 290, 544, 321, 179, 408, 430, 654, 216, 517, 381, 318, 437, 289, 374, 459, 346, 295, 272, 347, 460, 157, 436, 405, 375, 156, 495, 351, 462, 317, 433, 458, 155, 570, 345, 409, 300, 154, 158, 660, 625, 376, 373, 489, 401, 348, 461, 488, 523, 429, 551, 406, 597, 579, 434, 153, 377, 323, 516, 288, 487, 205, 349, 328, 457, 260, 316, 543, 486, 653, 378, 322, 344, 188, 356, 569, 372, 485, 178, 400, 384, 515, 245, 412, 634, 152, 232, 440, 159, 542, 350, 428, 468, 607, 624, 685, 541, 684, 273, 496, 514, 596, 524, 513, 456, 217, 552, 683, 484, 568, 686, 661, 540, 512, 301, 580, 371, 343, 399, 315, 151, 652, 287, 127, 128, 682, 427, 204, 595, 567, 455, 177, 687, 259, 129, 539, 483, 160, 497, 126, 511, 189, 525, 329, 469, 623, 441, 635, 553, 130, 125, 246, 231, 357, 608, 413, 274, 385, 150, 681, 566, 662, 688, 538, 218, 510, 594, 482, 454, 426, 398, 302, 370, 581, 131, 124, 470, 176, 342, 498, 651, 203, 161, 314, 526, 442, 286, 622, 190, 149, 330, 258, 689, 414, 636, 554, 537, 123, 680, 509, 565, 663, 358, 609, 386, 230, 132, 481, 247, 593, 453, 175, 275, 713, 712, 471, 219, 425, 443, 582, 711, 499, 162, 714, 148, 122, 202, 690, 397, 650, 303, 415, 710, 527, 369, 99, 715, 341, 313, 100, 387, 191, 621, 331, 98, 679, 637, 285, 664, 133, 359, 257, 101, 716, 555, 709, 97, 610, 121, 229, 102, 536, 174, 564, 691, 508, 147, 96, 717, 103, 163, 248, 480, 444, 472, 583, 592, 201, 708, 220, 276, 416, 95, 134, 500, 665, 649, 120, 638, 388, 452, 718, 678, 304, 740, 692, 192, 741, 528, 104, 739, 256, 284, 424, 742, 360, 312, 611, 146, 332, 738, 94, 620, 228, 173, 396, 340, 71, 368, 743, 707, 556, 719, 72, 744, 737, 70, 119, 164, 666, 73, 693, 105, 93, 584, 639, 249, 200, 135, 69, 745, 221, 563, 74, 535, 68, 736, 677, 277, 648, 720, 591, 145, 75, 706, 507, 193, 255, 283, 612, 746, 67, 92, 501, 473, 445, 305, 311, 172, 694, 227, 118, 667, 106, 529, 5, 735, 76, 479, 66, 417, 619, 721, 339, 136, 557, 747, 7, 4, 640, 451, 91, 367, 6, 333, 165, 389, 585, 423, 3, 77, 65, 676, 748, 770, 734, 199, 395, 705, 771, 695, 769, 361, 282, 117, 107, 768, 722, 254, 222, 43, 250, 772, 310, 144, 613, 647, 64, 767, 278, 668, 78, 562, 194, 90, 226, 766, 42, 749, 40, 44, 534, 773, 41, 590, 1, 306, 63, 338, 171, 2, 39, 108, 137, 733, 502, 765, 474, 723, 38, 704, 45, 618, 506, 446, 641, 478, 334, 366, 675, 558, 764, 774, 696, 46, 530], "train_min_nll": 56.10128402709961, "test_nll": 63.07518768310547}
\ No newline at end of file
diff --git a/mnist_0_mps_feature_600.safetensors b/mnist_0_mps_feature_600.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..ab0ad0b4a44621b87d1c03ddefff2f9f2c238121
--- /dev/null
+++ b/mnist_0_mps_feature_600.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a90b109854fc3a4bf3deafc964c91f7e17ddd47523552f65720057eaf639bd14
+size 19355504
diff --git a/mnist_1_losses.png b/mnist_1_losses.png
new file mode 100644
index 0000000000000000000000000000000000000000..3f741673b3baab3734e78a7aadd321d805459b0f
Binary files /dev/null and b/mnist_1_losses.png differ
diff --git a/mnist_1_mps.safetensors b/mnist_1_mps.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..ca54efc874527801c3a6ad6e75685ab8378121b7
--- /dev/null
+++ b/mnist_1_mps.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7f938395d75ffae71723957e8d3de7cfa493165629175388957678c47c7f0c35
+size 25398616
diff --git a/mnist_1_mps_feature_100.json b/mnist_1_mps_feature_100.json
new file mode 100644
index 0000000000000000000000000000000000000000..93dcc1f1dfb5d80a39211cf834aaf7acb2b951a9
--- /dev/null
+++ b/mnist_1_mps_feature_100.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 100, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521, 238, 628, 237, 210, 207, 213, 239, 493, 265, 240, 632, 599, 212, 214, 241, 211, 605, 549, 236, 266, 235, 577, 242, 185, 494, 181, 572, 352, 234, 466, 627, 186, 518, 293, 268, 215, 269, 545, 182, 297, 184, 491, 292, 492, 264, 243, 180, 270, 267, 324, 438, 325, 183, 296, 658, 522, 380, 383, 410, 206, 465, 382, 271, 657, 298, 411, 355, 353, 262, 463, 578, 659, 263, 432, 598, 656, 354, 626, 299, 633], "train_min_nll": 9.956103324890137, "test_nll": 10.403436660766602}
\ No newline at end of file
diff --git a/mnist_1_mps_feature_100.safetensors b/mnist_1_mps_feature_100.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..35bc43a1bc7a1b1b3965f127d274a101bde8b0d6
--- /dev/null
+++ b/mnist_1_mps_feature_100.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:93fceaca0adc20d0640548182322bae2fe3da0949424df1a0eef1cf1237daeb0
+size 2934416
diff --git a/mnist_1_mps_feature_15.json b/mnist_1_mps_feature_15.json
new file mode 100644
index 0000000000000000000000000000000000000000..54667664281cfc96d5b8bce406ffea19b2dca80d
--- /dev/null
+++ b/mnist_1_mps_feature_15.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 15, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573], "train_min_nll": 3.013178586959839, "test_nll": 3.05098557472229}
\ No newline at end of file
diff --git a/mnist_1_mps_feature_15.safetensors b/mnist_1_mps_feature_15.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..78d9fb50fc65504d3e7c4b98599451fbfa417d59
--- /dev/null
+++ b/mnist_1_mps_feature_15.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:34da4e48cf21be3e089b9fb6242dfb513a32ab243dba2a9977303c7b07f83c4e
+size 143032
diff --git a/mnist_1_mps_feature_20.json b/mnist_1_mps_feature_20.json
new file mode 100644
index 0000000000000000000000000000000000000000..4314259d2b24c5cf208d5939ae8826e9a48eedec
--- /dev/null
+++ b/mnist_1_mps_feature_20.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 20, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521], "train_min_nll": 3.41808819770813, "test_nll": 3.518298625946045}
\ No newline at end of file
diff --git a/mnist_1_mps_feature_20.safetensors b/mnist_1_mps_feature_20.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..8cd92d662a0b3cc5edb3a97a0d5f1ab4ee1c79e0
--- /dev/null
+++ b/mnist_1_mps_feature_20.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:da6ae8a5c2ef50e3322cae5eccad01f451ab427cf7aba9e3d1363f1a01d2f6ba
+size 307248
diff --git a/mnist_1_mps_feature_200.json b/mnist_1_mps_feature_200.json
new file mode 100644
index 0000000000000000000000000000000000000000..060743489be8b7ed1ebc7618b827bab536810999
--- /dev/null
+++ b/mnist_1_mps_feature_200.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 200, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521, 238, 628, 237, 210, 207, 213, 239, 493, 265, 240, 632, 599, 212, 214, 241, 211, 605, 549, 236, 266, 235, 577, 242, 185, 494, 181, 572, 352, 234, 466, 627, 186, 518, 293, 268, 215, 269, 545, 182, 297, 184, 491, 292, 492, 264, 243, 180, 270, 267, 324, 438, 325, 183, 296, 658, 522, 380, 383, 410, 206, 465, 382, 271, 657, 298, 411, 355, 353, 262, 463, 578, 659, 263, 432, 598, 656, 354, 626, 299, 633, 439, 187, 320, 655, 291, 431, 571, 550, 327, 490, 261, 326, 404, 233, 319, 435, 294, 403, 379, 407, 606, 467, 244, 464, 402, 290, 544, 321, 179, 408, 430, 654, 216, 517, 381, 318, 437, 289, 374, 459, 346, 295, 272, 347, 460, 157, 436, 405, 375, 156, 495, 351, 462, 317, 433, 458, 155, 570, 345, 409, 300, 154, 158, 660, 625, 376, 373, 489, 401, 348, 461, 488, 523, 429, 551, 406, 597, 579, 434, 153, 377, 323, 516, 288, 487, 205, 349, 328, 457, 260, 316, 543, 486, 653, 378, 322, 344, 188, 356, 569], "train_min_nll": 14.880247116088867, "test_nll": 15.606005668640137}
\ No newline at end of file
diff --git a/mnist_1_mps_feature_200.safetensors b/mnist_1_mps_feature_200.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..eba3081db25fe267cef2c07772516e257ec8debf
--- /dev/null
+++ b/mnist_1_mps_feature_200.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b933a1ea122e8eeafc91883bbfd30d98ff43101d2e09f7559814cb19423e9165
+size 6218528
diff --git a/mnist_1_mps_feature_300.json b/mnist_1_mps_feature_300.json
new file mode 100644
index 0000000000000000000000000000000000000000..23568b887825d17a5a107a882866f2ce9a4d3fbe
--- /dev/null
+++ b/mnist_1_mps_feature_300.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 300, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521, 238, 628, 237, 210, 207, 213, 239, 493, 265, 240, 632, 599, 212, 214, 241, 211, 605, 549, 236, 266, 235, 577, 242, 185, 494, 181, 572, 352, 234, 466, 627, 186, 518, 293, 268, 215, 269, 545, 182, 297, 184, 491, 292, 492, 264, 243, 180, 270, 267, 324, 438, 325, 183, 296, 658, 522, 380, 383, 410, 206, 465, 382, 271, 657, 298, 411, 355, 353, 262, 463, 578, 659, 263, 432, 598, 656, 354, 626, 299, 633, 439, 187, 320, 655, 291, 431, 571, 550, 327, 490, 261, 326, 404, 233, 319, 435, 294, 403, 379, 407, 606, 467, 244, 464, 402, 290, 544, 321, 179, 408, 430, 654, 216, 517, 381, 318, 437, 289, 374, 459, 346, 295, 272, 347, 460, 157, 436, 405, 375, 156, 495, 351, 462, 317, 433, 458, 155, 570, 345, 409, 300, 154, 158, 660, 625, 376, 373, 489, 401, 348, 461, 488, 523, 429, 551, 406, 597, 579, 434, 153, 377, 323, 516, 288, 487, 205, 349, 328, 457, 260, 316, 543, 486, 653, 378, 322, 344, 188, 356, 569, 372, 485, 178, 400, 384, 515, 245, 412, 634, 152, 232, 440, 159, 542, 350, 428, 468, 607, 624, 685, 541, 684, 273, 496, 514, 596, 524, 513, 456, 217, 552, 683, 484, 568, 686, 661, 540, 512, 301, 580, 371, 343, 399, 315, 151, 652, 287, 127, 128, 682, 427, 204, 595, 567, 455, 177, 687, 259, 129, 539, 483, 160, 497, 126, 511, 189, 525, 329, 469, 623, 441, 635, 553, 130, 125, 246, 231, 357, 608, 413, 274, 385, 150, 681, 566, 662, 688, 538, 218, 510, 594, 482, 454, 426, 398, 302, 370, 581, 131, 124], "train_min_nll": 17.932340621948242, "test_nll": 19.254934310913086}
\ No newline at end of file
diff --git a/mnist_1_mps_feature_300.safetensors b/mnist_1_mps_feature_300.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..643b24cee2abdec765a1088e2a421e82b1a271a0
--- /dev/null
+++ b/mnist_1_mps_feature_300.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f1d5aceb240fa0dab268573d3d18bbee4eb1ff29b0e991f4d5728171be83aa23
+size 9502624
diff --git a/mnist_1_mps_feature_400.json b/mnist_1_mps_feature_400.json
new file mode 100644
index 0000000000000000000000000000000000000000..59a72153f040b5edbf2eb92e74ebabea293798a8
--- /dev/null
+++ b/mnist_1_mps_feature_400.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 400, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521, 238, 628, 237, 210, 207, 213, 239, 493, 265, 240, 632, 599, 212, 214, 241, 211, 605, 549, 236, 266, 235, 577, 242, 185, 494, 181, 572, 352, 234, 466, 627, 186, 518, 293, 268, 215, 269, 545, 182, 297, 184, 491, 292, 492, 264, 243, 180, 270, 267, 324, 438, 325, 183, 296, 658, 522, 380, 383, 410, 206, 465, 382, 271, 657, 298, 411, 355, 353, 262, 463, 578, 659, 263, 432, 598, 656, 354, 626, 299, 633, 439, 187, 320, 655, 291, 431, 571, 550, 327, 490, 261, 326, 404, 233, 319, 435, 294, 403, 379, 407, 606, 467, 244, 464, 402, 290, 544, 321, 179, 408, 430, 654, 216, 517, 381, 318, 437, 289, 374, 459, 346, 295, 272, 347, 460, 157, 436, 405, 375, 156, 495, 351, 462, 317, 433, 458, 155, 570, 345, 409, 300, 154, 158, 660, 625, 376, 373, 489, 401, 348, 461, 488, 523, 429, 551, 406, 597, 579, 434, 153, 377, 323, 516, 288, 487, 205, 349, 328, 457, 260, 316, 543, 486, 653, 378, 322, 344, 188, 356, 569, 372, 485, 178, 400, 384, 515, 245, 412, 634, 152, 232, 440, 159, 542, 350, 428, 468, 607, 624, 685, 541, 684, 273, 496, 514, 596, 524, 513, 456, 217, 552, 683, 484, 568, 686, 661, 540, 512, 301, 580, 371, 343, 399, 315, 151, 652, 287, 127, 128, 682, 427, 204, 595, 567, 455, 177, 687, 259, 129, 539, 483, 160, 497, 126, 511, 189, 525, 329, 469, 623, 441, 635, 553, 130, 125, 246, 231, 357, 608, 413, 274, 385, 150, 681, 566, 662, 688, 538, 218, 510, 594, 482, 454, 426, 398, 302, 370, 581, 131, 124, 470, 176, 342, 498, 651, 203, 161, 314, 526, 442, 286, 622, 190, 149, 330, 258, 689, 414, 636, 554, 537, 123, 680, 509, 565, 663, 358, 609, 386, 230, 132, 481, 247, 593, 453, 175, 275, 713, 712, 471, 219, 425, 443, 582, 711, 499, 162, 714, 148, 122, 202, 690, 397, 650, 303, 415, 710, 527, 369, 99, 715, 341, 313, 100, 387, 191, 621, 331, 98, 679, 637, 285, 664, 133, 359, 257, 101, 716, 555, 709, 97, 610, 121, 229, 102, 536, 174, 564, 691, 508, 147, 96, 717, 103, 163, 248, 480, 444, 472, 583], "train_min_nll": 17.70745086669922, "test_nll": 19.01351547241211}
\ No newline at end of file
diff --git a/mnist_1_mps_feature_400.safetensors b/mnist_1_mps_feature_400.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..f7b29764150556385e8fe5274dee39270b1bb8b5
--- /dev/null
+++ b/mnist_1_mps_feature_400.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:25239190b11e58b8504f53bdf86d62f930d82b4aa5a3c3726bce6a1d0f0fc83e
+size 12786904
diff --git a/mnist_1_mps_feature_50.json b/mnist_1_mps_feature_50.json
new file mode 100644
index 0000000000000000000000000000000000000000..f8349b1b77a27cbe551caf802980eaf941146004
--- /dev/null
+++ b/mnist_1_mps_feature_50.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 50, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521, 238, 628, 237, 210, 207, 213, 239, 493, 265, 240, 632, 599, 212, 214, 241, 211, 605, 549, 236, 266, 235, 577, 242, 185, 494, 181, 572, 352, 234, 466], "train_min_nll": 6.056156635284424, "test_nll": 6.304147243499756}
\ No newline at end of file
diff --git a/mnist_1_mps_feature_50.safetensors b/mnist_1_mps_feature_50.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..ad57227ade39ca6b7cad79e6a675abc02e32331e
--- /dev/null
+++ b/mnist_1_mps_feature_50.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7d7366295d773477700ef310a45f7a9ca90d89ad3510f8bb6ffc0b8e277f7267
+size 1292416
diff --git a/mnist_1_mps_feature_500.json b/mnist_1_mps_feature_500.json
new file mode 100644
index 0000000000000000000000000000000000000000..bfd89ef951f927935e8edd963f1cf4a98c402514
--- /dev/null
+++ b/mnist_1_mps_feature_500.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 500, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521, 238, 628, 237, 210, 207, 213, 239, 493, 265, 240, 632, 599, 212, 214, 241, 211, 605, 549, 236, 266, 235, 577, 242, 185, 494, 181, 572, 352, 234, 466, 627, 186, 518, 293, 268, 215, 269, 545, 182, 297, 184, 491, 292, 492, 264, 243, 180, 270, 267, 324, 438, 325, 183, 296, 658, 522, 380, 383, 410, 206, 465, 382, 271, 657, 298, 411, 355, 353, 262, 463, 578, 659, 263, 432, 598, 656, 354, 626, 299, 633, 439, 187, 320, 655, 291, 431, 571, 550, 327, 490, 261, 326, 404, 233, 319, 435, 294, 403, 379, 407, 606, 467, 244, 464, 402, 290, 544, 321, 179, 408, 430, 654, 216, 517, 381, 318, 437, 289, 374, 459, 346, 295, 272, 347, 460, 157, 436, 405, 375, 156, 495, 351, 462, 317, 433, 458, 155, 570, 345, 409, 300, 154, 158, 660, 625, 376, 373, 489, 401, 348, 461, 488, 523, 429, 551, 406, 597, 579, 434, 153, 377, 323, 516, 288, 487, 205, 349, 328, 457, 260, 316, 543, 486, 653, 378, 322, 344, 188, 356, 569, 372, 485, 178, 400, 384, 515, 245, 412, 634, 152, 232, 440, 159, 542, 350, 428, 468, 607, 624, 685, 541, 684, 273, 496, 514, 596, 524, 513, 456, 217, 552, 683, 484, 568, 686, 661, 540, 512, 301, 580, 371, 343, 399, 315, 151, 652, 287, 127, 128, 682, 427, 204, 595, 567, 455, 177, 687, 259, 129, 539, 483, 160, 497, 126, 511, 189, 525, 329, 469, 623, 441, 635, 553, 130, 125, 246, 231, 357, 608, 413, 274, 385, 150, 681, 566, 662, 688, 538, 218, 510, 594, 482, 454, 426, 398, 302, 370, 581, 131, 124, 470, 176, 342, 498, 651, 203, 161, 314, 526, 442, 286, 622, 190, 149, 330, 258, 689, 414, 636, 554, 537, 123, 680, 509, 565, 663, 358, 609, 386, 230, 132, 481, 247, 593, 453, 175, 275, 713, 712, 471, 219, 425, 443, 582, 711, 499, 162, 714, 148, 122, 202, 690, 397, 650, 303, 415, 710, 527, 369, 99, 715, 341, 313, 100, 387, 191, 621, 331, 98, 679, 637, 285, 664, 133, 359, 257, 101, 716, 555, 709, 97, 610, 121, 229, 102, 536, 174, 564, 691, 508, 147, 96, 717, 103, 163, 248, 480, 444, 472, 583, 592, 201, 708, 220, 276, 416, 95, 134, 500, 665, 649, 120, 638, 388, 452, 718, 678, 304, 740, 692, 192, 741, 528, 104, 739, 256, 284, 424, 742, 360, 312, 611, 146, 332, 738, 94, 620, 228, 173, 396, 340, 71, 368, 743, 707, 556, 719, 72, 744, 737, 70, 119, 164, 666, 73, 693, 105, 93, 584, 639, 249, 200, 135, 69, 745, 221, 563, 74, 535, 68, 736, 677, 277, 648, 720, 591, 145, 75, 706, 507, 193, 255, 283, 612, 746, 67, 92, 501, 473, 445, 305, 311, 172, 694, 227, 118, 667, 106, 529, 5], "train_min_nll": 17.70111656188965, "test_nll": 19.006507873535156}
\ No newline at end of file
diff --git a/mnist_1_mps_feature_500.safetensors b/mnist_1_mps_feature_500.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..d0bdb0a7698c8d80b5544797523a755f540970c4
--- /dev/null
+++ b/mnist_1_mps_feature_500.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4ebcd4525f65b7204bd0ffbb81bf2c9bafd7b7933bc55a8b6a10fe56917d0b41
+size 16071208
diff --git a/mnist_1_mps_feature_600.json b/mnist_1_mps_feature_600.json
new file mode 100644
index 0000000000000000000000000000000000000000..461486bbb9cf9936ef91806366030c9242cba64c
--- /dev/null
+++ b/mnist_1_mps_feature_600.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 600, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521, 238, 628, 237, 210, 207, 213, 239, 493, 265, 240, 632, 599, 212, 214, 241, 211, 605, 549, 236, 266, 235, 577, 242, 185, 494, 181, 572, 352, 234, 466, 627, 186, 518, 293, 268, 215, 269, 545, 182, 297, 184, 491, 292, 492, 264, 243, 180, 270, 267, 324, 438, 325, 183, 296, 658, 522, 380, 383, 410, 206, 465, 382, 271, 657, 298, 411, 355, 353, 262, 463, 578, 659, 263, 432, 598, 656, 354, 626, 299, 633, 439, 187, 320, 655, 291, 431, 571, 550, 327, 490, 261, 326, 404, 233, 319, 435, 294, 403, 379, 407, 606, 467, 244, 464, 402, 290, 544, 321, 179, 408, 430, 654, 216, 517, 381, 318, 437, 289, 374, 459, 346, 295, 272, 347, 460, 157, 436, 405, 375, 156, 495, 351, 462, 317, 433, 458, 155, 570, 345, 409, 300, 154, 158, 660, 625, 376, 373, 489, 401, 348, 461, 488, 523, 429, 551, 406, 597, 579, 434, 153, 377, 323, 516, 288, 487, 205, 349, 328, 457, 260, 316, 543, 486, 653, 378, 322, 344, 188, 356, 569, 372, 485, 178, 400, 384, 515, 245, 412, 634, 152, 232, 440, 159, 542, 350, 428, 468, 607, 624, 685, 541, 684, 273, 496, 514, 596, 524, 513, 456, 217, 552, 683, 484, 568, 686, 661, 540, 512, 301, 580, 371, 343, 399, 315, 151, 652, 287, 127, 128, 682, 427, 204, 595, 567, 455, 177, 687, 259, 129, 539, 483, 160, 497, 126, 511, 189, 525, 329, 469, 623, 441, 635, 553, 130, 125, 246, 231, 357, 608, 413, 274, 385, 150, 681, 566, 662, 688, 538, 218, 510, 594, 482, 454, 426, 398, 302, 370, 581, 131, 124, 470, 176, 342, 498, 651, 203, 161, 314, 526, 442, 286, 622, 190, 149, 330, 258, 689, 414, 636, 554, 537, 123, 680, 509, 565, 663, 358, 609, 386, 230, 132, 481, 247, 593, 453, 175, 275, 713, 712, 471, 219, 425, 443, 582, 711, 499, 162, 714, 148, 122, 202, 690, 397, 650, 303, 415, 710, 527, 369, 99, 715, 341, 313, 100, 387, 191, 621, 331, 98, 679, 637, 285, 664, 133, 359, 257, 101, 716, 555, 709, 97, 610, 121, 229, 102, 536, 174, 564, 691, 508, 147, 96, 717, 103, 163, 248, 480, 444, 472, 583, 592, 201, 708, 220, 276, 416, 95, 134, 500, 665, 649, 120, 638, 388, 452, 718, 678, 304, 740, 692, 192, 741, 528, 104, 739, 256, 284, 424, 742, 360, 312, 611, 146, 332, 738, 94, 620, 228, 173, 396, 340, 71, 368, 743, 707, 556, 719, 72, 744, 737, 70, 119, 164, 666, 73, 693, 105, 93, 584, 639, 249, 200, 135, 69, 745, 221, 563, 74, 535, 68, 736, 677, 277, 648, 720, 591, 145, 75, 706, 507, 193, 255, 283, 612, 746, 67, 92, 501, 473, 445, 305, 311, 172, 694, 227, 118, 667, 106, 529, 5, 735, 76, 479, 66, 417, 619, 721, 339, 136, 557, 747, 7, 4, 640, 451, 91, 367, 6, 333, 165, 389, 585, 423, 3, 77, 65, 676, 748, 770, 734, 199, 395, 705, 771, 695, 769, 361, 282, 117, 107, 768, 722, 254, 222, 43, 250, 772, 310, 144, 613, 647, 64, 767, 278, 668, 78, 562, 194, 90, 226, 766, 42, 749, 40, 44, 534, 773, 41, 590, 1, 306, 63, 338, 171, 2, 39, 108, 137, 733, 502, 765, 474, 723, 38, 704, 45, 618, 506, 446, 641, 478, 334, 366, 675, 558, 764, 774, 696, 46, 530], "train_min_nll": 17.446922302246094, "test_nll": 18.656747817993164}
\ No newline at end of file
diff --git a/mnist_1_mps_feature_600.safetensors b/mnist_1_mps_feature_600.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..328bff82819f03667a985aa8c1e436e5b60498e8
--- /dev/null
+++ b/mnist_1_mps_feature_600.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:32d39ffee8d1f63141e03e5d2d7b3f14566e8185a5edd72a348035f250205ec9
+size 19355504
diff --git a/mnist_2_losses.png b/mnist_2_losses.png
new file mode 100644
index 0000000000000000000000000000000000000000..4e4b7c032d859aa52e5b3b8756eb48213a8980fe
Binary files /dev/null and b/mnist_2_losses.png differ
diff --git a/mnist_2_mps.safetensors b/mnist_2_mps.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..a130064a5f8322eeb61f698f19bf84e52be91a09
--- /dev/null
+++ b/mnist_2_mps.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:06b485ff235a1a588ac5f0fcda4aa9163847999e90cd18c52bdcb8e4e873c8e1
+size 25398616
diff --git a/mnist_2_mps_feature_100.json b/mnist_2_mps_feature_100.json
new file mode 100644
index 0000000000000000000000000000000000000000..2cd71b0b612c1b321d82fed54c5f85c342593460
--- /dev/null
+++ b/mnist_2_mps_feature_100.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 100, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521, 238, 628, 237, 210, 207, 213, 239, 493, 265, 240, 632, 599, 212, 214, 241, 211, 605, 549, 236, 266, 235, 577, 242, 185, 494, 181, 572, 352, 234, 466, 627, 186, 518, 293, 268, 215, 269, 545, 182, 297, 184, 491, 292, 492, 264, 243, 180, 270, 267, 324, 438, 325, 183, 296, 658, 522, 380, 383, 410, 206, 465, 382, 271, 657, 298, 411, 355, 353, 262, 463, 578, 659, 263, 432, 598, 656, 354, 626, 299, 633], "train_min_nll": 23.115591049194336, "test_nll": 25.322721481323242}
\ No newline at end of file
diff --git a/mnist_2_mps_feature_100.safetensors b/mnist_2_mps_feature_100.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..7db1bac83d6783ccceb84162da02ed1c5c7e6c2e
--- /dev/null
+++ b/mnist_2_mps_feature_100.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:74de1b13c8840be707b2a308831c97fcfdd8dd9abff65af128226401c3831b00
+size 2934416
diff --git a/mnist_2_mps_feature_15.json b/mnist_2_mps_feature_15.json
new file mode 100644
index 0000000000000000000000000000000000000000..5047fd704de620db8baa85c71602473feb253b7e
--- /dev/null
+++ b/mnist_2_mps_feature_15.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 15, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573], "train_min_nll": 4.04571533203125, "test_nll": 4.04901647567749}
\ No newline at end of file
diff --git a/mnist_2_mps_feature_15.safetensors b/mnist_2_mps_feature_15.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..0ceadc75d8b96038d828a1b0433e51afe5226586
--- /dev/null
+++ b/mnist_2_mps_feature_15.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d1efde1a51191599e3a2d36505719677c5be57aa3084c4883aa3468519337abc
+size 143032
diff --git a/mnist_2_mps_feature_20.json b/mnist_2_mps_feature_20.json
new file mode 100644
index 0000000000000000000000000000000000000000..68b894640ea7a16364937a0dbebb43c40c313e85
--- /dev/null
+++ b/mnist_2_mps_feature_20.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 20, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521], "train_min_nll": 5.3108391761779785, "test_nll": 5.446595668792725}
\ No newline at end of file
diff --git a/mnist_2_mps_feature_20.safetensors b/mnist_2_mps_feature_20.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..78e989dfd6b5303188a2320bd784b270c25c0010
--- /dev/null
+++ b/mnist_2_mps_feature_20.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:de172215e5ae979a5fbac6c27e6e944bdb4680a0c9ad24fcce9dbf880bda7d7a
+size 307248
diff --git a/mnist_2_mps_feature_200.json b/mnist_2_mps_feature_200.json
new file mode 100644
index 0000000000000000000000000000000000000000..f6e7bcc02ff4ffff51acd1db903e3d25bb0cb6e1
--- /dev/null
+++ b/mnist_2_mps_feature_200.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 200, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521, 238, 628, 237, 210, 207, 213, 239, 493, 265, 240, 632, 599, 212, 214, 241, 211, 605, 549, 236, 266, 235, 577, 242, 185, 494, 181, 572, 352, 234, 466, 627, 186, 518, 293, 268, 215, 269, 545, 182, 297, 184, 491, 292, 492, 264, 243, 180, 270, 267, 324, 438, 325, 183, 296, 658, 522, 380, 383, 410, 206, 465, 382, 271, 657, 298, 411, 355, 353, 262, 463, 578, 659, 263, 432, 598, 656, 354, 626, 299, 633, 439, 187, 320, 655, 291, 431, 571, 550, 327, 490, 261, 326, 404, 233, 319, 435, 294, 403, 379, 407, 606, 467, 244, 464, 402, 290, 544, 321, 179, 408, 430, 654, 216, 517, 381, 318, 437, 289, 374, 459, 346, 295, 272, 347, 460, 157, 436, 405, 375, 156, 495, 351, 462, 317, 433, 458, 155, 570, 345, 409, 300, 154, 158, 660, 625, 376, 373, 489, 401, 348, 461, 488, 523, 429, 551, 406, 597, 579, 434, 153, 377, 323, 516, 288, 487, 205, 349, 328, 457, 260, 316, 543, 486, 653, 378, 322, 344, 188, 356, 569], "train_min_nll": 44.23739242553711, "test_nll": 49.481292724609375}
\ No newline at end of file
diff --git a/mnist_2_mps_feature_200.safetensors b/mnist_2_mps_feature_200.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..280da4ed7d3a5853775e7b0af75c713892e5e73f
--- /dev/null
+++ b/mnist_2_mps_feature_200.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4eafc8b23f77d08ba49da02b78c918974b520ff2fd9f41275a5dcb44f454c5cd
+size 6218528
diff --git a/mnist_2_mps_feature_300.json b/mnist_2_mps_feature_300.json
new file mode 100644
index 0000000000000000000000000000000000000000..a133e4cb0d5660ed7902e1517f9b21487ce091b3
--- /dev/null
+++ b/mnist_2_mps_feature_300.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 300, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521, 238, 628, 237, 210, 207, 213, 239, 493, 265, 240, 632, 599, 212, 214, 241, 211, 605, 549, 236, 266, 235, 577, 242, 185, 494, 181, 572, 352, 234, 466, 627, 186, 518, 293, 268, 215, 269, 545, 182, 297, 184, 491, 292, 492, 264, 243, 180, 270, 267, 324, 438, 325, 183, 296, 658, 522, 380, 383, 410, 206, 465, 382, 271, 657, 298, 411, 355, 353, 262, 463, 578, 659, 263, 432, 598, 656, 354, 626, 299, 633, 439, 187, 320, 655, 291, 431, 571, 550, 327, 490, 261, 326, 404, 233, 319, 435, 294, 403, 379, 407, 606, 467, 244, 464, 402, 290, 544, 321, 179, 408, 430, 654, 216, 517, 381, 318, 437, 289, 374, 459, 346, 295, 272, 347, 460, 157, 436, 405, 375, 156, 495, 351, 462, 317, 433, 458, 155, 570, 345, 409, 300, 154, 158, 660, 625, 376, 373, 489, 401, 348, 461, 488, 523, 429, 551, 406, 597, 579, 434, 153, 377, 323, 516, 288, 487, 205, 349, 328, 457, 260, 316, 543, 486, 653, 378, 322, 344, 188, 356, 569, 372, 485, 178, 400, 384, 515, 245, 412, 634, 152, 232, 440, 159, 542, 350, 428, 468, 607, 624, 685, 541, 684, 273, 496, 514, 596, 524, 513, 456, 217, 552, 683, 484, 568, 686, 661, 540, 512, 301, 580, 371, 343, 399, 315, 151, 652, 287, 127, 128, 682, 427, 204, 595, 567, 455, 177, 687, 259, 129, 539, 483, 160, 497, 126, 511, 189, 525, 329, 469, 623, 441, 635, 553, 130, 125, 246, 231, 357, 608, 413, 274, 385, 150, 681, 566, 662, 688, 538, 218, 510, 594, 482, 454, 426, 398, 302, 370, 581, 131, 124], "train_min_nll": 60.986175537109375, "test_nll": 68.40010070800781}
\ No newline at end of file
diff --git a/mnist_2_mps_feature_300.safetensors b/mnist_2_mps_feature_300.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..14e007db8887b815765f922f38960f79e6b368cf
--- /dev/null
+++ b/mnist_2_mps_feature_300.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:df087ba3b20cbd744e50ff95b3e0b85062af1f52f2ecb4c1e3c6e3e181aa0a0c
+size 9502624
diff --git a/mnist_2_mps_feature_400.json b/mnist_2_mps_feature_400.json
new file mode 100644
index 0000000000000000000000000000000000000000..b03033e4e7bab6066071860e0b7b121ef0aa44bc
--- /dev/null
+++ b/mnist_2_mps_feature_400.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 400, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521, 238, 628, 237, 210, 207, 213, 239, 493, 265, 240, 632, 599, 212, 214, 241, 211, 605, 549, 236, 266, 235, 577, 242, 185, 494, 181, 572, 352, 234, 466, 627, 186, 518, 293, 268, 215, 269, 545, 182, 297, 184, 491, 292, 492, 264, 243, 180, 270, 267, 324, 438, 325, 183, 296, 658, 522, 380, 383, 410, 206, 465, 382, 271, 657, 298, 411, 355, 353, 262, 463, 578, 659, 263, 432, 598, 656, 354, 626, 299, 633, 439, 187, 320, 655, 291, 431, 571, 550, 327, 490, 261, 326, 404, 233, 319, 435, 294, 403, 379, 407, 606, 467, 244, 464, 402, 290, 544, 321, 179, 408, 430, 654, 216, 517, 381, 318, 437, 289, 374, 459, 346, 295, 272, 347, 460, 157, 436, 405, 375, 156, 495, 351, 462, 317, 433, 458, 155, 570, 345, 409, 300, 154, 158, 660, 625, 376, 373, 489, 401, 348, 461, 488, 523, 429, 551, 406, 597, 579, 434, 153, 377, 323, 516, 288, 487, 205, 349, 328, 457, 260, 316, 543, 486, 653, 378, 322, 344, 188, 356, 569, 372, 485, 178, 400, 384, 515, 245, 412, 634, 152, 232, 440, 159, 542, 350, 428, 468, 607, 624, 685, 541, 684, 273, 496, 514, 596, 524, 513, 456, 217, 552, 683, 484, 568, 686, 661, 540, 512, 301, 580, 371, 343, 399, 315, 151, 652, 287, 127, 128, 682, 427, 204, 595, 567, 455, 177, 687, 259, 129, 539, 483, 160, 497, 126, 511, 189, 525, 329, 469, 623, 441, 635, 553, 130, 125, 246, 231, 357, 608, 413, 274, 385, 150, 681, 566, 662, 688, 538, 218, 510, 594, 482, 454, 426, 398, 302, 370, 581, 131, 124, 470, 176, 342, 498, 651, 203, 161, 314, 526, 442, 286, 622, 190, 149, 330, 258, 689, 414, 636, 554, 537, 123, 680, 509, 565, 663, 358, 609, 386, 230, 132, 481, 247, 593, 453, 175, 275, 713, 712, 471, 219, 425, 443, 582, 711, 499, 162, 714, 148, 122, 202, 690, 397, 650, 303, 415, 710, 527, 369, 99, 715, 341, 313, 100, 387, 191, 621, 331, 98, 679, 637, 285, 664, 133, 359, 257, 101, 716, 555, 709, 97, 610, 121, 229, 102, 536, 174, 564, 691, 508, 147, 96, 717, 103, 163, 248, 480, 444, 472, 583], "train_min_nll": 68.79653930664062, "test_nll": 77.2430191040039}
\ No newline at end of file
diff --git a/mnist_2_mps_feature_400.safetensors b/mnist_2_mps_feature_400.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..5c6c827549a3aec1dff560efcbd899b398b6d0b1
--- /dev/null
+++ b/mnist_2_mps_feature_400.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6b3b63c02728f40b4262364f3c3c1fc0588957fa6903212234035f41c169ee3c
+size 12786904
diff --git a/mnist_2_mps_feature_50.json b/mnist_2_mps_feature_50.json
new file mode 100644
index 0000000000000000000000000000000000000000..ae090e9543e49b71bd0f0892d75d18cc406fb477
--- /dev/null
+++ b/mnist_2_mps_feature_50.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 50, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521, 238, 628, 237, 210, 207, 213, 239, 493, 265, 240, 632, 599, 212, 214, 241, 211, 605, 549, 236, 266, 235, 577, 242, 185, 494, 181, 572, 352, 234, 466], "train_min_nll": 12.698987007141113, "test_nll": 13.628695487976074}
\ No newline at end of file
diff --git a/mnist_2_mps_feature_50.safetensors b/mnist_2_mps_feature_50.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..8162bcabcd844fd08b60e14119db6874e8e25ef3
--- /dev/null
+++ b/mnist_2_mps_feature_50.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ba44bbb31d60828b6dbf7aeafac6b716b3026624620fccf4a662dd077149138d
+size 1292416
diff --git a/mnist_2_mps_feature_500.json b/mnist_2_mps_feature_500.json
new file mode 100644
index 0000000000000000000000000000000000000000..85f196c786762d4937953ad439a51b0e4e5f34d0
--- /dev/null
+++ b/mnist_2_mps_feature_500.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 500, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521, 238, 628, 237, 210, 207, 213, 239, 493, 265, 240, 632, 599, 212, 214, 241, 211, 605, 549, 236, 266, 235, 577, 242, 185, 494, 181, 572, 352, 234, 466, 627, 186, 518, 293, 268, 215, 269, 545, 182, 297, 184, 491, 292, 492, 264, 243, 180, 270, 267, 324, 438, 325, 183, 296, 658, 522, 380, 383, 410, 206, 465, 382, 271, 657, 298, 411, 355, 353, 262, 463, 578, 659, 263, 432, 598, 656, 354, 626, 299, 633, 439, 187, 320, 655, 291, 431, 571, 550, 327, 490, 261, 326, 404, 233, 319, 435, 294, 403, 379, 407, 606, 467, 244, 464, 402, 290, 544, 321, 179, 408, 430, 654, 216, 517, 381, 318, 437, 289, 374, 459, 346, 295, 272, 347, 460, 157, 436, 405, 375, 156, 495, 351, 462, 317, 433, 458, 155, 570, 345, 409, 300, 154, 158, 660, 625, 376, 373, 489, 401, 348, 461, 488, 523, 429, 551, 406, 597, 579, 434, 153, 377, 323, 516, 288, 487, 205, 349, 328, 457, 260, 316, 543, 486, 653, 378, 322, 344, 188, 356, 569, 372, 485, 178, 400, 384, 515, 245, 412, 634, 152, 232, 440, 159, 542, 350, 428, 468, 607, 624, 685, 541, 684, 273, 496, 514, 596, 524, 513, 456, 217, 552, 683, 484, 568, 686, 661, 540, 512, 301, 580, 371, 343, 399, 315, 151, 652, 287, 127, 128, 682, 427, 204, 595, 567, 455, 177, 687, 259, 129, 539, 483, 160, 497, 126, 511, 189, 525, 329, 469, 623, 441, 635, 553, 130, 125, 246, 231, 357, 608, 413, 274, 385, 150, 681, 566, 662, 688, 538, 218, 510, 594, 482, 454, 426, 398, 302, 370, 581, 131, 124, 470, 176, 342, 498, 651, 203, 161, 314, 526, 442, 286, 622, 190, 149, 330, 258, 689, 414, 636, 554, 537, 123, 680, 509, 565, 663, 358, 609, 386, 230, 132, 481, 247, 593, 453, 175, 275, 713, 712, 471, 219, 425, 443, 582, 711, 499, 162, 714, 148, 122, 202, 690, 397, 650, 303, 415, 710, 527, 369, 99, 715, 341, 313, 100, 387, 191, 621, 331, 98, 679, 637, 285, 664, 133, 359, 257, 101, 716, 555, 709, 97, 610, 121, 229, 102, 536, 174, 564, 691, 508, 147, 96, 717, 103, 163, 248, 480, 444, 472, 583, 592, 201, 708, 220, 276, 416, 95, 134, 500, 665, 649, 120, 638, 388, 452, 718, 678, 304, 740, 692, 192, 741, 528, 104, 739, 256, 284, 424, 742, 360, 312, 611, 146, 332, 738, 94, 620, 228, 173, 396, 340, 71, 368, 743, 707, 556, 719, 72, 744, 737, 70, 119, 164, 666, 73, 693, 105, 93, 584, 639, 249, 200, 135, 69, 745, 221, 563, 74, 535, 68, 736, 677, 277, 648, 720, 591, 145, 75, 706, 507, 193, 255, 283, 612, 746, 67, 92, 501, 473, 445, 305, 311, 172, 694, 227, 118, 667, 106, 529, 5], "train_min_nll": 70.3482894897461, "test_nll": 79.00808715820312}
\ No newline at end of file
diff --git a/mnist_2_mps_feature_500.safetensors b/mnist_2_mps_feature_500.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..e41995c85bb22b8aedf4b82510a89fbe400bb11d
--- /dev/null
+++ b/mnist_2_mps_feature_500.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1714ccd29fcb46f7b9fa6b22d6d68a5a13f302dc60164b5f12631e59830734f9
+size 16071208
diff --git a/mnist_2_mps_feature_600.json b/mnist_2_mps_feature_600.json
new file mode 100644
index 0000000000000000000000000000000000000000..d59dd691e33f4242da89ca47e97c9688c3d0d922
--- /dev/null
+++ b/mnist_2_mps_feature_600.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 600, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521, 238, 628, 237, 210, 207, 213, 239, 493, 265, 240, 632, 599, 212, 214, 241, 211, 605, 549, 236, 266, 235, 577, 242, 185, 494, 181, 572, 352, 234, 466, 627, 186, 518, 293, 268, 215, 269, 545, 182, 297, 184, 491, 292, 492, 264, 243, 180, 270, 267, 324, 438, 325, 183, 296, 658, 522, 380, 383, 410, 206, 465, 382, 271, 657, 298, 411, 355, 353, 262, 463, 578, 659, 263, 432, 598, 656, 354, 626, 299, 633, 439, 187, 320, 655, 291, 431, 571, 550, 327, 490, 261, 326, 404, 233, 319, 435, 294, 403, 379, 407, 606, 467, 244, 464, 402, 290, 544, 321, 179, 408, 430, 654, 216, 517, 381, 318, 437, 289, 374, 459, 346, 295, 272, 347, 460, 157, 436, 405, 375, 156, 495, 351, 462, 317, 433, 458, 155, 570, 345, 409, 300, 154, 158, 660, 625, 376, 373, 489, 401, 348, 461, 488, 523, 429, 551, 406, 597, 579, 434, 153, 377, 323, 516, 288, 487, 205, 349, 328, 457, 260, 316, 543, 486, 653, 378, 322, 344, 188, 356, 569, 372, 485, 178, 400, 384, 515, 245, 412, 634, 152, 232, 440, 159, 542, 350, 428, 468, 607, 624, 685, 541, 684, 273, 496, 514, 596, 524, 513, 456, 217, 552, 683, 484, 568, 686, 661, 540, 512, 301, 580, 371, 343, 399, 315, 151, 652, 287, 127, 128, 682, 427, 204, 595, 567, 455, 177, 687, 259, 129, 539, 483, 160, 497, 126, 511, 189, 525, 329, 469, 623, 441, 635, 553, 130, 125, 246, 231, 357, 608, 413, 274, 385, 150, 681, 566, 662, 688, 538, 218, 510, 594, 482, 454, 426, 398, 302, 370, 581, 131, 124, 470, 176, 342, 498, 651, 203, 161, 314, 526, 442, 286, 622, 190, 149, 330, 258, 689, 414, 636, 554, 537, 123, 680, 509, 565, 663, 358, 609, 386, 230, 132, 481, 247, 593, 453, 175, 275, 713, 712, 471, 219, 425, 443, 582, 711, 499, 162, 714, 148, 122, 202, 690, 397, 650, 303, 415, 710, 527, 369, 99, 715, 341, 313, 100, 387, 191, 621, 331, 98, 679, 637, 285, 664, 133, 359, 257, 101, 716, 555, 709, 97, 610, 121, 229, 102, 536, 174, 564, 691, 508, 147, 96, 717, 103, 163, 248, 480, 444, 472, 583, 592, 201, 708, 220, 276, 416, 95, 134, 500, 665, 649, 120, 638, 388, 452, 718, 678, 304, 740, 692, 192, 741, 528, 104, 739, 256, 284, 424, 742, 360, 312, 611, 146, 332, 738, 94, 620, 228, 173, 396, 340, 71, 368, 743, 707, 556, 719, 72, 744, 737, 70, 119, 164, 666, 73, 693, 105, 93, 584, 639, 249, 200, 135, 69, 745, 221, 563, 74, 535, 68, 736, 677, 277, 648, 720, 591, 145, 75, 706, 507, 193, 255, 283, 612, 746, 67, 92, 501, 473, 445, 305, 311, 172, 694, 227, 118, 667, 106, 529, 5, 735, 76, 479, 66, 417, 619, 721, 339, 136, 557, 747, 7, 4, 640, 451, 91, 367, 6, 333, 165, 389, 585, 423, 3, 77, 65, 676, 748, 770, 734, 199, 395, 705, 771, 695, 769, 361, 282, 117, 107, 768, 722, 254, 222, 43, 250, 772, 310, 144, 613, 647, 64, 767, 278, 668, 78, 562, 194, 90, 226, 766, 42, 749, 40, 44, 534, 773, 41, 590, 1, 306, 63, 338, 171, 2, 39, 108, 137, 733, 502, 765, 474, 723, 38, 704, 45, 618, 506, 446, 641, 478, 334, 366, 675, 558, 764, 774, 696, 46, 530], "train_min_nll": 70.83859252929688, "test_nll": 79.60649108886719}
\ No newline at end of file
diff --git a/mnist_2_mps_feature_600.safetensors b/mnist_2_mps_feature_600.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..1ffa307729258ddfaf7a80aaf388f7f31b142eb4
--- /dev/null
+++ b/mnist_2_mps_feature_600.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9eecd04745e4bd404adb11f9b8ec7d69228ae571846d341f3ec0915a39ecf403
+size 19355504
diff --git a/mnist_3_losses.png b/mnist_3_losses.png
new file mode 100644
index 0000000000000000000000000000000000000000..36ebb5a7e4655f0f12b52ef1387f75cbc5243834
Binary files /dev/null and b/mnist_3_losses.png differ
diff --git a/mnist_3_mps.safetensors b/mnist_3_mps.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..c8f546b60abc780d7a3c60ca2ee5e16b8941c790
--- /dev/null
+++ b/mnist_3_mps.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:28ea69df1fdcdd50de56876877588d46594b65ce0e0cbc8ef71519eb7841ebcb
+size 25398616
diff --git a/mnist_3_mps_feature_100.json b/mnist_3_mps_feature_100.json
new file mode 100644
index 0000000000000000000000000000000000000000..f9b91ff30728b1a31a28caa04e202bcf3c6779e2
--- /dev/null
+++ b/mnist_3_mps_feature_100.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 100, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521, 238, 628, 237, 210, 207, 213, 239, 493, 265, 240, 632, 599, 212, 214, 241, 211, 605, 549, 236, 266, 235, 577, 242, 185, 494, 181, 572, 352, 234, 466, 627, 186, 518, 293, 268, 215, 269, 545, 182, 297, 184, 491, 292, 492, 264, 243, 180, 270, 267, 324, 438, 325, 183, 296, 658, 522, 380, 383, 410, 206, 465, 382, 271, 657, 298, 411, 355, 353, 262, 463, 578, 659, 263, 432, 598, 656, 354, 626, 299, 633], "train_min_nll": 22.55504035949707, "test_nll": 25.01523780822754}
\ No newline at end of file
diff --git a/mnist_3_mps_feature_100.safetensors b/mnist_3_mps_feature_100.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..7d0a4e2b15af5a1de6ebd5c962dc9797c03bef8b
--- /dev/null
+++ b/mnist_3_mps_feature_100.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d743ef32ad6ccd4c1db5b4f0d7aa0928f10e936ca519bfd48a8f596fad194239
+size 2934416
diff --git a/mnist_3_mps_feature_15.json b/mnist_3_mps_feature_15.json
new file mode 100644
index 0000000000000000000000000000000000000000..c1943e3eb0be71925e5be5dcc47b3364e6bd51e0
--- /dev/null
+++ b/mnist_3_mps_feature_15.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 15, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573], "train_min_nll": 3.370436906814575, "test_nll": 3.3700966835021973}
\ No newline at end of file
diff --git a/mnist_3_mps_feature_15.safetensors b/mnist_3_mps_feature_15.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..89ed6d1e025f0e4065a62ef05a54f56cc649061d
--- /dev/null
+++ b/mnist_3_mps_feature_15.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e800de97e3d75acc17e58182f227897e07b217b8e0011589f865b9621d59d2d9
+size 143032
diff --git a/mnist_3_mps_feature_20.json b/mnist_3_mps_feature_20.json
new file mode 100644
index 0000000000000000000000000000000000000000..1de4c1b07a7be07b7c3181148ca224178b035ec7
--- /dev/null
+++ b/mnist_3_mps_feature_20.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 20, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521], "train_min_nll": 4.661399841308594, "test_nll": 4.702103137969971}
\ No newline at end of file
diff --git a/mnist_3_mps_feature_20.safetensors b/mnist_3_mps_feature_20.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..35981798e50516294056c75446146edefb73df87
--- /dev/null
+++ b/mnist_3_mps_feature_20.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a90abc1cdcf567522f70cbd70e1d90b97bf6080f63351e68d000128c29716157
+size 307248
diff --git a/mnist_3_mps_feature_200.json b/mnist_3_mps_feature_200.json
new file mode 100644
index 0000000000000000000000000000000000000000..c929299fdb339ec264b2301dadbe578b11c4bda1
--- /dev/null
+++ b/mnist_3_mps_feature_200.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 200, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521, 238, 628, 237, 210, 207, 213, 239, 493, 265, 240, 632, 599, 212, 214, 241, 211, 605, 549, 236, 266, 235, 577, 242, 185, 494, 181, 572, 352, 234, 466, 627, 186, 518, 293, 268, 215, 269, 545, 182, 297, 184, 491, 292, 492, 264, 243, 180, 270, 267, 324, 438, 325, 183, 296, 658, 522, 380, 383, 410, 206, 465, 382, 271, 657, 298, 411, 355, 353, 262, 463, 578, 659, 263, 432, 598, 656, 354, 626, 299, 633, 439, 187, 320, 655, 291, 431, 571, 550, 327, 490, 261, 326, 404, 233, 319, 435, 294, 403, 379, 407, 606, 467, 244, 464, 402, 290, 544, 321, 179, 408, 430, 654, 216, 517, 381, 318, 437, 289, 374, 459, 346, 295, 272, 347, 460, 157, 436, 405, 375, 156, 495, 351, 462, 317, 433, 458, 155, 570, 345, 409, 300, 154, 158, 660, 625, 376, 373, 489, 401, 348, 461, 488, 523, 429, 551, 406, 597, 579, 434, 153, 377, 323, 516, 288, 487, 205, 349, 328, 457, 260, 316, 543, 486, 653, 378, 322, 344, 188, 356, 569], "train_min_nll": 44.20879364013672, "test_nll": 49.7523193359375}
\ No newline at end of file
diff --git a/mnist_3_mps_feature_200.safetensors b/mnist_3_mps_feature_200.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..32193eb6cfff25f75dd54447e05d07f17d300e65
--- /dev/null
+++ b/mnist_3_mps_feature_200.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:aa6e9f1b702aeb26bf514ca7ef8a3d056d17916c866ed96cfdf134bcbd281ec2
+size 6218528
diff --git a/mnist_3_mps_feature_300.json b/mnist_3_mps_feature_300.json
new file mode 100644
index 0000000000000000000000000000000000000000..aeec697d67b5e6986ff61336562a9cd5659f45e3
--- /dev/null
+++ b/mnist_3_mps_feature_300.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 300, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521, 238, 628, 237, 210, 207, 213, 239, 493, 265, 240, 632, 599, 212, 214, 241, 211, 605, 549, 236, 266, 235, 577, 242, 185, 494, 181, 572, 352, 234, 466, 627, 186, 518, 293, 268, 215, 269, 545, 182, 297, 184, 491, 292, 492, 264, 243, 180, 270, 267, 324, 438, 325, 183, 296, 658, 522, 380, 383, 410, 206, 465, 382, 271, 657, 298, 411, 355, 353, 262, 463, 578, 659, 263, 432, 598, 656, 354, 626, 299, 633, 439, 187, 320, 655, 291, 431, 571, 550, 327, 490, 261, 326, 404, 233, 319, 435, 294, 403, 379, 407, 606, 467, 244, 464, 402, 290, 544, 321, 179, 408, 430, 654, 216, 517, 381, 318, 437, 289, 374, 459, 346, 295, 272, 347, 460, 157, 436, 405, 375, 156, 495, 351, 462, 317, 433, 458, 155, 570, 345, 409, 300, 154, 158, 660, 625, 376, 373, 489, 401, 348, 461, 488, 523, 429, 551, 406, 597, 579, 434, 153, 377, 323, 516, 288, 487, 205, 349, 328, 457, 260, 316, 543, 486, 653, 378, 322, 344, 188, 356, 569, 372, 485, 178, 400, 384, 515, 245, 412, 634, 152, 232, 440, 159, 542, 350, 428, 468, 607, 624, 685, 541, 684, 273, 496, 514, 596, 524, 513, 456, 217, 552, 683, 484, 568, 686, 661, 540, 512, 301, 580, 371, 343, 399, 315, 151, 652, 287, 127, 128, 682, 427, 204, 595, 567, 455, 177, 687, 259, 129, 539, 483, 160, 497, 126, 511, 189, 525, 329, 469, 623, 441, 635, 553, 130, 125, 246, 231, 357, 608, 413, 274, 385, 150, 681, 566, 662, 688, 538, 218, 510, 594, 482, 454, 426, 398, 302, 370, 581, 131, 124], "train_min_nll": 57.198429107666016, "test_nll": 64.73218536376953}
\ No newline at end of file
diff --git a/mnist_3_mps_feature_300.safetensors b/mnist_3_mps_feature_300.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..fbcec944b65f0c8e1d8b01e59780fb64814060ca
--- /dev/null
+++ b/mnist_3_mps_feature_300.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ecf51d58eb9ddaf0503253f4213c2a68b0492a5a532651a41d406082eab1a0d1
+size 9502624
diff --git a/mnist_3_mps_feature_400.json b/mnist_3_mps_feature_400.json
new file mode 100644
index 0000000000000000000000000000000000000000..26f19020a25802376fd8d7722f6a980b9b771d2b
--- /dev/null
+++ b/mnist_3_mps_feature_400.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 400, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521, 238, 628, 237, 210, 207, 213, 239, 493, 265, 240, 632, 599, 212, 214, 241, 211, 605, 549, 236, 266, 235, 577, 242, 185, 494, 181, 572, 352, 234, 466, 627, 186, 518, 293, 268, 215, 269, 545, 182, 297, 184, 491, 292, 492, 264, 243, 180, 270, 267, 324, 438, 325, 183, 296, 658, 522, 380, 383, 410, 206, 465, 382, 271, 657, 298, 411, 355, 353, 262, 463, 578, 659, 263, 432, 598, 656, 354, 626, 299, 633, 439, 187, 320, 655, 291, 431, 571, 550, 327, 490, 261, 326, 404, 233, 319, 435, 294, 403, 379, 407, 606, 467, 244, 464, 402, 290, 544, 321, 179, 408, 430, 654, 216, 517, 381, 318, 437, 289, 374, 459, 346, 295, 272, 347, 460, 157, 436, 405, 375, 156, 495, 351, 462, 317, 433, 458, 155, 570, 345, 409, 300, 154, 158, 660, 625, 376, 373, 489, 401, 348, 461, 488, 523, 429, 551, 406, 597, 579, 434, 153, 377, 323, 516, 288, 487, 205, 349, 328, 457, 260, 316, 543, 486, 653, 378, 322, 344, 188, 356, 569, 372, 485, 178, 400, 384, 515, 245, 412, 634, 152, 232, 440, 159, 542, 350, 428, 468, 607, 624, 685, 541, 684, 273, 496, 514, 596, 524, 513, 456, 217, 552, 683, 484, 568, 686, 661, 540, 512, 301, 580, 371, 343, 399, 315, 151, 652, 287, 127, 128, 682, 427, 204, 595, 567, 455, 177, 687, 259, 129, 539, 483, 160, 497, 126, 511, 189, 525, 329, 469, 623, 441, 635, 553, 130, 125, 246, 231, 357, 608, 413, 274, 385, 150, 681, 566, 662, 688, 538, 218, 510, 594, 482, 454, 426, 398, 302, 370, 581, 131, 124, 470, 176, 342, 498, 651, 203, 161, 314, 526, 442, 286, 622, 190, 149, 330, 258, 689, 414, 636, 554, 537, 123, 680, 509, 565, 663, 358, 609, 386, 230, 132, 481, 247, 593, 453, 175, 275, 713, 712, 471, 219, 425, 443, 582, 711, 499, 162, 714, 148, 122, 202, 690, 397, 650, 303, 415, 710, 527, 369, 99, 715, 341, 313, 100, 387, 191, 621, 331, 98, 679, 637, 285, 664, 133, 359, 257, 101, 716, 555, 709, 97, 610, 121, 229, 102, 536, 174, 564, 691, 508, 147, 96, 717, 103, 163, 248, 480, 444, 472, 583], "train_min_nll": 61.988441467285156, "test_nll": 70.14553833007812}
\ No newline at end of file
diff --git a/mnist_3_mps_feature_400.safetensors b/mnist_3_mps_feature_400.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..66a6a4223e49c174dd5188924e0bdc6f3fecf998
--- /dev/null
+++ b/mnist_3_mps_feature_400.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:75030032dfef2a784b6a9ea2fdb79c349548cc1c6d2dafdfacde22531ef894ee
+size 12786904
diff --git a/mnist_3_mps_feature_50.json b/mnist_3_mps_feature_50.json
new file mode 100644
index 0000000000000000000000000000000000000000..bc1f400a58254224f5dbbdce94a075a4eb34a364
--- /dev/null
+++ b/mnist_3_mps_feature_50.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 50, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521, 238, 628, 237, 210, 207, 213, 239, 493, 265, 240, 632, 599, 212, 214, 241, 211, 605, 549, 236, 266, 235, 577, 242, 185, 494, 181, 572, 352, 234, 466], "train_min_nll": 12.115177154541016, "test_nll": 13.280707359313965}
\ No newline at end of file
diff --git a/mnist_3_mps_feature_50.safetensors b/mnist_3_mps_feature_50.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..695b722237e778c7d4acf884fb97bfa6c8f33806
--- /dev/null
+++ b/mnist_3_mps_feature_50.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:acd0b9cb9d7d6cb47b134a7dac75a956222046cc9e6bb65302a0e2f223b5d423
+size 1292416
diff --git a/mnist_3_mps_feature_500.json b/mnist_3_mps_feature_500.json
new file mode 100644
index 0000000000000000000000000000000000000000..3218721f21afbdb595b33a3f8a3b61c3dbf8279d
--- /dev/null
+++ b/mnist_3_mps_feature_500.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 500, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521, 238, 628, 237, 210, 207, 213, 239, 493, 265, 240, 632, 599, 212, 214, 241, 211, 605, 549, 236, 266, 235, 577, 242, 185, 494, 181, 572, 352, 234, 466, 627, 186, 518, 293, 268, 215, 269, 545, 182, 297, 184, 491, 292, 492, 264, 243, 180, 270, 267, 324, 438, 325, 183, 296, 658, 522, 380, 383, 410, 206, 465, 382, 271, 657, 298, 411, 355, 353, 262, 463, 578, 659, 263, 432, 598, 656, 354, 626, 299, 633, 439, 187, 320, 655, 291, 431, 571, 550, 327, 490, 261, 326, 404, 233, 319, 435, 294, 403, 379, 407, 606, 467, 244, 464, 402, 290, 544, 321, 179, 408, 430, 654, 216, 517, 381, 318, 437, 289, 374, 459, 346, 295, 272, 347, 460, 157, 436, 405, 375, 156, 495, 351, 462, 317, 433, 458, 155, 570, 345, 409, 300, 154, 158, 660, 625, 376, 373, 489, 401, 348, 461, 488, 523, 429, 551, 406, 597, 579, 434, 153, 377, 323, 516, 288, 487, 205, 349, 328, 457, 260, 316, 543, 486, 653, 378, 322, 344, 188, 356, 569, 372, 485, 178, 400, 384, 515, 245, 412, 634, 152, 232, 440, 159, 542, 350, 428, 468, 607, 624, 685, 541, 684, 273, 496, 514, 596, 524, 513, 456, 217, 552, 683, 484, 568, 686, 661, 540, 512, 301, 580, 371, 343, 399, 315, 151, 652, 287, 127, 128, 682, 427, 204, 595, 567, 455, 177, 687, 259, 129, 539, 483, 160, 497, 126, 511, 189, 525, 329, 469, 623, 441, 635, 553, 130, 125, 246, 231, 357, 608, 413, 274, 385, 150, 681, 566, 662, 688, 538, 218, 510, 594, 482, 454, 426, 398, 302, 370, 581, 131, 124, 470, 176, 342, 498, 651, 203, 161, 314, 526, 442, 286, 622, 190, 149, 330, 258, 689, 414, 636, 554, 537, 123, 680, 509, 565, 663, 358, 609, 386, 230, 132, 481, 247, 593, 453, 175, 275, 713, 712, 471, 219, 425, 443, 582, 711, 499, 162, 714, 148, 122, 202, 690, 397, 650, 303, 415, 710, 527, 369, 99, 715, 341, 313, 100, 387, 191, 621, 331, 98, 679, 637, 285, 664, 133, 359, 257, 101, 716, 555, 709, 97, 610, 121, 229, 102, 536, 174, 564, 691, 508, 147, 96, 717, 103, 163, 248, 480, 444, 472, 583, 592, 201, 708, 220, 276, 416, 95, 134, 500, 665, 649, 120, 638, 388, 452, 718, 678, 304, 740, 692, 192, 741, 528, 104, 739, 256, 284, 424, 742, 360, 312, 611, 146, 332, 738, 94, 620, 228, 173, 396, 340, 71, 368, 743, 707, 556, 719, 72, 744, 737, 70, 119, 164, 666, 73, 693, 105, 93, 584, 639, 249, 200, 135, 69, 745, 221, 563, 74, 535, 68, 736, 677, 277, 648, 720, 591, 145, 75, 706, 507, 193, 255, 283, 612, 746, 67, 92, 501, 473, 445, 305, 311, 172, 694, 227, 118, 667, 106, 529, 5], "train_min_nll": 62.92741775512695, "test_nll": 71.11933898925781}
\ No newline at end of file
diff --git a/mnist_3_mps_feature_500.safetensors b/mnist_3_mps_feature_500.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..092d2b09add85ebd9f90d8d6464d9c249a095a2b
--- /dev/null
+++ b/mnist_3_mps_feature_500.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:730577b34cec8c424e0f4e3b74405dfa046acc5193d115447bbe6792f127f3eb
+size 16071208
diff --git a/mnist_3_mps_feature_600.json b/mnist_3_mps_feature_600.json
new file mode 100644
index 0000000000000000000000000000000000000000..e94beb285ae9479ec7afb7870954908c3dc8ebab
--- /dev/null
+++ b/mnist_3_mps_feature_600.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 600, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521, 238, 628, 237, 210, 207, 213, 239, 493, 265, 240, 632, 599, 212, 214, 241, 211, 605, 549, 236, 266, 235, 577, 242, 185, 494, 181, 572, 352, 234, 466, 627, 186, 518, 293, 268, 215, 269, 545, 182, 297, 184, 491, 292, 492, 264, 243, 180, 270, 267, 324, 438, 325, 183, 296, 658, 522, 380, 383, 410, 206, 465, 382, 271, 657, 298, 411, 355, 353, 262, 463, 578, 659, 263, 432, 598, 656, 354, 626, 299, 633, 439, 187, 320, 655, 291, 431, 571, 550, 327, 490, 261, 326, 404, 233, 319, 435, 294, 403, 379, 407, 606, 467, 244, 464, 402, 290, 544, 321, 179, 408, 430, 654, 216, 517, 381, 318, 437, 289, 374, 459, 346, 295, 272, 347, 460, 157, 436, 405, 375, 156, 495, 351, 462, 317, 433, 458, 155, 570, 345, 409, 300, 154, 158, 660, 625, 376, 373, 489, 401, 348, 461, 488, 523, 429, 551, 406, 597, 579, 434, 153, 377, 323, 516, 288, 487, 205, 349, 328, 457, 260, 316, 543, 486, 653, 378, 322, 344, 188, 356, 569, 372, 485, 178, 400, 384, 515, 245, 412, 634, 152, 232, 440, 159, 542, 350, 428, 468, 607, 624, 685, 541, 684, 273, 496, 514, 596, 524, 513, 456, 217, 552, 683, 484, 568, 686, 661, 540, 512, 301, 580, 371, 343, 399, 315, 151, 652, 287, 127, 128, 682, 427, 204, 595, 567, 455, 177, 687, 259, 129, 539, 483, 160, 497, 126, 511, 189, 525, 329, 469, 623, 441, 635, 553, 130, 125, 246, 231, 357, 608, 413, 274, 385, 150, 681, 566, 662, 688, 538, 218, 510, 594, 482, 454, 426, 398, 302, 370, 581, 131, 124, 470, 176, 342, 498, 651, 203, 161, 314, 526, 442, 286, 622, 190, 149, 330, 258, 689, 414, 636, 554, 537, 123, 680, 509, 565, 663, 358, 609, 386, 230, 132, 481, 247, 593, 453, 175, 275, 713, 712, 471, 219, 425, 443, 582, 711, 499, 162, 714, 148, 122, 202, 690, 397, 650, 303, 415, 710, 527, 369, 99, 715, 341, 313, 100, 387, 191, 621, 331, 98, 679, 637, 285, 664, 133, 359, 257, 101, 716, 555, 709, 97, 610, 121, 229, 102, 536, 174, 564, 691, 508, 147, 96, 717, 103, 163, 248, 480, 444, 472, 583, 592, 201, 708, 220, 276, 416, 95, 134, 500, 665, 649, 120, 638, 388, 452, 718, 678, 304, 740, 692, 192, 741, 528, 104, 739, 256, 284, 424, 742, 360, 312, 611, 146, 332, 738, 94, 620, 228, 173, 396, 340, 71, 368, 743, 707, 556, 719, 72, 744, 737, 70, 119, 164, 666, 73, 693, 105, 93, 584, 639, 249, 200, 135, 69, 745, 221, 563, 74, 535, 68, 736, 677, 277, 648, 720, 591, 145, 75, 706, 507, 193, 255, 283, 612, 746, 67, 92, 501, 473, 445, 305, 311, 172, 694, 227, 118, 667, 106, 529, 5, 735, 76, 479, 66, 417, 619, 721, 339, 136, 557, 747, 7, 4, 640, 451, 91, 367, 6, 333, 165, 389, 585, 423, 3, 77, 65, 676, 748, 770, 734, 199, 395, 705, 771, 695, 769, 361, 282, 117, 107, 768, 722, 254, 222, 43, 250, 772, 310, 144, 613, 647, 64, 767, 278, 668, 78, 562, 194, 90, 226, 766, 42, 749, 40, 44, 534, 773, 41, 590, 1, 306, 63, 338, 171, 2, 39, 108, 137, 733, 502, 765, 474, 723, 38, 704, 45, 618, 506, 446, 641, 478, 334, 366, 675, 558, 764, 774, 696, 46, 530], "train_min_nll": 62.563480377197266, "test_nll": 70.5661849975586}
\ No newline at end of file
diff --git a/mnist_3_mps_feature_600.safetensors b/mnist_3_mps_feature_600.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..6bfee3c364558964c88ed6905bd332ef07640fd1
--- /dev/null
+++ b/mnist_3_mps_feature_600.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5063c6d3a40a81c5b415199ce76e7b34a30dfb54e52b2ba8e58f464710169410
+size 19355504
diff --git a/mnist_4_losses.png b/mnist_4_losses.png
new file mode 100644
index 0000000000000000000000000000000000000000..2f829ac2416245c7936b8711651a38d0066a7d02
Binary files /dev/null and b/mnist_4_losses.png differ
diff --git a/mnist_4_mps.safetensors b/mnist_4_mps.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..f625e7f7bb2e58002faf5a3014535a9e643f30a5
--- /dev/null
+++ b/mnist_4_mps.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:77aba9f28fa8b074dbfe89d843bc9428df81c64d73dbd0c4c619aa06428818fe
+size 25398616
diff --git a/mnist_4_mps_feature_100.json b/mnist_4_mps_feature_100.json
new file mode 100644
index 0000000000000000000000000000000000000000..1fa9a883f80f440cca6d79b027f3239ec825e088
--- /dev/null
+++ b/mnist_4_mps_feature_100.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 100, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521, 238, 628, 237, 210, 207, 213, 239, 493, 265, 240, 632, 599, 212, 214, 241, 211, 605, 549, 236, 266, 235, 577, 242, 185, 494, 181, 572, 352, 234, 466, 627, 186, 518, 293, 268, 215, 269, 545, 182, 297, 184, 491, 292, 492, 264, 243, 180, 270, 267, 324, 438, 325, 183, 296, 658, 522, 380, 383, 410, 206, 465, 382, 271, 657, 298, 411, 355, 353, 262, 463, 578, 659, 263, 432, 598, 656, 354, 626, 299, 633], "train_min_nll": 18.90455436706543, "test_nll": 20.683218002319336}
\ No newline at end of file
diff --git a/mnist_4_mps_feature_100.safetensors b/mnist_4_mps_feature_100.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..577103e4ca33264ed353dc5a34eee9b62c43fe14
--- /dev/null
+++ b/mnist_4_mps_feature_100.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:43c0700cd8c5a41f525f8b0ee70380c4bfb68518ec10fa5cade28518573c1e78
+size 2934416
diff --git a/mnist_4_mps_feature_15.json b/mnist_4_mps_feature_15.json
new file mode 100644
index 0000000000000000000000000000000000000000..9a65d346715dd93984e01120e5b8b64e0e4d4e9d
--- /dev/null
+++ b/mnist_4_mps_feature_15.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 15, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573], "train_min_nll": 3.3762950897216797, "test_nll": 3.3462376594543457}
\ No newline at end of file
diff --git a/mnist_4_mps_feature_15.safetensors b/mnist_4_mps_feature_15.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..ea6f794b958cc88f2a958e499ca89ea3798384cc
--- /dev/null
+++ b/mnist_4_mps_feature_15.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:57809daf91b6c6b1cc8b79eab280f8d278381e3faff4ba792d740d6b9a0ca869
+size 143032
diff --git a/mnist_4_mps_feature_20.json b/mnist_4_mps_feature_20.json
new file mode 100644
index 0000000000000000000000000000000000000000..66e24a7f3eb077264ce1d8c2b5edf1f23a16624f
--- /dev/null
+++ b/mnist_4_mps_feature_20.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 20, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521], "train_min_nll": 4.237612247467041, "test_nll": 4.268853664398193}
\ No newline at end of file
diff --git a/mnist_4_mps_feature_20.safetensors b/mnist_4_mps_feature_20.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..ac8f11e7cd58ad52c926d4594bf2307a12d6f14a
--- /dev/null
+++ b/mnist_4_mps_feature_20.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:aa5b416a4258392abf9cf45ad406f15ba374ddf57ee5470328e56ecca424fb3a
+size 307248
diff --git a/mnist_4_mps_feature_200.json b/mnist_4_mps_feature_200.json
new file mode 100644
index 0000000000000000000000000000000000000000..bd68c2083570429a63b37b208afc6d285917dbe7
--- /dev/null
+++ b/mnist_4_mps_feature_200.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 200, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521, 238, 628, 237, 210, 207, 213, 239, 493, 265, 240, 632, 599, 212, 214, 241, 211, 605, 549, 236, 266, 235, 577, 242, 185, 494, 181, 572, 352, 234, 466, 627, 186, 518, 293, 268, 215, 269, 545, 182, 297, 184, 491, 292, 492, 264, 243, 180, 270, 267, 324, 438, 325, 183, 296, 658, 522, 380, 383, 410, 206, 465, 382, 271, 657, 298, 411, 355, 353, 262, 463, 578, 659, 263, 432, 598, 656, 354, 626, 299, 633, 439, 187, 320, 655, 291, 431, 571, 550, 327, 490, 261, 326, 404, 233, 319, 435, 294, 403, 379, 407, 606, 467, 244, 464, 402, 290, 544, 321, 179, 408, 430, 654, 216, 517, 381, 318, 437, 289, 374, 459, 346, 295, 272, 347, 460, 157, 436, 405, 375, 156, 495, 351, 462, 317, 433, 458, 155, 570, 345, 409, 300, 154, 158, 660, 625, 376, 373, 489, 401, 348, 461, 488, 523, 429, 551, 406, 597, 579, 434, 153, 377, 323, 516, 288, 487, 205, 349, 328, 457, 260, 316, 543, 486, 653, 378, 322, 344, 188, 356, 569], "train_min_nll": 38.21324157714844, "test_nll": 42.76984786987305}
\ No newline at end of file
diff --git a/mnist_4_mps_feature_200.safetensors b/mnist_4_mps_feature_200.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..41348db7f3e5cbb5399d03d7e8dcf6c26c3b893e
--- /dev/null
+++ b/mnist_4_mps_feature_200.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:45c216fc0482fe0120fa40d3a6dc354e4f4bed12ac1eb21da86d018fb6c232c6
+size 6218528
diff --git a/mnist_4_mps_feature_300.json b/mnist_4_mps_feature_300.json
new file mode 100644
index 0000000000000000000000000000000000000000..9b75a0e449e377412afa8fd4cbe5e24b78fe575e
--- /dev/null
+++ b/mnist_4_mps_feature_300.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 300, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521, 238, 628, 237, 210, 207, 213, 239, 493, 265, 240, 632, 599, 212, 214, 241, 211, 605, 549, 236, 266, 235, 577, 242, 185, 494, 181, 572, 352, 234, 466, 627, 186, 518, 293, 268, 215, 269, 545, 182, 297, 184, 491, 292, 492, 264, 243, 180, 270, 267, 324, 438, 325, 183, 296, 658, 522, 380, 383, 410, 206, 465, 382, 271, 657, 298, 411, 355, 353, 262, 463, 578, 659, 263, 432, 598, 656, 354, 626, 299, 633, 439, 187, 320, 655, 291, 431, 571, 550, 327, 490, 261, 326, 404, 233, 319, 435, 294, 403, 379, 407, 606, 467, 244, 464, 402, 290, 544, 321, 179, 408, 430, 654, 216, 517, 381, 318, 437, 289, 374, 459, 346, 295, 272, 347, 460, 157, 436, 405, 375, 156, 495, 351, 462, 317, 433, 458, 155, 570, 345, 409, 300, 154, 158, 660, 625, 376, 373, 489, 401, 348, 461, 488, 523, 429, 551, 406, 597, 579, 434, 153, 377, 323, 516, 288, 487, 205, 349, 328, 457, 260, 316, 543, 486, 653, 378, 322, 344, 188, 356, 569, 372, 485, 178, 400, 384, 515, 245, 412, 634, 152, 232, 440, 159, 542, 350, 428, 468, 607, 624, 685, 541, 684, 273, 496, 514, 596, 524, 513, 456, 217, 552, 683, 484, 568, 686, 661, 540, 512, 301, 580, 371, 343, 399, 315, 151, 652, 287, 127, 128, 682, 427, 204, 595, 567, 455, 177, 687, 259, 129, 539, 483, 160, 497, 126, 511, 189, 525, 329, 469, 623, 441, 635, 553, 130, 125, 246, 231, 357, 608, 413, 274, 385, 150, 681, 566, 662, 688, 538, 218, 510, 594, 482, 454, 426, 398, 302, 370, 581, 131, 124], "train_min_nll": 49.8540153503418, "test_nll": 56.35111999511719}
\ No newline at end of file
diff --git a/mnist_4_mps_feature_300.safetensors b/mnist_4_mps_feature_300.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..dcdc0348ec27d2154d358f927c96545df7d0df91
--- /dev/null
+++ b/mnist_4_mps_feature_300.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f054a2b846999b0aa6736ddbd2cb850ad5fa5c762beb741aeb1bca1e647f2960
+size 9502624
diff --git a/mnist_4_mps_feature_400.json b/mnist_4_mps_feature_400.json
new file mode 100644
index 0000000000000000000000000000000000000000..befd0a0ed1706d5bff573f22fc1be6aa4ce2144a
--- /dev/null
+++ b/mnist_4_mps_feature_400.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 400, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521, 238, 628, 237, 210, 207, 213, 239, 493, 265, 240, 632, 599, 212, 214, 241, 211, 605, 549, 236, 266, 235, 577, 242, 185, 494, 181, 572, 352, 234, 466, 627, 186, 518, 293, 268, 215, 269, 545, 182, 297, 184, 491, 292, 492, 264, 243, 180, 270, 267, 324, 438, 325, 183, 296, 658, 522, 380, 383, 410, 206, 465, 382, 271, 657, 298, 411, 355, 353, 262, 463, 578, 659, 263, 432, 598, 656, 354, 626, 299, 633, 439, 187, 320, 655, 291, 431, 571, 550, 327, 490, 261, 326, 404, 233, 319, 435, 294, 403, 379, 407, 606, 467, 244, 464, 402, 290, 544, 321, 179, 408, 430, 654, 216, 517, 381, 318, 437, 289, 374, 459, 346, 295, 272, 347, 460, 157, 436, 405, 375, 156, 495, 351, 462, 317, 433, 458, 155, 570, 345, 409, 300, 154, 158, 660, 625, 376, 373, 489, 401, 348, 461, 488, 523, 429, 551, 406, 597, 579, 434, 153, 377, 323, 516, 288, 487, 205, 349, 328, 457, 260, 316, 543, 486, 653, 378, 322, 344, 188, 356, 569, 372, 485, 178, 400, 384, 515, 245, 412, 634, 152, 232, 440, 159, 542, 350, 428, 468, 607, 624, 685, 541, 684, 273, 496, 514, 596, 524, 513, 456, 217, 552, 683, 484, 568, 686, 661, 540, 512, 301, 580, 371, 343, 399, 315, 151, 652, 287, 127, 128, 682, 427, 204, 595, 567, 455, 177, 687, 259, 129, 539, 483, 160, 497, 126, 511, 189, 525, 329, 469, 623, 441, 635, 553, 130, 125, 246, 231, 357, 608, 413, 274, 385, 150, 681, 566, 662, 688, 538, 218, 510, 594, 482, 454, 426, 398, 302, 370, 581, 131, 124, 470, 176, 342, 498, 651, 203, 161, 314, 526, 442, 286, 622, 190, 149, 330, 258, 689, 414, 636, 554, 537, 123, 680, 509, 565, 663, 358, 609, 386, 230, 132, 481, 247, 593, 453, 175, 275, 713, 712, 471, 219, 425, 443, 582, 711, 499, 162, 714, 148, 122, 202, 690, 397, 650, 303, 415, 710, 527, 369, 99, 715, 341, 313, 100, 387, 191, 621, 331, 98, 679, 637, 285, 664, 133, 359, 257, 101, 716, 555, 709, 97, 610, 121, 229, 102, 536, 174, 564, 691, 508, 147, 96, 717, 103, 163, 248, 480, 444, 472, 583], "train_min_nll": 53.794464111328125, "test_nll": 60.782958984375}
\ No newline at end of file
diff --git a/mnist_4_mps_feature_400.safetensors b/mnist_4_mps_feature_400.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..ece877f916962e71c43e7bb393dd43b0bd36bc6b
--- /dev/null
+++ b/mnist_4_mps_feature_400.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:40749efb3a42d09c7845d77df82be81df365b4939235b87055223c5977dc20d2
+size 12786904
diff --git a/mnist_4_mps_feature_50.json b/mnist_4_mps_feature_50.json
new file mode 100644
index 0000000000000000000000000000000000000000..aaaae04ca9a56e0946b833f821b8a1b40cf4e2a2
--- /dev/null
+++ b/mnist_4_mps_feature_50.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 50, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521, 238, 628, 237, 210, 207, 213, 239, 493, 265, 240, 632, 599, 212, 214, 241, 211, 605, 549, 236, 266, 235, 577, 242, 185, 494, 181, 572, 352, 234, 466], "train_min_nll": 9.5249662399292, "test_nll": 10.373994827270508}
\ No newline at end of file
diff --git a/mnist_4_mps_feature_50.safetensors b/mnist_4_mps_feature_50.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..54dce758ea84ba287e06417f1e3802b8a1549b18
--- /dev/null
+++ b/mnist_4_mps_feature_50.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2d0ac563d24e17ad4ee5a5fed871d5172a9d407a14f0dbf3790a76a7303d2aa4
+size 1292416
diff --git a/mnist_4_mps_feature_500.json b/mnist_4_mps_feature_500.json
new file mode 100644
index 0000000000000000000000000000000000000000..3fb7013c7297b64951274408d2d3b1909e10794f
--- /dev/null
+++ b/mnist_4_mps_feature_500.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 500, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521, 238, 628, 237, 210, 207, 213, 239, 493, 265, 240, 632, 599, 212, 214, 241, 211, 605, 549, 236, 266, 235, 577, 242, 185, 494, 181, 572, 352, 234, 466, 627, 186, 518, 293, 268, 215, 269, 545, 182, 297, 184, 491, 292, 492, 264, 243, 180, 270, 267, 324, 438, 325, 183, 296, 658, 522, 380, 383, 410, 206, 465, 382, 271, 657, 298, 411, 355, 353, 262, 463, 578, 659, 263, 432, 598, 656, 354, 626, 299, 633, 439, 187, 320, 655, 291, 431, 571, 550, 327, 490, 261, 326, 404, 233, 319, 435, 294, 403, 379, 407, 606, 467, 244, 464, 402, 290, 544, 321, 179, 408, 430, 654, 216, 517, 381, 318, 437, 289, 374, 459, 346, 295, 272, 347, 460, 157, 436, 405, 375, 156, 495, 351, 462, 317, 433, 458, 155, 570, 345, 409, 300, 154, 158, 660, 625, 376, 373, 489, 401, 348, 461, 488, 523, 429, 551, 406, 597, 579, 434, 153, 377, 323, 516, 288, 487, 205, 349, 328, 457, 260, 316, 543, 486, 653, 378, 322, 344, 188, 356, 569, 372, 485, 178, 400, 384, 515, 245, 412, 634, 152, 232, 440, 159, 542, 350, 428, 468, 607, 624, 685, 541, 684, 273, 496, 514, 596, 524, 513, 456, 217, 552, 683, 484, 568, 686, 661, 540, 512, 301, 580, 371, 343, 399, 315, 151, 652, 287, 127, 128, 682, 427, 204, 595, 567, 455, 177, 687, 259, 129, 539, 483, 160, 497, 126, 511, 189, 525, 329, 469, 623, 441, 635, 553, 130, 125, 246, 231, 357, 608, 413, 274, 385, 150, 681, 566, 662, 688, 538, 218, 510, 594, 482, 454, 426, 398, 302, 370, 581, 131, 124, 470, 176, 342, 498, 651, 203, 161, 314, 526, 442, 286, 622, 190, 149, 330, 258, 689, 414, 636, 554, 537, 123, 680, 509, 565, 663, 358, 609, 386, 230, 132, 481, 247, 593, 453, 175, 275, 713, 712, 471, 219, 425, 443, 582, 711, 499, 162, 714, 148, 122, 202, 690, 397, 650, 303, 415, 710, 527, 369, 99, 715, 341, 313, 100, 387, 191, 621, 331, 98, 679, 637, 285, 664, 133, 359, 257, 101, 716, 555, 709, 97, 610, 121, 229, 102, 536, 174, 564, 691, 508, 147, 96, 717, 103, 163, 248, 480, 444, 472, 583, 592, 201, 708, 220, 276, 416, 95, 134, 500, 665, 649, 120, 638, 388, 452, 718, 678, 304, 740, 692, 192, 741, 528, 104, 739, 256, 284, 424, 742, 360, 312, 611, 146, 332, 738, 94, 620, 228, 173, 396, 340, 71, 368, 743, 707, 556, 719, 72, 744, 737, 70, 119, 164, 666, 73, 693, 105, 93, 584, 639, 249, 200, 135, 69, 745, 221, 563, 74, 535, 68, 736, 677, 277, 648, 720, 591, 145, 75, 706, 507, 193, 255, 283, 612, 746, 67, 92, 501, 473, 445, 305, 311, 172, 694, 227, 118, 667, 106, 529, 5], "train_min_nll": 54.992408752441406, "test_nll": 62.79642105102539}
\ No newline at end of file
diff --git a/mnist_4_mps_feature_500.safetensors b/mnist_4_mps_feature_500.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..b8eb993b153840cb6238b8702c446201c02f30c9
--- /dev/null
+++ b/mnist_4_mps_feature_500.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3a610a6ccab89714357a608d6beaeccd6422113622726fdad459c94d04cbc7f4
+size 16071208
diff --git a/mnist_4_mps_feature_600.json b/mnist_4_mps_feature_600.json
new file mode 100644
index 0000000000000000000000000000000000000000..62693a9eae0ec29dbb8f7676d52a905696c94548
--- /dev/null
+++ b/mnist_4_mps_feature_600.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 600, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521, 238, 628, 237, 210, 207, 213, 239, 493, 265, 240, 632, 599, 212, 214, 241, 211, 605, 549, 236, 266, 235, 577, 242, 185, 494, 181, 572, 352, 234, 466, 627, 186, 518, 293, 268, 215, 269, 545, 182, 297, 184, 491, 292, 492, 264, 243, 180, 270, 267, 324, 438, 325, 183, 296, 658, 522, 380, 383, 410, 206, 465, 382, 271, 657, 298, 411, 355, 353, 262, 463, 578, 659, 263, 432, 598, 656, 354, 626, 299, 633, 439, 187, 320, 655, 291, 431, 571, 550, 327, 490, 261, 326, 404, 233, 319, 435, 294, 403, 379, 407, 606, 467, 244, 464, 402, 290, 544, 321, 179, 408, 430, 654, 216, 517, 381, 318, 437, 289, 374, 459, 346, 295, 272, 347, 460, 157, 436, 405, 375, 156, 495, 351, 462, 317, 433, 458, 155, 570, 345, 409, 300, 154, 158, 660, 625, 376, 373, 489, 401, 348, 461, 488, 523, 429, 551, 406, 597, 579, 434, 153, 377, 323, 516, 288, 487, 205, 349, 328, 457, 260, 316, 543, 486, 653, 378, 322, 344, 188, 356, 569, 372, 485, 178, 400, 384, 515, 245, 412, 634, 152, 232, 440, 159, 542, 350, 428, 468, 607, 624, 685, 541, 684, 273, 496, 514, 596, 524, 513, 456, 217, 552, 683, 484, 568, 686, 661, 540, 512, 301, 580, 371, 343, 399, 315, 151, 652, 287, 127, 128, 682, 427, 204, 595, 567, 455, 177, 687, 259, 129, 539, 483, 160, 497, 126, 511, 189, 525, 329, 469, 623, 441, 635, 553, 130, 125, 246, 231, 357, 608, 413, 274, 385, 150, 681, 566, 662, 688, 538, 218, 510, 594, 482, 454, 426, 398, 302, 370, 581, 131, 124, 470, 176, 342, 498, 651, 203, 161, 314, 526, 442, 286, 622, 190, 149, 330, 258, 689, 414, 636, 554, 537, 123, 680, 509, 565, 663, 358, 609, 386, 230, 132, 481, 247, 593, 453, 175, 275, 713, 712, 471, 219, 425, 443, 582, 711, 499, 162, 714, 148, 122, 202, 690, 397, 650, 303, 415, 710, 527, 369, 99, 715, 341, 313, 100, 387, 191, 621, 331, 98, 679, 637, 285, 664, 133, 359, 257, 101, 716, 555, 709, 97, 610, 121, 229, 102, 536, 174, 564, 691, 508, 147, 96, 717, 103, 163, 248, 480, 444, 472, 583, 592, 201, 708, 220, 276, 416, 95, 134, 500, 665, 649, 120, 638, 388, 452, 718, 678, 304, 740, 692, 192, 741, 528, 104, 739, 256, 284, 424, 742, 360, 312, 611, 146, 332, 738, 94, 620, 228, 173, 396, 340, 71, 368, 743, 707, 556, 719, 72, 744, 737, 70, 119, 164, 666, 73, 693, 105, 93, 584, 639, 249, 200, 135, 69, 745, 221, 563, 74, 535, 68, 736, 677, 277, 648, 720, 591, 145, 75, 706, 507, 193, 255, 283, 612, 746, 67, 92, 501, 473, 445, 305, 311, 172, 694, 227, 118, 667, 106, 529, 5, 735, 76, 479, 66, 417, 619, 721, 339, 136, 557, 747, 7, 4, 640, 451, 91, 367, 6, 333, 165, 389, 585, 423, 3, 77, 65, 676, 748, 770, 734, 199, 395, 705, 771, 695, 769, 361, 282, 117, 107, 768, 722, 254, 222, 43, 250, 772, 310, 144, 613, 647, 64, 767, 278, 668, 78, 562, 194, 90, 226, 766, 42, 749, 40, 44, 534, 773, 41, 590, 1, 306, 63, 338, 171, 2, 39, 108, 137, 733, 502, 765, 474, 723, 38, 704, 45, 618, 506, 446, 641, 478, 334, 366, 675, 558, 764, 774, 696, 46, 530], "train_min_nll": 54.49281692504883, "test_nll": 61.686119079589844}
\ No newline at end of file
diff --git a/mnist_4_mps_feature_600.safetensors b/mnist_4_mps_feature_600.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..2a5b979b13646f6996426d93ceed45e248aafd3e
--- /dev/null
+++ b/mnist_4_mps_feature_600.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6d4cb97a3b5ed4b1278ac8bd78cce234205ba41839312411b258a769bce96696
+size 19355504
diff --git a/mnist_5_losses.png b/mnist_5_losses.png
new file mode 100644
index 0000000000000000000000000000000000000000..09fc6bec0e1ca6182d5585ef63a1f3958948ed3b
Binary files /dev/null and b/mnist_5_losses.png differ
diff --git a/mnist_5_mps.safetensors b/mnist_5_mps.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..e4cdaaf7e522065d34596bc8666d2f640480662e
--- /dev/null
+++ b/mnist_5_mps.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d645046a22f2bab56f494067317eb8aab9af0bc5733b674deeda3bf3b44ca164
+size 25398616
diff --git a/mnist_5_mps_feature_100.json b/mnist_5_mps_feature_100.json
new file mode 100644
index 0000000000000000000000000000000000000000..e4ce33ee01634dff560366af35d7e100c93bd23f
--- /dev/null
+++ b/mnist_5_mps_feature_100.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 100, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521, 238, 628, 237, 210, 207, 213, 239, 493, 265, 240, 632, 599, 212, 214, 241, 211, 605, 549, 236, 266, 235, 577, 242, 185, 494, 181, 572, 352, 234, 466, 627, 186, 518, 293, 268, 215, 269, 545, 182, 297, 184, 491, 292, 492, 264, 243, 180, 270, 267, 324, 438, 325, 183, 296, 658, 522, 380, 383, 410, 206, 465, 382, 271, 657, 298, 411, 355, 353, 262, 463, 578, 659, 263, 432, 598, 656, 354, 626, 299, 633], "train_min_nll": 21.341197967529297, "test_nll": 24.41822624206543}
\ No newline at end of file
diff --git a/mnist_5_mps_feature_100.safetensors b/mnist_5_mps_feature_100.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..73d1a95a9b41de06aef94f064e86cf5b19d8901b
--- /dev/null
+++ b/mnist_5_mps_feature_100.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6dc2f7a358203fcc276f7cc02eb6e09fd298fd8c75bf79d9839447b1959148b3
+size 2934416
diff --git a/mnist_5_mps_feature_15.json b/mnist_5_mps_feature_15.json
new file mode 100644
index 0000000000000000000000000000000000000000..c5af65e533d8362ae3928772933b94f35d528c0a
--- /dev/null
+++ b/mnist_5_mps_feature_15.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 15, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573], "train_min_nll": 4.030087471008301, "test_nll": 4.127983093261719}
\ No newline at end of file
diff --git a/mnist_5_mps_feature_15.safetensors b/mnist_5_mps_feature_15.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..e3221d94baa26c57499c6aac16eef12290663130
--- /dev/null
+++ b/mnist_5_mps_feature_15.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e2d8a331c50d9e8379d805682e291f8cd79415009d854fe328d997b0a5587401
+size 143032
diff --git a/mnist_5_mps_feature_20.json b/mnist_5_mps_feature_20.json
new file mode 100644
index 0000000000000000000000000000000000000000..7e428805e7a2efe268a34a8f5540318ef41946df
--- /dev/null
+++ b/mnist_5_mps_feature_20.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 20, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521], "train_min_nll": 4.782350540161133, "test_nll": 4.945647716522217}
\ No newline at end of file
diff --git a/mnist_5_mps_feature_20.safetensors b/mnist_5_mps_feature_20.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..4c7ffd6cefc2e7eb233712919062de3978defc27
--- /dev/null
+++ b/mnist_5_mps_feature_20.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:872d48c5675dd1dac46379167b5f29b51b3a51175048877da205ebc3677cf9da
+size 307248
diff --git a/mnist_5_mps_feature_200.json b/mnist_5_mps_feature_200.json
new file mode 100644
index 0000000000000000000000000000000000000000..32b2fe7afa800bd4e2501b12f1dba0e4bbbf5148
--- /dev/null
+++ b/mnist_5_mps_feature_200.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 200, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521, 238, 628, 237, 210, 207, 213, 239, 493, 265, 240, 632, 599, 212, 214, 241, 211, 605, 549, 236, 266, 235, 577, 242, 185, 494, 181, 572, 352, 234, 466, 627, 186, 518, 293, 268, 215, 269, 545, 182, 297, 184, 491, 292, 492, 264, 243, 180, 270, 267, 324, 438, 325, 183, 296, 658, 522, 380, 383, 410, 206, 465, 382, 271, 657, 298, 411, 355, 353, 262, 463, 578, 659, 263, 432, 598, 656, 354, 626, 299, 633, 439, 187, 320, 655, 291, 431, 571, 550, 327, 490, 261, 326, 404, 233, 319, 435, 294, 403, 379, 407, 606, 467, 244, 464, 402, 290, 544, 321, 179, 408, 430, 654, 216, 517, 381, 318, 437, 289, 374, 459, 346, 295, 272, 347, 460, 157, 436, 405, 375, 156, 495, 351, 462, 317, 433, 458, 155, 570, 345, 409, 300, 154, 158, 660, 625, 376, 373, 489, 401, 348, 461, 488, 523, 429, 551, 406, 597, 579, 434, 153, 377, 323, 516, 288, 487, 205, 349, 328, 457, 260, 316, 543, 486, 653, 378, 322, 344, 188, 356, 569], "train_min_nll": 43.83343505859375, "test_nll": 50.594749450683594}
\ No newline at end of file
diff --git a/mnist_5_mps_feature_200.safetensors b/mnist_5_mps_feature_200.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..74dd4c4fa15611f129d45693d8a43a748acda8ff
--- /dev/null
+++ b/mnist_5_mps_feature_200.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:afdf9d8bf03f6d3324f655d567ab1069f5e339988ab06fb02adbea9587878410
+size 6218528
diff --git a/mnist_5_mps_feature_300.json b/mnist_5_mps_feature_300.json
new file mode 100644
index 0000000000000000000000000000000000000000..d94800dd5b086a0234c23851ab85bc6cc344f0a9
--- /dev/null
+++ b/mnist_5_mps_feature_300.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 300, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521, 238, 628, 237, 210, 207, 213, 239, 493, 265, 240, 632, 599, 212, 214, 241, 211, 605, 549, 236, 266, 235, 577, 242, 185, 494, 181, 572, 352, 234, 466, 627, 186, 518, 293, 268, 215, 269, 545, 182, 297, 184, 491, 292, 492, 264, 243, 180, 270, 267, 324, 438, 325, 183, 296, 658, 522, 380, 383, 410, 206, 465, 382, 271, 657, 298, 411, 355, 353, 262, 463, 578, 659, 263, 432, 598, 656, 354, 626, 299, 633, 439, 187, 320, 655, 291, 431, 571, 550, 327, 490, 261, 326, 404, 233, 319, 435, 294, 403, 379, 407, 606, 467, 244, 464, 402, 290, 544, 321, 179, 408, 430, 654, 216, 517, 381, 318, 437, 289, 374, 459, 346, 295, 272, 347, 460, 157, 436, 405, 375, 156, 495, 351, 462, 317, 433, 458, 155, 570, 345, 409, 300, 154, 158, 660, 625, 376, 373, 489, 401, 348, 461, 488, 523, 429, 551, 406, 597, 579, 434, 153, 377, 323, 516, 288, 487, 205, 349, 328, 457, 260, 316, 543, 486, 653, 378, 322, 344, 188, 356, 569, 372, 485, 178, 400, 384, 515, 245, 412, 634, 152, 232, 440, 159, 542, 350, 428, 468, 607, 624, 685, 541, 684, 273, 496, 514, 596, 524, 513, 456, 217, 552, 683, 484, 568, 686, 661, 540, 512, 301, 580, 371, 343, 399, 315, 151, 652, 287, 127, 128, 682, 427, 204, 595, 567, 455, 177, 687, 259, 129, 539, 483, 160, 497, 126, 511, 189, 525, 329, 469, 623, 441, 635, 553, 130, 125, 246, 231, 357, 608, 413, 274, 385, 150, 681, 566, 662, 688, 538, 218, 510, 594, 482, 454, 426, 398, 302, 370, 581, 131, 124], "train_min_nll": 56.740379333496094, "test_nll": 65.79816436767578}
\ No newline at end of file
diff --git a/mnist_5_mps_feature_300.safetensors b/mnist_5_mps_feature_300.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..1ef583cdc9131ce4b94f8c15ef945f463dc86728
--- /dev/null
+++ b/mnist_5_mps_feature_300.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:396d33165c3166d70b0d2ea62543c2e021797d692e3e6d1eb6e13b623a234c2e
+size 9502624
diff --git a/mnist_5_mps_feature_400.json b/mnist_5_mps_feature_400.json
new file mode 100644
index 0000000000000000000000000000000000000000..4824d06e5745e0249775a6edb1ca49becc484068
--- /dev/null
+++ b/mnist_5_mps_feature_400.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 400, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521, 238, 628, 237, 210, 207, 213, 239, 493, 265, 240, 632, 599, 212, 214, 241, 211, 605, 549, 236, 266, 235, 577, 242, 185, 494, 181, 572, 352, 234, 466, 627, 186, 518, 293, 268, 215, 269, 545, 182, 297, 184, 491, 292, 492, 264, 243, 180, 270, 267, 324, 438, 325, 183, 296, 658, 522, 380, 383, 410, 206, 465, 382, 271, 657, 298, 411, 355, 353, 262, 463, 578, 659, 263, 432, 598, 656, 354, 626, 299, 633, 439, 187, 320, 655, 291, 431, 571, 550, 327, 490, 261, 326, 404, 233, 319, 435, 294, 403, 379, 407, 606, 467, 244, 464, 402, 290, 544, 321, 179, 408, 430, 654, 216, 517, 381, 318, 437, 289, 374, 459, 346, 295, 272, 347, 460, 157, 436, 405, 375, 156, 495, 351, 462, 317, 433, 458, 155, 570, 345, 409, 300, 154, 158, 660, 625, 376, 373, 489, 401, 348, 461, 488, 523, 429, 551, 406, 597, 579, 434, 153, 377, 323, 516, 288, 487, 205, 349, 328, 457, 260, 316, 543, 486, 653, 378, 322, 344, 188, 356, 569, 372, 485, 178, 400, 384, 515, 245, 412, 634, 152, 232, 440, 159, 542, 350, 428, 468, 607, 624, 685, 541, 684, 273, 496, 514, 596, 524, 513, 456, 217, 552, 683, 484, 568, 686, 661, 540, 512, 301, 580, 371, 343, 399, 315, 151, 652, 287, 127, 128, 682, 427, 204, 595, 567, 455, 177, 687, 259, 129, 539, 483, 160, 497, 126, 511, 189, 525, 329, 469, 623, 441, 635, 553, 130, 125, 246, 231, 357, 608, 413, 274, 385, 150, 681, 566, 662, 688, 538, 218, 510, 594, 482, 454, 426, 398, 302, 370, 581, 131, 124, 470, 176, 342, 498, 651, 203, 161, 314, 526, 442, 286, 622, 190, 149, 330, 258, 689, 414, 636, 554, 537, 123, 680, 509, 565, 663, 358, 609, 386, 230, 132, 481, 247, 593, 453, 175, 275, 713, 712, 471, 219, 425, 443, 582, 711, 499, 162, 714, 148, 122, 202, 690, 397, 650, 303, 415, 710, 527, 369, 99, 715, 341, 313, 100, 387, 191, 621, 331, 98, 679, 637, 285, 664, 133, 359, 257, 101, 716, 555, 709, 97, 610, 121, 229, 102, 536, 174, 564, 691, 508, 147, 96, 717, 103, 163, 248, 480, 444, 472, 583], "train_min_nll": 60.984493255615234, "test_nll": 71.03687286376953}
\ No newline at end of file
diff --git a/mnist_5_mps_feature_400.safetensors b/mnist_5_mps_feature_400.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..5dadc8866bd35a50159cced9819f1c5b0a5259f5
--- /dev/null
+++ b/mnist_5_mps_feature_400.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:41c8ca3f173052245e9fe25e934000c948dae65b588193013a87253412b04295
+size 12786904
diff --git a/mnist_5_mps_feature_50.json b/mnist_5_mps_feature_50.json
new file mode 100644
index 0000000000000000000000000000000000000000..501b21c30192b7581c59169d62e86901a0dcf05c
--- /dev/null
+++ b/mnist_5_mps_feature_50.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 50, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521, 238, 628, 237, 210, 207, 213, 239, 493, 265, 240, 632, 599, 212, 214, 241, 211, 605, 549, 236, 266, 235, 577, 242, 185, 494, 181, 572, 352, 234, 466], "train_min_nll": 11.600281715393066, "test_nll": 13.054468154907227}
\ No newline at end of file
diff --git a/mnist_5_mps_feature_50.safetensors b/mnist_5_mps_feature_50.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..3fa3df05e202d5ee839cfffef08d0c4f3f09301f
--- /dev/null
+++ b/mnist_5_mps_feature_50.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:eb32353f65a9fd1437e03f9aeb82bd19b39924e0a10040d7951c013581f02675
+size 1292416
diff --git a/mnist_5_mps_feature_500.json b/mnist_5_mps_feature_500.json
new file mode 100644
index 0000000000000000000000000000000000000000..a1707621f7194d467ed854c44f000faee721dcdf
--- /dev/null
+++ b/mnist_5_mps_feature_500.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 500, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521, 238, 628, 237, 210, 207, 213, 239, 493, 265, 240, 632, 599, 212, 214, 241, 211, 605, 549, 236, 266, 235, 577, 242, 185, 494, 181, 572, 352, 234, 466, 627, 186, 518, 293, 268, 215, 269, 545, 182, 297, 184, 491, 292, 492, 264, 243, 180, 270, 267, 324, 438, 325, 183, 296, 658, 522, 380, 383, 410, 206, 465, 382, 271, 657, 298, 411, 355, 353, 262, 463, 578, 659, 263, 432, 598, 656, 354, 626, 299, 633, 439, 187, 320, 655, 291, 431, 571, 550, 327, 490, 261, 326, 404, 233, 319, 435, 294, 403, 379, 407, 606, 467, 244, 464, 402, 290, 544, 321, 179, 408, 430, 654, 216, 517, 381, 318, 437, 289, 374, 459, 346, 295, 272, 347, 460, 157, 436, 405, 375, 156, 495, 351, 462, 317, 433, 458, 155, 570, 345, 409, 300, 154, 158, 660, 625, 376, 373, 489, 401, 348, 461, 488, 523, 429, 551, 406, 597, 579, 434, 153, 377, 323, 516, 288, 487, 205, 349, 328, 457, 260, 316, 543, 486, 653, 378, 322, 344, 188, 356, 569, 372, 485, 178, 400, 384, 515, 245, 412, 634, 152, 232, 440, 159, 542, 350, 428, 468, 607, 624, 685, 541, 684, 273, 496, 514, 596, 524, 513, 456, 217, 552, 683, 484, 568, 686, 661, 540, 512, 301, 580, 371, 343, 399, 315, 151, 652, 287, 127, 128, 682, 427, 204, 595, 567, 455, 177, 687, 259, 129, 539, 483, 160, 497, 126, 511, 189, 525, 329, 469, 623, 441, 635, 553, 130, 125, 246, 231, 357, 608, 413, 274, 385, 150, 681, 566, 662, 688, 538, 218, 510, 594, 482, 454, 426, 398, 302, 370, 581, 131, 124, 470, 176, 342, 498, 651, 203, 161, 314, 526, 442, 286, 622, 190, 149, 330, 258, 689, 414, 636, 554, 537, 123, 680, 509, 565, 663, 358, 609, 386, 230, 132, 481, 247, 593, 453, 175, 275, 713, 712, 471, 219, 425, 443, 582, 711, 499, 162, 714, 148, 122, 202, 690, 397, 650, 303, 415, 710, 527, 369, 99, 715, 341, 313, 100, 387, 191, 621, 331, 98, 679, 637, 285, 664, 133, 359, 257, 101, 716, 555, 709, 97, 610, 121, 229, 102, 536, 174, 564, 691, 508, 147, 96, 717, 103, 163, 248, 480, 444, 472, 583, 592, 201, 708, 220, 276, 416, 95, 134, 500, 665, 649, 120, 638, 388, 452, 718, 678, 304, 740, 692, 192, 741, 528, 104, 739, 256, 284, 424, 742, 360, 312, 611, 146, 332, 738, 94, 620, 228, 173, 396, 340, 71, 368, 743, 707, 556, 719, 72, 744, 737, 70, 119, 164, 666, 73, 693, 105, 93, 584, 639, 249, 200, 135, 69, 745, 221, 563, 74, 535, 68, 736, 677, 277, 648, 720, 591, 145, 75, 706, 507, 193, 255, 283, 612, 746, 67, 92, 501, 473, 445, 305, 311, 172, 694, 227, 118, 667, 106, 529, 5], "train_min_nll": 61.824398040771484, "test_nll": 71.47968292236328}
\ No newline at end of file
diff --git a/mnist_5_mps_feature_500.safetensors b/mnist_5_mps_feature_500.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..1a69e2fd595277129515e3d40c571d6a913ffc2e
--- /dev/null
+++ b/mnist_5_mps_feature_500.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:201f4d7ceaa321582d163e7a188cc3b484c5c16211b86a4185308c8186206ee9
+size 16071208
diff --git a/mnist_5_mps_feature_600.json b/mnist_5_mps_feature_600.json
new file mode 100644
index 0000000000000000000000000000000000000000..f0cbe6584c2fce29bc5f9e2a864a2504a9485558
--- /dev/null
+++ b/mnist_5_mps_feature_600.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 600, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521, 238, 628, 237, 210, 207, 213, 239, 493, 265, 240, 632, 599, 212, 214, 241, 211, 605, 549, 236, 266, 235, 577, 242, 185, 494, 181, 572, 352, 234, 466, 627, 186, 518, 293, 268, 215, 269, 545, 182, 297, 184, 491, 292, 492, 264, 243, 180, 270, 267, 324, 438, 325, 183, 296, 658, 522, 380, 383, 410, 206, 465, 382, 271, 657, 298, 411, 355, 353, 262, 463, 578, 659, 263, 432, 598, 656, 354, 626, 299, 633, 439, 187, 320, 655, 291, 431, 571, 550, 327, 490, 261, 326, 404, 233, 319, 435, 294, 403, 379, 407, 606, 467, 244, 464, 402, 290, 544, 321, 179, 408, 430, 654, 216, 517, 381, 318, 437, 289, 374, 459, 346, 295, 272, 347, 460, 157, 436, 405, 375, 156, 495, 351, 462, 317, 433, 458, 155, 570, 345, 409, 300, 154, 158, 660, 625, 376, 373, 489, 401, 348, 461, 488, 523, 429, 551, 406, 597, 579, 434, 153, 377, 323, 516, 288, 487, 205, 349, 328, 457, 260, 316, 543, 486, 653, 378, 322, 344, 188, 356, 569, 372, 485, 178, 400, 384, 515, 245, 412, 634, 152, 232, 440, 159, 542, 350, 428, 468, 607, 624, 685, 541, 684, 273, 496, 514, 596, 524, 513, 456, 217, 552, 683, 484, 568, 686, 661, 540, 512, 301, 580, 371, 343, 399, 315, 151, 652, 287, 127, 128, 682, 427, 204, 595, 567, 455, 177, 687, 259, 129, 539, 483, 160, 497, 126, 511, 189, 525, 329, 469, 623, 441, 635, 553, 130, 125, 246, 231, 357, 608, 413, 274, 385, 150, 681, 566, 662, 688, 538, 218, 510, 594, 482, 454, 426, 398, 302, 370, 581, 131, 124, 470, 176, 342, 498, 651, 203, 161, 314, 526, 442, 286, 622, 190, 149, 330, 258, 689, 414, 636, 554, 537, 123, 680, 509, 565, 663, 358, 609, 386, 230, 132, 481, 247, 593, 453, 175, 275, 713, 712, 471, 219, 425, 443, 582, 711, 499, 162, 714, 148, 122, 202, 690, 397, 650, 303, 415, 710, 527, 369, 99, 715, 341, 313, 100, 387, 191, 621, 331, 98, 679, 637, 285, 664, 133, 359, 257, 101, 716, 555, 709, 97, 610, 121, 229, 102, 536, 174, 564, 691, 508, 147, 96, 717, 103, 163, 248, 480, 444, 472, 583, 592, 201, 708, 220, 276, 416, 95, 134, 500, 665, 649, 120, 638, 388, 452, 718, 678, 304, 740, 692, 192, 741, 528, 104, 739, 256, 284, 424, 742, 360, 312, 611, 146, 332, 738, 94, 620, 228, 173, 396, 340, 71, 368, 743, 707, 556, 719, 72, 744, 737, 70, 119, 164, 666, 73, 693, 105, 93, 584, 639, 249, 200, 135, 69, 745, 221, 563, 74, 535, 68, 736, 677, 277, 648, 720, 591, 145, 75, 706, 507, 193, 255, 283, 612, 746, 67, 92, 501, 473, 445, 305, 311, 172, 694, 227, 118, 667, 106, 529, 5, 735, 76, 479, 66, 417, 619, 721, 339, 136, 557, 747, 7, 4, 640, 451, 91, 367, 6, 333, 165, 389, 585, 423, 3, 77, 65, 676, 748, 770, 734, 199, 395, 705, 771, 695, 769, 361, 282, 117, 107, 768, 722, 254, 222, 43, 250, 772, 310, 144, 613, 647, 64, 767, 278, 668, 78, 562, 194, 90, 226, 766, 42, 749, 40, 44, 534, 773, 41, 590, 1, 306, 63, 338, 171, 2, 39, 108, 137, 733, 502, 765, 474, 723, 38, 704, 45, 618, 506, 446, 641, 478, 334, 366, 675, 558, 764, 774, 696, 46, 530], "train_min_nll": 62.51140213012695, "test_nll": 72.74060821533203}
\ No newline at end of file
diff --git a/mnist_5_mps_feature_600.safetensors b/mnist_5_mps_feature_600.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..e6588c44aaaf37a8e1e9c919d59fd656dc96570d
--- /dev/null
+++ b/mnist_5_mps_feature_600.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:cb3ff2d3a95e75119d1f59e2399e2c744c040f38e5fa947c744efc5237215b8e
+size 19355504
diff --git a/mnist_6_losses.png b/mnist_6_losses.png
new file mode 100644
index 0000000000000000000000000000000000000000..036866b0dd9605dd349d0c600e05c353a972a853
Binary files /dev/null and b/mnist_6_losses.png differ
diff --git a/mnist_6_mps.safetensors b/mnist_6_mps.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..6431fe5c5a44fee0dddb02963a8f721e730400ef
--- /dev/null
+++ b/mnist_6_mps.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a9dc58b3ba41b469205c47ff5651704fea1df86410c32fa56b21fd695bbcd1b9
+size 25398616
diff --git a/mnist_6_mps_feature_100.json b/mnist_6_mps_feature_100.json
new file mode 100644
index 0000000000000000000000000000000000000000..03b7a0af3eff1061be35bc556b0e1812b400cf5b
--- /dev/null
+++ b/mnist_6_mps_feature_100.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 100, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521, 238, 628, 237, 210, 207, 213, 239, 493, 265, 240, 632, 599, 212, 214, 241, 211, 605, 549, 236, 266, 235, 577, 242, 185, 494, 181, 572, 352, 234, 466, 627, 186, 518, 293, 268, 215, 269, 545, 182, 297, 184, 491, 292, 492, 264, 243, 180, 270, 267, 324, 438, 325, 183, 296, 658, 522, 380, 383, 410, 206, 465, 382, 271, 657, 298, 411, 355, 353, 262, 463, 578, 659, 263, 432, 598, 656, 354, 626, 299, 633], "train_min_nll": 18.385791778564453, "test_nll": 20.67178726196289}
\ No newline at end of file
diff --git a/mnist_6_mps_feature_100.safetensors b/mnist_6_mps_feature_100.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..dadf5eb97d971baf00250ce57ab5bfabe483710a
--- /dev/null
+++ b/mnist_6_mps_feature_100.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:34582cddaa6007a10f7fae89f979d74873068c404bc0d48fa8521df07b46ad9d
+size 2934416
diff --git a/mnist_6_mps_feature_15.json b/mnist_6_mps_feature_15.json
new file mode 100644
index 0000000000000000000000000000000000000000..ff5a0bcfa7d666478ca6ea23b1893165fbe26739
--- /dev/null
+++ b/mnist_6_mps_feature_15.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 15, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573], "train_min_nll": 4.493940353393555, "test_nll": 4.558094024658203}
\ No newline at end of file
diff --git a/mnist_6_mps_feature_15.safetensors b/mnist_6_mps_feature_15.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..62fcb5f79f1dedf6d88c68135d374f32a4295121
--- /dev/null
+++ b/mnist_6_mps_feature_15.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:83161a129cd79fe5547d177296375c025944d3991e1a6a857300ccbbe6ef9154
+size 143032
diff --git a/mnist_6_mps_feature_20.json b/mnist_6_mps_feature_20.json
new file mode 100644
index 0000000000000000000000000000000000000000..7b26457a67e9da9bad4e39d808b784163a6ee84c
--- /dev/null
+++ b/mnist_6_mps_feature_20.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 20, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521], "train_min_nll": 5.57802152633667, "test_nll": 5.920148849487305}
\ No newline at end of file
diff --git a/mnist_6_mps_feature_20.safetensors b/mnist_6_mps_feature_20.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..540f6d10f2539a8db3709a036ab65df238c35a1c
--- /dev/null
+++ b/mnist_6_mps_feature_20.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:74052a514fa3f9f58fe5d5c492fab5b37c79eaad16a0f3b1687661337f9e00e4
+size 307248
diff --git a/mnist_6_mps_feature_200.json b/mnist_6_mps_feature_200.json
new file mode 100644
index 0000000000000000000000000000000000000000..3a318d03433b81a9cdb6070788c6ba2a43341896
--- /dev/null
+++ b/mnist_6_mps_feature_200.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 200, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521, 238, 628, 237, 210, 207, 213, 239, 493, 265, 240, 632, 599, 212, 214, 241, 211, 605, 549, 236, 266, 235, 577, 242, 185, 494, 181, 572, 352, 234, 466, 627, 186, 518, 293, 268, 215, 269, 545, 182, 297, 184, 491, 292, 492, 264, 243, 180, 270, 267, 324, 438, 325, 183, 296, 658, 522, 380, 383, 410, 206, 465, 382, 271, 657, 298, 411, 355, 353, 262, 463, 578, 659, 263, 432, 598, 656, 354, 626, 299, 633, 439, 187, 320, 655, 291, 431, 571, 550, 327, 490, 261, 326, 404, 233, 319, 435, 294, 403, 379, 407, 606, 467, 244, 464, 402, 290, 544, 321, 179, 408, 430, 654, 216, 517, 381, 318, 437, 289, 374, 459, 346, 295, 272, 347, 460, 157, 436, 405, 375, 156, 495, 351, 462, 317, 433, 458, 155, 570, 345, 409, 300, 154, 158, 660, 625, 376, 373, 489, 401, 348, 461, 488, 523, 429, 551, 406, 597, 579, 434, 153, 377, 323, 516, 288, 487, 205, 349, 328, 457, 260, 316, 543, 486, 653, 378, 322, 344, 188, 356, 569], "train_min_nll": 37.81410598754883, "test_nll": 43.23678970336914}
\ No newline at end of file
diff --git a/mnist_6_mps_feature_200.safetensors b/mnist_6_mps_feature_200.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..3df0c9ea36b5cb3e3c7ed414bc2c93f45198281f
--- /dev/null
+++ b/mnist_6_mps_feature_200.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:34207a959dea4a707af9a29d7e690b4ebb53c12c7e5f08f055ced612b79a686a
+size 6218528
diff --git a/mnist_6_mps_feature_300.json b/mnist_6_mps_feature_300.json
new file mode 100644
index 0000000000000000000000000000000000000000..4499fc7d66288e9b7f72bf53615defdca6c36af5
--- /dev/null
+++ b/mnist_6_mps_feature_300.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 300, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521, 238, 628, 237, 210, 207, 213, 239, 493, 265, 240, 632, 599, 212, 214, 241, 211, 605, 549, 236, 266, 235, 577, 242, 185, 494, 181, 572, 352, 234, 466, 627, 186, 518, 293, 268, 215, 269, 545, 182, 297, 184, 491, 292, 492, 264, 243, 180, 270, 267, 324, 438, 325, 183, 296, 658, 522, 380, 383, 410, 206, 465, 382, 271, 657, 298, 411, 355, 353, 262, 463, 578, 659, 263, 432, 598, 656, 354, 626, 299, 633, 439, 187, 320, 655, 291, 431, 571, 550, 327, 490, 261, 326, 404, 233, 319, 435, 294, 403, 379, 407, 606, 467, 244, 464, 402, 290, 544, 321, 179, 408, 430, 654, 216, 517, 381, 318, 437, 289, 374, 459, 346, 295, 272, 347, 460, 157, 436, 405, 375, 156, 495, 351, 462, 317, 433, 458, 155, 570, 345, 409, 300, 154, 158, 660, 625, 376, 373, 489, 401, 348, 461, 488, 523, 429, 551, 406, 597, 579, 434, 153, 377, 323, 516, 288, 487, 205, 349, 328, 457, 260, 316, 543, 486, 653, 378, 322, 344, 188, 356, 569, 372, 485, 178, 400, 384, 515, 245, 412, 634, 152, 232, 440, 159, 542, 350, 428, 468, 607, 624, 685, 541, 684, 273, 496, 514, 596, 524, 513, 456, 217, 552, 683, 484, 568, 686, 661, 540, 512, 301, 580, 371, 343, 399, 315, 151, 652, 287, 127, 128, 682, 427, 204, 595, 567, 455, 177, 687, 259, 129, 539, 483, 160, 497, 126, 511, 189, 525, 329, 469, 623, 441, 635, 553, 130, 125, 246, 231, 357, 608, 413, 274, 385, 150, 681, 566, 662, 688, 538, 218, 510, 594, 482, 454, 426, 398, 302, 370, 581, 131, 124], "train_min_nll": 47.449127197265625, "test_nll": 54.839500427246094}
\ No newline at end of file
diff --git a/mnist_6_mps_feature_300.safetensors b/mnist_6_mps_feature_300.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..bb70347d77018b17ccf281ad0db7d3716a02ef5e
--- /dev/null
+++ b/mnist_6_mps_feature_300.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9f5de3e7b4f31730af3e3196fd834fe97dc66157b3f026830eaae52c0e6759df
+size 9502624
diff --git a/mnist_6_mps_feature_400.json b/mnist_6_mps_feature_400.json
new file mode 100644
index 0000000000000000000000000000000000000000..9b88ffc065c3dcd48a73cdd6e7c148870039a4f7
--- /dev/null
+++ b/mnist_6_mps_feature_400.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 400, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521, 238, 628, 237, 210, 207, 213, 239, 493, 265, 240, 632, 599, 212, 214, 241, 211, 605, 549, 236, 266, 235, 577, 242, 185, 494, 181, 572, 352, 234, 466, 627, 186, 518, 293, 268, 215, 269, 545, 182, 297, 184, 491, 292, 492, 264, 243, 180, 270, 267, 324, 438, 325, 183, 296, 658, 522, 380, 383, 410, 206, 465, 382, 271, 657, 298, 411, 355, 353, 262, 463, 578, 659, 263, 432, 598, 656, 354, 626, 299, 633, 439, 187, 320, 655, 291, 431, 571, 550, 327, 490, 261, 326, 404, 233, 319, 435, 294, 403, 379, 407, 606, 467, 244, 464, 402, 290, 544, 321, 179, 408, 430, 654, 216, 517, 381, 318, 437, 289, 374, 459, 346, 295, 272, 347, 460, 157, 436, 405, 375, 156, 495, 351, 462, 317, 433, 458, 155, 570, 345, 409, 300, 154, 158, 660, 625, 376, 373, 489, 401, 348, 461, 488, 523, 429, 551, 406, 597, 579, 434, 153, 377, 323, 516, 288, 487, 205, 349, 328, 457, 260, 316, 543, 486, 653, 378, 322, 344, 188, 356, 569, 372, 485, 178, 400, 384, 515, 245, 412, 634, 152, 232, 440, 159, 542, 350, 428, 468, 607, 624, 685, 541, 684, 273, 496, 514, 596, 524, 513, 456, 217, 552, 683, 484, 568, 686, 661, 540, 512, 301, 580, 371, 343, 399, 315, 151, 652, 287, 127, 128, 682, 427, 204, 595, 567, 455, 177, 687, 259, 129, 539, 483, 160, 497, 126, 511, 189, 525, 329, 469, 623, 441, 635, 553, 130, 125, 246, 231, 357, 608, 413, 274, 385, 150, 681, 566, 662, 688, 538, 218, 510, 594, 482, 454, 426, 398, 302, 370, 581, 131, 124, 470, 176, 342, 498, 651, 203, 161, 314, 526, 442, 286, 622, 190, 149, 330, 258, 689, 414, 636, 554, 537, 123, 680, 509, 565, 663, 358, 609, 386, 230, 132, 481, 247, 593, 453, 175, 275, 713, 712, 471, 219, 425, 443, 582, 711, 499, 162, 714, 148, 122, 202, 690, 397, 650, 303, 415, 710, 527, 369, 99, 715, 341, 313, 100, 387, 191, 621, 331, 98, 679, 637, 285, 664, 133, 359, 257, 101, 716, 555, 709, 97, 610, 121, 229, 102, 536, 174, 564, 691, 508, 147, 96, 717, 103, 163, 248, 480, 444, 472, 583], "train_min_nll": 50.92384338378906, "test_nll": 59.317440032958984}
\ No newline at end of file
diff --git a/mnist_6_mps_feature_400.safetensors b/mnist_6_mps_feature_400.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..826ab618d1c3d443e0f7478ade1d09d26e37e71e
--- /dev/null
+++ b/mnist_6_mps_feature_400.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9eb87c51e90421270c16eae33a0a4554bfa23f5d25378516d05ef8ffe16975b5
+size 12786904
diff --git a/mnist_6_mps_feature_50.json b/mnist_6_mps_feature_50.json
new file mode 100644
index 0000000000000000000000000000000000000000..666e61ab19e76a13499cdb1858778e583d810897
--- /dev/null
+++ b/mnist_6_mps_feature_50.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 50, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521, 238, 628, 237, 210, 207, 213, 239, 493, 265, 240, 632, 599, 212, 214, 241, 211, 605, 549, 236, 266, 235, 577, 242, 185, 494, 181, 572, 352, 234, 466], "train_min_nll": 9.834566116333008, "test_nll": 10.723784446716309}
\ No newline at end of file
diff --git a/mnist_6_mps_feature_50.safetensors b/mnist_6_mps_feature_50.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..d7c1d6e0a391bd98efa96b6c5841215e9bb8c4b1
--- /dev/null
+++ b/mnist_6_mps_feature_50.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8df257bb474903d60eafb3e41b795901366d7b489aeb9dcfc98fd67d6b821688
+size 1292416
diff --git a/mnist_6_mps_feature_500.json b/mnist_6_mps_feature_500.json
new file mode 100644
index 0000000000000000000000000000000000000000..d52423ee0d0758f75e5257323cc3873c55a63d3a
--- /dev/null
+++ b/mnist_6_mps_feature_500.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 500, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521, 238, 628, 237, 210, 207, 213, 239, 493, 265, 240, 632, 599, 212, 214, 241, 211, 605, 549, 236, 266, 235, 577, 242, 185, 494, 181, 572, 352, 234, 466, 627, 186, 518, 293, 268, 215, 269, 545, 182, 297, 184, 491, 292, 492, 264, 243, 180, 270, 267, 324, 438, 325, 183, 296, 658, 522, 380, 383, 410, 206, 465, 382, 271, 657, 298, 411, 355, 353, 262, 463, 578, 659, 263, 432, 598, 656, 354, 626, 299, 633, 439, 187, 320, 655, 291, 431, 571, 550, 327, 490, 261, 326, 404, 233, 319, 435, 294, 403, 379, 407, 606, 467, 244, 464, 402, 290, 544, 321, 179, 408, 430, 654, 216, 517, 381, 318, 437, 289, 374, 459, 346, 295, 272, 347, 460, 157, 436, 405, 375, 156, 495, 351, 462, 317, 433, 458, 155, 570, 345, 409, 300, 154, 158, 660, 625, 376, 373, 489, 401, 348, 461, 488, 523, 429, 551, 406, 597, 579, 434, 153, 377, 323, 516, 288, 487, 205, 349, 328, 457, 260, 316, 543, 486, 653, 378, 322, 344, 188, 356, 569, 372, 485, 178, 400, 384, 515, 245, 412, 634, 152, 232, 440, 159, 542, 350, 428, 468, 607, 624, 685, 541, 684, 273, 496, 514, 596, 524, 513, 456, 217, 552, 683, 484, 568, 686, 661, 540, 512, 301, 580, 371, 343, 399, 315, 151, 652, 287, 127, 128, 682, 427, 204, 595, 567, 455, 177, 687, 259, 129, 539, 483, 160, 497, 126, 511, 189, 525, 329, 469, 623, 441, 635, 553, 130, 125, 246, 231, 357, 608, 413, 274, 385, 150, 681, 566, 662, 688, 538, 218, 510, 594, 482, 454, 426, 398, 302, 370, 581, 131, 124, 470, 176, 342, 498, 651, 203, 161, 314, 526, 442, 286, 622, 190, 149, 330, 258, 689, 414, 636, 554, 537, 123, 680, 509, 565, 663, 358, 609, 386, 230, 132, 481, 247, 593, 453, 175, 275, 713, 712, 471, 219, 425, 443, 582, 711, 499, 162, 714, 148, 122, 202, 690, 397, 650, 303, 415, 710, 527, 369, 99, 715, 341, 313, 100, 387, 191, 621, 331, 98, 679, 637, 285, 664, 133, 359, 257, 101, 716, 555, 709, 97, 610, 121, 229, 102, 536, 174, 564, 691, 508, 147, 96, 717, 103, 163, 248, 480, 444, 472, 583, 592, 201, 708, 220, 276, 416, 95, 134, 500, 665, 649, 120, 638, 388, 452, 718, 678, 304, 740, 692, 192, 741, 528, 104, 739, 256, 284, 424, 742, 360, 312, 611, 146, 332, 738, 94, 620, 228, 173, 396, 340, 71, 368, 743, 707, 556, 719, 72, 744, 737, 70, 119, 164, 666, 73, 693, 105, 93, 584, 639, 249, 200, 135, 69, 745, 221, 563, 74, 535, 68, 736, 677, 277, 648, 720, 591, 145, 75, 706, 507, 193, 255, 283, 612, 746, 67, 92, 501, 473, 445, 305, 311, 172, 694, 227, 118, 667, 106, 529, 5], "train_min_nll": 52.27257537841797, "test_nll": 61.244014739990234}
\ No newline at end of file
diff --git a/mnist_6_mps_feature_500.safetensors b/mnist_6_mps_feature_500.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..6674fdbe7ba0bf7346d94009f9fe61c665f280c4
--- /dev/null
+++ b/mnist_6_mps_feature_500.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d85d55644352331e313d2b735225db6e0c9ff9cd836a67ae4ba1b9a29b6ebdc1
+size 16071208
diff --git a/mnist_6_mps_feature_600.json b/mnist_6_mps_feature_600.json
new file mode 100644
index 0000000000000000000000000000000000000000..b41c69557476198606a75bce7c4b00e74691b150
--- /dev/null
+++ b/mnist_6_mps_feature_600.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 600, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521, 238, 628, 237, 210, 207, 213, 239, 493, 265, 240, 632, 599, 212, 214, 241, 211, 605, 549, 236, 266, 235, 577, 242, 185, 494, 181, 572, 352, 234, 466, 627, 186, 518, 293, 268, 215, 269, 545, 182, 297, 184, 491, 292, 492, 264, 243, 180, 270, 267, 324, 438, 325, 183, 296, 658, 522, 380, 383, 410, 206, 465, 382, 271, 657, 298, 411, 355, 353, 262, 463, 578, 659, 263, 432, 598, 656, 354, 626, 299, 633, 439, 187, 320, 655, 291, 431, 571, 550, 327, 490, 261, 326, 404, 233, 319, 435, 294, 403, 379, 407, 606, 467, 244, 464, 402, 290, 544, 321, 179, 408, 430, 654, 216, 517, 381, 318, 437, 289, 374, 459, 346, 295, 272, 347, 460, 157, 436, 405, 375, 156, 495, 351, 462, 317, 433, 458, 155, 570, 345, 409, 300, 154, 158, 660, 625, 376, 373, 489, 401, 348, 461, 488, 523, 429, 551, 406, 597, 579, 434, 153, 377, 323, 516, 288, 487, 205, 349, 328, 457, 260, 316, 543, 486, 653, 378, 322, 344, 188, 356, 569, 372, 485, 178, 400, 384, 515, 245, 412, 634, 152, 232, 440, 159, 542, 350, 428, 468, 607, 624, 685, 541, 684, 273, 496, 514, 596, 524, 513, 456, 217, 552, 683, 484, 568, 686, 661, 540, 512, 301, 580, 371, 343, 399, 315, 151, 652, 287, 127, 128, 682, 427, 204, 595, 567, 455, 177, 687, 259, 129, 539, 483, 160, 497, 126, 511, 189, 525, 329, 469, 623, 441, 635, 553, 130, 125, 246, 231, 357, 608, 413, 274, 385, 150, 681, 566, 662, 688, 538, 218, 510, 594, 482, 454, 426, 398, 302, 370, 581, 131, 124, 470, 176, 342, 498, 651, 203, 161, 314, 526, 442, 286, 622, 190, 149, 330, 258, 689, 414, 636, 554, 537, 123, 680, 509, 565, 663, 358, 609, 386, 230, 132, 481, 247, 593, 453, 175, 275, 713, 712, 471, 219, 425, 443, 582, 711, 499, 162, 714, 148, 122, 202, 690, 397, 650, 303, 415, 710, 527, 369, 99, 715, 341, 313, 100, 387, 191, 621, 331, 98, 679, 637, 285, 664, 133, 359, 257, 101, 716, 555, 709, 97, 610, 121, 229, 102, 536, 174, 564, 691, 508, 147, 96, 717, 103, 163, 248, 480, 444, 472, 583, 592, 201, 708, 220, 276, 416, 95, 134, 500, 665, 649, 120, 638, 388, 452, 718, 678, 304, 740, 692, 192, 741, 528, 104, 739, 256, 284, 424, 742, 360, 312, 611, 146, 332, 738, 94, 620, 228, 173, 396, 340, 71, 368, 743, 707, 556, 719, 72, 744, 737, 70, 119, 164, 666, 73, 693, 105, 93, 584, 639, 249, 200, 135, 69, 745, 221, 563, 74, 535, 68, 736, 677, 277, 648, 720, 591, 145, 75, 706, 507, 193, 255, 283, 612, 746, 67, 92, 501, 473, 445, 305, 311, 172, 694, 227, 118, 667, 106, 529, 5, 735, 76, 479, 66, 417, 619, 721, 339, 136, 557, 747, 7, 4, 640, 451, 91, 367, 6, 333, 165, 389, 585, 423, 3, 77, 65, 676, 748, 770, 734, 199, 395, 705, 771, 695, 769, 361, 282, 117, 107, 768, 722, 254, 222, 43, 250, 772, 310, 144, 613, 647, 64, 767, 278, 668, 78, 562, 194, 90, 226, 766, 42, 749, 40, 44, 534, 773, 41, 590, 1, 306, 63, 338, 171, 2, 39, 108, 137, 733, 502, 765, 474, 723, 38, 704, 45, 618, 506, 446, 641, 478, 334, 366, 675, 558, 764, 774, 696, 46, 530], "train_min_nll": 52.61896896362305, "test_nll": 61.4886360168457}
\ No newline at end of file
diff --git a/mnist_6_mps_feature_600.safetensors b/mnist_6_mps_feature_600.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..19b9b3ca4b400beec02a23a770921ba431cecfc3
--- /dev/null
+++ b/mnist_6_mps_feature_600.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f157cbd61905dde2a83bf6c548cc4554ee66e99297a436b07527772fca87f38d
+size 19355504
diff --git a/mnist_7_losses.png b/mnist_7_losses.png
new file mode 100644
index 0000000000000000000000000000000000000000..cd72e6580af57005b0ebcc9ba209ef8b895f7033
Binary files /dev/null and b/mnist_7_losses.png differ
diff --git a/mnist_7_mps.safetensors b/mnist_7_mps.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..3e820d0ff65bcf08e390e9f214279beebd6c1e63
--- /dev/null
+++ b/mnist_7_mps.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0fb5b01ccf910360229838b5d297ed5399e26c7f8ac21d4d616080b44486bf1b
+size 25398616
diff --git a/mnist_7_mps_feature_100.json b/mnist_7_mps_feature_100.json
new file mode 100644
index 0000000000000000000000000000000000000000..acda53c4c5b9cce77c7a1d5f6f89e834019084cb
--- /dev/null
+++ b/mnist_7_mps_feature_100.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 100, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521, 238, 628, 237, 210, 207, 213, 239, 493, 265, 240, 632, 599, 212, 214, 241, 211, 605, 549, 236, 266, 235, 577, 242, 185, 494, 181, 572, 352, 234, 466, 627, 186, 518, 293, 268, 215, 269, 545, 182, 297, 184, 491, 292, 492, 264, 243, 180, 270, 267, 324, 438, 325, 183, 296, 658, 522, 380, 383, 410, 206, 465, 382, 271, 657, 298, 411, 355, 353, 262, 463, 578, 659, 263, 432, 598, 656, 354, 626, 299, 633], "train_min_nll": 18.166107177734375, "test_nll": 20.229049682617188}
\ No newline at end of file
diff --git a/mnist_7_mps_feature_100.safetensors b/mnist_7_mps_feature_100.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..e4f8a1a2c301eead3f4fe9ad460efd35a8733227
--- /dev/null
+++ b/mnist_7_mps_feature_100.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c6c08921001b936b52cca9055d629d84024d6e1615aad53c27c87218650c5285
+size 2934416
diff --git a/mnist_7_mps_feature_15.json b/mnist_7_mps_feature_15.json
new file mode 100644
index 0000000000000000000000000000000000000000..fcb7d4447a14f2b23de08e7d85300f624ca923fe
--- /dev/null
+++ b/mnist_7_mps_feature_15.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 15, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573], "train_min_nll": 3.452531337738037, "test_nll": 3.601685047149658}
\ No newline at end of file
diff --git a/mnist_7_mps_feature_15.safetensors b/mnist_7_mps_feature_15.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..a6b0f1f10b205e98ecf0326ba17561d0c9b83653
--- /dev/null
+++ b/mnist_7_mps_feature_15.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d606bb34f06f86efa4257c4a5dc55843b2ce5b47eed0f1fb02d4d948921fd6a5
+size 143032
diff --git a/mnist_7_mps_feature_20.json b/mnist_7_mps_feature_20.json
new file mode 100644
index 0000000000000000000000000000000000000000..1e68df0d2a1a70dafef4816cb655af0beed29a53
--- /dev/null
+++ b/mnist_7_mps_feature_20.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 20, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521], "train_min_nll": 4.324412822723389, "test_nll": 4.482158660888672}
\ No newline at end of file
diff --git a/mnist_7_mps_feature_20.safetensors b/mnist_7_mps_feature_20.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..3ebe6c03a974a7428de88fe9aa6232b19b639847
--- /dev/null
+++ b/mnist_7_mps_feature_20.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:266493a8690e3ae820fb0504a7ab44330d34daab8820e5ea463c38163d3021f1
+size 307248
diff --git a/mnist_7_mps_feature_200.json b/mnist_7_mps_feature_200.json
new file mode 100644
index 0000000000000000000000000000000000000000..5145a5ca9b642eb32f4bd92b41f1971a5361205e
--- /dev/null
+++ b/mnist_7_mps_feature_200.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 200, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521, 238, 628, 237, 210, 207, 213, 239, 493, 265, 240, 632, 599, 212, 214, 241, 211, 605, 549, 236, 266, 235, 577, 242, 185, 494, 181, 572, 352, 234, 466, 627, 186, 518, 293, 268, 215, 269, 545, 182, 297, 184, 491, 292, 492, 264, 243, 180, 270, 267, 324, 438, 325, 183, 296, 658, 522, 380, 383, 410, 206, 465, 382, 271, 657, 298, 411, 355, 353, 262, 463, 578, 659, 263, 432, 598, 656, 354, 626, 299, 633, 439, 187, 320, 655, 291, 431, 571, 550, 327, 490, 261, 326, 404, 233, 319, 435, 294, 403, 379, 407, 606, 467, 244, 464, 402, 290, 544, 321, 179, 408, 430, 654, 216, 517, 381, 318, 437, 289, 374, 459, 346, 295, 272, 347, 460, 157, 436, 405, 375, 156, 495, 351, 462, 317, 433, 458, 155, 570, 345, 409, 300, 154, 158, 660, 625, 376, 373, 489, 401, 348, 461, 488, 523, 429, 551, 406, 597, 579, 434, 153, 377, 323, 516, 288, 487, 205, 349, 328, 457, 260, 316, 543, 486, 653, 378, 322, 344, 188, 356, 569], "train_min_nll": 31.213092803955078, "test_nll": 35.18926239013672}
\ No newline at end of file
diff --git a/mnist_7_mps_feature_200.safetensors b/mnist_7_mps_feature_200.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..3f575500667ebd7bbb80115a5a3205349f64ee32
--- /dev/null
+++ b/mnist_7_mps_feature_200.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d2ab39b422079d37a4d78ee7cb81336a1d98440a226d250e264e7c025100a51e
+size 6218528
diff --git a/mnist_7_mps_feature_300.json b/mnist_7_mps_feature_300.json
new file mode 100644
index 0000000000000000000000000000000000000000..17ff10f29e439403345e0072261b21278cc5f664
--- /dev/null
+++ b/mnist_7_mps_feature_300.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 300, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521, 238, 628, 237, 210, 207, 213, 239, 493, 265, 240, 632, 599, 212, 214, 241, 211, 605, 549, 236, 266, 235, 577, 242, 185, 494, 181, 572, 352, 234, 466, 627, 186, 518, 293, 268, 215, 269, 545, 182, 297, 184, 491, 292, 492, 264, 243, 180, 270, 267, 324, 438, 325, 183, 296, 658, 522, 380, 383, 410, 206, 465, 382, 271, 657, 298, 411, 355, 353, 262, 463, 578, 659, 263, 432, 598, 656, 354, 626, 299, 633, 439, 187, 320, 655, 291, 431, 571, 550, 327, 490, 261, 326, 404, 233, 319, 435, 294, 403, 379, 407, 606, 467, 244, 464, 402, 290, 544, 321, 179, 408, 430, 654, 216, 517, 381, 318, 437, 289, 374, 459, 346, 295, 272, 347, 460, 157, 436, 405, 375, 156, 495, 351, 462, 317, 433, 458, 155, 570, 345, 409, 300, 154, 158, 660, 625, 376, 373, 489, 401, 348, 461, 488, 523, 429, 551, 406, 597, 579, 434, 153, 377, 323, 516, 288, 487, 205, 349, 328, 457, 260, 316, 543, 486, 653, 378, 322, 344, 188, 356, 569, 372, 485, 178, 400, 384, 515, 245, 412, 634, 152, 232, 440, 159, 542, 350, 428, 468, 607, 624, 685, 541, 684, 273, 496, 514, 596, 524, 513, 456, 217, 552, 683, 484, 568, 686, 661, 540, 512, 301, 580, 371, 343, 399, 315, 151, 652, 287, 127, 128, 682, 427, 204, 595, 567, 455, 177, 687, 259, 129, 539, 483, 160, 497, 126, 511, 189, 525, 329, 469, 623, 441, 635, 553, 130, 125, 246, 231, 357, 608, 413, 274, 385, 150, 681, 566, 662, 688, 538, 218, 510, 594, 482, 454, 426, 398, 302, 370, 581, 131, 124], "train_min_nll": 38.26948547363281, "test_nll": 43.04433059692383}
\ No newline at end of file
diff --git a/mnist_7_mps_feature_300.safetensors b/mnist_7_mps_feature_300.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..29e0d4bc2715efe8ec0347c9fd91a65800da338f
--- /dev/null
+++ b/mnist_7_mps_feature_300.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:78d2487ce681cf372a0932378c1e33e6c4d34e2ede86c098b6cd107351e077a3
+size 9502624
diff --git a/mnist_7_mps_feature_400.json b/mnist_7_mps_feature_400.json
new file mode 100644
index 0000000000000000000000000000000000000000..41b693f914aa056a492dbb5ce206c461b28c2b75
--- /dev/null
+++ b/mnist_7_mps_feature_400.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 400, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521, 238, 628, 237, 210, 207, 213, 239, 493, 265, 240, 632, 599, 212, 214, 241, 211, 605, 549, 236, 266, 235, 577, 242, 185, 494, 181, 572, 352, 234, 466, 627, 186, 518, 293, 268, 215, 269, 545, 182, 297, 184, 491, 292, 492, 264, 243, 180, 270, 267, 324, 438, 325, 183, 296, 658, 522, 380, 383, 410, 206, 465, 382, 271, 657, 298, 411, 355, 353, 262, 463, 578, 659, 263, 432, 598, 656, 354, 626, 299, 633, 439, 187, 320, 655, 291, 431, 571, 550, 327, 490, 261, 326, 404, 233, 319, 435, 294, 403, 379, 407, 606, 467, 244, 464, 402, 290, 544, 321, 179, 408, 430, 654, 216, 517, 381, 318, 437, 289, 374, 459, 346, 295, 272, 347, 460, 157, 436, 405, 375, 156, 495, 351, 462, 317, 433, 458, 155, 570, 345, 409, 300, 154, 158, 660, 625, 376, 373, 489, 401, 348, 461, 488, 523, 429, 551, 406, 597, 579, 434, 153, 377, 323, 516, 288, 487, 205, 349, 328, 457, 260, 316, 543, 486, 653, 378, 322, 344, 188, 356, 569, 372, 485, 178, 400, 384, 515, 245, 412, 634, 152, 232, 440, 159, 542, 350, 428, 468, 607, 624, 685, 541, 684, 273, 496, 514, 596, 524, 513, 456, 217, 552, 683, 484, 568, 686, 661, 540, 512, 301, 580, 371, 343, 399, 315, 151, 652, 287, 127, 128, 682, 427, 204, 595, 567, 455, 177, 687, 259, 129, 539, 483, 160, 497, 126, 511, 189, 525, 329, 469, 623, 441, 635, 553, 130, 125, 246, 231, 357, 608, 413, 274, 385, 150, 681, 566, 662, 688, 538, 218, 510, 594, 482, 454, 426, 398, 302, 370, 581, 131, 124, 470, 176, 342, 498, 651, 203, 161, 314, 526, 442, 286, 622, 190, 149, 330, 258, 689, 414, 636, 554, 537, 123, 680, 509, 565, 663, 358, 609, 386, 230, 132, 481, 247, 593, 453, 175, 275, 713, 712, 471, 219, 425, 443, 582, 711, 499, 162, 714, 148, 122, 202, 690, 397, 650, 303, 415, 710, 527, 369, 99, 715, 341, 313, 100, 387, 191, 621, 331, 98, 679, 637, 285, 664, 133, 359, 257, 101, 716, 555, 709, 97, 610, 121, 229, 102, 536, 174, 564, 691, 508, 147, 96, 717, 103, 163, 248, 480, 444, 472, 583], "train_min_nll": 42.88633346557617, "test_nll": 48.643348693847656}
\ No newline at end of file
diff --git a/mnist_7_mps_feature_400.safetensors b/mnist_7_mps_feature_400.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..fb583faf1708dff17d3cfd89a499d6ba008bc8cd
--- /dev/null
+++ b/mnist_7_mps_feature_400.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1353e16f235e2fae7b5390568104d5b629bb1f3b58e04d58acc3df1c248da684
+size 12786904
diff --git a/mnist_7_mps_feature_50.json b/mnist_7_mps_feature_50.json
new file mode 100644
index 0000000000000000000000000000000000000000..b3810cc7dd803baa0cb999f8ef351f6ea48cc4ba
--- /dev/null
+++ b/mnist_7_mps_feature_50.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 50, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521, 238, 628, 237, 210, 207, 213, 239, 493, 265, 240, 632, 599, 212, 214, 241, 211, 605, 549, 236, 266, 235, 577, 242, 185, 494, 181, 572, 352, 234, 466], "train_min_nll": 9.180594444274902, "test_nll": 9.934714317321777}
\ No newline at end of file
diff --git a/mnist_7_mps_feature_50.safetensors b/mnist_7_mps_feature_50.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..0d37a70024ef9fb9f4e49e0d86689bd3dc5c1d8a
--- /dev/null
+++ b/mnist_7_mps_feature_50.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:234790d8c2d6cf5089810811d97c481bd6bd0b2735436f191133a06d0c530278
+size 1292416
diff --git a/mnist_7_mps_feature_500.json b/mnist_7_mps_feature_500.json
new file mode 100644
index 0000000000000000000000000000000000000000..62c1918fd500e10b75fd7b58ea33700665403883
--- /dev/null
+++ b/mnist_7_mps_feature_500.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 500, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521, 238, 628, 237, 210, 207, 213, 239, 493, 265, 240, 632, 599, 212, 214, 241, 211, 605, 549, 236, 266, 235, 577, 242, 185, 494, 181, 572, 352, 234, 466, 627, 186, 518, 293, 268, 215, 269, 545, 182, 297, 184, 491, 292, 492, 264, 243, 180, 270, 267, 324, 438, 325, 183, 296, 658, 522, 380, 383, 410, 206, 465, 382, 271, 657, 298, 411, 355, 353, 262, 463, 578, 659, 263, 432, 598, 656, 354, 626, 299, 633, 439, 187, 320, 655, 291, 431, 571, 550, 327, 490, 261, 326, 404, 233, 319, 435, 294, 403, 379, 407, 606, 467, 244, 464, 402, 290, 544, 321, 179, 408, 430, 654, 216, 517, 381, 318, 437, 289, 374, 459, 346, 295, 272, 347, 460, 157, 436, 405, 375, 156, 495, 351, 462, 317, 433, 458, 155, 570, 345, 409, 300, 154, 158, 660, 625, 376, 373, 489, 401, 348, 461, 488, 523, 429, 551, 406, 597, 579, 434, 153, 377, 323, 516, 288, 487, 205, 349, 328, 457, 260, 316, 543, 486, 653, 378, 322, 344, 188, 356, 569, 372, 485, 178, 400, 384, 515, 245, 412, 634, 152, 232, 440, 159, 542, 350, 428, 468, 607, 624, 685, 541, 684, 273, 496, 514, 596, 524, 513, 456, 217, 552, 683, 484, 568, 686, 661, 540, 512, 301, 580, 371, 343, 399, 315, 151, 652, 287, 127, 128, 682, 427, 204, 595, 567, 455, 177, 687, 259, 129, 539, 483, 160, 497, 126, 511, 189, 525, 329, 469, 623, 441, 635, 553, 130, 125, 246, 231, 357, 608, 413, 274, 385, 150, 681, 566, 662, 688, 538, 218, 510, 594, 482, 454, 426, 398, 302, 370, 581, 131, 124, 470, 176, 342, 498, 651, 203, 161, 314, 526, 442, 286, 622, 190, 149, 330, 258, 689, 414, 636, 554, 537, 123, 680, 509, 565, 663, 358, 609, 386, 230, 132, 481, 247, 593, 453, 175, 275, 713, 712, 471, 219, 425, 443, 582, 711, 499, 162, 714, 148, 122, 202, 690, 397, 650, 303, 415, 710, 527, 369, 99, 715, 341, 313, 100, 387, 191, 621, 331, 98, 679, 637, 285, 664, 133, 359, 257, 101, 716, 555, 709, 97, 610, 121, 229, 102, 536, 174, 564, 691, 508, 147, 96, 717, 103, 163, 248, 480, 444, 472, 583, 592, 201, 708, 220, 276, 416, 95, 134, 500, 665, 649, 120, 638, 388, 452, 718, 678, 304, 740, 692, 192, 741, 528, 104, 739, 256, 284, 424, 742, 360, 312, 611, 146, 332, 738, 94, 620, 228, 173, 396, 340, 71, 368, 743, 707, 556, 719, 72, 744, 737, 70, 119, 164, 666, 73, 693, 105, 93, 584, 639, 249, 200, 135, 69, 745, 221, 563, 74, 535, 68, 736, 677, 277, 648, 720, 591, 145, 75, 706, 507, 193, 255, 283, 612, 746, 67, 92, 501, 473, 445, 305, 311, 172, 694, 227, 118, 667, 106, 529, 5], "train_min_nll": 44.17021560668945, "test_nll": 50.253753662109375}
\ No newline at end of file
diff --git a/mnist_7_mps_feature_500.safetensors b/mnist_7_mps_feature_500.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..5caca5c165c316abc2267dccb128bfc1f48eb059
--- /dev/null
+++ b/mnist_7_mps_feature_500.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:54b7df06a08a308f94ecbce7fa9f9f3c20cfd02445c34c93ee6fe512058fd3aa
+size 16071208
diff --git a/mnist_7_mps_feature_600.json b/mnist_7_mps_feature_600.json
new file mode 100644
index 0000000000000000000000000000000000000000..0ec8a0fa854492ca3316c8ea100f94da5a7e26f7
--- /dev/null
+++ b/mnist_7_mps_feature_600.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 600, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521, 238, 628, 237, 210, 207, 213, 239, 493, 265, 240, 632, 599, 212, 214, 241, 211, 605, 549, 236, 266, 235, 577, 242, 185, 494, 181, 572, 352, 234, 466, 627, 186, 518, 293, 268, 215, 269, 545, 182, 297, 184, 491, 292, 492, 264, 243, 180, 270, 267, 324, 438, 325, 183, 296, 658, 522, 380, 383, 410, 206, 465, 382, 271, 657, 298, 411, 355, 353, 262, 463, 578, 659, 263, 432, 598, 656, 354, 626, 299, 633, 439, 187, 320, 655, 291, 431, 571, 550, 327, 490, 261, 326, 404, 233, 319, 435, 294, 403, 379, 407, 606, 467, 244, 464, 402, 290, 544, 321, 179, 408, 430, 654, 216, 517, 381, 318, 437, 289, 374, 459, 346, 295, 272, 347, 460, 157, 436, 405, 375, 156, 495, 351, 462, 317, 433, 458, 155, 570, 345, 409, 300, 154, 158, 660, 625, 376, 373, 489, 401, 348, 461, 488, 523, 429, 551, 406, 597, 579, 434, 153, 377, 323, 516, 288, 487, 205, 349, 328, 457, 260, 316, 543, 486, 653, 378, 322, 344, 188, 356, 569, 372, 485, 178, 400, 384, 515, 245, 412, 634, 152, 232, 440, 159, 542, 350, 428, 468, 607, 624, 685, 541, 684, 273, 496, 514, 596, 524, 513, 456, 217, 552, 683, 484, 568, 686, 661, 540, 512, 301, 580, 371, 343, 399, 315, 151, 652, 287, 127, 128, 682, 427, 204, 595, 567, 455, 177, 687, 259, 129, 539, 483, 160, 497, 126, 511, 189, 525, 329, 469, 623, 441, 635, 553, 130, 125, 246, 231, 357, 608, 413, 274, 385, 150, 681, 566, 662, 688, 538, 218, 510, 594, 482, 454, 426, 398, 302, 370, 581, 131, 124, 470, 176, 342, 498, 651, 203, 161, 314, 526, 442, 286, 622, 190, 149, 330, 258, 689, 414, 636, 554, 537, 123, 680, 509, 565, 663, 358, 609, 386, 230, 132, 481, 247, 593, 453, 175, 275, 713, 712, 471, 219, 425, 443, 582, 711, 499, 162, 714, 148, 122, 202, 690, 397, 650, 303, 415, 710, 527, 369, 99, 715, 341, 313, 100, 387, 191, 621, 331, 98, 679, 637, 285, 664, 133, 359, 257, 101, 716, 555, 709, 97, 610, 121, 229, 102, 536, 174, 564, 691, 508, 147, 96, 717, 103, 163, 248, 480, 444, 472, 583, 592, 201, 708, 220, 276, 416, 95, 134, 500, 665, 649, 120, 638, 388, 452, 718, 678, 304, 740, 692, 192, 741, 528, 104, 739, 256, 284, 424, 742, 360, 312, 611, 146, 332, 738, 94, 620, 228, 173, 396, 340, 71, 368, 743, 707, 556, 719, 72, 744, 737, 70, 119, 164, 666, 73, 693, 105, 93, 584, 639, 249, 200, 135, 69, 745, 221, 563, 74, 535, 68, 736, 677, 277, 648, 720, 591, 145, 75, 706, 507, 193, 255, 283, 612, 746, 67, 92, 501, 473, 445, 305, 311, 172, 694, 227, 118, 667, 106, 529, 5, 735, 76, 479, 66, 417, 619, 721, 339, 136, 557, 747, 7, 4, 640, 451, 91, 367, 6, 333, 165, 389, 585, 423, 3, 77, 65, 676, 748, 770, 734, 199, 395, 705, 771, 695, 769, 361, 282, 117, 107, 768, 722, 254, 222, 43, 250, 772, 310, 144, 613, 647, 64, 767, 278, 668, 78, 562, 194, 90, 226, 766, 42, 749, 40, 44, 534, 773, 41, 590, 1, 306, 63, 338, 171, 2, 39, 108, 137, 733, 502, 765, 474, 723, 38, 704, 45, 618, 506, 446, 641, 478, 334, 366, 675, 558, 764, 774, 696, 46, 530], "train_min_nll": 44.12428665161133, "test_nll": 49.84828186035156}
\ No newline at end of file
diff --git a/mnist_7_mps_feature_600.safetensors b/mnist_7_mps_feature_600.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..36b17c88ea31a91a159c6d7b9a89f01b8b4a41cb
--- /dev/null
+++ b/mnist_7_mps_feature_600.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:cb6cee5a47ce6c4bfb3ac3d74481e9afc8b9a0999965bccdb91e05478827f8f6
+size 19355504
diff --git a/mnist_8_losses.png b/mnist_8_losses.png
new file mode 100644
index 0000000000000000000000000000000000000000..8fc18356d7789e7d478435a9d5f50d0424767223
Binary files /dev/null and b/mnist_8_losses.png differ
diff --git a/mnist_8_mps.safetensors b/mnist_8_mps.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..d761a5aaba2861a9168982c1139b7c9b3ba36b91
--- /dev/null
+++ b/mnist_8_mps.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:70a3d1b88cd6472dce61367b78c94c81083b1b8f50fd9642abfe8dc683e9a5f2
+size 25398616
diff --git a/mnist_8_mps_feature_100.json b/mnist_8_mps_feature_100.json
new file mode 100644
index 0000000000000000000000000000000000000000..7cf1aabdae8f6b0e07f76715d2119fccf703b84d
--- /dev/null
+++ b/mnist_8_mps_feature_100.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 100, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521, 238, 628, 237, 210, 207, 213, 239, 493, 265, 240, 632, 599, 212, 214, 241, 211, 605, 549, 236, 266, 235, 577, 242, 185, 494, 181, 572, 352, 234, 466, 627, 186, 518, 293, 268, 215, 269, 545, 182, 297, 184, 491, 292, 492, 264, 243, 180, 270, 267, 324, 438, 325, 183, 296, 658, 522, 380, 383, 410, 206, 465, 382, 271, 657, 298, 411, 355, 353, 262, 463, 578, 659, 263, 432, 598, 656, 354, 626, 299, 633], "train_min_nll": 25.681318283081055, "test_nll": 30.149761199951172}
\ No newline at end of file
diff --git a/mnist_8_mps_feature_100.safetensors b/mnist_8_mps_feature_100.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..c4c08abecef520a801a598d5512453cb378a0723
--- /dev/null
+++ b/mnist_8_mps_feature_100.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:fe35fcc11472f97a3ff8c1b02da54f0d9387873d6519543b7a3deee4c47dd2d0
+size 2934416
diff --git a/mnist_8_mps_feature_15.json b/mnist_8_mps_feature_15.json
new file mode 100644
index 0000000000000000000000000000000000000000..c5b66bfb09cad18abade22bd82cb07bbde688297
--- /dev/null
+++ b/mnist_8_mps_feature_15.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 15, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573], "train_min_nll": 4.636969089508057, "test_nll": 4.740553855895996}
\ No newline at end of file
diff --git a/mnist_8_mps_feature_15.safetensors b/mnist_8_mps_feature_15.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..ecfc289ce776d1e2d6a51f736ae776ef544697f9
--- /dev/null
+++ b/mnist_8_mps_feature_15.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:713f3b541c17fce922571ced21b1ccfbee56604a24c143127afd2e0e50a7c9fc
+size 143032
diff --git a/mnist_8_mps_feature_20.json b/mnist_8_mps_feature_20.json
new file mode 100644
index 0000000000000000000000000000000000000000..69537c70e33caabb019f6ee0fe5c56180f9a4342
--- /dev/null
+++ b/mnist_8_mps_feature_20.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 20, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521], "train_min_nll": 5.709591388702393, "test_nll": 5.970897674560547}
\ No newline at end of file
diff --git a/mnist_8_mps_feature_20.safetensors b/mnist_8_mps_feature_20.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..0e10e6645b38e813998b9db7d52a181d1f9cd5e0
--- /dev/null
+++ b/mnist_8_mps_feature_20.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:72f5f78e7f361580ea691c734954561444b39aedffebb064d00a2ac780574aea
+size 307248
diff --git a/mnist_8_mps_feature_200.json b/mnist_8_mps_feature_200.json
new file mode 100644
index 0000000000000000000000000000000000000000..a684c6820bff9dc48f24edaba85caa099022566d
--- /dev/null
+++ b/mnist_8_mps_feature_200.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 200, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521, 238, 628, 237, 210, 207, 213, 239, 493, 265, 240, 632, 599, 212, 214, 241, 211, 605, 549, 236, 266, 235, 577, 242, 185, 494, 181, 572, 352, 234, 466, 627, 186, 518, 293, 268, 215, 269, 545, 182, 297, 184, 491, 292, 492, 264, 243, 180, 270, 267, 324, 438, 325, 183, 296, 658, 522, 380, 383, 410, 206, 465, 382, 271, 657, 298, 411, 355, 353, 262, 463, 578, 659, 263, 432, 598, 656, 354, 626, 299, 633, 439, 187, 320, 655, 291, 431, 571, 550, 327, 490, 261, 326, 404, 233, 319, 435, 294, 403, 379, 407, 606, 467, 244, 464, 402, 290, 544, 321, 179, 408, 430, 654, 216, 517, 381, 318, 437, 289, 374, 459, 346, 295, 272, 347, 460, 157, 436, 405, 375, 156, 495, 351, 462, 317, 433, 458, 155, 570, 345, 409, 300, 154, 158, 660, 625, 376, 373, 489, 401, 348, 461, 488, 523, 429, 551, 406, 597, 579, 434, 153, 377, 323, 516, 288, 487, 205, 349, 328, 457, 260, 316, 543, 486, 653, 378, 322, 344, 188, 356, 569], "train_min_nll": 50.4657096862793, "test_nll": 59.45723342895508}
\ No newline at end of file
diff --git a/mnist_8_mps_feature_200.safetensors b/mnist_8_mps_feature_200.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..e5cfaf21ac7abd1cbb12b81fde2aae40087a603a
--- /dev/null
+++ b/mnist_8_mps_feature_200.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b02cae66eda166e964f259f1ed59f8649f48fa7c6285630a782ba644ab3de772
+size 6218528
diff --git a/mnist_8_mps_feature_300.json b/mnist_8_mps_feature_300.json
new file mode 100644
index 0000000000000000000000000000000000000000..4d5a89117e9425865dcf24fbccba99e62092479d
--- /dev/null
+++ b/mnist_8_mps_feature_300.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 300, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521, 238, 628, 237, 210, 207, 213, 239, 493, 265, 240, 632, 599, 212, 214, 241, 211, 605, 549, 236, 266, 235, 577, 242, 185, 494, 181, 572, 352, 234, 466, 627, 186, 518, 293, 268, 215, 269, 545, 182, 297, 184, 491, 292, 492, 264, 243, 180, 270, 267, 324, 438, 325, 183, 296, 658, 522, 380, 383, 410, 206, 465, 382, 271, 657, 298, 411, 355, 353, 262, 463, 578, 659, 263, 432, 598, 656, 354, 626, 299, 633, 439, 187, 320, 655, 291, 431, 571, 550, 327, 490, 261, 326, 404, 233, 319, 435, 294, 403, 379, 407, 606, 467, 244, 464, 402, 290, 544, 321, 179, 408, 430, 654, 216, 517, 381, 318, 437, 289, 374, 459, 346, 295, 272, 347, 460, 157, 436, 405, 375, 156, 495, 351, 462, 317, 433, 458, 155, 570, 345, 409, 300, 154, 158, 660, 625, 376, 373, 489, 401, 348, 461, 488, 523, 429, 551, 406, 597, 579, 434, 153, 377, 323, 516, 288, 487, 205, 349, 328, 457, 260, 316, 543, 486, 653, 378, 322, 344, 188, 356, 569, 372, 485, 178, 400, 384, 515, 245, 412, 634, 152, 232, 440, 159, 542, 350, 428, 468, 607, 624, 685, 541, 684, 273, 496, 514, 596, 524, 513, 456, 217, 552, 683, 484, 568, 686, 661, 540, 512, 301, 580, 371, 343, 399, 315, 151, 652, 287, 127, 128, 682, 427, 204, 595, 567, 455, 177, 687, 259, 129, 539, 483, 160, 497, 126, 511, 189, 525, 329, 469, 623, 441, 635, 553, 130, 125, 246, 231, 357, 608, 413, 274, 385, 150, 681, 566, 662, 688, 538, 218, 510, 594, 482, 454, 426, 398, 302, 370, 581, 131, 124], "train_min_nll": 62.62789535522461, "test_nll": 73.63768768310547}
\ No newline at end of file
diff --git a/mnist_8_mps_feature_300.safetensors b/mnist_8_mps_feature_300.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..e152434854d22c7329be100e26a51a2d05c4eee6
--- /dev/null
+++ b/mnist_8_mps_feature_300.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4b68264e154f8d741a1efbd6213efba2d11254166679736229f46f8f22ff408a
+size 9502624
diff --git a/mnist_8_mps_feature_400.json b/mnist_8_mps_feature_400.json
new file mode 100644
index 0000000000000000000000000000000000000000..b5e173a72ce4b66ead61d241d43fdcbffda318d6
--- /dev/null
+++ b/mnist_8_mps_feature_400.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 400, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521, 238, 628, 237, 210, 207, 213, 239, 493, 265, 240, 632, 599, 212, 214, 241, 211, 605, 549, 236, 266, 235, 577, 242, 185, 494, 181, 572, 352, 234, 466, 627, 186, 518, 293, 268, 215, 269, 545, 182, 297, 184, 491, 292, 492, 264, 243, 180, 270, 267, 324, 438, 325, 183, 296, 658, 522, 380, 383, 410, 206, 465, 382, 271, 657, 298, 411, 355, 353, 262, 463, 578, 659, 263, 432, 598, 656, 354, 626, 299, 633, 439, 187, 320, 655, 291, 431, 571, 550, 327, 490, 261, 326, 404, 233, 319, 435, 294, 403, 379, 407, 606, 467, 244, 464, 402, 290, 544, 321, 179, 408, 430, 654, 216, 517, 381, 318, 437, 289, 374, 459, 346, 295, 272, 347, 460, 157, 436, 405, 375, 156, 495, 351, 462, 317, 433, 458, 155, 570, 345, 409, 300, 154, 158, 660, 625, 376, 373, 489, 401, 348, 461, 488, 523, 429, 551, 406, 597, 579, 434, 153, 377, 323, 516, 288, 487, 205, 349, 328, 457, 260, 316, 543, 486, 653, 378, 322, 344, 188, 356, 569, 372, 485, 178, 400, 384, 515, 245, 412, 634, 152, 232, 440, 159, 542, 350, 428, 468, 607, 624, 685, 541, 684, 273, 496, 514, 596, 524, 513, 456, 217, 552, 683, 484, 568, 686, 661, 540, 512, 301, 580, 371, 343, 399, 315, 151, 652, 287, 127, 128, 682, 427, 204, 595, 567, 455, 177, 687, 259, 129, 539, 483, 160, 497, 126, 511, 189, 525, 329, 469, 623, 441, 635, 553, 130, 125, 246, 231, 357, 608, 413, 274, 385, 150, 681, 566, 662, 688, 538, 218, 510, 594, 482, 454, 426, 398, 302, 370, 581, 131, 124, 470, 176, 342, 498, 651, 203, 161, 314, 526, 442, 286, 622, 190, 149, 330, 258, 689, 414, 636, 554, 537, 123, 680, 509, 565, 663, 358, 609, 386, 230, 132, 481, 247, 593, 453, 175, 275, 713, 712, 471, 219, 425, 443, 582, 711, 499, 162, 714, 148, 122, 202, 690, 397, 650, 303, 415, 710, 527, 369, 99, 715, 341, 313, 100, 387, 191, 621, 331, 98, 679, 637, 285, 664, 133, 359, 257, 101, 716, 555, 709, 97, 610, 121, 229, 102, 536, 174, 564, 691, 508, 147, 96, 717, 103, 163, 248, 480, 444, 472, 583], "train_min_nll": 65.30487060546875, "test_nll": 77.10811614990234}
\ No newline at end of file
diff --git a/mnist_8_mps_feature_400.safetensors b/mnist_8_mps_feature_400.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..9aa07c57026f3da1ce05d87a62d002cf47bce21b
--- /dev/null
+++ b/mnist_8_mps_feature_400.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f85e37d8a5e8b00364f9f51a459d1d230518c21f5f6fa6998213e0b761839b71
+size 12786904
diff --git a/mnist_8_mps_feature_50.json b/mnist_8_mps_feature_50.json
new file mode 100644
index 0000000000000000000000000000000000000000..0142ad2a20c9fca3c2f8bb8e63cf02e8e1b9a4bd
--- /dev/null
+++ b/mnist_8_mps_feature_50.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 50, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521, 238, 628, 237, 210, 207, 213, 239, 493, 265, 240, 632, 599, 212, 214, 241, 211, 605, 549, 236, 266, 235, 577, 242, 185, 494, 181, 572, 352, 234, 466], "train_min_nll": 13.304944038391113, "test_nll": 15.418815612792969}
\ No newline at end of file
diff --git a/mnist_8_mps_feature_50.safetensors b/mnist_8_mps_feature_50.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..ee1eae5f2245fac298296beac99f807f162c74f1
--- /dev/null
+++ b/mnist_8_mps_feature_50.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:85a380efe85e44f76082bc9f9de0b20388f287d401de097dcc54dc0be68dde91
+size 1292416
diff --git a/mnist_8_mps_feature_500.json b/mnist_8_mps_feature_500.json
new file mode 100644
index 0000000000000000000000000000000000000000..82cde546ef937c48b8dadceb26b02df68f138b93
--- /dev/null
+++ b/mnist_8_mps_feature_500.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 500, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521, 238, 628, 237, 210, 207, 213, 239, 493, 265, 240, 632, 599, 212, 214, 241, 211, 605, 549, 236, 266, 235, 577, 242, 185, 494, 181, 572, 352, 234, 466, 627, 186, 518, 293, 268, 215, 269, 545, 182, 297, 184, 491, 292, 492, 264, 243, 180, 270, 267, 324, 438, 325, 183, 296, 658, 522, 380, 383, 410, 206, 465, 382, 271, 657, 298, 411, 355, 353, 262, 463, 578, 659, 263, 432, 598, 656, 354, 626, 299, 633, 439, 187, 320, 655, 291, 431, 571, 550, 327, 490, 261, 326, 404, 233, 319, 435, 294, 403, 379, 407, 606, 467, 244, 464, 402, 290, 544, 321, 179, 408, 430, 654, 216, 517, 381, 318, 437, 289, 374, 459, 346, 295, 272, 347, 460, 157, 436, 405, 375, 156, 495, 351, 462, 317, 433, 458, 155, 570, 345, 409, 300, 154, 158, 660, 625, 376, 373, 489, 401, 348, 461, 488, 523, 429, 551, 406, 597, 579, 434, 153, 377, 323, 516, 288, 487, 205, 349, 328, 457, 260, 316, 543, 486, 653, 378, 322, 344, 188, 356, 569, 372, 485, 178, 400, 384, 515, 245, 412, 634, 152, 232, 440, 159, 542, 350, 428, 468, 607, 624, 685, 541, 684, 273, 496, 514, 596, 524, 513, 456, 217, 552, 683, 484, 568, 686, 661, 540, 512, 301, 580, 371, 343, 399, 315, 151, 652, 287, 127, 128, 682, 427, 204, 595, 567, 455, 177, 687, 259, 129, 539, 483, 160, 497, 126, 511, 189, 525, 329, 469, 623, 441, 635, 553, 130, 125, 246, 231, 357, 608, 413, 274, 385, 150, 681, 566, 662, 688, 538, 218, 510, 594, 482, 454, 426, 398, 302, 370, 581, 131, 124, 470, 176, 342, 498, 651, 203, 161, 314, 526, 442, 286, 622, 190, 149, 330, 258, 689, 414, 636, 554, 537, 123, 680, 509, 565, 663, 358, 609, 386, 230, 132, 481, 247, 593, 453, 175, 275, 713, 712, 471, 219, 425, 443, 582, 711, 499, 162, 714, 148, 122, 202, 690, 397, 650, 303, 415, 710, 527, 369, 99, 715, 341, 313, 100, 387, 191, 621, 331, 98, 679, 637, 285, 664, 133, 359, 257, 101, 716, 555, 709, 97, 610, 121, 229, 102, 536, 174, 564, 691, 508, 147, 96, 717, 103, 163, 248, 480, 444, 472, 583, 592, 201, 708, 220, 276, 416, 95, 134, 500, 665, 649, 120, 638, 388, 452, 718, 678, 304, 740, 692, 192, 741, 528, 104, 739, 256, 284, 424, 742, 360, 312, 611, 146, 332, 738, 94, 620, 228, 173, 396, 340, 71, 368, 743, 707, 556, 719, 72, 744, 737, 70, 119, 164, 666, 73, 693, 105, 93, 584, 639, 249, 200, 135, 69, 745, 221, 563, 74, 535, 68, 736, 677, 277, 648, 720, 591, 145, 75, 706, 507, 193, 255, 283, 612, 746, 67, 92, 501, 473, 445, 305, 311, 172, 694, 227, 118, 667, 106, 529, 5], "train_min_nll": 65.45540618896484, "test_nll": 76.9846420288086}
\ No newline at end of file
diff --git a/mnist_8_mps_feature_500.safetensors b/mnist_8_mps_feature_500.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..1c0cb2573b34398c718656504209c4605eba3b5b
--- /dev/null
+++ b/mnist_8_mps_feature_500.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e423db88c50a467916923c009c4abcfcc64bf072f597fed46870a5c1708cadd4
+size 16071208
diff --git a/mnist_8_mps_feature_600.json b/mnist_8_mps_feature_600.json
new file mode 100644
index 0000000000000000000000000000000000000000..6e576ebf50f4736ec762d49f399b852fe50aab30
--- /dev/null
+++ b/mnist_8_mps_feature_600.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 600, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521, 238, 628, 237, 210, 207, 213, 239, 493, 265, 240, 632, 599, 212, 214, 241, 211, 605, 549, 236, 266, 235, 577, 242, 185, 494, 181, 572, 352, 234, 466, 627, 186, 518, 293, 268, 215, 269, 545, 182, 297, 184, 491, 292, 492, 264, 243, 180, 270, 267, 324, 438, 325, 183, 296, 658, 522, 380, 383, 410, 206, 465, 382, 271, 657, 298, 411, 355, 353, 262, 463, 578, 659, 263, 432, 598, 656, 354, 626, 299, 633, 439, 187, 320, 655, 291, 431, 571, 550, 327, 490, 261, 326, 404, 233, 319, 435, 294, 403, 379, 407, 606, 467, 244, 464, 402, 290, 544, 321, 179, 408, 430, 654, 216, 517, 381, 318, 437, 289, 374, 459, 346, 295, 272, 347, 460, 157, 436, 405, 375, 156, 495, 351, 462, 317, 433, 458, 155, 570, 345, 409, 300, 154, 158, 660, 625, 376, 373, 489, 401, 348, 461, 488, 523, 429, 551, 406, 597, 579, 434, 153, 377, 323, 516, 288, 487, 205, 349, 328, 457, 260, 316, 543, 486, 653, 378, 322, 344, 188, 356, 569, 372, 485, 178, 400, 384, 515, 245, 412, 634, 152, 232, 440, 159, 542, 350, 428, 468, 607, 624, 685, 541, 684, 273, 496, 514, 596, 524, 513, 456, 217, 552, 683, 484, 568, 686, 661, 540, 512, 301, 580, 371, 343, 399, 315, 151, 652, 287, 127, 128, 682, 427, 204, 595, 567, 455, 177, 687, 259, 129, 539, 483, 160, 497, 126, 511, 189, 525, 329, 469, 623, 441, 635, 553, 130, 125, 246, 231, 357, 608, 413, 274, 385, 150, 681, 566, 662, 688, 538, 218, 510, 594, 482, 454, 426, 398, 302, 370, 581, 131, 124, 470, 176, 342, 498, 651, 203, 161, 314, 526, 442, 286, 622, 190, 149, 330, 258, 689, 414, 636, 554, 537, 123, 680, 509, 565, 663, 358, 609, 386, 230, 132, 481, 247, 593, 453, 175, 275, 713, 712, 471, 219, 425, 443, 582, 711, 499, 162, 714, 148, 122, 202, 690, 397, 650, 303, 415, 710, 527, 369, 99, 715, 341, 313, 100, 387, 191, 621, 331, 98, 679, 637, 285, 664, 133, 359, 257, 101, 716, 555, 709, 97, 610, 121, 229, 102, 536, 174, 564, 691, 508, 147, 96, 717, 103, 163, 248, 480, 444, 472, 583, 592, 201, 708, 220, 276, 416, 95, 134, 500, 665, 649, 120, 638, 388, 452, 718, 678, 304, 740, 692, 192, 741, 528, 104, 739, 256, 284, 424, 742, 360, 312, 611, 146, 332, 738, 94, 620, 228, 173, 396, 340, 71, 368, 743, 707, 556, 719, 72, 744, 737, 70, 119, 164, 666, 73, 693, 105, 93, 584, 639, 249, 200, 135, 69, 745, 221, 563, 74, 535, 68, 736, 677, 277, 648, 720, 591, 145, 75, 706, 507, 193, 255, 283, 612, 746, 67, 92, 501, 473, 445, 305, 311, 172, 694, 227, 118, 667, 106, 529, 5, 735, 76, 479, 66, 417, 619, 721, 339, 136, 557, 747, 7, 4, 640, 451, 91, 367, 6, 333, 165, 389, 585, 423, 3, 77, 65, 676, 748, 770, 734, 199, 395, 705, 771, 695, 769, 361, 282, 117, 107, 768, 722, 254, 222, 43, 250, 772, 310, 144, 613, 647, 64, 767, 278, 668, 78, 562, 194, 90, 226, 766, 42, 749, 40, 44, 534, 773, 41, 590, 1, 306, 63, 338, 171, 2, 39, 108, 137, 733, 502, 765, 474, 723, 38, 704, 45, 618, 506, 446, 641, 478, 334, 366, 675, 558, 764, 774, 696, 46, 530], "train_min_nll": 65.89892578125, "test_nll": 77.50455474853516}
\ No newline at end of file
diff --git a/mnist_8_mps_feature_600.safetensors b/mnist_8_mps_feature_600.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..26ceb49ff1cb8ca56f074eb4a4a00bd163a7f6e1
--- /dev/null
+++ b/mnist_8_mps_feature_600.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:213e6c7c9af617d484919b35538f5edd0e9c787634aa1f13a3366147f25cc37e
+size 19355504
diff --git a/mnist_9_losses.png b/mnist_9_losses.png
new file mode 100644
index 0000000000000000000000000000000000000000..696988abc3c48e692dc1c2323194845bf9b9efcb
Binary files /dev/null and b/mnist_9_losses.png differ
diff --git a/mnist_9_mps.safetensors b/mnist_9_mps.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..238561ca0d322e90671c8ff0442558b31bdcd814
--- /dev/null
+++ b/mnist_9_mps.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:dfb9559c5235f34c1d8df256cc6227c1ba08617c4a0fab2cb174c5e775caeef0
+size 25398616
diff --git a/mnist_9_mps_feature_100.json b/mnist_9_mps_feature_100.json
new file mode 100644
index 0000000000000000000000000000000000000000..00140e47dbc67b57c4b2b84655aa9f1e32166f47
--- /dev/null
+++ b/mnist_9_mps_feature_100.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 100, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521, 238, 628, 237, 210, 207, 213, 239, 493, 265, 240, 632, 599, 212, 214, 241, 211, 605, 549, 236, 266, 235, 577, 242, 185, 494, 181, 572, 352, 234, 466, 627, 186, 518, 293, 268, 215, 269, 545, 182, 297, 184, 491, 292, 492, 264, 243, 180, 270, 267, 324, 438, 325, 183, 296, 658, 522, 380, 383, 410, 206, 465, 382, 271, 657, 298, 411, 355, 353, 262, 463, 578, 659, 263, 432, 598, 656, 354, 626, 299, 633], "train_min_nll": 20.026180267333984, "test_nll": 22.2977294921875}
\ No newline at end of file
diff --git a/mnist_9_mps_feature_100.safetensors b/mnist_9_mps_feature_100.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..8628adfea7de1c82dc49843326b73780995330ea
--- /dev/null
+++ b/mnist_9_mps_feature_100.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:71be179c5c96cc75829a3a5c00a11c874b356af94a4d93abd08bf7bf66f77aa0
+size 2934416
diff --git a/mnist_9_mps_feature_15.json b/mnist_9_mps_feature_15.json
new file mode 100644
index 0000000000000000000000000000000000000000..be19b7fb4583015af62a8cc1efbaea83323ad674
--- /dev/null
+++ b/mnist_9_mps_feature_15.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 15, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573], "train_min_nll": 3.143733501434326, "test_nll": 3.1358907222747803}
\ No newline at end of file
diff --git a/mnist_9_mps_feature_15.safetensors b/mnist_9_mps_feature_15.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..72920d733f4852ddacf1fed3ab3ba6dcaad08703
--- /dev/null
+++ b/mnist_9_mps_feature_15.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:363bb548f12b311db30210abcaf1139d8e0d2d68ec109c7d50bac4073c56690b
+size 143032
diff --git a/mnist_9_mps_feature_20.json b/mnist_9_mps_feature_20.json
new file mode 100644
index 0000000000000000000000000000000000000000..0e3b7607187fa2c76613a74604859be9343fb83c
--- /dev/null
+++ b/mnist_9_mps_feature_20.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 20, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521], "train_min_nll": 4.326898097991943, "test_nll": 4.50715970993042}
\ No newline at end of file
diff --git a/mnist_9_mps_feature_20.safetensors b/mnist_9_mps_feature_20.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..c8e902b1c8f3729aaa9bf8c962aa9d85782e1c39
--- /dev/null
+++ b/mnist_9_mps_feature_20.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f54368982bb5df4ffe0c212845ca0af35dd959d2c7005345582888bd32d92e83
+size 307248
diff --git a/mnist_9_mps_feature_200.json b/mnist_9_mps_feature_200.json
new file mode 100644
index 0000000000000000000000000000000000000000..62448d699f88ac6e8a0f0778e724d682bed20345
--- /dev/null
+++ b/mnist_9_mps_feature_200.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 200, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521, 238, 628, 237, 210, 207, 213, 239, 493, 265, 240, 632, 599, 212, 214, 241, 211, 605, 549, 236, 266, 235, 577, 242, 185, 494, 181, 572, 352, 234, 466, 627, 186, 518, 293, 268, 215, 269, 545, 182, 297, 184, 491, 292, 492, 264, 243, 180, 270, 267, 324, 438, 325, 183, 296, 658, 522, 380, 383, 410, 206, 465, 382, 271, 657, 298, 411, 355, 353, 262, 463, 578, 659, 263, 432, 598, 656, 354, 626, 299, 633, 439, 187, 320, 655, 291, 431, 571, 550, 327, 490, 261, 326, 404, 233, 319, 435, 294, 403, 379, 407, 606, 467, 244, 464, 402, 290, 544, 321, 179, 408, 430, 654, 216, 517, 381, 318, 437, 289, 374, 459, 346, 295, 272, 347, 460, 157, 436, 405, 375, 156, 495, 351, 462, 317, 433, 458, 155, 570, 345, 409, 300, 154, 158, 660, 625, 376, 373, 489, 401, 348, 461, 488, 523, 429, 551, 406, 597, 579, 434, 153, 377, 323, 516, 288, 487, 205, 349, 328, 457, 260, 316, 543, 486, 653, 378, 322, 344, 188, 356, 569], "train_min_nll": 37.631473541259766, "test_nll": 42.20743179321289}
\ No newline at end of file
diff --git a/mnist_9_mps_feature_200.safetensors b/mnist_9_mps_feature_200.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..bceafd9b8fa09a8eff73f8fa0cd766c02b774d46
--- /dev/null
+++ b/mnist_9_mps_feature_200.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5b530113385282ec60abdab5ee51b324734900c68486f651746764d922a4f34d
+size 6218528
diff --git a/mnist_9_mps_feature_300.json b/mnist_9_mps_feature_300.json
new file mode 100644
index 0000000000000000000000000000000000000000..9e3a6839db153a1ebb411f0e84ff53a07ed555dc
--- /dev/null
+++ b/mnist_9_mps_feature_300.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 300, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521, 238, 628, 237, 210, 207, 213, 239, 493, 265, 240, 632, 599, 212, 214, 241, 211, 605, 549, 236, 266, 235, 577, 242, 185, 494, 181, 572, 352, 234, 466, 627, 186, 518, 293, 268, 215, 269, 545, 182, 297, 184, 491, 292, 492, 264, 243, 180, 270, 267, 324, 438, 325, 183, 296, 658, 522, 380, 383, 410, 206, 465, 382, 271, 657, 298, 411, 355, 353, 262, 463, 578, 659, 263, 432, 598, 656, 354, 626, 299, 633, 439, 187, 320, 655, 291, 431, 571, 550, 327, 490, 261, 326, 404, 233, 319, 435, 294, 403, 379, 407, 606, 467, 244, 464, 402, 290, 544, 321, 179, 408, 430, 654, 216, 517, 381, 318, 437, 289, 374, 459, 346, 295, 272, 347, 460, 157, 436, 405, 375, 156, 495, 351, 462, 317, 433, 458, 155, 570, 345, 409, 300, 154, 158, 660, 625, 376, 373, 489, 401, 348, 461, 488, 523, 429, 551, 406, 597, 579, 434, 153, 377, 323, 516, 288, 487, 205, 349, 328, 457, 260, 316, 543, 486, 653, 378, 322, 344, 188, 356, 569, 372, 485, 178, 400, 384, 515, 245, 412, 634, 152, 232, 440, 159, 542, 350, 428, 468, 607, 624, 685, 541, 684, 273, 496, 514, 596, 524, 513, 456, 217, 552, 683, 484, 568, 686, 661, 540, 512, 301, 580, 371, 343, 399, 315, 151, 652, 287, 127, 128, 682, 427, 204, 595, 567, 455, 177, 687, 259, 129, 539, 483, 160, 497, 126, 511, 189, 525, 329, 469, 623, 441, 635, 553, 130, 125, 246, 231, 357, 608, 413, 274, 385, 150, 681, 566, 662, 688, 538, 218, 510, 594, 482, 454, 426, 398, 302, 370, 581, 131, 124], "train_min_nll": 44.27325439453125, "test_nll": 50.27799987792969}
\ No newline at end of file
diff --git a/mnist_9_mps_feature_300.safetensors b/mnist_9_mps_feature_300.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..fb3142d75ab93ecb038d27367dc8410d8f550bb6
--- /dev/null
+++ b/mnist_9_mps_feature_300.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4ee0b60331cdf048b87e6005cb7e350f88ea380b0b2a4d1cb3843c214589534f
+size 9502624
diff --git a/mnist_9_mps_feature_400.json b/mnist_9_mps_feature_400.json
new file mode 100644
index 0000000000000000000000000000000000000000..a343dbad21b11d3cd84655738eb723e509e5a55c
--- /dev/null
+++ b/mnist_9_mps_feature_400.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 400, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521, 238, 628, 237, 210, 207, 213, 239, 493, 265, 240, 632, 599, 212, 214, 241, 211, 605, 549, 236, 266, 235, 577, 242, 185, 494, 181, 572, 352, 234, 466, 627, 186, 518, 293, 268, 215, 269, 545, 182, 297, 184, 491, 292, 492, 264, 243, 180, 270, 267, 324, 438, 325, 183, 296, 658, 522, 380, 383, 410, 206, 465, 382, 271, 657, 298, 411, 355, 353, 262, 463, 578, 659, 263, 432, 598, 656, 354, 626, 299, 633, 439, 187, 320, 655, 291, 431, 571, 550, 327, 490, 261, 326, 404, 233, 319, 435, 294, 403, 379, 407, 606, 467, 244, 464, 402, 290, 544, 321, 179, 408, 430, 654, 216, 517, 381, 318, 437, 289, 374, 459, 346, 295, 272, 347, 460, 157, 436, 405, 375, 156, 495, 351, 462, 317, 433, 458, 155, 570, 345, 409, 300, 154, 158, 660, 625, 376, 373, 489, 401, 348, 461, 488, 523, 429, 551, 406, 597, 579, 434, 153, 377, 323, 516, 288, 487, 205, 349, 328, 457, 260, 316, 543, 486, 653, 378, 322, 344, 188, 356, 569, 372, 485, 178, 400, 384, 515, 245, 412, 634, 152, 232, 440, 159, 542, 350, 428, 468, 607, 624, 685, 541, 684, 273, 496, 514, 596, 524, 513, 456, 217, 552, 683, 484, 568, 686, 661, 540, 512, 301, 580, 371, 343, 399, 315, 151, 652, 287, 127, 128, 682, 427, 204, 595, 567, 455, 177, 687, 259, 129, 539, 483, 160, 497, 126, 511, 189, 525, 329, 469, 623, 441, 635, 553, 130, 125, 246, 231, 357, 608, 413, 274, 385, 150, 681, 566, 662, 688, 538, 218, 510, 594, 482, 454, 426, 398, 302, 370, 581, 131, 124, 470, 176, 342, 498, 651, 203, 161, 314, 526, 442, 286, 622, 190, 149, 330, 258, 689, 414, 636, 554, 537, 123, 680, 509, 565, 663, 358, 609, 386, 230, 132, 481, 247, 593, 453, 175, 275, 713, 712, 471, 219, 425, 443, 582, 711, 499, 162, 714, 148, 122, 202, 690, 397, 650, 303, 415, 710, 527, 369, 99, 715, 341, 313, 100, 387, 191, 621, 331, 98, 679, 637, 285, 664, 133, 359, 257, 101, 716, 555, 709, 97, 610, 121, 229, 102, 536, 174, 564, 691, 508, 147, 96, 717, 103, 163, 248, 480, 444, 472, 583], "train_min_nll": 47.04791259765625, "test_nll": 53.65601348876953}
\ No newline at end of file
diff --git a/mnist_9_mps_feature_400.safetensors b/mnist_9_mps_feature_400.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..2a372ccaff87a70ebb8085352a3420abca9f0cb0
--- /dev/null
+++ b/mnist_9_mps_feature_400.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8a0f77354e5524a9d365d72fbe3caaeb64ae97ff0f1c3f73e4ee3039ff8ceff2
+size 12786904
diff --git a/mnist_9_mps_feature_50.json b/mnist_9_mps_feature_50.json
new file mode 100644
index 0000000000000000000000000000000000000000..38d491d45afc7f7b11073ba7e7c39b2ef9f88568
--- /dev/null
+++ b/mnist_9_mps_feature_50.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 50, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521, 238, 628, 237, 210, 207, 213, 239, 493, 265, 240, 632, 599, 212, 214, 241, 211, 605, 549, 236, 266, 235, 577, 242, 185, 494, 181, 572, 352, 234, 466], "train_min_nll": 9.539341926574707, "test_nll": 10.259583473205566}
\ No newline at end of file
diff --git a/mnist_9_mps_feature_50.safetensors b/mnist_9_mps_feature_50.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..01bbb9d4f5920b023a1985bb62d9e476709ad2df
--- /dev/null
+++ b/mnist_9_mps_feature_50.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e61982a42f637df0979e29459d9b5f273f02a506efa6f956b912ff94f9e0ab9a
+size 1292416
diff --git a/mnist_9_mps_feature_500.json b/mnist_9_mps_feature_500.json
new file mode 100644
index 0000000000000000000000000000000000000000..66a96b8da681bd4b6334e75b134f2839ef340731
--- /dev/null
+++ b/mnist_9_mps_feature_500.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 500, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521, 238, 628, 237, 210, 207, 213, 239, 493, 265, 240, 632, 599, 212, 214, 241, 211, 605, 549, 236, 266, 235, 577, 242, 185, 494, 181, 572, 352, 234, 466, 627, 186, 518, 293, 268, 215, 269, 545, 182, 297, 184, 491, 292, 492, 264, 243, 180, 270, 267, 324, 438, 325, 183, 296, 658, 522, 380, 383, 410, 206, 465, 382, 271, 657, 298, 411, 355, 353, 262, 463, 578, 659, 263, 432, 598, 656, 354, 626, 299, 633, 439, 187, 320, 655, 291, 431, 571, 550, 327, 490, 261, 326, 404, 233, 319, 435, 294, 403, 379, 407, 606, 467, 244, 464, 402, 290, 544, 321, 179, 408, 430, 654, 216, 517, 381, 318, 437, 289, 374, 459, 346, 295, 272, 347, 460, 157, 436, 405, 375, 156, 495, 351, 462, 317, 433, 458, 155, 570, 345, 409, 300, 154, 158, 660, 625, 376, 373, 489, 401, 348, 461, 488, 523, 429, 551, 406, 597, 579, 434, 153, 377, 323, 516, 288, 487, 205, 349, 328, 457, 260, 316, 543, 486, 653, 378, 322, 344, 188, 356, 569, 372, 485, 178, 400, 384, 515, 245, 412, 634, 152, 232, 440, 159, 542, 350, 428, 468, 607, 624, 685, 541, 684, 273, 496, 514, 596, 524, 513, 456, 217, 552, 683, 484, 568, 686, 661, 540, 512, 301, 580, 371, 343, 399, 315, 151, 652, 287, 127, 128, 682, 427, 204, 595, 567, 455, 177, 687, 259, 129, 539, 483, 160, 497, 126, 511, 189, 525, 329, 469, 623, 441, 635, 553, 130, 125, 246, 231, 357, 608, 413, 274, 385, 150, 681, 566, 662, 688, 538, 218, 510, 594, 482, 454, 426, 398, 302, 370, 581, 131, 124, 470, 176, 342, 498, 651, 203, 161, 314, 526, 442, 286, 622, 190, 149, 330, 258, 689, 414, 636, 554, 537, 123, 680, 509, 565, 663, 358, 609, 386, 230, 132, 481, 247, 593, 453, 175, 275, 713, 712, 471, 219, 425, 443, 582, 711, 499, 162, 714, 148, 122, 202, 690, 397, 650, 303, 415, 710, 527, 369, 99, 715, 341, 313, 100, 387, 191, 621, 331, 98, 679, 637, 285, 664, 133, 359, 257, 101, 716, 555, 709, 97, 610, 121, 229, 102, 536, 174, 564, 691, 508, 147, 96, 717, 103, 163, 248, 480, 444, 472, 583, 592, 201, 708, 220, 276, 416, 95, 134, 500, 665, 649, 120, 638, 388, 452, 718, 678, 304, 740, 692, 192, 741, 528, 104, 739, 256, 284, 424, 742, 360, 312, 611, 146, 332, 738, 94, 620, 228, 173, 396, 340, 71, 368, 743, 707, 556, 719, 72, 744, 737, 70, 119, 164, 666, 73, 693, 105, 93, 584, 639, 249, 200, 135, 69, 745, 221, 563, 74, 535, 68, 736, 677, 277, 648, 720, 591, 145, 75, 706, 507, 193, 255, 283, 612, 746, 67, 92, 501, 473, 445, 305, 311, 172, 694, 227, 118, 667, 106, 529, 5], "train_min_nll": 47.40895462036133, "test_nll": 53.81501388549805}
\ No newline at end of file
diff --git a/mnist_9_mps_feature_500.safetensors b/mnist_9_mps_feature_500.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..106727073473bd595accd48be133b255f45845e1
--- /dev/null
+++ b/mnist_9_mps_feature_500.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ce09359389fd79ab6f461f6dd46fa1015b3f35e6363766df9601251e5c1b01c8
+size 16071208
diff --git a/mnist_9_mps_feature_600.json b/mnist_9_mps_feature_600.json
new file mode 100644
index 0000000000000000000000000000000000000000..476001dbb0adfbed4cf9fae8501ce455d2f73d29
--- /dev/null
+++ b/mnist_9_mps_feature_600.json
@@ -0,0 +1 @@
+{"hyperparameters": {"feature_num": 600, "physical_dim": 2, "virtual_dim": 64}, "selected_feature_indices": [547, 602, 575, 574, 601, 603, 548, 600, 604, 546, 520, 576, 630, 208, 573, 631, 209, 629, 519, 521, 238, 628, 237, 210, 207, 213, 239, 493, 265, 240, 632, 599, 212, 214, 241, 211, 605, 549, 236, 266, 235, 577, 242, 185, 494, 181, 572, 352, 234, 466, 627, 186, 518, 293, 268, 215, 269, 545, 182, 297, 184, 491, 292, 492, 264, 243, 180, 270, 267, 324, 438, 325, 183, 296, 658, 522, 380, 383, 410, 206, 465, 382, 271, 657, 298, 411, 355, 353, 262, 463, 578, 659, 263, 432, 598, 656, 354, 626, 299, 633, 439, 187, 320, 655, 291, 431, 571, 550, 327, 490, 261, 326, 404, 233, 319, 435, 294, 403, 379, 407, 606, 467, 244, 464, 402, 290, 544, 321, 179, 408, 430, 654, 216, 517, 381, 318, 437, 289, 374, 459, 346, 295, 272, 347, 460, 157, 436, 405, 375, 156, 495, 351, 462, 317, 433, 458, 155, 570, 345, 409, 300, 154, 158, 660, 625, 376, 373, 489, 401, 348, 461, 488, 523, 429, 551, 406, 597, 579, 434, 153, 377, 323, 516, 288, 487, 205, 349, 328, 457, 260, 316, 543, 486, 653, 378, 322, 344, 188, 356, 569, 372, 485, 178, 400, 384, 515, 245, 412, 634, 152, 232, 440, 159, 542, 350, 428, 468, 607, 624, 685, 541, 684, 273, 496, 514, 596, 524, 513, 456, 217, 552, 683, 484, 568, 686, 661, 540, 512, 301, 580, 371, 343, 399, 315, 151, 652, 287, 127, 128, 682, 427, 204, 595, 567, 455, 177, 687, 259, 129, 539, 483, 160, 497, 126, 511, 189, 525, 329, 469, 623, 441, 635, 553, 130, 125, 246, 231, 357, 608, 413, 274, 385, 150, 681, 566, 662, 688, 538, 218, 510, 594, 482, 454, 426, 398, 302, 370, 581, 131, 124, 470, 176, 342, 498, 651, 203, 161, 314, 526, 442, 286, 622, 190, 149, 330, 258, 689, 414, 636, 554, 537, 123, 680, 509, 565, 663, 358, 609, 386, 230, 132, 481, 247, 593, 453, 175, 275, 713, 712, 471, 219, 425, 443, 582, 711, 499, 162, 714, 148, 122, 202, 690, 397, 650, 303, 415, 710, 527, 369, 99, 715, 341, 313, 100, 387, 191, 621, 331, 98, 679, 637, 285, 664, 133, 359, 257, 101, 716, 555, 709, 97, 610, 121, 229, 102, 536, 174, 564, 691, 508, 147, 96, 717, 103, 163, 248, 480, 444, 472, 583, 592, 201, 708, 220, 276, 416, 95, 134, 500, 665, 649, 120, 638, 388, 452, 718, 678, 304, 740, 692, 192, 741, 528, 104, 739, 256, 284, 424, 742, 360, 312, 611, 146, 332, 738, 94, 620, 228, 173, 396, 340, 71, 368, 743, 707, 556, 719, 72, 744, 737, 70, 119, 164, 666, 73, 693, 105, 93, 584, 639, 249, 200, 135, 69, 745, 221, 563, 74, 535, 68, 736, 677, 277, 648, 720, 591, 145, 75, 706, 507, 193, 255, 283, 612, 746, 67, 92, 501, 473, 445, 305, 311, 172, 694, 227, 118, 667, 106, 529, 5, 735, 76, 479, 66, 417, 619, 721, 339, 136, 557, 747, 7, 4, 640, 451, 91, 367, 6, 333, 165, 389, 585, 423, 3, 77, 65, 676, 748, 770, 734, 199, 395, 705, 771, 695, 769, 361, 282, 117, 107, 768, 722, 254, 222, 43, 250, 772, 310, 144, 613, 647, 64, 767, 278, 668, 78, 562, 194, 90, 226, 766, 42, 749, 40, 44, 534, 773, 41, 590, 1, 306, 63, 338, 171, 2, 39, 108, 137, 733, 502, 765, 474, 723, 38, 704, 45, 618, 506, 446, 641, 478, 334, 366, 675, 558, 764, 774, 696, 46, 530], "train_min_nll": 47.605079650878906, "test_nll": 54.15461730957031}
\ No newline at end of file
diff --git a/mnist_9_mps_feature_600.safetensors b/mnist_9_mps_feature_600.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..c375c4a99a30da735304055c3f73282bf1639035
--- /dev/null
+++ b/mnist_9_mps_feature_600.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:21c8e771f0dd2be29b14ee713e4aa7295ae1860257fac6617bda160f2503f2c1
+size 19355504
diff --git a/mnist_experimental_mps.safetensors b/mnist_experimental_mps.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..56c12e02b0833078ebc33194fa9485d2ba4ab343
--- /dev/null
+++ b/mnist_experimental_mps.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:148b12bd7149224b96a430ea97514a565c2bba8c22a3caeb6527c49aa7839110
+size 25529992