torchzero 0.3.13__py3-none-any.whl → 0.3.15__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- tests/test_opts.py +4 -10
- torchzero/core/__init__.py +4 -1
- torchzero/core/chain.py +50 -0
- torchzero/core/functional.py +37 -0
- torchzero/core/modular.py +237 -0
- torchzero/core/module.py +12 -599
- torchzero/core/reformulation.py +3 -1
- torchzero/core/transform.py +7 -5
- torchzero/core/var.py +376 -0
- torchzero/modules/__init__.py +0 -1
- torchzero/modules/adaptive/adahessian.py +2 -2
- torchzero/modules/adaptive/esgd.py +2 -2
- torchzero/modules/adaptive/matrix_momentum.py +1 -1
- torchzero/modules/adaptive/sophia_h.py +2 -2
- torchzero/modules/conjugate_gradient/cg.py +16 -16
- torchzero/modules/experimental/__init__.py +1 -0
- torchzero/modules/experimental/newtonnewton.py +5 -5
- torchzero/modules/experimental/spsa1.py +93 -0
- torchzero/modules/functional.py +7 -0
- torchzero/modules/grad_approximation/__init__.py +1 -1
- torchzero/modules/grad_approximation/forward_gradient.py +2 -5
- torchzero/modules/grad_approximation/rfdm.py +27 -110
- torchzero/modules/line_search/__init__.py +1 -1
- torchzero/modules/line_search/_polyinterp.py +3 -1
- torchzero/modules/line_search/adaptive.py +3 -3
- torchzero/modules/line_search/backtracking.py +1 -1
- torchzero/modules/line_search/interpolation.py +160 -0
- torchzero/modules/line_search/line_search.py +11 -20
- torchzero/modules/line_search/scipy.py +15 -3
- torchzero/modules/line_search/strong_wolfe.py +3 -5
- torchzero/modules/misc/misc.py +2 -2
- torchzero/modules/misc/multistep.py +13 -13
- torchzero/modules/quasi_newton/__init__.py +2 -0
- torchzero/modules/quasi_newton/quasi_newton.py +15 -6
- torchzero/modules/quasi_newton/sg2.py +292 -0
- torchzero/modules/restarts/restars.py +5 -4
- torchzero/modules/second_order/__init__.py +6 -3
- torchzero/modules/second_order/ifn.py +89 -0
- torchzero/modules/second_order/inm.py +105 -0
- torchzero/modules/second_order/newton.py +103 -193
- torchzero/modules/second_order/newton_cg.py +86 -110
- torchzero/modules/second_order/nystrom.py +1 -1
- torchzero/modules/second_order/rsn.py +227 -0
- torchzero/modules/trust_region/levenberg_marquardt.py +2 -2
- torchzero/modules/trust_region/trust_cg.py +6 -4
- torchzero/modules/wrappers/optim_wrapper.py +49 -42
- torchzero/modules/zeroth_order/__init__.py +1 -1
- torchzero/modules/zeroth_order/cd.py +1 -238
- torchzero/utils/derivatives.py +19 -19
- torchzero/utils/linalg/linear_operator.py +50 -2
- torchzero/utils/optimizer.py +2 -2
- torchzero/utils/python_tools.py +1 -0
- {torchzero-0.3.13.dist-info → torchzero-0.3.15.dist-info}/METADATA +1 -1
- {torchzero-0.3.13.dist-info → torchzero-0.3.15.dist-info}/RECORD +57 -48
- torchzero/modules/higher_order/__init__.py +0 -1
- /torchzero/modules/{higher_order → experimental}/higher_order_newton.py +0 -0
- {torchzero-0.3.13.dist-info → torchzero-0.3.15.dist-info}/WHEEL +0 -0
- {torchzero-0.3.13.dist-info → torchzero-0.3.15.dist-info}/top_level.txt +0 -0
|
@@ -1,28 +1,32 @@
|
|
|
1
1
|
tests/test_identical.py,sha256=PJnQtSO3aHZYMQolHmoB26BEUPD_Gpmdh2_M0tfUfm0,11502
|
|
2
2
|
tests/test_module.py,sha256=qX3rjdSJsbA8JO17bPTUIDspe7bg2dogqxMw__KV7SU,2039
|
|
3
|
-
tests/test_opts.py,sha256=
|
|
3
|
+
tests/test_opts.py,sha256=vxNVKPPQP4MMbLm61R8DfLOSuEbSQNsCjIaZGTEMixQ,43770
|
|
4
4
|
tests/test_tensorlist.py,sha256=pWXQE-vEq08EGJSKWgsTgo-7QjjkavOJ5BlWUm241qI,72434
|
|
5
5
|
tests/test_utils_optimizer.py,sha256=bvC0Ehvs2L8fohpyIF5Vfr9OKTycpnODWLPflXilU1c,8414
|
|
6
6
|
tests/test_vars.py,sha256=2BoawNdDAnnNh_vv49_peJMnHvaQjp_sfnca1nosTWY,6766
|
|
7
7
|
torchzero/__init__.py,sha256=aIH-cCTXnDr90cKUPhM8bv-uE69Hzjlf0jlYspYf0ZM,120
|
|
8
|
-
torchzero/core/__init__.py,sha256=
|
|
9
|
-
torchzero/core/
|
|
10
|
-
torchzero/core/
|
|
11
|
-
torchzero/core/
|
|
12
|
-
torchzero/
|
|
13
|
-
torchzero/
|
|
8
|
+
torchzero/core/__init__.py,sha256=dqj1sEvy9vAnO19rQKaYyaIjp2yZjPUFFRvtICD2chM,205
|
|
9
|
+
torchzero/core/chain.py,sha256=yGNYQhTDqulpgiKla3lzhc5yrIpOj7aszXM362xdzDQ,1723
|
|
10
|
+
torchzero/core/functional.py,sha256=Ij29pNW0uy58vZzPlP8foGxd74c7cdEAsHVqykjDuKI,1192
|
|
11
|
+
torchzero/core/modular.py,sha256=ZK5ZzWGuVxMDOlJRTuONVhKwOfJFfmolHb1kqKyAFYU,9902
|
|
12
|
+
torchzero/core/module.py,sha256=C0v4AKwr_1UjaAO893EvxeIdTLnNJ5BNgDTv8MAsEeo,14709
|
|
13
|
+
torchzero/core/reformulation.py,sha256=fc8e01KORAoLeAfjMELG1nbwqY-1qP8RyuzqbUsZuVY,2419
|
|
14
|
+
torchzero/core/transform.py,sha256=NeRCcR_QsU0bJ7LPMJCfn2TKC3lcd9bQ_FxvfPO_vAQ,17131
|
|
15
|
+
torchzero/core/var.py,sha256=RC05ZrmOefioM3bgJZOZJBEY_6PWRivjz566naTnPog,16949
|
|
16
|
+
torchzero/modules/__init__.py,sha256=k79l7dMEmfxvikxfG8iUinsvtkngErSgKhHV3uqzLyc,587
|
|
17
|
+
torchzero/modules/functional.py,sha256=XIwk1j-0cfQFE2B6j5ppgfMkAVFILRBVDR08n5UeKLo,8121
|
|
14
18
|
torchzero/modules/adaptive/__init__.py,sha256=5L2dlEJV6HKBnYhgd7wo2yGi0WPd9qmpw9XS5wOQOq8,944
|
|
15
19
|
torchzero/modules/adaptive/adagrad.py,sha256=0qXC5F4PuOsgLjRXQUWBoiq0AUixsvOP1uDbEeRIcNs,12531
|
|
16
|
-
torchzero/modules/adaptive/adahessian.py,sha256=
|
|
20
|
+
torchzero/modules/adaptive/adahessian.py,sha256=XFSPLEIx5cEzSHrlnDLq1Q4exf27bz8tzwnR6zAFFqE,8680
|
|
17
21
|
torchzero/modules/adaptive/adam.py,sha256=4lWSe__tdyRv0rfkUda1qa_NH36DIW3sd8td98bK6XI,3829
|
|
18
22
|
torchzero/modules/adaptive/adan.py,sha256=Dt_gibyrGtWDIUCaSF6RFIxu2xwiF9fCfrpkoD-CaUM,2825
|
|
19
23
|
torchzero/modules/adaptive/adaptive_heavyball.py,sha256=xQQw1Vx-NgQd_ouK14J1p5ijd5lEn3sAN0hVJAL0j8U,2024
|
|
20
24
|
torchzero/modules/adaptive/aegd.py,sha256=_4ASgDX8__DPnnBE_RncnMqM4rItM7Eji4EZzSGGq5I,1876
|
|
21
|
-
torchzero/modules/adaptive/esgd.py,sha256=
|
|
25
|
+
torchzero/modules/adaptive/esgd.py,sha256=jGNOARWblfeaN64G-VLoN08o2XRQnLiwlKmGpiL6LKA,6426
|
|
22
26
|
torchzero/modules/adaptive/lion.py,sha256=XFyglRNdnP1l8CmEZ7L_ZB8HWiR03BsZ_PEFCvHijb8,1127
|
|
23
27
|
torchzero/modules/adaptive/lmadagrad.py,sha256=rMs7vrgiwOJgWo-OZXkGu32X561edEhhaxwgbY2NTnk,7176
|
|
24
28
|
torchzero/modules/adaptive/mars.py,sha256=iOkyY3r52btp7Cry7WN0AB4arpn4N9b_Hg6S55XC6Q8,2255
|
|
25
|
-
torchzero/modules/adaptive/matrix_momentum.py,sha256=
|
|
29
|
+
torchzero/modules/adaptive/matrix_momentum.py,sha256=5a64tHRuUJrQhzpsiqVGc8Rbf_7kn2jv6Q5ONOoPrHY,6703
|
|
26
30
|
torchzero/modules/adaptive/msam.py,sha256=locqM2jiC3AbGCDCo6T40GF3iaVV2svrkzoi0hD2cJI,6663
|
|
27
31
|
torchzero/modules/adaptive/muon.py,sha256=5Asgj03s6JXXrO-p5Qgn3D8bVwbDEqCq7hxNy4joQDE,10335
|
|
28
32
|
torchzero/modules/adaptive/natural_gradient.py,sha256=5qRehh-iAeZ4hjfOR-gfsObbMsJZnipzCkG-yptkrH0,6349
|
|
@@ -32,48 +36,49 @@ torchzero/modules/adaptive/rprop.py,sha256=VDnLPKxw8ECihyUeNVE8cyDll_Ut3k3_NqoLg
|
|
|
32
36
|
torchzero/modules/adaptive/sam.py,sha256=LnOPNZnIUsis0402RHnA-fTPkNM8baUR9HR50pF_BtM,5696
|
|
33
37
|
torchzero/modules/adaptive/shampoo.py,sha256=r7V4I5_Ve1YVOS3HhO2k5cZvJT1lPHTVApV3iVJVceA,9711
|
|
34
38
|
torchzero/modules/adaptive/soap.py,sha256=roQLBthNNNmTYgeJPi_LxZY-r4m6REeUo0_DZknYU50,10662
|
|
35
|
-
torchzero/modules/adaptive/sophia_h.py,sha256=
|
|
39
|
+
torchzero/modules/adaptive/sophia_h.py,sha256=kdvSTraaTCxydyCLKEXSiIsHUXE2F7bvOb418VImrRk,7189
|
|
36
40
|
torchzero/modules/clipping/__init__.py,sha256=ZaffMF7mIRK6hZSfuZadgjNTX6hF5ANiLBny2w3S7I8,250
|
|
37
41
|
torchzero/modules/clipping/clipping.py,sha256=t98M3QKZKqXJ3_tzXXIiG4EOYMaHqLYrMZ-6zmRuy-k,14331
|
|
38
42
|
torchzero/modules/clipping/ema_clipping.py,sha256=Ki0LPNUwPoE825A5rSE7SxGQMiI3nO3iwnjKQ486iaI,6611
|
|
39
43
|
torchzero/modules/clipping/growth_clipping.py,sha256=mHn6BQqfHFWnfVjYG_Dokb3VjfSK2QVGsy2rf0Z-RMY,6657
|
|
40
44
|
torchzero/modules/conjugate_gradient/__init__.py,sha256=G5WcVoiQYupRBeqjI4lCraGeXNSvWT-_-ynpcE6NQS8,184
|
|
41
|
-
torchzero/modules/conjugate_gradient/cg.py,sha256=
|
|
42
|
-
torchzero/modules/experimental/__init__.py,sha256=
|
|
45
|
+
torchzero/modules/conjugate_gradient/cg.py,sha256=L49p_wBeyQ3pZmaPM1vLx7uWmZBKtIhMf7Uv2ggUbI4,14534
|
|
46
|
+
torchzero/modules/experimental/__init__.py,sha256=SD4DXMZPqqXsrp50xsf5X-N88_2QXc3VODXdxl-6YtM,780
|
|
43
47
|
torchzero/modules/experimental/curveball.py,sha256=JdgojuSYLNe9u3bmqcYrFm8brUD4kvKm9XYx78GzpKI,3257
|
|
44
48
|
torchzero/modules/experimental/dct.py,sha256=CW-Y2gcjlHlxtIx7SekUOfw2EzujA6v0LcjDYGAfh6M,2433
|
|
45
49
|
torchzero/modules/experimental/fft.py,sha256=s95EzvK4-ZJdwZbVhtqwirY9eVy7v6mFDRMgoLY9wjo,3020
|
|
46
50
|
torchzero/modules/experimental/gradmin.py,sha256=hKTOG7tk6FnG8t-7OmTAhGTGSDdONzP1JvCRPRqaKt0,3740
|
|
51
|
+
torchzero/modules/experimental/higher_order_newton.py,sha256=2r1wuhdi57pbo8akQE88O8R-Y79BtiwD1WQIShh1rjQ,12967
|
|
47
52
|
torchzero/modules/experimental/l_infinity.py,sha256=nhYusM3YYbc0ptaSf9zlrsqY8EgxlHm9OejJ6VV0qtM,4750
|
|
48
53
|
torchzero/modules/experimental/momentum.py,sha256=VqZc14EGVO_KUABPLRIBlvHdgg-64o-4heMQH0vW5vY,5233
|
|
49
54
|
torchzero/modules/experimental/newton_solver.py,sha256=0HnDBlrBLvUgS4hWmkJqyw0M7UPFp3kU3SFq2xZVYhQ,5454
|
|
50
|
-
torchzero/modules/experimental/newtonnewton.py,sha256=
|
|
55
|
+
torchzero/modules/experimental/newtonnewton.py,sha256=gUDqx_IWtHuflzPvMu6IUnL6hGyHNJJ7Vj8OUlrUpCY,3816
|
|
51
56
|
torchzero/modules/experimental/reduce_outward_lr.py,sha256=ui_39wNdf5J2FOQtQFk0WUA8DuicwEp0kepccbq8rI0,1309
|
|
52
57
|
torchzero/modules/experimental/scipy_newton_cg.py,sha256=8nKBaHHmqdU9F1kVPn2QVFUTx2_I8Jsfqxix1v-qoL0,4073
|
|
58
|
+
torchzero/modules/experimental/spsa1.py,sha256=fi50Uk68vl9I56_mL3WqefWnbiJ3OBLNMdHQqTYh31s,3589
|
|
53
59
|
torchzero/modules/experimental/structural_projections.py,sha256=rxJFG5F23dOiK_8KqKyvoSMLWqAOXtVGHSwfRqH22Wg,4185
|
|
54
|
-
torchzero/modules/grad_approximation/__init__.py,sha256=
|
|
60
|
+
torchzero/modules/grad_approximation/__init__.py,sha256=_mQ2sWvnMfqc3RQcVmZuBlphtLZCO7z819abGY6kYuM,196
|
|
55
61
|
torchzero/modules/grad_approximation/fdm.py,sha256=zx70GZDQmhe43bZP5Mbbl31xsMOsGO43kznoQDbqxJo,4372
|
|
56
|
-
torchzero/modules/grad_approximation/forward_gradient.py,sha256=
|
|
62
|
+
torchzero/modules/grad_approximation/forward_gradient.py,sha256=pCOsvt4ZZtzsIlAeXLmpS8vNXYewA_Gh7uyz1_1yROs,4011
|
|
57
63
|
torchzero/modules/grad_approximation/grad_approximator.py,sha256=x8vlweBrfJ6SmhMHvI_C8UZGzlS3AnmlulvqnSzm6iY,4437
|
|
58
|
-
torchzero/modules/grad_approximation/rfdm.py,sha256=
|
|
59
|
-
torchzero/modules/higher_order/__init__.py,sha256=iaoIrmR9DJE9QHt9PeZNCWqIYDe-86h1IjkaumR4qF0,51
|
|
60
|
-
torchzero/modules/higher_order/higher_order_newton.py,sha256=2r1wuhdi57pbo8akQE88O8R-Y79BtiwD1WQIShh1rjQ,12967
|
|
64
|
+
torchzero/modules/grad_approximation/rfdm.py,sha256=zgzZYJDAq6HsOMeHePUWlnnJWYRpqey77nHMaHh8140,19217
|
|
61
65
|
torchzero/modules/least_squares/__init__.py,sha256=mJwE2IXVB3mn_7BzsmDNKhfyViCV8GOrqHJJjz04HR4,41
|
|
62
66
|
torchzero/modules/least_squares/gn.py,sha256=23AB6AWAl5IuBj4Vd3boQ6ndk0pO3ovaF9EiY1a1XWs,5094
|
|
63
|
-
torchzero/modules/line_search/__init__.py,sha256=
|
|
64
|
-
torchzero/modules/line_search/_polyinterp.py,sha256=
|
|
65
|
-
torchzero/modules/line_search/adaptive.py,sha256=
|
|
66
|
-
torchzero/modules/line_search/backtracking.py,sha256=
|
|
67
|
-
torchzero/modules/line_search/
|
|
68
|
-
torchzero/modules/line_search/
|
|
69
|
-
torchzero/modules/line_search/
|
|
67
|
+
torchzero/modules/line_search/__init__.py,sha256=_QjxUJmNC8OqtUuyTJp9wDfHNFKZBZqj6lttWKhG-cI,217
|
|
68
|
+
torchzero/modules/line_search/_polyinterp.py,sha256=i3sNl6SFAUJi4oxhhjBlcxJY9KRunIZjJ8sGdaJOVjc,10990
|
|
69
|
+
torchzero/modules/line_search/adaptive.py,sha256=YNabP6-01dhAUDAOuHRPZCwiV5xTRdHmkN667HQ6V3w,3798
|
|
70
|
+
torchzero/modules/line_search/backtracking.py,sha256=oJeYdNVRodvDXhYTDtOle1VC7X_oJ33OwQ0IJ6log1s,9044
|
|
71
|
+
torchzero/modules/line_search/interpolation.py,sha256=tHXlZD1MgfYaymhXY75k9CocltwSYDYZg6ENDCEUiss,4942
|
|
72
|
+
torchzero/modules/line_search/line_search.py,sha256=F7Wax39JILOwm66gQldoTrHb1MOlB8d-soeqLa7c-jY,12763
|
|
73
|
+
torchzero/modules/line_search/scipy.py,sha256=xQ80h9cSyF4Iorq_1NoJglu_Bx4_KeojulBIxvwU6gQ,2836
|
|
74
|
+
torchzero/modules/line_search/strong_wolfe.py,sha256=8Li0VF2GH3qu_VdWpcQPCv3cy_DsIXUX4O-OosbZX2g,14973
|
|
70
75
|
torchzero/modules/misc/__init__.py,sha256=UYY9CeNepnC8H1LnFa829ux5MEjtGZ9zql624IbCFX8,825
|
|
71
76
|
torchzero/modules/misc/debug.py,sha256=6pFAGYANjCPGIZH_4ghpUYYTEsT5jr7PMB9KLuPP4p8,1532
|
|
72
77
|
torchzero/modules/misc/escape.py,sha256=qfEdKLD5rejqrmvyHrI5BRQq8js9UF2-Axs_C0KFyWA,1866
|
|
73
78
|
torchzero/modules/misc/gradient_accumulation.py,sha256=mBWa5CBCZwp4TrtOyjWI3VnHag4gum4WBM2WFhvHqW4,4891
|
|
74
79
|
torchzero/modules/misc/homotopy.py,sha256=hihLETE4dNZ27zatqPR_qT3kGX-AXbC7oBWRDbFQo58,1939
|
|
75
|
-
torchzero/modules/misc/misc.py,sha256=
|
|
76
|
-
torchzero/modules/misc/multistep.py,sha256=
|
|
80
|
+
torchzero/modules/misc/misc.py,sha256=AYeLYJWiZ4D3LwaRUFweMGhs0o4-MYfdTDcV0sRlqpo,15421
|
|
81
|
+
torchzero/modules/misc/multistep.py,sha256=Izgicn1L732ED54qfg6LD0ySWBgxBG-JR5Xu2jDLY1M,6288
|
|
77
82
|
torchzero/modules/misc/regularization.py,sha256=SkQ0_Ybtv9IEGI9QGdvNZaja5bAyc1x-j_1gvYIVepI,6105
|
|
78
83
|
torchzero/modules/misc/split.py,sha256=JcXVB4xk3h55YT2OAdepVsRoE1PD7bqX6NmJ2IxBgAI,4013
|
|
79
84
|
torchzero/modules/misc/switch.py,sha256=p758heAnv-PkoslpafL35Yp7mlvPmDVSe1mWiuuD8Mk,3711
|
|
@@ -93,19 +98,23 @@ torchzero/modules/projections/__init__.py,sha256=4LfmBEu_eM4YWmcWQVH4CdI1H0ucCIH
|
|
|
93
98
|
torchzero/modules/projections/cast.py,sha256=FJx2Tt1lbQRnOC5wxx3LbOnacLfUluFP6QOXLUCIEPY,2174
|
|
94
99
|
torchzero/modules/projections/galore.py,sha256=GDJ7hf6cdk_Iu2qW0rWaQwYLQAxQEe27FEfOiZvFXHo,252
|
|
95
100
|
torchzero/modules/projections/projection.py,sha256=itkkb2UmMqbdtWKjUUg6gbFJfCEIZAskC0HCvom-6sc,14084
|
|
96
|
-
torchzero/modules/quasi_newton/__init__.py,sha256=
|
|
101
|
+
torchzero/modules/quasi_newton/__init__.py,sha256=scn1qRpS1dtUv0u4tQPbjPy9Db66KrxyjS8YNDiKDYQ,543
|
|
97
102
|
torchzero/modules/quasi_newton/damping.py,sha256=K1DVqqKiAs6-F3JQh5jlKNb79oJdObqnKWwHHRl6boQ,2813
|
|
98
103
|
torchzero/modules/quasi_newton/diagonal_quasi_newton.py,sha256=Zx-tlFRa89GhoSP7RFJdLQJPiqPCL7rWaV7WJoQ1YCs,6930
|
|
99
104
|
torchzero/modules/quasi_newton/lbfgs.py,sha256=fzCjV5YsLo_uJTVG3vosPHsvDc97mLKueK6fxOHLb8I,11195
|
|
100
105
|
torchzero/modules/quasi_newton/lsr1.py,sha256=D3_yV5xtgklMlU4fAL1-sH82-1tNl3K2F12ZBZyLQGM,8512
|
|
101
|
-
torchzero/modules/quasi_newton/quasi_newton.py,sha256
|
|
106
|
+
torchzero/modules/quasi_newton/quasi_newton.py,sha256=DP2mxjWdfgpiQ3KiZBcW85iXpXbfovubvFhzHTB2phM,45473
|
|
107
|
+
torchzero/modules/quasi_newton/sg2.py,sha256=bSMc1Bf0FDiCJKqKjxKianbLcpfNkUwW9mUMwC9ZLbg,7711
|
|
102
108
|
torchzero/modules/restarts/__init__.py,sha256=7282ePwN_I0vSeLPYS4TTclE9ZU7pL6UpyRp5ydgdSg,134
|
|
103
|
-
torchzero/modules/restarts/restars.py,sha256=
|
|
104
|
-
torchzero/modules/second_order/__init__.py,sha256=
|
|
109
|
+
torchzero/modules/restarts/restars.py,sha256=ZN8kzufkOKredI54HJ9cSANAYY2yN4gAggqAVtEBCyA,9084
|
|
110
|
+
torchzero/modules/second_order/__init__.py,sha256=hnlxH2LkejWRdwkLxQDF9RorgN1pakaeOVp2AmM11fE,291
|
|
111
|
+
torchzero/modules/second_order/ifn.py,sha256=k7-5sZUdPTJwqiOxM_ImmLkoswp6ooVnwgHGz4iEHTY,3352
|
|
112
|
+
torchzero/modules/second_order/inm.py,sha256=BtR7LXkXVUHYHP2RgTNi68h9T5qYXKLZpptNC1yMF1I,3478
|
|
105
113
|
torchzero/modules/second_order/multipoint.py,sha256=Ilzo0Ddd3iApegceu7cHSMGim9ZH5QS4-2uBtrKXC6k,8581
|
|
106
|
-
torchzero/modules/second_order/newton.py,sha256=
|
|
107
|
-
torchzero/modules/second_order/newton_cg.py,sha256=
|
|
108
|
-
torchzero/modules/second_order/nystrom.py,sha256=
|
|
114
|
+
torchzero/modules/second_order/newton.py,sha256=MPZ3pOxmXLQ0uS_h9JaxOsRTYvE3j2Ai--K8neUnZX8,11778
|
|
115
|
+
torchzero/modules/second_order/newton_cg.py,sha256=6KmYr-U5gmuTGoVZA8Gt2-rfLwnjNakjKkBKEuBffTk,16738
|
|
116
|
+
torchzero/modules/second_order/nystrom.py,sha256=jtuzIIzKebRu06twvUsuC828KNKeB3DRQR9WjJKp2C8,11352
|
|
117
|
+
torchzero/modules/second_order/rsn.py,sha256=rHfBCmlFjE7MIaU0vO8qV4nHhigQBUX89w_dZyWzrX0,10118
|
|
109
118
|
torchzero/modules/smoothing/__init__.py,sha256=RYxCLLfG2onBbMUToaoedsr20rXaayyBt7Ov8OxULrU,80
|
|
110
119
|
torchzero/modules/smoothing/laplacian.py,sha256=05Y6ft0GHRGkfSDV-g8vlaTlZTiXMr79xDagJb126ug,5302
|
|
111
120
|
torchzero/modules/smoothing/sampling.py,sha256=zI5bATytQmCqm_UgAQbfA9tNRgrZaKLfUb0B-kzKRHU,12867
|
|
@@ -117,17 +126,17 @@ torchzero/modules/termination/termination.py,sha256=BXU3R04caBc8rFJ4v_yJjgGi1X4i
|
|
|
117
126
|
torchzero/modules/trust_region/__init__.py,sha256=kWke9FB41-EpjdXCPk8VBwZhpgYalOWSKDI1XWe0yYg,204
|
|
118
127
|
torchzero/modules/trust_region/cubic_regularization.py,sha256=gbKOR5zBo3t9i-sW23DCtTQwZrBubuFy_VuafrLaeUw,6718
|
|
119
128
|
torchzero/modules/trust_region/dogleg.py,sha256=zwFR49gghxztVGEETF2D4AkeGgHkQRbHGGelav3GuFg,3619
|
|
120
|
-
torchzero/modules/trust_region/levenberg_marquardt.py,sha256=
|
|
121
|
-
torchzero/modules/trust_region/trust_cg.py,sha256=
|
|
129
|
+
torchzero/modules/trust_region/levenberg_marquardt.py,sha256=s4XBXK8LwHuwyanOJFtgmOLkxEiBMhBVzch0J2_dIFk,5055
|
|
130
|
+
torchzero/modules/trust_region/trust_cg.py,sha256=F6G0hTXv6Ry0swO_4fx3ecxYWxMr72nrmwJPbFDpqH4,4459
|
|
122
131
|
torchzero/modules/trust_region/trust_region.py,sha256=eimCFViJSzoubrRmDluCon6mfcyT7PQA0yRPu4FlO2Q,12872
|
|
123
132
|
torchzero/modules/variance_reduction/__init__.py,sha256=3pwPWZpjgz1btfLJ3rEaK7Wl8B1pDh0HIf0kvD_NJH8,22
|
|
124
133
|
torchzero/modules/variance_reduction/svrg.py,sha256=9pBjPY4EMkGyfj68gXqPi1GJIolUVl5zyNtlZInCKKo,8635
|
|
125
134
|
torchzero/modules/weight_decay/__init__.py,sha256=7UHAiiimsbQ_dHlxxcW87G5cCQFom9Uh_733W_23PWU,93
|
|
126
135
|
torchzero/modules/weight_decay/weight_decay.py,sha256=Y7kE_j0GRF8ceJ9SS6qykQ8a23X2OTDCjJ9VklOQSEw,5415
|
|
127
136
|
torchzero/modules/wrappers/__init__.py,sha256=6b5Ac-8u18IVp_Jnw1T1xQExwpQhpQ0JwNV9GyC_Yj8,31
|
|
128
|
-
torchzero/modules/wrappers/optim_wrapper.py,sha256=
|
|
129
|
-
torchzero/modules/zeroth_order/__init__.py,sha256=
|
|
130
|
-
torchzero/modules/zeroth_order/cd.py,sha256=
|
|
137
|
+
torchzero/modules/wrappers/optim_wrapper.py,sha256=6sLqAPt2xfnx81V0iiCWV5eeclF9qOnpE7kFYXZnY9I,4617
|
|
138
|
+
torchzero/modules/zeroth_order/__init__.py,sha256=1ADUiOHVHzvIP4TpH7_ILmeW2heidfikbf6d5g_1RzY,18
|
|
139
|
+
torchzero/modules/zeroth_order/cd.py,sha256=SwjwoAqX86-JnVHIwKAE7g_tqm0EvEUUNuLM4T5mKXE,4876
|
|
131
140
|
torchzero/optim/__init__.py,sha256=aXf7EkywqYiR50I4QeeVXro9aBhKiqfbY_BCia59sgU,46
|
|
132
141
|
torchzero/optim/root.py,sha256=gGtAJ9qBoSNV58EKzUGZ8J3lyKGUF8BEw34Zfprppdo,2273
|
|
133
142
|
torchzero/optim/utility/__init__.py,sha256=pUacok4XmebfxofE-QWZLgViajsU-3JkXcWi9OS-Jrw,24
|
|
@@ -142,25 +151,25 @@ torchzero/optim/wrappers/optuna.py,sha256=pIXkC5NVmEnUQ4jsGaz6Gv9uYOZM9rnxME4UGk
|
|
|
142
151
|
torchzero/optim/wrappers/scipy.py,sha256=A4yeQRdB9f65UrJ2g80NfqqMc6zUyr9js40TUESCHPg,21535
|
|
143
152
|
torchzero/utils/__init__.py,sha256=7S4VRTkfS-0uI8HOR0EFIjiEcKrmYK7LEhTocIgki6c,1112
|
|
144
153
|
torchzero/utils/compile.py,sha256=Dozox91tcShUJ3L320TTbJrcuA-l4WVegLAQujRqy94,5132
|
|
145
|
-
torchzero/utils/derivatives.py,sha256=
|
|
154
|
+
torchzero/utils/derivatives.py,sha256=pwE7BYO6utXraLwXAelw52JHrsDXLp4rbWwZdxr9Pnk,17230
|
|
146
155
|
torchzero/utils/metrics.py,sha256=XPpOvY257tb4mN3Sje1AVNlQkOXiW24_lXXdtd0JYok,3130
|
|
147
156
|
torchzero/utils/numberlist.py,sha256=iMoqz4IzXy-aE9bqVYJ21GV6pl0z-NeTsXR-LaI8C24,6229
|
|
148
157
|
torchzero/utils/ops.py,sha256=n4Su1sbgTzlHczuPEHkuWenTtNBCa_MvlQ_hCZkIPnQ,314
|
|
149
|
-
torchzero/utils/optimizer.py,sha256=
|
|
158
|
+
torchzero/utils/optimizer.py,sha256=pzGIddJtsR_oH3mb3GOsWHi02Sb62ms4NCaWUOk0Clo,12470
|
|
150
159
|
torchzero/utils/optuna_tools.py,sha256=F-1Xg0n_29MVEb6lqgUFFNIl9BNJ6MOdIJPduoNH4JU,1325
|
|
151
160
|
torchzero/utils/params.py,sha256=nQo270aOURU7rJ_D102y2pSXbzhJPK0Z_ehx4mZBMes,5784
|
|
152
|
-
torchzero/utils/python_tools.py,sha256=
|
|
161
|
+
torchzero/utils/python_tools.py,sha256=QxLZ2PKgp4R2zI_C1qNOX_u4eIcuT0wNBpBM5YEIYuU,3428
|
|
153
162
|
torchzero/utils/tensorlist.py,sha256=nIWBME3fUQPsr4buvtV3LaJgSXPEG_Xb58KAzfjwK-k,56064
|
|
154
163
|
torchzero/utils/torch_tools.py,sha256=DsHaSRGZ3-IuySZJTrkojTbaMMlttJFe0hFvB2xnl2U,5069
|
|
155
164
|
torchzero/utils/linalg/__init__.py,sha256=cNoTJOPeqbNn9l7_HAAen2rlehGS3DyY5SveInG3Stc,328
|
|
156
165
|
torchzero/utils/linalg/benchmark.py,sha256=wiIMn-GY2xxWbHVf8CPbJddUPeUPq9OUDkvbp1iILYI,479
|
|
157
|
-
torchzero/utils/linalg/linear_operator.py,sha256=
|
|
166
|
+
torchzero/utils/linalg/linear_operator.py,sha256=KpI-qkDPtdeLhvr5k1DitAyk55CJszVvSTObTPhOuhw,14611
|
|
158
167
|
torchzero/utils/linalg/matrix_funcs.py,sha256=BKQK_oIG35R6yGxU80eBG0VkyY2EgxywqbhvU7JhWm4,3109
|
|
159
168
|
torchzero/utils/linalg/orthogonalize.py,sha256=BpuDiAPrsJMUpTNBMCntBNA8-O2nozLxY5ZbCfRlEFY,444
|
|
160
169
|
torchzero/utils/linalg/qr.py,sha256=5tbPEV9I6X69r5ACWF9XeqjZTUtUql2145uoGjlJNDs,2517
|
|
161
170
|
torchzero/utils/linalg/solve.py,sha256=R5lPTzHn2sgvRy4MRp-Ngl0sypSGLRLHJjf1oKKAJD0,14395
|
|
162
171
|
torchzero/utils/linalg/svd.py,sha256=wBxl-JSciINV-N6zvM4SGdveqMr6idq51h68LyQQRYg,660
|
|
163
|
-
torchzero-0.3.
|
|
164
|
-
torchzero-0.3.
|
|
165
|
-
torchzero-0.3.
|
|
166
|
-
torchzero-0.3.
|
|
172
|
+
torchzero-0.3.15.dist-info/METADATA,sha256=LOdUII1vQJzYSiX1ICG0cwp5Tn9TAdXNDB1uIoD_alI,565
|
|
173
|
+
torchzero-0.3.15.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
|
|
174
|
+
torchzero-0.3.15.dist-info/top_level.txt,sha256=ETW_iE2ubg0oMyef_h-ayB5i1OOZZd4SNdR3ltIbHe0,16
|
|
175
|
+
torchzero-0.3.15.dist-info/RECORD,,
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
from .higher_order_newton import HigherOrderNewton
|
|
File without changes
|
|
File without changes
|
|
File without changes
|