Wheelodex » Projects » pax3

pax3

View on PyPI — Reverse Dependencies (1)

0.5.9 pax3-0.5.9-py3-none-any.whl

Wheel Details

Project: pax3
Version: 0.5.9
Filename: pax3-0.5.9-py3-none-any.whl
Download: [link]
Size: 71123
MD5: b1f3bbae7251671fb27ce086784bc3a8
SHA256: f0f0fe4318cd47ae93ba6449e8c9d0eb8ba6bc01be9f49606ec812cc4c2b6eaa
Uploaded: 2022-07-30 10:12:17 +0000

dist-info

METADATA · WHEEL · RECORD · top_level.txt

METADATA

Metadata-Version: 2.1
Name: pax3
Version: 0.5.9
Summary: A stateful pytree library for training neural networks.
Author: Thông Nguyễn
Home-Page: https://github.com/ntt123/pax
Keywords: deep-learning,jax
Requires-Python: >=3.7
Requires-Dist: jax (>=0.2.21)
Requires-Dist: jmp (>=0.0.2)
Requires-Dist: chex; extra == "test"
Requires-Dist: dm-haiku; extra == "test"
Requires-Dist: fire; extra == "test"
Requires-Dist: opax; extra == "test"
Requires-Dist: pytest; extra == "test"
Requires-Dist: pytype; extra == "test"
Requires-Dist: tqdm; extra == "test"
Provides-Extra: test
Description-Content-Type: text/markdown
License-File: LICENSE
[Description omitted; length: 7416 characters]

WHEEL

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

RECORD

Path Digest Size
docs/conf.py sha256=55A7otUtR8n7Ru4W4vxRm1JidXHl808_YGImujMQT-Q 5158
examples/denoising_diffusion/data_loader.py sha256=Ec0gBtcT_YmQ55wdPZ-YdOgzclheTA7l38WtAuatvwE 2818
examples/denoising_diffusion/model.py sha256=R9pqJWUmDlb7iVPAw_InZK787py8jNMC5WLCcJxPFiY 14430
examples/denoising_diffusion/train.py sha256=DXeKDwzwzUGwqrOy8Rg83fS6UDdM8I88eeLNu3j0Vz8 2612
examples/notebooks/pretrained_resnet18.py sha256=lTckbl8peQ9qV-24gGWbLnrgY8PyK_3AplH-pFWyOQM 4028
examples/transformer/data.py sha256=u_IhJeqq3pziVEsid_T3OqA6P5bkUFmbXX6W7bSfA_s 2565
examples/transformer/model.py sha256=7t8Vc9ASbhWWq7oQZe-d2eCG3MH_7zW4Q0jhoUhh7co 2395
examples/transformer/train.py sha256=zLa2c9PyW-2z5KHZIauxxSXqmPL4eemdfX_3mv5fDtg 4039
examples/wave_gru/data_loader.py sha256=vP1YqZ1SZmykwV1QO03pyTbNRE-5G3B5R3Rf11aPEFk 1540
examples/wave_gru/model.py sha256=Iox2BxWAjk49aQQwE31hV_HaDYXw-ozoXB9tkpb6wzw 3044
examples/wave_gru/train.py sha256=R7XWVdqYTeIFqZOer6gjIkG1160RgsqvXryMuvHYc6A 2971
pax/__init__.py sha256=KASwn9tTOCmmV7P_OA7x43K1EikamvKC-dMpZmdZV-w 2084
pax/nets.py sha256=sLRDXu2lWVRLT-j3EFT-0XDwqmzgVQUNzmeS0B94wGU 287
pax/utils.py sha256=D8npAu9biLn-CyDbwN9cn6UsUVu9435G_YIubebBhJg 150
pax/_src/__init__.py sha256=0kvAr_HqBQjhKPcCxyAxyBh3E9tkyyvENKENQmjWAlk 23
pax/_src/utils.py sha256=P2FBMV_29QMjzPsALOlwJAcFNIMKVyvEr5QSscNXJnI 8484
pax/_src/core/__init__.py sha256=-r8s6uL4GnLwx4sgrXeXhBV-eOa3x1fsJww26LqKuF4 606
pax/_src/core/base.py sha256=53hN3a5i7YU0HLlYB_66WsQ-QdD3pmZh6hrlAWrnl4A 5796
pax/_src/core/graph_module.py sha256=VizbNbmdTJWgKdkFRlFeYgtqQKGmRBYtAlhI323EwIc 7769
pax/_src/core/mixed_precision.py sha256=7LoH-cK-W7F0D_DiJRfqFEYqugeZ5iPMm1i93iNnrVo 5399
pax/_src/core/module.py sha256=1gL81FJXoQT8z8F1neLbCUQF7INrq079inuEbwPjze0 14518
pax/_src/core/module_and_value.py sha256=AiI95W7ydOmMcBqeNlLYxE-mt0TnN4aZo-XvXwrW0nw 1616
pax/_src/core/mutable.py sha256=Pi1GvtMTGYNms9cOY9sgahuXlbofrDFufVKB9cp0kdU 695
pax/_src/core/pure.py sha256=AoXbHgXduVF6ekrYhwCVWPB7yq5Y6KbG8tldLMrj60A 2856
pax/_src/core/rng.py sha256=rtDtvY_We3Cyee17E92bxFaRmREQoqghP6Ovd4XbkJU 169
pax/_src/core/safe_module.py sha256=uIyZKX1bK38XowZ5N7aUVquwQCG-rNlxzzICu-Ws94g 4992
pax/_src/core/threading_local.py sha256=sFChwIyUNqHPOQQdD54NDPehtzzggWAhFQLmQKUG0zw 3210
pax/_src/core/transforms.py sha256=Du3VSIfsuSS-isawryWQSsjoqTT3ir3cDuGQulEuVAM 1343
pax/_src/core/utility_modules.py sha256=geQwD1DLRrX4RB-_72Sk_feMvgiPdBdlrQC_wjyOuMM 5468
pax/_src/core/utils.py sha256=SOBQB6prRSRTWKHn2O_PPXe43SMQzG54lBpH-p_1HWA 1287
pax/_src/nets/__init__.py sha256=rWFULImc7MHKdcFcxefONVDr41_c2QUtqBBwVPAimlQ 180
pax/_src/nets/resnet.py sha256=Jv1XlbR9qb5e2oXqYWjnq4ExYLIxV5nJ7vkFA30SlGg 11199
pax/_src/nets/transformer.py sha256=jbrBdlp3x7bvqv391eNZGBPaPcorvx4edDh1NX8g7_k 3961
pax/_src/nn/__init__.py sha256=CM3GJbhBtIvNRnxyZqxz-OzNzUq2NEIFYmdzbfhGNFk 592
pax/_src/nn/attention.py sha256=Zj4w_6HjEt-vhjV1aac1sA6TfUVq8N-RYU6whkuT9bs 2922
pax/_src/nn/batch_norm.py sha256=pWPQTfONrz52tlHY2r2_kB7dNv2ML-tLCp4wzonLfkg 6020
pax/_src/nn/conv.py sha256=A0nKqx8s9wZzzq9xw4ic97CscZSGaRgh_mdXLkFPLBE 18709
pax/_src/nn/dropout.py sha256=0NIc9IIObNPdLnyYQ1KKeHziNGGOQ_xHK8ek_N1jE0w 1696
pax/_src/nn/ema.py sha256=e95PeZg1I-WkvfMnAy1djTHG_RsZ6VgzGJ55rlkl67M 2643
pax/_src/nn/embed.py sha256=GG4Xpi6euMgGympqAc04ix-oNYUaTvOC9kgQwwYO6PQ 1481
pax/_src/nn/group_norm.py sha256=mQSxymBuvcLcFX4uH98t2_wW-DCWtUxPmoPPGBN_8go 9107
pax/_src/nn/identity.py sha256=_OxHNoXNZ42Zn1kGLn45AYegSLXjtCrUpd-Rn2AJqhg 185
pax/_src/nn/lambda_module.py sha256=X1KlpmwR77pBrPyP4hoh1isIEgmbYKkwsvTo3WBsBug 87
pax/_src/nn/layer_norm.py sha256=3QcmEUGtBYbSlT8--pW8mpLkGFYOXCzll2bk-hx1ZYY 6016
pax/_src/nn/linear.py sha256=iArmhYJ5ZEXJ4ccpJ2y6vKd-LiqKTePMeq_BbS9yQy0 2285
pax/_src/nn/pool.py sha256=WAhdRh1_ektlLnD5VveHpqNj8i0PesYNy1GWbXk8CmE 5391
pax/_src/nn/recurrent.py sha256=jx78a6ZIeFCAewoUDR1vL8L58VLLJMfPyvIvGTNbHyQ 6087
pax/_src/nn/rng_seq.py sha256=qxmYC8cmcg_1oXeSnGpGDZJ4JEMLc2NlCSN1A3Hw4B8 1607
pax/_src/nn/sequential.py sha256=fwh1SPehA3sOJtJXFav8xmdOWAcgdsx-4J7ll4rlkXc 1840
pax/experimental/__init__.py sha256=RaLiUoElcISYyxNL1armA8wGyxRfN7PDrwmssj_y7yU 430
pax/experimental/graph.py sha256=WP8anhOhN0szXp1P4NX_dUPw9_JEq0pPT3DjTWrQYA0 208
pax3-0.5.9.dist-info/LICENSE sha256=ekSaJpMHAKwDB64Gzs2G09xoMALKR1A4Zt6mni8mnZc 1072
pax3-0.5.9.dist-info/METADATA sha256=cQPPa2saJpGauLtPp87X4ccexqDRwJ38M5Z3yhy6f20 8109
pax3-0.5.9.dist-info/WHEEL sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA 92
pax3-0.5.9.dist-info/top_level.txt sha256=30faRJqr9W5KTaJBuR1QXOffBiv1B4pyhvO9CkFiDf8 25
pax3-0.5.9.dist-info/RECORD — —

top_level.txt

docs
examples
images
pax