Wheelodex » Projects » psgd-jax » psgd_jax-0.2.8-py3-none-any.whl

psgd-jax

View on PyPI — Reverse Dependencies (1)

0.2.8 psgd_jax-0.2.8-py3-none-any.whl

Wheel Details

Project: psgd-jax
Version: 0.2.8
Filename: psgd_jax-0.2.8-py3-none-any.whl
Download: [link]
Size: 34960
MD5: cd02e833fc2ecce5419fea083f949bd7
SHA256: 198a0f0197630b2524913d1c83b7037adb10bca8da8e4b384fcb349c2c7db472
Uploaded: 2024-12-08 20:46:31 +0000

dist-info

METADATA · WHEEL · RECORD

METADATA

Metadata-Version: 2.3
Name: psgd-jax
Version: 0.2.8
Summary: An implementation of PSGD optimizer in JAX.
Author: Evan Walters, Omead Pooladzandi, Xi-Lin Li
Project-Url: homepage, https://github.com/evanatyourservice/psgd_jax
Project-Url: repository, https://github.com/evanatyourservice/psgd_jax
Keywords: python,machine learning,optimization,jax
Classifier: Environment :: Console
Classifier: Programming Language :: Python
Classifier: Intended Audience :: Developers
Classifier: Operating System :: OS Independent
Classifier: Programming Language :: Python :: 3
Classifier: Intended Audience :: Science/Research
Classifier: Development Status :: 4 - Beta
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Classifier: Topic :: Software Development :: Libraries :: Python Modules
Requires-Python: >=3.9
Requires-Dist: numpy
Requires-Dist: matplotlib
Requires-Dist: jax
Requires-Dist: optax
Description-Content-Type: text/markdown
[Description omitted; length: 12059 characters]

WHEEL

Wheel-Version: 1.0
Generator: flit 3.10.1
Root-Is-Purelib: true
Tag: py3-none-any

RECORD

Path Digest Size
psgd_jax/.DS_Store sha256=hIJc1FbywebgzmnOLS_7BVhc5DLiQLYQckZ68zN3i10 6148
psgd_jax/__init__.py sha256=k_rrOhdLfjxkdba3UxADbA_sSRVTOMzzw4FQuEqMjmg 42
psgd_jax/affine.py sha256=uU1r53NDjgYXydvtRw7d2I1NMQ-KHpDwy4WMs4oqNOg 25869
psgd_jax/kron.py sha256=aHx-3GY59-WQIk2zxsejx8usuDmGZ98r9eZ-8dkeYkg 25513
psgd_jax/low_rank_approximation.py sha256=lbIurWaeFSx_N5cBKeFBRosZ4De06PQ4RZC4t_oOqm0 14328
psgd_jax/psgd_test.py sha256=jVZ2JpQ5g73kQ7JSXG2Vd6DmydjSvLpDo7zzFSjKaEE 8303
psgd_jax/utils.py sha256=xc4-p4g9PtaV8vrUFLwqSTE6vAYoZ5HHR_Z8na-cusk 3563
psgd_jax/xmat.py sha256=phdQBpPPALidH-hP0gPw9uos7uNXkq2ZgfGe6TlJXqA 10421
psgd_jax-0.2.8.dist-info/LICENSE sha256=nl8bPGELnC2lwxO_gdV3p9Gs7Ghr2wOE7e-m3w-QzZQ 18656
psgd_jax-0.2.8.dist-info/WHEEL sha256=CpUCUxeHQbRN5UGRQHYRJorO5Af-Qy_fHMctcQ8DSGI 82
psgd_jax-0.2.8.dist-info/METADATA sha256=XRbgJWjHBx0GKDmTqbpReULicEwWmvSorNeLABtWZ7A 13011
psgd_jax-0.2.8.dist-info/RECORD — —