TorchOpt

View on PyPIReverse Dependencies (4)

0.7.3 torchopt-0.7.3-py3-none-any.whl
torchopt-0.7.3-cp39-cp39-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
torchopt-0.7.3-cp38-cp38-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
torchopt-0.7.3-cp311-cp311-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
torchopt-0.7.3-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl

Wheel Details

Project: TorchOpt
Version: 0.7.3
Filename: torchopt-0.7.3-py3-none-any.whl
Download: [link]
Size: 168166
MD5: af0fccef2380a706dccd8922e3c51ef7
SHA256: ff84581d3c335d83996e6541650170bd22b3fe7eca2215593720df5902dcdc75
Uploaded: 2023-11-10 06:32:37 +0000

dist-info

METADATA

Metadata-Version: 2.1
Name: torchopt
Version: 0.7.3
Summary: An efficient library for differentiable optimization for PyTorch.
Author: TorchOpt Contributors
Author-Email: Jie Ren <jieren9806[at]gmail.com>, Xidong Feng <xidong.feng.20[at]ucl.ac.uk>, Bo Liu <benjaminliu.eecs[at]gmail.com>, Xuehai Pan <XuehaiPan[at]pku.edu.cn>
Project-Url: Homepage, https://github.com/metaopt/torchopt
Project-Url: Repository, https://github.com/metaopt/torchopt
Project-Url: Documentation, https://torchopt.readthedocs.io
Project-Url: Bug Report, https://github.com/metaopt/torchopt/issues
License: Apache License, Version 2.0
Keywords: PyTorch,functorch,JAX,Meta-Learning,Optimizer,Differentiable Optimizer,Functional Programming
Classifier: Development Status :: 4 - Beta
Classifier: License :: OSI Approved :: Apache Software License
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3.8
Classifier: Programming Language :: Python :: 3.9
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: 3.11
Classifier: Operating System :: Microsoft :: Windows
Classifier: Operating System :: POSIX :: Linux
Classifier: Operating System :: MacOS
Classifier: Environment :: GPU
Classifier: Environment :: GPU :: NVIDIA CUDA
Classifier: Intended Audience :: Developers
Classifier: Intended Audience :: Education
Classifier: Intended Audience :: Science/Research
Classifier: Topic :: Scientific/Engineering
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Requires-Python: >=3.8
Requires-Dist: torch (>=1.13)
Requires-Dist: optree (>=0.4.1)
Requires-Dist: numpy
Requires-Dist: graphviz
Requires-Dist: typing-extensions (>=4.0.0)
Requires-Dist: isort; extra == "lint"
Requires-Dist: black[jupyter] (>=22.6.0); extra == "lint"
Requires-Dist: pylint[spelling] (>=2.15.0); extra == "lint"
Requires-Dist: mypy (>=0.990); extra == "lint"
Requires-Dist: flake8; extra == "lint"
Requires-Dist: flake8-bugbear; extra == "lint"
Requires-Dist: flake8-comprehensions; extra == "lint"
Requires-Dist: flake8-docstrings; extra == "lint"
Requires-Dist: flake8-pyi; extra == "lint"
Requires-Dist: flake8-simplify; extra == "lint"
Requires-Dist: ruff; extra == "lint"
Requires-Dist: doc8; extra == "lint"
Requires-Dist: pydocstyle[toml]; extra == "lint"
Requires-Dist: pyenchant; extra == "lint"
Requires-Dist: cpplint; extra == "lint"
Requires-Dist: pre-commit; extra == "lint"
Requires-Dist: pytest; extra == "test"
Requires-Dist: pytest-cov; extra == "test"
Requires-Dist: pytest-xdist; extra == "test"
Requires-Dist: jax[cpu] (>=0.3); platform_system != "Windows" and extra == "test"
Requires-Dist: jaxopt; platform_system != "Windows" and extra == "test"
Requires-Dist: optax; platform_system != "Windows" and extra == "test"
Provides-Extra: lint
Provides-Extra: test
Description-Content-Type: text/markdown
License-File: LICENSE
[Description omitted; length: 25851 characters]

WHEEL

Wheel-Version: 1.0
Generator: bdist_wheel (0.41.3)
Root-Is-Purelib: true
Tag: py3-none-any

RECORD

Path Digest Size
torchopt/__init__.py sha256=R30wYsAkM_JS5z-KnILBYla98TcPtQ0b3_-qFZWn_2s 2821
torchopt/base.py sha256=Aq-UiWbyEuKMcHnPM0VYxLBaJPsQa-s7v1Vbp7doTsI 10785
torchopt/clip.py sha256=RIKhlx30LpFEnFR782jcx-gMgh3tl1mJMIHt_bhuexk 4068
torchopt/combine.py sha256=3u9AtpNg0LC16S0GGP81il40u4CTDMWjpBKmkhp0UW0 4064
torchopt/hook.py sha256=xnVoMt-dkWJYWO1a7XvQjRVAV47C1vukKU4udnRp-4k 2497
torchopt/py.typed sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
torchopt/pytree.py sha256=MDTStlhLK0byfXvdsiqPhBnjrcpgAdwsIwUqKPparGI 6389
torchopt/typing.py sha256=4lT0QciInY7vNrYn1rohmjpcBYNgqz7vnJNL3X4w91s 4346
torchopt/update.py sha256=ZoeFQ628OYWZ3fMprRmTsiFTuN1H1cutBYeVLqzDp6g 3188
torchopt/utils.py sha256=lc3LiHGenGnb-qwbfTjct1tmGItziHr2p-rM-CpCBfo 18438
torchopt/version.py sha256=Y3mwNP_IQ6RHBMWal8URduoPEkf1bEKetlOe8HMcDBY 1815
torchopt/visual.py sha256=2BBE49DM764KKiKXahKYvVdLb0msbaWdUoMLBVX9O3w 9511
torchopt/_C/adam_op.pyi sha256=05IKd6n0Xs0IISXwQv7BCkRDudqPs8XaPQWfzmW3_Mw 1839
torchopt/accelerated_op/__init__.py sha256=mdgU_fpe5YSQKSsVX6WfVbzk2KkHSYtEfBl9B3FQ1tA 1669
torchopt/accelerated_op/adam_op.py sha256=FQ0W4DY9UpNzVGpiF64xePTqsnLjBdG5Uz0Plby9EXg 6640
torchopt/accelerated_op/_src/__init__.py sha256=T1RVg1_nA5-cZLT4DfQZ3SUWQdcTZyRiZrr-KJE4sPU 731
torchopt/accelerated_op/_src/adam_op.py sha256=hlkRvtjeHkd3Sia8QRvxPrApu6TUIOloM8b9IyM8Lyo 3421
torchopt/alias/__init__.py sha256=hU940gVBko5kjDJG7sBPjZZliGIdYbYkpfVJsIguuzs 2051
torchopt/alias/adadelta.py sha256=ksqXkX6Qi8R-n_bGqf8MyUzA-c-n17VBq8VTroq4L5U 3932
torchopt/alias/adagrad.py sha256=ubYooIPxaHKYZnzNOI8WgNoyIF_f1yZZUc1yenmUDd0 6782
torchopt/alias/adam.py sha256=LuxhMfImNgIAF-oIZQETlt9TvzwbgivKSccud664iZk 5955
torchopt/alias/adamax.py sha256=CjGIMjV4PZtecRi50Vq9VuNmkGDJDdO80TXFwhQ7XAw 4048
torchopt/alias/adamw.py sha256=ZZ4AjkAEztlFE61tP8jTw0pc41JZlfog9e_p-t7pg2Y 7242
torchopt/alias/radam.py sha256=aoNFpnIdnxHtyvijfPCDYilxIYneEqrStsL53189kgo 4105
torchopt/alias/rmsprop.py sha256=zmje7cRymrV6xyAGufoyrdfjxmdP3RMDCUwCuPhsk1Q 6165
torchopt/alias/sgd.py sha256=3AvPjj-NrH9S6Yslo0vmef70BqwSyzrukr8kMiPV7UU 5203
torchopt/alias/utils.py sha256=53cRijIYhW8XERj0GhYowIvU2sM4NVrBnxtjNTUfxJU 7421
torchopt/diff/__init__.py sha256=NszsfaRG6pl179H_kJLyxpUkY8BcaN5yGEfVVV64dhk 897
torchopt/diff/implicit/__init__.py sha256=XfIw3NYy8hH60MLWS7Jb7g77wtEgoRy8Dd6SbPCRh88 928
torchopt/diff/implicit/decorator.py sha256=JvOgOiTqumRuGiG_12GbKLEfKULUawmHYM1RvkT7xP0 19663
torchopt/diff/implicit/nn/__init__.py sha256=-ZfxJCtrIFyCrMT5UCjfpwumo9VstrzMgu-yOyDpH40 953
torchopt/diff/implicit/nn/module.py sha256=SU5qT6Oc_xHST0Olp0geMVySMT-4licGOHBeOupBZ00 11846
torchopt/diff/zero_order/__init__.py sha256=tXOLA-xbFpcTJWs6cyo1ceAInTKl9XT18PMhO0WmqtQ 1526
torchopt/diff/zero_order/decorator.py sha256=y5Xba8MbCaarf3KnOn045zw_q-VKnp_AD6oecoWTk20 17682
torchopt/diff/zero_order/nn/__init__.py sha256=s15hB-3MHAwwdBW1qqWTYMG10CLSQt6LukHlL_KWKuk 931
torchopt/diff/zero_order/nn/module.py sha256=PqatVqXyz74wUFUadbLTgJhQiCofD18Whw_Q9DMW5qY 3722
torchopt/distributed/__init__.py sha256=OrrrN8G1KuuBPzTe6Qdt1LcXtTFwC7Q9MewdA2Fm6Ag 1173
torchopt/distributed/api.py sha256=v7XpX6JUsq3ECMHUaYGelQA0j2DlQbZ7oiML5bKT8nk 18766
torchopt/distributed/autograd.py sha256=VyC5mDpKDutdzTYZgz6_dpkCwSmMkHKmx2nouA0lTIw 5821
torchopt/distributed/world.py sha256=jbwBwNqYZJsR7lliXMfb2eTz9QZNx2kzZOxd4zobRwQ 7513
torchopt/linalg/__init__.py sha256=IVD-WfQxCCrXr0HW7Zs4sI-rEqAb7NW4eXJrqVjmIOg 1664
torchopt/linalg/cg.py sha256=LuBAlEPPzMRyzE18YHMtjC06fpbllOaaZixGPX_j5vw 7699
torchopt/linalg/ns.py sha256=bAGAifLBbI-cxUcxd1gKYU4lAUcmr-9yjTFjtgkIR0k 6318
torchopt/linalg/utils.py sha256=86uX7S9mBWvmx1NcOraKH7Qe5oVBnXPmRGTRksxbB_k 2105
torchopt/linear_solve/__init__.py sha256=Xep3Txn4zsU9Btw8xg1tOLet_Y_dbEhqiYOtTXlrcvU 1761
torchopt/linear_solve/cg.py sha256=_8jWrXr3gM7pqbeQmslsvBkz2ACJK4wWS-fH1LGsTTo 4672
torchopt/linear_solve/inv.py sha256=Mu1hSJtmNw6hHNoosaFha56b__kpuauDyNTQFNulmng 5224
torchopt/linear_solve/normal_cg.py sha256=33hYxOXpVBRQNLhhw09cb2f05tUB0DNev8RQA-_Sey4 5209
torchopt/linear_solve/utils.py sha256=MniNCELzNI0dEtXypTQWtamA3DOROeG7S3WzNhPJlV4 4332
torchopt/nn/__init__.py sha256=VnaI--SvLwvf5xZQnRgtboLBbxtC2w_Zs_6Goxm9NvQ 1254
torchopt/nn/module.py sha256=Wt1bGSWELmTCv8KA-4jcKz5KqQyLlh5HqIJd0galaBw 20378
torchopt/nn/stateless.py sha256=5175Fy3C_qivZKvKPNbaOYtz9prNBqfiYlLvwcO6c7g 3401
torchopt/optim/__init__.py sha256=86JrAqqt7U0bhUwhYswI9gegRvqXDmmv9x_5OGyyp8o 1212
torchopt/optim/adadelta.py sha256=e1rMYzPqurQaSMHu-dz9_Tl4PSgNcGiYgRfXxZK2ggk 2620
torchopt/optim/adagrad.py sha256=h_B_HEa6OrCXgD9MJIqLq6xjos_vTLDELvpbkRBUXsg 3006
torchopt/optim/adam.py sha256=0CZPPoVL8E7GsLY8GrwU_UAb_QlmmC0eaNrOib_rmx8 3641
torchopt/optim/adamax.py sha256=7G-qnwLx7FZaYtoWjxL6l1JNSWAcmcuvwoc3_6jH650 2641
torchopt/optim/adamw.py sha256=ZoEJlevjL4aUR86Gx-QfQIUljhBBGdxFvFK-qc9j9Ks 4626
torchopt/optim/base.py sha256=Z73vaU-gUNwACn7a1DQ43KLbSt-3Cj1avy7XQK3A-FU 5188
torchopt/optim/radam.py sha256=ZZA_HlqunXqT2dLQg2QhAYs4YPM-I4NlhfHI-YfsLcI 2566
torchopt/optim/rmsprop.py sha256=8kPUDHJ86YhaMT3Atmfqr-Iw7IQSXxjUwjtU-_Q7qr0 3755
torchopt/optim/sgd.py sha256=92sVNBWrxCANe310TLVQ_25Wc-V5eocvCwSd28-CHJQ 2981
torchopt/optim/func/__init__.py sha256=83z6Ijverld_EF18JYZxKgHXS7oqN69qcxDdekxQFRs 768
torchopt/optim/func/base.py sha256=sRIUQZxU2hqvXmV0UglP6dwZrReypSHK7t_Ht174FMg 4705
torchopt/optim/meta/__init__.py sha256=xlv_6aGIsas091b05TiviNCSi7GKAvBI2IXdFEQGenM 1220
torchopt/optim/meta/adadelta.py sha256=g8pXi3lnu4_D1jLnmtGTsIuTt1EZTVBgcuBHJPvdmD4 2874
torchopt/optim/meta/adagrad.py sha256=r6oEClFKyZrOrJSsTRoNDn7FqS1espI0gqHA6pZIUas 2955
torchopt/optim/meta/adam.py sha256=hXbzTDsXr7uqjLO5cZtAQ9t1ScvijhbX1vWda5f_POQ 3631
torchopt/optim/meta/adamax.py sha256=DLOPLn4GT7Yn_loVec17ni9V6HJsxW1Otj1Ug5nPBbA 2895
torchopt/optim/meta/adamw.py sha256=Sy6FV8XIdO8Q7ujAkUQNXOcRJI-p9HJb7h1xPsEB-Do 4641
torchopt/optim/meta/base.py sha256=qjmT1vrcbV6SJawbYvQtGIK5P8uwNAive_DrNAm6bm4 4930
torchopt/optim/meta/radam.py sha256=hG5mylrt230jeYmaADF37sdtl2vZNrmWltFz1mnHqdE 2809
torchopt/optim/meta/rmsprop.py sha256=x0S038iY3Vrmx8g-EmFMILNQuapDvds5gdkpdF5pZvM 3704
torchopt/optim/meta/sgd.py sha256=PaJr97e3q56hWL8gqVIUGKE6F1w5R8lHsog4fTXKCy8 3000
torchopt/schedule/__init__.py sha256=p-wHuBV5BMLFmAHBvnrK7JbfGeFtasHpFGqMS7E7y7Q 1814
torchopt/schedule/exponential_decay.py sha256=jdV6TPHRosRyduA1VMHZQdG19x7FJPd-x2OrxM1w9J4 5052
torchopt/schedule/polynomial.py sha256=vPRLQZ6F8rr-AZ-LqUK-c4a8xug5hQCpB-zInDT7F8k 4387
torchopt/transform/__init__.py sha256=s5jQC6oq7clI2TBzpEpvODPDB13aj1qjJnUxaZnOOGY 2646
torchopt/transform/add_decayed_weights.py sha256=jNTN4VPM5sQTragsUqbSBlsX2q5TSH3Xkq5SW0u54NI 9557
torchopt/transform/nan_to_num.py sha256=Gqufhz8vgOQzZ0EK_iL-WOI8BuZMeGSNjFfxiLf60RI 2022
torchopt/transform/scale.py sha256=hSdRKUMlREso8BsxV8mFkX5-ehRKZ2MqqAwEpIgFcwo 3581
torchopt/transform/scale_by_adadelta.py sha256=aBTi7Rhi2krnl1UBUYQAOVWSADJphDPAnOTTviZoNu0 4838
torchopt/transform/scale_by_adam.py sha256=Xham3L3VZCvXjTHirG4IBcMQh6Ey7gv6OWwrGwwHDD4 14645
torchopt/transform/scale_by_adamax.py sha256=9e6Q8d-xOYiDZ5PvyAT7U7dUkUELXi_tGS8CbL-YOeo 4942
torchopt/transform/scale_by_radam.py sha256=PIm1ABM7eOEja2HIOCXJDBIX0355_s5t6sO_yUKRWlM 6192
torchopt/transform/scale_by_rms.py sha256=VtDAXG-nyAc4d1yeBJxbGny7s-mG8ycil6sPgYtiUFw 5256
torchopt/transform/scale_by_rss.py sha256=w_-Unk3-fNq7pmQg920gBrTkUFgY1bdvCiebS7lP_Rk 5180
torchopt/transform/scale_by_schedule.py sha256=NKPZjLIMmRPpEbX7TH4TflbF0fFonU5MegSFOwHyd4g 4771
torchopt/transform/scale_by_stddev.py sha256=abtuZsfQlDnFWN4Eb43ZykXb1SdrZuy0AO4a9TI50os 5789
torchopt/transform/trace.py sha256=eIu3aURQVOXO5HsAd2IC74p7e30m8pbxoOkAIcly7xw 7410
torchopt/transform/utils.py sha256=k6SdBluLP1-i8DPWSnODoEy0JO2Frcp6_cyvf0FZWAc 6868
torchopt-0.7.3.dist-info/LICENSE sha256=2p1yHZ2-Wggk7ku5MVp59TvIUeCBZB8nd7RwtdlJPL0 11374
torchopt-0.7.3.dist-info/METADATA sha256=F8Yr_bRKOBT3BnDtdNqsLh5al7_k2SzdMk3g4PboSbo 28753
torchopt-0.7.3.dist-info/WHEEL sha256=Xo9-1PvkuimrydujYJAjF7pCkriuXBpUPEjma1nZyJ0 92
torchopt-0.7.3.dist-info/top_level.txt sha256=6sv9eroZeCxdNZqhXZl9bYWJhSwr4xG7s3AzIPXmJM4 9
torchopt-0.7.3.dist-info/RECORD

top_level.txt

torchopt