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,,
