Wheelodex » Projects » kfac-jax

kfac-jax

View on PyPI — Reverse Dependencies (1)

0.0.6 kfac_jax-0.0.6-py3-none-any.whl

Wheel Details

Project: kfac-jax
Version: 0.0.6
Filename: kfac_jax-0.0.6-py3-none-any.whl
Download: [link]
Size: 146249
MD5: edf7b7bbcb2bb7e11cbd665d48547b9b
SHA256: bbace8bc3670d332c5e18b1fb86a0f6cd7af5732f4713803a3ff9c418016c96c
Uploaded: 2024-04-04 10:59:13 +0000

dist-info

METADATA · WHEEL · RECORD · top_level.txt

METADATA

Metadata-Version: 2.1
Name: kfac-jax
Version: 0.0.6
Summary: A Jax package for approximate curvature estimation and optimization using KFAC.
Author: DeepMind
Author-Email: kfac-jax-dev[at]google.com
Home-Page: https://github.com/google-deepmind/kfac-jax
License: Apache 2.0
Classifier: Development Status :: 4 - Beta
Classifier: Intended Audience :: Developers
Classifier: Intended Audience :: Education
Classifier: Intended Audience :: Science/Research
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: Topic :: Scientific/Engineering :: Artificial Intelligence
Classifier: Topic :: Scientific/Engineering :: Mathematics
Classifier: Topic :: Software Development :: Libraries
Classifier: Topic :: Software Development :: Libraries :: Python Modules
Requires-Dist: absl-py (>=0.12.0)
Requires-Dist: immutabledict (>=2.2.1)
Requires-Dist: numpy (>=1.21)
Requires-Dist: distrax (>=0.1.3)
Requires-Dist: jax (>=0.4.7)
Requires-Dist: jaxlib (>=0.4.7)
Requires-Dist: dm-tree (>=0.1.7)
Requires-Dist: optax (>=0.1.4)
Requires-Dist: typing-extensions (>=4.2.0); python_version < "3.10"
Requires-Dist: pytest-xdist; extra == "tests"
Requires-Dist: absl-py (==0.12.0); extra == "tests"
Requires-Dist: immutabledict (==2.2.1); extra == "tests"
Requires-Dist: numpy (==1.21); extra == "tests"
Requires-Dist: distrax (==0.1.3); extra == "tests"
Requires-Dist: jax (==0.4.7); extra == "tests"
Requires-Dist: jaxlib (==0.4.7); extra == "tests"
Requires-Dist: dm-haiku (==0.0.9); extra == "tests"
Requires-Dist: dm-tree (==0.1.7); extra == "tests"
Requires-Dist: optax (==0.1.4); extra == "tests"
Provides-Extra: tests
Description-Content-Type: text/markdown
License-File: LICENSE
[Description omitted; length: 7879 characters]

WHEEL

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

RECORD

Path Digest Size
docs/conf.py sha256=6wLnJ0FSXSVsSWjDMcqih1ZD3RabRR4WcpGxUcsJzhU 6726
docs/ext/coverage_check.py sha256=alMYHgnwMtZFgbTk6yjXB2liTOQ5M8O0pUbnylvvWOc 2586
examples/autoencoder_mnist/experiment.py sha256=BIZF52wr5C7GZLFQmzEc9IVMiV4_jfya6G4ReYorCsA 3276
examples/autoencoder_mnist/pipeline.py sha256=rnIlHIHHjVNrkUSI1WnEEcXLnqDypreOrHzEo5FTuxE 3027
examples/classifier_mnist/experiment.py sha256=iQFyeDvyRyp_Q9UUMcNYvvqzBAjl1UtUz1SAEh_sIIs 3037
examples/classifier_mnist/pipeline.py sha256=lJDnn8vM0_35VrwYDfutCWAdZ0LooP39HCpOvqRly78 3038
examples/lrelunet101_imagenet/experiment.py sha256=f7NJoNVz-hrAtdrYBo7dKvYVweHR2UzPntr8-fq9378 12373
examples/lrelunet101_imagenet/pipeline.py sha256=HcXgQJCYNfdsXCq9YxK6s9BPnmlNObcu6USIyvUUkUI 3761
examples/resnet50_imagenet/experiment.py sha256=n506ldrrzpJDLecRc7DcRHcn41mMMmQt6QlKjRdcHmg 3737
examples/resnet50_imagenet/pipeline.py sha256=0c6nS-4hc8VQOZIF60OYC3TMtFPaJ5006g787g46l_M 3518
kfac_jax/__init__.py sha256=unlXCMLP5Mnnu31jGOLj6Iz2Uz4YxxsceGdriNCDBjk 7221
kfac_jax/py.typed sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
kfac_jax/_src/__init__.py sha256=WIVmD14Sr1HfmJddHhkHkq6FPL1k3Dpk09EDS0ET13M 615
kfac_jax/_src/curvature_blocks.py sha256=bLZTY9ZRWCKkeT4O_MNvkNvil0osRlDk6z3Cl38xdPY 68034
kfac_jax/_src/curvature_estimator.py sha256=uysXZHfUCKhCaI2H8PeZzdk9kuA3YTc1Eg5Yyc9SOFw 63406
kfac_jax/_src/layers_and_loss_tags.py sha256=y74CcPd4IKgR29IjBC1YfnMQYu7Zc-MpHfXtUNPhA7Y 12925
kfac_jax/_src/loss_functions.py sha256=LWHMZhUbyffDzUCfW67U4rRcW_CiX0NZwlfR1whwI9M 52339
kfac_jax/_src/optimizer.py sha256=K-FcWhztI31eeO2T6b5zDcoJSecrQZfEHnEEb7OOHvs 66190
kfac_jax/_src/patches_second_moment.py sha256=vc5NCk-AA-7hT9KC1Og9T7q3eCzL1K5Qd-GdU9kbi9Q 36908
kfac_jax/_src/tag_graph_matcher.py sha256=waY3UbovqFKFbPrTQPknDydSZxVO0r9c3DAfa96KMTU 59297
kfac_jax/_src/tracer.py sha256=WdPtDJMxpApta5CnwAYWyy4BGxIF6mnG873C7NqVCf8 38422
kfac_jax/_src/utils/__init__.py sha256=mKcgSUfXzqOjn36R3E5HmmLZYqwep4KphSNa2Du9Ow8 5179
kfac_jax/_src/utils/accumulators.py sha256=Kjt8_yuxU1HC5nrDnKfcFdP8Un7BIqehDZyvMwt4z1w 8948
kfac_jax/_src/utils/math.py sha256=7cBqTW8wawsL_twRD2SNPLTBRYXTBeOpQ0soXsA10P8 37111
kfac_jax/_src/utils/misc.py sha256=Dmu5QXDX_1Up-HCC7sC9vq6V_KHMiGeMCud8w50gJ7s 11166
kfac_jax/_src/utils/parallel.py sha256=ArkTPlRLpzzxlbFarRM86CUGwYqYsqM__a2W4jNoDPo 9914
kfac_jax/_src/utils/staging.py sha256=bOnmtl47eeNsiOX4t3T-ckmf8myR_qw4yfF3sSGuosM 8813
kfac_jax/_src/utils/types.py sha256=CQg9_GhXV7BvIqaMJQVlWGVnxU4oX4D2Ph79u5rDFMI 2937
kfac_jax-0.0.6.dist-info/LICENSE sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA 11358
kfac_jax-0.0.6.dist-info/METADATA sha256=IOW6amnwC46efOpsmI9Q0mP6pLtxx5nCvMxkgLG3EAs 9691
kfac_jax-0.0.6.dist-info/WHEEL sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ 92
kfac_jax-0.0.6.dist-info/top_level.txt sha256=RAlYDjMHjt-k59AY-9d8um4rfHbjJnzQmteCppp4qcs 23
kfac_jax-0.0.6.dist-info/RECORD — —

top_level.txt

docs
examples
kfac_jax