penzai/__init__.py,sha256=qk-0VFzY293ti-p944VeUjTLoyspv9D_5qQ2v5nhMp8,692
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=bd330wHFKLaIgTX5nOCgOcK6mVKN1bNfl47V9c0misI,80779
penzai/core/partitioning.py,sha256=zYRrWNliZJLg3chhi2ApVMbH9LXcPWXovvqhED_Np7M,5305
penzai/core/random_stream.py,sha256=RXY5BLbe-_CTuug2VMA7yb770B2XpUE0u-qihvVbwJo,1869
penzai/core/selectors.py,sha256=MMvSpDJti72kn5x4BiTtJnc42I2xdcswYsRMaQ5evVY,55632
penzai/core/shapecheck.py,sha256=pE9FVsZBDSMGa41qs0FCXwQjl-Lf4WiCBPDHnzHmLPA,43761
penzai/core/struct.py,sha256=7tKNpfz0RziCv1vpzr1IMhOoSbjyrGLv_k7p9cROs2I,29331
penzai/core/syntactic_sugar.py,sha256=e-z6OF7_wWN37ROwzoeKvXIAnn-yOEl6TPqhyEKhGb8,1117
penzai/core/tree_util.py,sha256=Pqc1Z1eK4kl6EYkgoph7BHG1sSGtjXJjL8OQQ_IUhOU,2384
penzai/core/variables.py,sha256=4QLfn8gBth9IO4uNR_U6OsfxghCYpPWWzLR7AXiotME,28754
penzai/core/_treescope_handlers/__init__.py,sha256=5d1inGwgDwaTVLJ74sdUUK01_LEzHSvhvWQzoDy7aCo,635
penzai/core/_treescope_handlers/named_axes_handlers.py,sha256=H3Bsh1LWpbQUsEga1ENzv_lZFcUQGzrfmJ4mrmCOKRY,8795
penzai/core/_treescope_handlers/selection_rendering.py,sha256=u5l-pWnAL4oO89JHYmQDwWERD7mLWfGHsO8mRENbUHo,11433
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=3s7tpU_zxwnPYYFm9wxmExeKQuYkVRIuZgNx3ZK9kX0,9702
penzai/deprecated/v1/core/random_stream.py,sha256=noa2jHElBx23bUlkvdMDSPUUlwvbL3fAUWAumfszGxs,3782
penzai/deprecated/v1/core/_treescope_handlers/layer_handler.py,sha256=iPVO-UetB8KOSuCHRgdcVE6AiulFb03bcEv_3u9Zq6c,8212
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=UWpmG2X69N-K-X6GyM9DFM_jgbc4kiR8J76Upchm-2M,21356
penzai/deprecated/v1/data_effects/local_state.py,sha256=QGbDeG_gMeKLlaOu2Epz8ZIQbuIh3i1XCKMe7lZU2BE,24527
penzai/deprecated/v1/data_effects/random.py,sha256=oc9bmtfaSncVrbeM--aNBr3nGMX9BU1oKQKL4xzUJQQ,11970
penzai/deprecated/v1/data_effects/side_input.py,sha256=eYLhhPsmDPTvmRP-fnGmjBMkUIwdbcWahwg3O5cuqXo,14575
penzai/deprecated/v1/data_effects/side_output.py,sha256=JHpLYvHljIICLMmENsKswYkD14PMAti-H-qOWzLP3DE,7287
penzai/deprecated/v1/example_models/__init__.py,sha256=pr5uQLqWLt4vn80GFEpQOL3ImqQ2FRvW9I7PVeK9nPM,583
penzai/deprecated/v1/example_models/simple_mlp.py,sha256=LWc_OmUU39yDoEk0uvrLSBKWNudafHhbiMiQupNWMMs,2588
penzai/deprecated/v1/example_models/gemma/__init__.py,sha256=gW5JzWZLSzKR8Yo5rZL22kVFT-_tb3QndBMrIeplkoc,735
penzai/deprecated/v1/example_models/gemma/model_core.py,sha256=AesZ0T0e9K-vXMENVTpPDka2Z03nuzO_zmCTID7und8,23824
penzai/deprecated/v1/example_models/gemma/sampling_mode.py,sha256=jqO191x79-_Y5j-Q1sXPBgp9VJBYLLNkMoXEh6LkCq4,11271
penzai/deprecated/v1/example_models/gemma/simple_decoding_loop.py,sha256=WTekqISevck670GDsQxUB_baBou3SOW_RWog5SRpEGM,9524
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=OEYGxIVWzBWgfcSwxVrFpa7eUC5L30n5nCuwAwkHPXY,3332
penzai/deprecated/v1/nn/combinators.py,sha256=aYzSyDWBdnwv49btoiTixLsKSUfepbEYRE2LHDsttJQ,4031
penzai/deprecated/v1/nn/dropout.py,sha256=YmwaURxYjPgI1EudWQ3xzR2gbZWtDaNEOY6KkD7nrm4,5314
penzai/deprecated/v1/nn/embeddings.py,sha256=FO2CLQvD2UJKAb8MiXlhzIhENp-ycahHft3LhyDz4ks,12913
penzai/deprecated/v1/nn/grouping.py,sha256=xXobUs30cXCXHLqSRqEfnKhndbZG1ELt19NcnE8Dfw8,11873
penzai/deprecated/v1/nn/linear_and_affine.py,sha256=RpPAqBhiKjrIojI4pQrPOdTBR8vRg-0OxGs_a8CbCBQ,27220
penzai/deprecated/v1/nn/parameters.py,sha256=5DqBda43kfIwO6JynMAf4EopvxFdRLPf07XEAhrNPHE,23933
penzai/deprecated/v1/nn/standardization.py,sha256=wEirWISHBmAq1bJi3T8RyyD3p5Wwhm3848sb17JT-X4,6652
penzai/deprecated/v1/pz/__init__.py,sha256=lBw8KD_YR8nrU4xJwu3vMWdFOXJMfExIfhizoP7ik1o,1959
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=xg8qh-Dz290E4Ho7Ae-EagYQ8rNtdgnomRKkmwwDX-A,13557
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=Tm_0Sy6__wvCX3AJKIxg_GZhsU9ZYc-Aup5prpoqFAA,4755
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=EK6bCyi5-HSTIkryyvOENF9eHrIeb_Jr6M199VV-W80,18496
penzai/experimental/__init__.py,sha256=pr5uQLqWLt4vn80GFEpQOL3ImqQ2FRvW9I7PVeK9nPM,583
penzai/experimental/v2/__init__.py,sha256=4lfRf9BYqVZ1nljmlwBDlfRP4KqpfMZr5hLNFNrfpHI,6560
penzai/models/simple_mlp.py,sha256=qaIUDyn828zd8qT4TXFbm0i2pR2aBfxPLamshFRCUu0,2686
penzai/models/transformer/__init__.py,sha256=GwkKB-_D12iLGSLfJ09H_MB7SbTAbiGjV7_O1wfsDR8,745
penzai/models/transformer/model_parts.py,sha256=zi7mzeZ-IpZrtyFRQyKurw4Ed6enCA06_ZZLbVzb2_k,5512
penzai/models/transformer/sampling_mode.py,sha256=cP_lmwAIF7nLzFoJA-HQ_AMitzwUnt1xQ2GNKRAtYhc,8623
penzai/models/transformer/simple_decoding_loop.py,sha256=OTs4LksD974BqHh4mvH8YvuAFaio1lPo0Qssoab_8Is,3238
penzai/models/transformer/variants/__init__.py,sha256=zJblKGjiGAFgHmEygNWonIff_IV--GCbUB341pGXz58,738
penzai/models/transformer/variants/gemma.py,sha256=Gj1HIHgMpelh1Fe5cdZ1XaIUzODgqsTjXnYpgpiA7HI,7014
penzai/models/transformer/variants/gpt_neox.py,sha256=oEqRmwWndyLHcrWc00DHdzwHz_tv-0IXW-5PVLF57PE,19227
penzai/models/transformer/variants/llama.py,sha256=4plSEgyOBN4_V_5pTps53klwblwqrhRFKywVKhUhNbU,2707
penzai/models/transformer/variants/llamalike_common.py,sha256=6jwd9SRaY_KjzlJ1nGPyOwYvS5QDOLj3x9VJLj45Bvs,24915
penzai/models/transformer/variants/mistral.py,sha256=VRy3bNvT9f1KaI_tVCT6wD_Bhq_uSbj_fTdjriSe554,2962
penzai/nn/attention.py,sha256=9YREuZ8DSe_etVdF3RtRa2V-ir6TkxuqN77EkEE49Os,16779
penzai/nn/basic_ops.py,sha256=25-kUfd6rrmxG8DE-cfsv04dKqmYCJY2qiYINBEZrD0,2683
penzai/nn/combinators.py,sha256=xgQYpgWUK4rv5YACZZYpQFqxgdGyOQ0wa2svNeep7A4,4187
penzai/nn/dropout.py,sha256=HxfwVySMVMqvQ0TKbxxEQLpzVF_9DN6n4NbMSSsf_yw,5672
penzai/nn/embeddings.py,sha256=bEVk2Qmozzg8Wgnlpa0zgJGPl6HXeYAtq1p4zV2ItWk,12653
penzai/nn/grouping.py,sha256=SLE57i8KhZotTlXZ9jYJO34JWNHKF2m-Bc_9SFIfa0A,12319
penzai/nn/layer.py,sha256=XeeNM_IXqIIZ-_oe9uRQMxZSeCBJLDtvbzWf85KezsU,6661
penzai/nn/layer_stack.py,sha256=9hUqIY13F-v4bvdOEdUFXvzQHtfU7mUaZqn5H2w8WxE,14353
penzai/nn/linear_and_affine.py,sha256=OXEfQ7DFZEysnTcoYedOX1BpTddDvnWqKVR3Wyzw69A,25629
penzai/nn/parameters.py,sha256=S6vjzUVJLqJ43l0DwbZwKi8Tycp4Oczd9i1z7UsDzuM,4503
penzai/nn/standardization.py,sha256=O1YHT-nsrw-zS2iWh__jn5fsG_eib1G871BcktRBqyE,6087
penzai/nn/_treescope_handlers/layer_handler.py,sha256=_pscaWzV_nZymQ8WWAd5Oriig5E1Niwc7LT6aFUcV9Q,2905
penzai/pz/__init__.py,sha256=Y0voGEZZoMA-tONfkw5HIXFQWM3W40Blnsp36LnfHbY,2422
penzai/pz/nn.py,sha256=QhfZBHK6kYbgqlcaOEOnxe6fgNbIPvLDFV4t_YT3jqE,2322
penzai/pz/ts.py,sha256=1Rdnsx7ocFHbQgkqoz3AOtg7z1bnvOGR2227kSvMMdM,1516
penzai/toolshed/auto_nmap.py,sha256=KH3bxfB5oy8lIiLHOKFv0SJxUrCVE43RaQE14ar3_UE,3575
penzai/toolshed/basic_training.py,sha256=ssmJv5xUPSq09pKbG1GhKXUijwZt4y0Z3D0vfE5t6pA,8395
penzai/toolshed/gradient_checkpointing.py,sha256=kU9zhlpKVev5ND8IMQNQxdwZnEKKd0RVmuC6Aknns-o,2490
penzai/toolshed/isolate_submodel.py,sha256=SG9dzE4kGAAbAJysu2Ak9W7Keb0NPz4dUvBbxZ_sT60,6623
penzai/toolshed/jit_wrapper.py,sha256=ZKmaNTTiBw7sRauvEt_0Vx990OELuNu1xiffq6hpIWY,2024
penzai/toolshed/lora.py,sha256=Daz54dVhNptPZnJVGXVhA2muogE9ZLk89wv--5Yvd84,4757
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=tvJp_IptJ5dQj3i9vyV4cr4BqicCSCyRXKFlbtchqFA,6840
penzai/toolshed/unflaxify.py,sha256=6k_X2tNtI8AttNT7kCKrd_LCNyPdeSb37QUFAl8vxyI,18488
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.0.dist-info/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
penzai-0.2.0.dist-info/WHEEL,sha256=EZbGkh7Ie4PoZfRQ8I0ZuP9VklN_TvcZ6DSE5Uar4z4,81
penzai-0.2.0.dist-info/METADATA,sha256=l1vYVpaoy2_yUAibdJ_DVkz0sNRx2Zcya6l3lTKjZfU,8721
penzai-0.2.0.dist-info/RECORD,,
