docs/conf.py,sha256=BNPFzpHmzQmyG1kJopxJyZbbR4Z_oQKvRe_YXPcooXk,4781
examples/denoising_diffusion/data_loader.py,sha256=Ec0gBtcT_YmQ55wdPZ-YdOgzclheTA7l38WtAuatvwE,2818
examples/denoising_diffusion/model.py,sha256=RgSeCzXRaXRMmVVA6FRQJs2KCENbcEtXu2-q7x3mfDk,14544
examples/denoising_diffusion/train.py,sha256=kmTUFWKeueAkzNJrpKlTJAcD5Dh-Vb_jFhMMZMw4Jo4,2665
examples/notebooks/pretrained_resnet18.py,sha256=T9N2jqdrOWks4dlsMOZavgK_Q0P4iIL6m9rqqQULzjA,4029
examples/transformer/model.py,sha256=QB-XbOMtR-ZU9z-G6p8nhrWpr4HrbNJpyuzrhIvxgCA,2443
examples/transformer/train.py,sha256=jxZnfUNHad5IvqrfKtrzon-BkQ7KaHgwvKNsj9oiUOw,5947
examples/wave_gru/data_loader.py,sha256=G_JaD6BJs_sROGZzoF3p6SphjxwYzRnZT5PGB3Mo0jM,1501
examples/wave_gru/model.py,sha256=Ya-bl7Fkv8--eTzMmQehQUNUmMZLZKm933hIxMg4M_0,3722
examples/wave_gru/train.py,sha256=kloeGGqRCN1CaFbas2IclEtwR3O2HiuCjXOew3iIgCA,2994
pax/__init__.py,sha256=c1mbSulA2CCa0VpvwGT4jJpmiLzXpfwW8d2V2y___sU,1269
pax/initializers.py,sha256=gBcGEDbQZXoWp8SvaSUvrK2gZVGoGnHfqnSFYbeb0FA,167
pax/nets.py,sha256=92XvYYzRBBVCMPtFIVm0cAYLmQQk78tsixD4hZZjDzg,151
pax/nn.py,sha256=NgS422K7StwFcR0nFBTtqVlgNkj3V3scpxwI5vceXVw,380
pax/utils.py,sha256=lC6mCQZWqSN2dVKgMKKi2nJghOxhCsH8jHe4I1f-GF4,96
pax/_src/__init__.py,sha256=0kvAr_HqBQjhKPcCxyAxyBh3E9tkyyvENKENQmjWAlk,23
pax/_src/initializers.py,sha256=TUk3HUPG__hSONvs-F3591bKfxLEJ5I60QALLpaHRIU,4940
pax/_src/utils.py,sha256=PMPrVKfMjK8OnZiq8e_5Oxxm6-6LDcmLpfko6BWPTew,4977
pax/_src/core/__init__.py,sha256=GkF65CSBY5-ynWIEVjn1av7DjniqUC0Dz4JUQS2QCVY,459
pax/_src/core/base.py,sha256=SZovA9uBs2kr75bvJ6h4cgKuERsKZ8x8M-7p0S3BUBw,16446
pax/_src/core/flatten_module.py,sha256=kQr7ZiKmNZa293JCQUi60qmvu3dQiflPHzArgVDbt2I,2644
pax/_src/core/mixed_precision.py,sha256=8EuprQ_TAUOMFBrL5_uz8nN5RgxhLMUfB-22Rbs4nSM,4675
pax/_src/core/module.py,sha256=fVnXaavUVAlaij6QvphCBTNngJ-HDZKSyd76DSaz3ak,2695
pax/_src/core/module_and_value.py,sha256=fMP0GaHTH6dXTR_7jxYkt5pM3-tNzbm0vPnKtU5ESEg,1774
pax/_src/core/pure.py,sha256=v86fbdpgAKhbtBBsI6QfcS_iFOOBfYoiYfWSGBLmZf4,3510
pax/_src/core/rng.py,sha256=J379sMVKa0NiNKqfMPrr9Ew5TBn4vYykI9oWp_kQwwM,154
pax/_src/core/threading_local.py,sha256=7KJeBM3-LFL8KrTyZak11Wwle55buEQuzMF8SDanOxU,4319
pax/_src/core/transforms.py,sha256=-Bj6XLo7BnBxHzj4cxMUHQpKxA_EzK9kwbKn7sV6pNM,3605
pax/_src/core/utils.py,sha256=-HYXam0GYKdQC0T1XwGOKg2EMYSCEiohYt--ZRp-zE4,1453
pax/_src/nets/__init__.py,sha256=rWFULImc7MHKdcFcxefONVDr41_c2QUtqBBwVPAimlQ,180
pax/_src/nets/resnet.py,sha256=zrcPA_dhkE_z98nV7zpiu4EjgdKso5-wJRMZAy209BU,11099
pax/_src/nets/transformer.py,sha256=Ne0KlRE26VvDZzidiAq-8t7OTRBi6AW4kD3mSUhXGUE,3922
pax/_src/nn/__init__.py,sha256=u2ORikXXT5yBbygJVG-hn4Fb0qbi3uqMUc4I1DAzZRs,563
pax/_src/nn/attention.py,sha256=--oodYXF7lKf804JWCS0KQxln2vYlShbF92Zbx_2FyM,2876
pax/_src/nn/batch_norm.py,sha256=dMvO5kVgjQI-Jc5Co9ExoTKQ0oMKyKMBG0djaTDiO3w,5834
pax/_src/nn/conv.py,sha256=ZmYgSG7xpHaqTwPZ3-ypLavSKbPpH4Fz9kYv07kr6CA,18113
pax/_src/nn/dropout.py,sha256=jOEYmfdxn_VMNCgSXMnlPq3FUw5xd5WdMx6eMBuVxcg,1612
pax/_src/nn/ema.py,sha256=SdpkTuM9q2N2emsI5KVLQE4nxIqhe4fs68DVDV7Zb_o,1582
pax/_src/nn/embed.py,sha256=WKRvkA_PI2QlckbVoSyUFUtrNbx6M8BkipVxbnZg_4o,1526
pax/_src/nn/group_norm.py,sha256=M4wAFMsQzlYabljK6l5aPKrvuBkDvwWP9rN5VFpepVQ,9499
pax/_src/nn/identity.py,sha256=_OxHNoXNZ42Zn1kGLn45AYegSLXjtCrUpd-Rn2AJqhg,185
pax/_src/nn/lambda_module.py,sha256=FLbOaJkngHDLnfW8Sh7hfuCPA2lcW_Cyu4Qv72_KKRw,901
pax/_src/nn/layer_norm.py,sha256=_KUp_DlsQ6cdHuvKRaH5slq7Jsh6lujwovcn7O2ChZw,6095
pax/_src/nn/linear.py,sha256=41QHrxcH6UN9EOBbNXCmrVFkY60YJE2urXbGSLWibiA,2285
pax/_src/nn/pool.py,sha256=WAhdRh1_ektlLnD5VveHpqNj8i0PesYNy1GWbXk8CmE,5391
pax/_src/nn/recurrent.py,sha256=d79G7ItwQTXBcYNkT_qZX3pt_1gwFY9nXAam80mJtBo,4620
pax/_src/nn/rng_seq.py,sha256=55aG_tm4qI492eUv45eU0AXCzI35lzMkx_Lp8xn0b10,1768
pax/_src/nn/sequential.py,sha256=nGNIb36BHT0Ax0BQy88RlCV1GE4D3Zr8yQ433EJop5I,1478
pax3-0.4.1.dev0.dist-info/LICENSE,sha256=ekSaJpMHAKwDB64Gzs2G09xoMALKR1A4Zt6mni8mnZc,1072
pax3-0.4.1.dev0.dist-info/METADATA,sha256=KwcC-mIzaufyANnWkc5e-_ty5xvGe-c9hSPPOs4Yaco,9124
pax3-0.4.1.dev0.dist-info/WHEEL,sha256=ewwEueio1C2XeHTvT17n8dZUJgOvyCWCt0WVNLClP9o,92
pax3-0.4.1.dev0.dist-info/top_level.txt,sha256=stI1Zi37HS1S9Ka2KfCrzhDfnt80p8Bn4d6jIHzvptM,18
pax3-0.4.1.dev0.dist-info/RECORD,,
