optax-shampoo

View on PyPIReverse Dependencies (1)

0.0.6 optax_shampoo-0.0.6-py3-none-any.whl

Wheel Details

Project: optax-shampoo
Version: 0.0.6
Filename: optax_shampoo-0.0.6-py3-none-any.whl
Download: [link]
Size: 32044
MD5: 179f8497a911f496081b02a7683d8fb2
SHA256: 935bf236149365e6afbaef5de2384008c78cb5a7d2ac74b0c75181fadf14c6aa
Uploaded: 2022-10-07 02:28:56 +0000

dist-info

METADATA

Metadata-Version: 2.1
Name: optax-shampoo
Version: 0.0.6
Summary: Distributed Shampoo (Second order Optimizer for Deep Learning) Optax Optimizer
Author: rohananil
Author-Email: rohan.anil[at]gmail.com
Home-Page: https://github.com/google-research/google-research/tree/master/scalable_shampoo
License: Apache Software License
Keywords: optax distributed shampoo
Classifier: Programming Language :: Python :: 3
Classifier: License :: OSI Approved :: Apache Software License
Classifier: Operating System :: OS Independent
Classifier: Development Status :: 3 - Alpha
Requires-Python: >=3.6
Description-Content-Type: text/markdown
License-File: LICENSE
[Description omitted; length: 730 characters]

WHEEL

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

RECORD

Path Digest Size
optax_shampoo/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
optax_shampoo/distributed_shampoo.py sha256=KG9STjbnkQWTGTdV_IG6L01EZnzqvEcPq9pa9MbRdgM 95063
optax_shampoo/distributed_shampoo_test.py sha256=ta__77owwqOJUfMZloj-g0Z78ZDXmlFtrL7dNvtg9Eo 13511
optax_shampoo/quantization_utils.py sha256=4KdOlAlt-ijBS7clVlj9t9J7_GGgp6ZASSmcmVqsNqI 4247
optax_shampoo-0.0.6.dist-info/LICENSE sha256=gTXjZ4WSwVpgnHkvzVZu-N_pW6OXwg2cUBFM3uUlicM 11311
optax_shampoo-0.0.6.dist-info/METADATA sha256=2CgIabSxGPOyajI_Ul3wIvoVDQ8ipVkGJRo7_uqD0aE 1376
optax_shampoo-0.0.6.dist-info/WHEEL sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA 92
optax_shampoo-0.0.6.dist-info/top_level.txt sha256=Bx00fxZA_lSBqcOchyWRvZE9QAC-XUwYLjQG7mANxf4 14
optax_shampoo-0.0.6.dist-info/RECORD

top_level.txt

optax_shampoo