Wheelodex » Projects » penzai » penzai-0.2.4-py3-none-any.whl

penzai

View on PyPI — Reverse Dependencies (3)

0.2.4 penzai-0.2.4-py3-none-any.whl

Wheel Details

Project: penzai
Version: 0.2.4
Filename: penzai-0.2.4-py3-none-any.whl
Download: [link]
Size: 317839
MD5: 1c091d977c40d7da2a1cb8600fef7fdb
SHA256: 107b03be3e959bb1aeb0357cd1a7f30aab8264ef64b931157f2873118acdc85a
Uploaded: 2024-12-16 08:39:40 +0000

dist-info

METADATA · WHEEL · RECORD

METADATA

Metadata-Version: 2.3
Name: penzai
Version: 0.2.4
Summary: Penzai: A JAX research toolkit for building, editing, and visualizing neural networks.
Author-Email: The Penzai Authors <penzai-dev[at]google.com>
Project-Url: homepage, https://github.com/google-deepmind/penzai
Project-Url: repository, https://github.com/google-deepmind/penzai
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3 :: Only
Classifier: License :: OSI Approved :: Apache Software License
Classifier: Intended Audience :: Science/Research
Requires-Python: >=3.10
Requires-Dist: absl-py (>=1.4.0)
Requires-Dist: jax (>=0.4.23)
Requires-Dist: numpy (>=1.25.2)
Requires-Dist: ordered_set (>=4.1.0)
Requires-Dist: treescope (>=0.1.3)
Requires-Dist: typing_extensions (>=4.2)
Requires-Dist: pylint (>=2.6.0); extra == "dev"
Requires-Dist: pyink (>=24.3.0); extra == "dev"
Requires-Dist: ipython; extra == "dev"
Requires-Dist: jupyter; extra == "dev"
Requires-Dist: pytest (>=8.2.2); extra == "dev"
Requires-Dist: pytype; extra == "dev"
Requires-Dist: ipython; extra == "docs"
Requires-Dist: flax (>=0.8.2); extra == "docs"
Requires-Dist: optax; extra == "docs"
Requires-Dist: setuptools; extra == "docs"
Requires-Dist: sphinx (<7.3.0,>=6.0.0); extra == "docs"
Requires-Dist: sphinx-book-theme (>=1.0.1); extra == "docs"
Requires-Dist: sphinxcontrib-katex; extra == "docs"
Requires-Dist: ipython (>=8.8.0); extra == "docs"
Requires-Dist: myst-nb (>=1.0.0); extra == "docs"
Requires-Dist: myst-parser (>=3.0.1); extra == "docs"
Requires-Dist: matplotlib (>=3.5.0); extra == "docs"
Requires-Dist: packaging (==24.1); extra == "docs"
Requires-Dist: sphinx-collections (>=0.0.1); extra == "docs"
Requires-Dist: sphinx_contributors; extra == "docs"
Requires-Dist: sphinx-hoverxref; extra == "docs"
Requires-Dist: jax[cpu] (>=0.4.23); extra == "docs"
Requires-Dist: ipython; extra == "extras"
Requires-Dist: flax (>=0.8.2); extra == "extras"
Requires-Dist: optax; extra == "extras"
Requires-Dist: torch; extra == "extras"
Requires-Dist: transformers (>=4.41.2); extra == "extras"
Requires-Dist: ipython; extra == "notebook"
Requires-Dist: flax (>=0.8.2); extra == "notebook"
Requires-Dist: optax; extra == "notebook"
Requires-Dist: orbax-checkpoint; extra == "notebook"
Requires-Dist: palettable; extra == "notebook"
Provides-Extra: dev
Provides-Extra: docs
Provides-Extra: extras
Provides-Extra: notebook
Description-Content-Type: text/markdown
[Description omitted; length: 6708 characters]

WHEEL

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

RECORD

Path Digest Size
penzai/__init__.py sha256=QgxHfXI9zMkX5N8CEWtXg0w3dhgSDIv3dpGkgouDRs8 724
penzai/core/__init__.py sha256=Z3Pg9RHcIhHdMQWJVOsLVaf_2-osSfwjh6IqOTHd2kU 800
penzai/core/auto_order_types.py sha256=JJ_vsimGXp3rqTfmcMz3CvVTb4WRspkkiozq9tWmE1g 4861
penzai/core/named_axes.py sha256=-s8VhAJjNYfHB3YUmtjdXd-I7C5x5w82nVa-cXjqy_0 80814
penzai/core/partitioning.py sha256=zYRrWNliZJLg3chhi2ApVMbH9LXcPWXovvqhED_Np7M 5305
penzai/core/random_stream.py sha256=RXY5BLbe-_CTuug2VMA7yb770B2XpUE0u-qihvVbwJo 1869
penzai/core/selectors.py sha256=fgOOl8tT3uHdKvpMY1Tb8f4l3sIIZ25Sb9bBjF9mans 56056
penzai/core/shapecheck.py sha256=rz5vQ6DMhep88WhO9uVoCYk2yi5Lse3rWXEAelKz9wA 43769
penzai/core/struct.py sha256=JKxcCS00YcBkZWeUXgOFMcKvgMd3C8DVwbW16l_IlZI 29378
penzai/core/syntactic_sugar.py sha256=e-z6OF7_wWN37ROwzoeKvXIAnn-yOEl6TPqhyEKhGb8 1117
penzai/core/tree_util.py sha256=PZN2utt8AW-Yg7zblxiWg2tWCszO3I27ty7hGMai_kA 2634
penzai/core/variables.py sha256=3pX62qPm0Dza2wjxBKiHZrUOL6EeYp3u8SCPQHldLeo 29497
penzai/core/_treescope_handlers/__init__.py sha256=5d1inGwgDwaTVLJ74sdUUK01_LEzHSvhvWQzoDy7aCo 635
penzai/core/_treescope_handlers/named_axes_handlers.py sha256=It2irVw7wWbh1RZAPtFbuvU3ecWOD8XOShCY7WTBFoY 8914
penzai/core/_treescope_handlers/selection_rendering.py sha256=TfiKa23l2qJixYQ7dchhtpgAjOkYWgF3La5tzBKTSK0 11499
penzai/core/_treescope_handlers/shapecheck_handlers.py sha256=eUdPriwbjZa_uRrGG-vjlfoFqTVeb2fsNYYc92Db_qo 4868
penzai/core/_treescope_handlers/struct_handler.py sha256=APc7em0j8uscYpg1Ss5BGJCPNeyzeSRCh0h98lhHlsE 4727
penzai/deprecated/__init__.py sha256=pr5uQLqWLt4vn80GFEpQOL3ImqQ2FRvW9I7PVeK9nPM 583
penzai/deprecated/v1/__init__.py sha256=pr5uQLqWLt4vn80GFEpQOL3ImqQ2FRvW9I7PVeK9nPM 583
penzai/deprecated/v1/core/__init__.py sha256=pr5uQLqWLt4vn80GFEpQOL3ImqQ2FRvW9I7PVeK9nPM 583
penzai/deprecated/v1/core/layer.py sha256=P4xOW3Xp2odejoo4w3TycotSp54OA6NX_pI0iZAnjSw 9739
penzai/deprecated/v1/core/random_stream.py sha256=noa2jHElBx23bUlkvdMDSPUUlwvbL3fAUWAumfszGxs 3782
penzai/deprecated/v1/core/_treescope_handlers/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
penzai/deprecated/v1/core/_treescope_handlers/layer_handler.py sha256=y_WPtLz4c4G_d51sNHd28Bmk87xkEz3j22F4O117CuY 8245
penzai/deprecated/v1/data_effects/__init__.py sha256=kNR_MZ2jXAEiKfqjOmbIFkv_UAcBROe6JCxkMbl773A 744
penzai/deprecated/v1/data_effects/_treescope_handlers.py sha256=z3hUVymM8msbk_IvEnoe-XjlUodHwyRYoNSY1e4vmPc 5550
penzai/deprecated/v1/data_effects/effect_base.py sha256=hzYbFgChDk4bYcWuZthqQ4alF5D3KdOwmokksvlDDGQ 21372
penzai/deprecated/v1/data_effects/local_state.py sha256=Gp98fbxIKuELcgt5jo0DlrtAeUFfmz84cUk-lwHkpUs 24503
penzai/deprecated/v1/data_effects/random.py sha256=HiU5lrUGl3tdSffG3lmrGBnnqvuDFhZ8XDPaJirF2NE 11995
penzai/deprecated/v1/data_effects/side_input.py sha256=R8dD9jR8S4P_IK4kvGJnwL23sSEIDV8kCCkQdK0s6Xc 14596
penzai/deprecated/v1/data_effects/side_output.py sha256=j984iAv6xmv0nKGKUTyH6qXJXLRnnStg0iTg8E0f7rk 7279
penzai/deprecated/v1/example_models/__init__.py sha256=pr5uQLqWLt4vn80GFEpQOL3ImqQ2FRvW9I7PVeK9nPM 583
penzai/deprecated/v1/example_models/simple_mlp.py sha256=QSFwIaPekZcdJfxJcDPOszVGPS8sikX4W7vLUn67dPc 2730
penzai/deprecated/v1/example_models/gemma/__init__.py sha256=gW5JzWZLSzKR8Yo5rZL22kVFT-_tb3QndBMrIeplkoc 735
penzai/deprecated/v1/example_models/gemma/model_core.py sha256=vwge4kwJxen1zhi9e57VPSruqfKopXpgml4Ij0MNTVI 24071
penzai/deprecated/v1/example_models/gemma/sampling_mode.py sha256=g_PSEWchaGjDiZxBuxa-ptlibRrl78iph_568XkEZCA 11377
penzai/deprecated/v1/example_models/gemma/simple_decoding_loop.py sha256=fh7PG5l86sGKtMpcS1YryFPknL4U-akHG9WdoUZB-CU 9559
penzai/deprecated/v1/nn/__init__.py sha256=rBxHJiK02ToFkkW_Kq88P3TCPGvvm0cnmgKD9N_syaQ 867
penzai/deprecated/v1/nn/attention.py sha256=al1qFCgUg5MC3ySKxGuMKTp9n4eOqNBhzw9BoGxO3hk 11740
penzai/deprecated/v1/nn/basic_ops.py sha256=rAUD2w_uCWj5coI7_gawonwSshkfD6s5o8TxNI-Ag8s 3336
penzai/deprecated/v1/nn/combinators.py sha256=DNbglK60bpgfbesATkU5C7X5IM71IdIrPjSkEejUtm4 4026
penzai/deprecated/v1/nn/dropout.py sha256=YmwaURxYjPgI1EudWQ3xzR2gbZWtDaNEOY6KkD7nrm4 5314
penzai/deprecated/v1/nn/embeddings.py sha256=D9MteOYocAeYtfk0tqdbmFfOY1PVGHTX6dTaIGgPEME 12948
penzai/deprecated/v1/nn/grouping.py sha256=7ikgNsx6CjyPlVMjiUB_0ieN3V-4aJMiUM91o_wwDqI 11879
penzai/deprecated/v1/nn/linear_and_affine.py sha256=-L7R_sbC3rWGyHIQZfVZRwMDc55vwgkU_sqD3MVnmS0 27314
penzai/deprecated/v1/nn/parameters.py sha256=vSl3D_arIOBTOAhws3UBPPWLqIwr-DX7JTSy9qbDJ48 23965
penzai/deprecated/v1/nn/standardization.py sha256=wEirWISHBmAq1bJi3T8RyyD3p5Wwhm3848sb17JT-X4 6652
penzai/deprecated/v1/pz/__init__.py sha256=qYDslAy6PHdlZYzONxhz-zVgMlBeMIHNC_ry-Q1_o1Y 2591
penzai/deprecated/v1/pz/de.py sha256=VlPBqUTQT_WQT2YDtYCsdmtaBBYk6FqkHCIEot22qJA 2011
penzai/deprecated/v1/pz/nn.py sha256=aMmv-d80JMq5ivRgRsdN-AaTKJT5RxcYX7hWG714Qx8 2459
penzai/deprecated/v1/toolshed/annotate_shapes.py sha256=qazEicLotdaDWVnFTlsKo_n0nyue43AICRza4Liz_sE 5780
penzai/deprecated/v1/toolshed/basic_training.py sha256=akXo-wVXil89P7M6REK533JBe6d5LOQ5T9762D2cM6o 13563
penzai/deprecated/v1/toolshed/check_layers_by_tracing.py sha256=yNuswm8fdWCjCj8JaZ60fbuOISC6Za7dDKvhwDUsKAg 5376
penzai/deprecated/v1/toolshed/interleave_intermediates.py sha256=Si0Wb_VblwRVIy3n-e17-OzvIxM_ulQ_aoDgIgbvZpw 6459
penzai/deprecated/v1/toolshed/isolate_submodel.py sha256=Gf3qBxK2eYoCVqOqCasHjpZAbRNlxxhxY2KzlKNCeks 10323
penzai/deprecated/v1/toolshed/jit_wrapper.py sha256=UqsWVJJNuytjqVDQufjaRd3vMTJVNspxOtJ_NcJ4_qc 2770
penzai/deprecated/v1/toolshed/lora.py sha256=JiX7WEcMmuaCDZT7cZnQKL6j345189e25BY7AL1ujDU 4826
penzai/deprecated/v1/toolshed/model_rewiring.py sha256=zjP6oH8XOBQstySXyBSKohamYyC7WWvX9RwN9JXw_Oc 10553
penzai/deprecated/v1/toolshed/sharding_util.py sha256=7OX12LzMEZLyB2luQgwu6WfWwfgDBnfjKhmhJuV64Zw 4296
penzai/deprecated/v1/toolshed/unflaxify.py sha256=QQsXnaelN6F3D6C9Sg8pJwXDqqS_kDyaBeVpcuxnXBg 18572
penzai/experimental/__init__.py sha256=pr5uQLqWLt4vn80GFEpQOL3ImqQ2FRvW9I7PVeK9nPM 583
penzai/experimental/v2/__init__.py sha256=9bn-Zdqw7EHRv9vj82r2Kj5jJw3zarnrX2RlRYIX0CM 6568
penzai/models/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
penzai/models/simple_mlp.py sha256=r6QRnHBMWDdw-ZvREOqUAUh1CMe3t8dt-hF3sG4GQ50 2828
penzai/models/transformer/__init__.py sha256=GwkKB-_D12iLGSLfJ09H_MB7SbTAbiGjV7_O1wfsDR8 745
penzai/models/transformer/model_parts.py sha256=7pr6T7i-gLDYCGOSpjNDLkgVxfaYArMoSnTRai1IBvw 5790
penzai/models/transformer/sampling_mode.py sha256=cP_lmwAIF7nLzFoJA-HQ_AMitzwUnt1xQ2GNKRAtYhc 8623
penzai/models/transformer/simple_decoding_loop.py sha256=zQmTp8Fpm3G47BLCOyOdtTtwjEPaA0AWsustwCvAdXw 3273
penzai/models/transformer/variants/__init__.py sha256=zJblKGjiGAFgHmEygNWonIff_IV--GCbUB341pGXz58 738
penzai/models/transformer/variants/gemma.py sha256=LCFD3V4XZK6somxUlaMaexW95eF0ITZ2KBqPwwWm0fY 11508
penzai/models/transformer/variants/gpt_neox.py sha256=bfZqEGlv-fn6-r1UELCrcdZEXawzTWGgmVqiS_MDP30 20443
penzai/models/transformer/variants/llama.py sha256=oOp-mCjAHd-hStDi_a28t-q6-lw5sTgyapiPrOHBANQ 3595
penzai/models/transformer/variants/llamalike_common.py sha256=s9020EOB2q0VCRephlkPUjlcLqptXYRrBcnR1RG4XhU 26825
penzai/models/transformer/variants/mistral.py sha256=5w922WV5EjCavAld0R3Pi3UCvcwb6hX-W4lyLgwYKZs 3907
penzai/nn/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
penzai/nn/attention.py sha256=0gAOu8SjOAnyCCRAvEhIkibMXrRK5Y7-HeYHrwAMEpg 16876
penzai/nn/basic_ops.py sha256=Lcs0IHZiavnVoPX2NSOegFwV1Q5BP4KfIzI4MLE4tRQ 3065
penzai/nn/combinators.py sha256=Z0pV7KTun7Yy72zgf1lSwyuKVqC-UlPbtJMoXD3xoxM 4184
penzai/nn/dropout.py sha256=HxfwVySMVMqvQ0TKbxxEQLpzVF_9DN6n4NbMSSsf_yw 5672
penzai/nn/embeddings.py sha256=11V3k5I4cs3e_dYBRbAdI283_9XcB0cygG3gpH0KdIQ 12661
penzai/nn/grouping.py sha256=7gaBwSK7dyVM4IiQISOE4N5P0DJVrmEUZ1y4glTbUY4 12325
penzai/nn/layer.py sha256=mEEC7_H6Kr0e0GHL8F0C3zubTpXU3m132kD4dnYilyg 6665
penzai/nn/layer_stack.py sha256=TK-xKMuwUowG7mQuWgIDUM5z33FEyKwoseKsAts-xGM 14419
penzai/nn/linear_and_affine.py sha256=26Ex_TBbBcI9drFN-CUIJ8A9B-Fwl7KSvcNd4rX8X4Q 26236
penzai/nn/parameters.py sha256=S6vjzUVJLqJ43l0DwbZwKi8Tycp4Oczd9i1z7UsDzuM 4503
penzai/nn/standardization.py sha256=PvQSnciJo0bDjxmo1y8UBu7999uRndIkDXb7mOR_dMY 6076
penzai/nn/_treescope_handlers/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
penzai/nn/_treescope_handlers/layer_handler.py sha256=8l0kDj05afp0UyZe466QwsCEw9IRszaUweNPjN1aKp8 2973
penzai/pz/__init__.py sha256=FqKuawVUK0ESr7k-i08JHi3-O4wgxH38r08NoQqzoV8 3036
penzai/pz/nn.py sha256=awxB6Y8GP9veNsVJuEQRn7zvmZV8cUkxpR9PDbjJFfg 2339
penzai/pz/ts.py sha256=1Rdnsx7ocFHbQgkqoz3AOtg7z1bnvOGR2227kSvMMdM 1516
penzai/toolshed/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
penzai/toolshed/auto_nmap.py sha256=KH3bxfB5oy8lIiLHOKFv0SJxUrCVE43RaQE14ar3_UE 3575
penzai/toolshed/basic_training.py sha256=jKJ7xAxH2TYlvJ7yEhGwYEnCIZOhSTjvQhcGb6QGvbE 8427
penzai/toolshed/gradient_checkpointing.py sha256=kU9zhlpKVev5ND8IMQNQxdwZnEKKd0RVmuC6Aknns-o 2490
penzai/toolshed/isolate_submodel.py sha256=8hxbtEpPbSWV_VtvspRi9uaTfDA4ais9jdFAtisxYsw 6616
penzai/toolshed/jit_wrapper.py sha256=ZKmaNTTiBw7sRauvEt_0Vx990OELuNu1xiffq6hpIWY 2024
penzai/toolshed/lora.py sha256=oMBqv94k8_hjaQnlt5pxQHtqm0rdTeMTOse9FXPpuDM 4828
penzai/toolshed/model_rewiring.py sha256=XEerRoXtql9_VmB1YU8lYrfso7J593d8EPLES9Tp-0I 10766
penzai/toolshed/patch_ipdb.py sha256=3WitNX8sNur0xVjn9U0iM3a78s5jtgOUsULajuRjAm8 2461
penzai/toolshed/save_intermediates.py sha256=vqw0tMy_GoyT0vn2fClSBL35AQyjVpgnr-j2h6JTM1U 5558
penzai/toolshed/sharding_util.py sha256=BQk4PHIb15GFMufzzFzk8DfsDmV89-PBRIK46gHI2a4 9579
penzai/toolshed/token_visualization.py sha256=4yo0mjUfEI0kaQlZ2PuH0VaGozuUrjZeGYKOo_bbGKg 6816
penzai/toolshed/unflaxify.py sha256=47AgdYFjEJOkUjB34n808EaGEVVteFpr_EM7OvRCiRo 18610
penzai/treescope/__init__.py sha256=Hqgm1E5A8yMquAwVFlYPGD0G3GDs17_omZmGnSx3H8k 1038
penzai/treescope/_compatibility_setup.py sha256=lOjuHNFdgK4DzEEamAIanowzOb9viZWfB0FaoUgECqI 5852
penzai/treescope/formatting_util.py sha256=sbBjskCYAGfBS9we1bi_Uose84tdEm_IuuDNbmwR1xQ 917
penzai/treescope/repr_lib.py sha256=wrQGw9lVz2nqzZn_EDfB-0tNK48rzMfj-BlysiDcADU 966
penzai-0.2.4.dist-info/LICENSE sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA 11358
penzai-0.2.4.dist-info/WHEEL sha256=CpUCUxeHQbRN5UGRQHYRJorO5Af-Qy_fHMctcQ8DSGI 82
penzai-0.2.4.dist-info/METADATA sha256=lAYpaFrYEoeLr5byNeoHVFIz3ABu948_bZCugLjW_fI 9137
penzai-0.2.4.dist-info/RECORD — —