docs/__init__.py,sha256=4S3pYNuiw6yzPmNNm2iRi40_Jf9xlHj-oIRb9Gp4sxk,581
docs/array_refs.py,sha256=uYB8ObIteht9XolmMSRotHAfAnIGL0ZzNM16hC-rLrA,11857
docs/autodidax.py,sha256=toEmRLy-4UR0IUPVLT0mE_n4HqUYTrg7P9vk1bK7bFk,106360
docs/autodidax2_part1.py,sha256=Z27y_4D2aWjihUt0inEJRYwYO7Tu3QMQgMgagDjUcZc,19695
docs/conf.py,sha256=3hH10o0U2wBJREaMrLsgJe5t8UsHUM0cDQpYXeh_-jU,13255
docs/the-training-cookbook.py,sha256=Xknx8PDi4_969Acaosw4JEXDkbQgLUm2CGshef9msuw,8979
jax/__init__.py,sha256=OUgpnlKLM6N99R47uxgig3o4bhzF08bZ8PnijFdBnQo,9401
jax/ad_checkpoint.py,sha256=NXatsd0qUy0fw2VgkmGsyEWAGHCj04EOr59Xy4YZEWQ,1548
jax/api_util.py,sha256=avYenJ6674XMZHYtWzplyMPO0j2NXOk13BYb1o3vnCc,979
jax/cloud_tpu_init.py,sha256=Q12OEXt9BfKAPY2o7bbuDnVMz1QMHvzkxXODZ8-tSG4,1379
jax/collect_profile.py,sha256=Q5CuPjFBUCdZ7W7aSmYwG6BYyBOaF-tboex9ZZho2SU,6148
jax/core.py,sha256=PxonetroJreOIH4JoIBnV0JJ5ot6L9o1q5-Y_nTbZDg,4371
jax/custom_batching.py,sha256=MEoypZfGgYqvZrBKtnvL5sQjl8YrjVm11hiFWoTCtV8,691
jax/custom_derivatives.py,sha256=5ZlbQG085r56qJZkGi1tNxeQYTYufJvJYsGyN6N4Rks,1345
jax/custom_transpose.py,sha256=vGj3GjRJQpxl-MhvYztIedauNBiWMj9nD_aJHQiNaYc,664
jax/debug.py,sha256=-rWgSDyzFV9hHg6tCAC3b5aVjxfnOzDAz77gUNYmhJo,1149
jax/distributed.py,sha256=CP5IDXwA4QzxXilcTCM1F6435U3dYsdWkDzO4hr_evE,710
jax/dlpack.py,sha256=lPvnhizhMsUdKJzvDl4fUUPfoU_vzCNvoX3Xl3z3Bow,688
jax/dtypes.py,sha256=HRwqpSsU_JD5ZJCjZ8u3MFUzf7JiRXBTaBRedpcJXVM,1224
jax/errors.py,sha256=FyhOBvQW_mPV72z0FXLeqvZ9PPWFfeDYtoxWDVOJv2o,1314
jax/export.py,sha256=OzLiZq2zwjC4gDUs-3VcHVFF5IUZtIhMaaTdnV89qIA,1537
jax/ffi.py,sha256=ooQYELAXRz1D7RiwDhUhFYUqfwl3Pr-TV5Gw1KMGhAI,1160
jax/flatten_util.py,sha256=jGwQUB80s2jr78XuEaT1_OP2OUyVm8Pk-yAQq-C1SsY,645
jax/memory.py,sha256=uONVdGEZa7XodzKxzpCjwZ17cX3a1eRcrGnltSq7w40,625
jax/monitoring.py,sha256=ik0PsR4rbn1NswE4IJ-i6bpgmV4ueXuaLfAlagUMdNI,1695
jax/profiler.py,sha256=h7sMKCFMzMPSVlnmbXSj7w6bDt7Rf90S5U1x3WkMPCE,1276
jax/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
jax/random.py,sha256=W3juZxdpY4y0e3IwzDLdtqZEDHeW1DeCc0tsVUCI_6U,9433
jax/ref.py,sha256=Aym2Pm3RWG3CZjJADcyeM2VUcdTOPMnBqK9YdT9XBC0,1340
jax/sharding.py,sha256=B5cxsDZdoaoyPhO5FO19OP3HC-3Tb-_nPDOOsmPw92A,1909
jax/stages.py,sha256=KOQqCPRCnLKryQUrTRPVRiNuZqyV4chGi_oY2t3HpQ4,1286
jax/test_util.py,sha256=DVJAPxzZMKdM01Nhx7s2kPe5mcdv8ZL8a_37sgyZPUs,835
jax/tree.py,sha256=UMACxhh3xfcgrO6h9QE00cqAl4gMzyRkweFfBxPnlAQ,1152
jax/tree_util.py,sha256=Ko9em477wzRwOf0uXExrXJqTjC7msCy_RXvJlkYWs4I,3189
jax/typing.py,sha256=zijMsLWzFOqn-u8__DbGxxI7MyjePxpUwU5nFE_P78Y,3319
jax/version.py,sha256=fC-Y3wMpkjveJ8wOLYkNt233TkMBhZ3BkNkwm38sopg,6725
jax/_src/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/abstract_arrays.py,sha256=CRUiqcjfYqkhpH3PaCcU3RpXYdsBvZ6LnQZT8R3lItM,6166
jax/_src/ad_checkpoint.py,sha256=d51F8KJGjVUR4Hb49iYuJ6Wx4kk8mGIAy5BOruT80dg,41825
jax/_src/ad_util.py,sha256=2DORD9j3ol8FSsfTXuHHHlEfcvbree8er8W5LYRwfgM,4589
jax/_src/api.py,sha256=Wyh0TiS8Tb1_71K0Y8AhGMcNfJUoqIoq778rwOv7g2Q,136539
jax/_src/api_util.py,sha256=p3FvzyC1RHXy3axEhuTyY9HSwwtK9HfoJ0amDx-tK3w,30043
jax/_src/array.py,sha256=coV6tQM2e1w9bYuED7lxQz0v0W-rtZq3gki9kZ5teVc,52300
jax/_src/basearray.py,sha256=qX-m0BJHWhT4sYSK7YdSiE0UwSkuilQiqa9a98BjEJk,7422
jax/_src/basearray.pyi,sha256=EwBvM8g6xrNI46_hlX5e0NSK3huFZoNxt40H_gifpLw,14683
jax/_src/blocked_sampler.py,sha256=ZM9AAU8BUUji9XrrmGRPe4os4-9GXEcFGzRkSoUdU6Q,5880
jax/_src/buffer_callback.py,sha256=64T8iJ5KO4nfPPEmgY6cFLloIWPRIWy6GSe30XW_Zjc,10463
jax/_src/cache_key.py,sha256=Ndff_zDPcdrWCrbhBef1bKac8nlCc8_7jEtxefMnnGA,12895
jax/_src/callback.py,sha256=XF-cdyKUmAym7zKqlLS_nnJHXFsOo3IdDoecqGhj-B4,36186
jax/_src/checkify.py,sha256=zWHiZMv7BklwNMQQnPyvRvFl9dg_AI46hYbrrI8v_RU,58041
jax/_src/cloud_tpu_init.py,sha256=KIsKxtuqSCH-32l1FQ-4DblQSsMOPK-T1SQ81rPnvtU,5141
jax/_src/compilation_cache.py,sha256=BdXwMLQckPuHrpokGER9ti4z0jcJIDoYkOHgJZik1RY,12284
jax/_src/compilation_cache_interface.py,sha256=ONHvsIC3Kzoda0TcQWnhhjxtOWYixjkO9Or__BAW3_0,865
jax/_src/compiler.py,sha256=MjgjQAYMUoDA86VUZZiMWMd3r5C6EmoFbv179GaYlRA,32108
jax/_src/compute_on.py,sha256=wrPfM4YKEFzKk03yH5xZGdlorkCY0BY1D2qepOqDH_U,11265
jax/_src/config.py,sha256=YQtePSYT6o9fIS-Bi1rzwK2-N24mWviKHQV4l6YvqQQ,82238
jax/_src/core.py,sha256=xDgTHBPNmzh_3UTUMqaRsDT-z1CpbRsAYYl9v-hrKmo,142036
jax/_src/custom_api_util.py,sha256=uv6uyrXU1BHGzKuAe498W2PD9bcWt1PEa-8UAqr9Zaw,876
jax/_src/custom_batching.py,sha256=9FwMf5BPVsiWTMy3Wk9j7gCwhnJwhOAKYq1aJlwNARw,15277
jax/_src/custom_dce.py,sha256=UHUTZ9UZM4EyeTBrHDw3xp7RgPqeeoSb5d1lRLKIoUY,17470
jax/_src/custom_derivatives.py,sha256=G02dKOtb5fzuNJdugDun6Ac8SmyJB6KpOE6SV0ug42o,80691
jax/_src/custom_partitioning.py,sha256=B1JokmbCkvVH3qLsBSS9ZHn3N5r429UWIlWV_RoLMoM,28452
jax/_src/custom_partitioning_sharding_rule.py,sha256=uzVA5zwJH5XH3WJXVMuAkK07hWiXMtjRwsDjGCUbwfE,21207
jax/_src/custom_transpose.py,sha256=6-lfrre7KJKZ7BrXUl6QKR-_b3QZ4Gsuf_OwwGjsLuE,9204
jax/_src/debugging.py,sha256=yga9Ovs--KSZxXVgzH1qbHnVP98FbEZ7hGE0mPDtKRQ,34927
jax/_src/deprecations.py,sha256=eu61OgOFcIvKasgby9Qm87yjWtHMpEyJKCzzyMjzspQ,4605
jax/_src/dispatch.py,sha256=g5382eTfQb50Rq2Vtxy-lRfAtvfQmnVFNwDfS6MSjIs,29149
jax/_src/distributed.py,sha256=sSN5SEdoc-HV79Kpfoha4ltlJLz8uEs3PtD0gfKpllc,15657
jax/_src/dlpack.py,sha256=nx-ECIv5fiSe5cxWFymWD3qMmLiUTFiRJlzQnJEweVc,11414
jax/_src/dtypes.py,sha256=vdSTu-GRp3Mewz4K_lTPxJvEg0Nhcl7U7m4VabdYF3M,40469
jax/_src/earray.py,sha256=WS8XoNTCP6zQTC5DahYR0S6g_V9YHj3wKW3L9B8RIhU,4405
jax/_src/effects.py,sha256=qQCTznjR8_76HVCp_MAAKKXbs1HvIspt1Z3w3UYS5CA,5041
jax/_src/environment_info.py,sha256=Xc9-Y5_DvVjW3iEhen51ATJvIESluEIrWJIF_e-oqA8,2031
jax/_src/error_check.py,sha256=0nfYoRjEuIgFK5zSXm9Ocxh9ntHDwJ1WI00e_tnzViI,12960
jax/_src/errors.py,sha256=VXa9nNqjMBlt1elh_XywuqFRKST8zbfdtw_rDUy3XEE,24777
jax/_src/ffi.py,sha256=hfJf0qCVx6Iuf3SsMda0VXh0wI1BtsHJH5K_8ebJ_WE,30240
jax/_src/flatten_util.py,sha256=sG6qaC_yWTcNNbVjc1A1KrwVz3w2q0b1nm7SQGFmKBU,3971
jax/_src/frozen_dict.py,sha256=V9UhNofzdhy7pTM16OOf7i1_fIfmRqLAUH-GSWcU8R4,1457
jax/_src/hardware_utils.py,sha256=YQcgkyPgQN94grS6VvBIVVFL2R-LDCsdY8Tqj43jnVA,2349
jax/_src/hashable_array.py,sha256=4ff9eVDYR7DFK3l9nNF0btGecguXonn1AzeWviiJQcQ,1036
jax/_src/hijax.py,sha256=Czv331EfcWicsKP6NnfPHsY-kCRrXl-2PLXamj1wVnE,30503
jax/_src/jaxpr_util.py,sha256=aDZuCTqG23Bpvz5hKiSSfquuLn9_wOwIuLYpBnUrE8U,10572
jax/_src/lax_reference.py,sha256=GdcrRoijNaFZujR4l2hRDRp7QTRYc8JO5M_nY-Qpb2M,19847
jax/_src/layout.py,sha256=rNJTOUmy2b-dBxRr4ofx2cHGb0jmxmC7swkRzmenmMw,5797
jax/_src/lazy_loader.py,sha256=5rbYHfu4Lfu6cRZ92c1dAffoLMnk5aoDXjZ-zUjICmc,1843
jax/_src/linear_util.py,sha256=niNg2OolghIh_iFZ_JnZomLyGOVrqf-N6-pDaFgHxEE,18346
jax/_src/literals.py,sha256=_t8O6FIrM9fvxi4mQBmCaPfRJnnJWcUdNwM5fNgetdM,6667
jax/_src/logging_config.py,sha256=7pmMRwe82KfRm9lo-2UQmsZ4oCV8UsW0_vGUjVO60bg,4195
jax/_src/lru_cache.py,sha256=NejEgo8BmO7xy8uORQrAHsN0HpCnY8qsEUnuEhDCkPs,6741
jax/_src/memory.py,sha256=xku_pvnyIR2XpG3brybs0u1MjQ75xRh97AlXgf2w-ro,745
jax/_src/mesh.py,sha256=rw-IVAkPH75LkYJ4PYGtgJbesP2d8DCobMEEhgtczLo,21153
jax/_src/mesh_utils.py,sha256=AhV8qDbTBqxWDBXyoYTK9kkNPRfSQmnfuHSa7VFM9hE,34427
jax/_src/monitoring.py,sha256=OcXRp8maRJewgZDoKLE8D-9tHZnjINXA_m35ewm9uPY,5631
jax/_src/named_sharding.py,sha256=_h1cQtOHEGqhyy1Z4YoNOEj2JbJMhU6pph_oV0HWinw,20959
jax/_src/op_shardings.py,sha256=VvE1zwvbCBY5EyrKUygzCy_AiJCPz8pW3anx-i9qTHY,4388
jax/_src/partition_spec.py,sha256=Zpsp8cX6Ah1MPy5iTqTWq1VMmtrkumczwxpz-_w6E_E,6639
jax/_src/path.py,sha256=d9R4bFgk3cHS37ZUO8up_dHW10Vz3oR2NZTaRM28gdc,2162
jax/_src/pickle_util.py,sha256=gE8IrNWtg4hfu84T5rQ2DVAdpozh5xwSzAlRi7UIWcQ,2312
jax/_src/pjit.py,sha256=DtFDJgHBBdRJjGG-jYvOtNGDYUTHzVhDXTUFrIAU1nI,109031
jax/_src/pmap.py,sha256=6yMA6DC43SvZjDGnTE2RkSRPS1novCkX9WhHkigLc98,5116
jax/_src/pretty_printer.py,sha256=_UZ4oTCAFc55iKvstsMhquEW4AmuhfimYHXYchMnM50,3419
jax/_src/prng.py,sha256=r3K-HpgbT7euMzo3byosezbyr_MCZtherVg29Z82JU4,47161
jax/_src/profiler.py,sha256=k5ZYc-CLSyTd4F2r62qvs4AwGx-QRvdhW3ciCWDzWxM,17446
jax/_src/public_test_util.py,sha256=QTK_AirWntJ8cj_grBi6yqmA7zUFhf8rQAO3caBBTVg,13073
jax/_src/random.py,sha256=xj2HpU51MF2qh85WuXLEjGRdFalK4oCZ5vrvGpjlWFI,115596
jax/_src/ref.py,sha256=G_7qltcYY4hmq2J-a9IxnuspKOYcJc76bUqtxsDAkNs,1194
jax/_src/shard_alike.py,sha256=vY6VYE9jSN1ZeCyxG7ojmTZerG6ujY6ouvcocW7806I,4001
jax/_src/shard_map.py,sha256=m5zniya5azzirb3qEQwcF8sMH9rZpn-ARNRFJUup4GA,89333
jax/_src/sharding.py,sha256=XmIUQds44Y_BY-g-UbFM9ldUCgpl8Ve2xtl1WsBLJzI,8300
jax/_src/sharding_impls.py,sha256=ZQPtfVB9VF-rndNKFA2VFUkEM_11cB3CM_BulUh4ceE,48350
jax/_src/sharding_specs.py,sha256=ww54m5UDD0e6JBD7v33f8zItDggmp3-xTL9QuBNqWwo,8321
jax/_src/source_info_util.py,sha256=r_o6Xz4hvgZQDQNxKU7LcETmRV8ryyTckLNmcOl8ahM,10455
jax/_src/sourcemap.py,sha256=5s4zTThhX6PZwgQ4v4UbYL4norg4uHipqgUGdHBn76Y,6640
jax/_src/stages.py,sha256=83Q81HRuVGlVREdrA_zFwodGqDmgQEMONPLfrWYFUxk,39662
jax/_src/stateful_rng.py,sha256=eGpvQqMc5jJJKpM2kikeyjrlj20_KzPEPlaNHBo_RaM,10530
jax/_src/test_loader.py,sha256=wKRdyNQhSd56d6Ws6iCgDXNkNrAJCUVkIcBKKHyF20U,7621
jax/_src/test_multiprocess.py,sha256=kDUK7pLkcQMxYIbHDP8uVDKZb74jFYzj3Y2lUbsiPlc,15745
jax/_src/test_util.py,sha256=S5FwkFK9YKuJ6W7Y5o_0V_NvmEVCReOF8-Baopw2O3E,86542
jax/_src/test_warning_util.py,sha256=9CvuFdeyGcmTH5xfz3Wqs0qOUGpOVcOwh4SnJsp6FbE,4074
jax/_src/tpu_custom_call.py,sha256=S4oxDm3y_kGqQmd0NXSo18taQROMSKb_-k9p2KYF6lw,30432
jax/_src/traceback_util.py,sha256=IOLKpsQTsSv-NIyb22FZvGQWINXd41w7ElhBPhbORYc,9021
jax/_src/tree.py,sha256=jnVN52WvAbY2pCYuZ1JArn1sutuY8I_Dhj11NuIQdRI,15379
jax/_src/tree_util.py,sha256=kVHkBwHp4wgDfe-DEaL9NN7SPJIVuZmyPjRmM2knV5Y,58574
jax/_src/typing.py,sha256=jdaddrjvthRKzrF6USr77IS2CyOTb-3nj0X4JDq_DZY,3698
jax/_src/util.py,sha256=q8jkDMk97-7Df4kgGR4O8t8o2t4UTtd94-Xq08ZqmMs,27517
jax/_src/xla_bridge.py,sha256=VKKq8bRruK5U3zVBMPscDHfawItpWataksiE4Rqzuyg,45613
jax/_src/xla_metadata.py,sha256=ksuLeoH-TVxQl36uK8_RP9vUbAoS0N7P_55gRJlvQCk,4865
jax/_src/xla_metadata_lib.py,sha256=gyRBCCpMfUG-o51PZQ9dO_rafReE7Py_yZOs6cHSY_Q,1474
jax/_src/clusters/__init__.py,sha256=blIgNT92mYE9Bst7obP5_xhZ2fE2hKqwxH6R0cAQZ94,1283
jax/_src/clusters/cloud_tpu_cluster.py,sha256=lz2sXkgXUZXDdeDlUIZAfTsD5-fwReI6Z7ggnpQnSNA,9233
jax/_src/clusters/cluster.py,sha256=ZFG_OxHn5VjEKcnGhguM_oo6X7n4ijE9IjL5cUaaKb8,6017
jax/_src/clusters/k8s_cluster.py,sha256=XO5iiQ2L8s6zFkwtscffRVKl1geL7Dgly_nSwdpY1dk,9365
jax/_src/clusters/mpi4py_cluster.py,sha256=ZC4O9Wyktd4GobCjmN9AFS-W3XeOFFotghBHnD28qEQ,2760
jax/_src/clusters/ompi_cluster.py,sha256=57kq-MvL70Dt-C5WYgTeZ1Ivlg6wTEzWkD96jIfyk3s,2470
jax/_src/clusters/slurm_cluster.py,sha256=OI-OiVtZdO0u0-3ANX0NeP5mj_-gLWYDGaytsDYlV6c,2555
jax/_src/cudnn/__init__.py,sha256=03MR0G6UMqHXi-JBeyj6tipV6ioP9a-iSN1mW-8Ws5A,631
jax/_src/cudnn/fused_attention_stablehlo.py,sha256=ZRTl_j-38ogFO1tRANX9fomunyzUik1XSq5X9AGjJs4,81797
jax/_src/cudnn/fusion.py,sha256=0FF_DACwZ2HRuRUQA9LdAPgYKFADMhItn8AHed-jmZs,2866
jax/_src/cudnn/scaled_matmul_stablehlo.py,sha256=ETQh0yp9OnqIJFF1k0iAIRFUtaaKUb8g-Ar72ijbfK8,25882
jax/_src/debugger/__init__.py,sha256=iZU1X5nkpSBK5pwLSuWhZsPZgtLUovBF3tyWPmQoh1Q,899
jax/_src/debugger/cli_debugger.py,sha256=bzn-5Rd4C9WwGsVret8gbFNaLpXAf0SzvgltXNuZrcU,4778
jax/_src/debugger/colab_debugger.py,sha256=z49K89L1bTbGN-kRyCr1ElOhTuSxgpoEBpEI0L8YSyQ,7854
jax/_src/debugger/colab_lib.py,sha256=jCbwj2Q4f8SEo7CBr6AeiUtzlsIv4BfKa5CVbq4Bn2U,4298
jax/_src/debugger/core.py,sha256=sRN6dv-Vhuz6j5eenDq8s1ZTUbxYiOv-JxDWx9519ew,8682
jax/_src/debugger/web_debugger.py,sha256=Or-gJL40oBUfClu8mg1YUBoKA9cQytZgJAeGGrPPK_w,3433
jax/_src/export/__init__.py,sha256=mS3YSYUZFIG68T5Zo9CFGkT7wCyWRycnWV4L36CnHYU,581
jax/_src/export/_export.py,sha256=wcoRnXtJ1HqRdvzTkaU0k_bl4e5e10d-CQ5Lfp8lH98,78169
jax/_src/export/serialization.py,sha256=VVbS2zVYwyu2vZ2pHuUSgKtR6TZsY8ZV3FUmzFfdNyM,29303
jax/_src/export/serialization_generated.py,sha256=DS-CjaLEN_r5PVtH60Q2oNLD9pfAXEdix-UFy-m54eA,40489
jax/_src/export/shape_poly.py,sha256=JCrDDNdOGam_xPxB3Es36Y77rcHKKwDx87s_eBiWC2E,86151
jax/_src/export/shape_poly_decision.py,sha256=x24hHa1CGTskS83TMeo7xGILKsrgufYsGReiKLQAJtc,20476
jax/_src/extend/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/_src/extend/random.py,sha256=Rc68o6EA7zPLY3nK-cyF8Tq0WPdvnk81rg9sLb22QRg,1269
jax/_src/image/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/image/scale.py,sha256=5ZBF-5vRHhgLbkHA-0uhAr62HPLwVTAWW0Ia3k_YgzE,13841
jax/_src/interpreters/__init__.py,sha256=hnuhN7h1jY5__Kt2rBvTrk0o8nX4fQl244fDCIMh0qk,690
jax/_src/interpreters/ad.py,sha256=tHatheHfPtDRJO_iEgSyps53g61JaiyBeO10Tk78A0Q,62588
jax/_src/interpreters/batching.py,sha256=etWxUPVKo75Ql8obwzylJaNoYuAlTEHr0DRGCYGUi_s,35720
jax/_src/interpreters/mlir.py,sha256=hNhQMG17dHang10GNldrocmbR4lOB0GuWgt8wkXHe9I,138388
jax/_src/interpreters/partial_eval.py,sha256=blndQZ-f2JnXT7wC5cxma4QjUMq-DXNAeg1hzEgWbOI,113276
jax/_src/interpreters/pxla.py,sha256=Jvgg6F2VBoObqfu9sSAS8Qf3fpWeUXZDMzxrSwgFjlI,141240
jax/_src/lax/__init__.py,sha256=JIUypgx2zBPFvR_TV6qq3yXrs30Ns7UZNkoPWeBJFyg,8406
jax/_src/lax/ann.py,sha256=kZBr1f99HjbpzX58JCxwWhGTHFOBZSxyQx6FnjH4OPI,17043
jax/_src/lax/convolution.py,sha256=dK7VWwUeIZ8sZU5A1oW38pS9PYgyolub1Vw8yFvNP1w,49694
jax/_src/lax/fft.py,sha256=so4u_i_yimHveOuclhXi6uIHGn3oD_sRa4Dchcp0jGA,7259
jax/_src/lax/lax.py,sha256=pbNdZ4lheDS4-IGOVP9cTmODrraGFhOpAgfAzbn_OQ8,359589
jax/_src/lax/linalg.py,sha256=yC9N2SKxJLRMUwWghEGgzK2n6eAx9o3unFky_RY1dGk,107109
jax/_src/lax/other.py,sha256=E1n6P-LXfYNbi9maGIDHRBvw994z6XojQvYRC2Kt-lc,13863
jax/_src/lax/parallel.py,sha256=PY-2-gFodE0eI5Neenor0rFfXbv8i0B0DwXiJkWAf_E,119413
jax/_src/lax/slicing.py,sha256=albjjHQj5um8CDP4mt_cC3xXm-jJVnJ34XXnozeIoBA,150700
jax/_src/lax/special.py,sha256=9V-lSV5LJhy-thwO8-mfcWTT8FA_7QpkBhj63d6UYJc,28872
jax/_src/lax/utils.py,sha256=XnO6iqurwkLxn42VuRCPvqFs_rY6AzkX-7vrvYYXH8Q,11358
jax/_src/lax/windowed_reductions.py,sha256=SgUrwxbv-HcV-Am6sSOglHe0stopS2yG6ol-9Ymtrpo,47330
jax/_src/lax/control_flow/__init__.py,sha256=JsuWyikJkp-5IVosqvubQfR6K1qXVeKMmSgNWLDzjso,2093
jax/_src/lax/control_flow/common.py,sha256=0q38QiHlcDYfUs7aW01Bn02d46vkJbXoVyWVZzgngK4,6480
jax/_src/lax/control_flow/conditionals.py,sha256=MbeaGuRC59hCkBDfo2cM3icz7Foag30sTep1tm8JdKw,53519
jax/_src/lax/control_flow/loops.py,sha256=S1M96sAMKHykUCkl1JzD7oe38euNR2ykd0L8am-tf6Q,131983
jax/_src/lax/control_flow/solves.py,sha256=vQlDghpXDIkhOhmwxt2uE1OUNYL201qitIpC2TY6QkY,20727
jax/_src/lib/__init__.py,sha256=FtxaCYUz9-gwZ_-59mGUCEwrJL3q6iUCFUxF-PxwNX8,7577
jax/_src/lib/mosaic_gpu.py,sha256=andYvo1TC8Eru9Pd_rM76L66Ozami3Q5awjdTkdIE10,1017
jax/_src/lib/triton.py,sha256=lMwAguUTQxYs9HkdLqeracg8sz5RL7gm1W6PpzSrzdk,2202
jax/_src/lib/mlir/__init__.py,sha256=KyKFigc6V_REcOMXB95auEAFQaOuFbfnxlxeyLcqfJk,676
jax/_src/lib/mlir/dialects/__init__.py,sha256=zsqbkEnPPS1LY5XzzTp0TkAoCd9lAwICwX4g2YHBGtw,2089
jax/_src/nn/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/nn/functions.py,sha256=1hGDOhwERj5dUYpKzRPrfkIJ4_yNRTMSEHU7h58h59w,48631
jax/_src/nn/initializers.py,sha256=VcgC8CnVGD-VqBPqdG4rlLTvTKaNbXeLgFICJ1uMp8c,27202
jax/_src/numpy/__init__.py,sha256=K2etMeZNy0A7Da0vjnLD_OOwhm3VjbXlMWE65YknBE8,5610
jax/_src/numpy/array_api_metadata.py,sha256=291eTlGuTc9h7aHCrZbft933hlyRjiM-PwAozlr_BJ8,3617
jax/_src/numpy/array_constructors.py,sha256=TX5YEMsPhdYm1blQnDbOqrnl2A2Px0BY1vyeKfB5Vbk,15801
jax/_src/numpy/array_creation.py,sha256=Nyre8-4ILp5_3Q5TlxRw4jFW0teZ86whn5gpBZ79Pww,30971
jax/_src/numpy/array_methods.py,sha256=EclFjyjGsFJoeCdwjGaOgSpan9lIaVgM0hNJBJ1I_jc,50781
jax/_src/numpy/einsum.py,sha256=4qZ1yZqq5uc7HEeO15Y0HWsh--g4R2Bhzl77QWaebUs,24314
jax/_src/numpy/error.py,sha256=K8y90ly6UrjyJS1WszXtPAcjZmYUGlAlbSJIe8pMvok,6519
jax/_src/numpy/fft.py,sha256=Oo4Dl6F1Ol4L1dQH-YtYDPZimhsShoXI48f19uJL6hE,52323
jax/_src/numpy/index_tricks.py,sha256=VTGtSNYaMu6wzNwScnpeCbSiRySAcW2tIFW3wX39jM8,10034
jax/_src/numpy/indexing.py,sha256=RkqsXtMKM59W0QWWzbA3e9QgBNqpf2NjtkNrxM_RhnI,70485
jax/_src/numpy/lax_numpy.py,sha256=iUtaiN68T7I0GQ7Mmi0vIvcRUvbXpVL4lmhgpuXT5yU,338494
jax/_src/numpy/linalg.py,sha256=-dClQrVvYD3sz2-G89mjM-iV8Ob-E0kYIka7q2peOlM,79349
jax/_src/numpy/polynomial.py,sha256=FjTVJq0H2PsL06CsssGJHj8sVFLXRs4DsAC7EyrFi60,28434
jax/_src/numpy/reductions.py,sha256=fd49D-HAk3AtJyoaHM-WmFXHp0RYsJh2YRxTFP2Bh_I,109404
jax/_src/numpy/scalar_types.py,sha256=Y3ifLkxymbC4gILDHZKtdAenKrGd3q28fLrA0L77Hy0,3789
jax/_src/numpy/setops.py,sha256=s5aYjW-vzy28oWy_gGHCSX8oF08-Au1BFkBfyY-mQkA,47942
jax/_src/numpy/sorting.py,sha256=htittZJmxgDBvvdwpzAKlttPSNRAtbaOcGD2POj6KKk,16550
jax/_src/numpy/tensor_contractions.py,sha256=c3u1HtEwfBOdMyGKRIGaoZloG2zriXRllYSpEmNH9ZY,24597
jax/_src/numpy/ufunc_api.py,sha256=6mSK5p4yWR-s3G-uyOU25X1udAt0dCH1jtw_ev7BAmU,24292
jax/_src/numpy/ufuncs.py,sha256=AjrH8OL3xDb-3zkgdqqGKqwrygDRUx2ufbgSb1tCa9k,118226
jax/_src/numpy/util.py,sha256=0XV2c3KM40gmS5lt3dfxQez0YD1PO_l1WI-iUJveKqY,17371
jax/_src/numpy/vectorize.py,sha256=thLB2fBHVQBas7oaVWSsaOhtLgVWRwQekq_L5F7IjAA,14077
jax/_src/numpy/window_functions.py,sha256=jZsnPyv4nF6IXNHMCiX_9-gBGDdJfyb7V74fLN8TaJ0,5473
jax/_src/ops/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/ops/scatter.py,sha256=fjrVO2tRBFzc38ony19GHuWTzYnKcG-85c3XnHHT5m4,18162
jax/_src/ops/special.py,sha256=zYKBMe29eMqKXDeaZ5BK8YehQdpAxAHPFKTmNjwKBa4,4380
jax/_src/pallas/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/_src/pallas/core.py,sha256=b2i9dBogSTtoFpnYB1IFMgHWdOS17RfZa-7JjX_9Q0c,57855
jax/_src/pallas/cost_estimate.py,sha256=9WMlS2n3SS8K7EzuB8Ye11H_ClsS1fzQTB4jbYgxPaY,8425
jax/_src/pallas/helpers.py,sha256=XK9ot2TnQxP061RAoelwi03xDZdl4683qLt5ubv5Ckg,8379
jax/_src/pallas/hlo_interpreter.py,sha256=UdFbnytm45OmXsno03Jskg4isXrMAIQPq5WfOKXrKuQ,19277
jax/_src/pallas/pallas_call.py,sha256=ISVlqIesMcc3ovCvMNk1mcZQ9dbw55Hne0R_YRoYYFo,65302
jax/_src/pallas/pallas_test_util.py,sha256=YighL1QH0VJ10CH-47Tm22ftes8L3OIBhlXxM_K_6ho,2079
jax/_src/pallas/primitives.py,sha256=e4xGHh8PhAe-Fpg71sqKZeYlp9BL1NJCwu2NyWihAos,51662
jax/_src/pallas/utils.py,sha256=HUT6vuHkdGYeu2uLxyhSpyKcx3UK9u92a--2BYU6KeQ,15151
jax/_src/pallas/fuser/__init__.py,sha256=9vpH6Fvikt8NEg8EA0o3ryFBh1oz37RMm1MudOZ-WxY,1265
jax/_src/pallas/fuser/block_spec.py,sha256=UuMnwYS_pZc_VBpvLHc_aVyLbp3UI_8N5ZiGANKTHwc,78695
jax/_src/pallas/fuser/custom_evaluate.py,sha256=jv5JjL5BlUUK89ZKx9sufff0bVyqJA3s5ag6Bpus5ys,2823
jax/_src/pallas/fuser/custom_fusion_lib.py,sha256=6uQZTerfg8V9d_v2kDRNhQd5YhlPM5nO6L5gao25SY8,9031
jax/_src/pallas/fuser/fuser_utils.py,sha256=nu0u_l7Xv4mi0X3rRY2ZfIXvQogesdAAT0QIhBfirU4,1279
jax/_src/pallas/fuser/fusible.py,sha256=MxGUxbkRF6Jbl0STKoc2UUAbPWxc1Gc_vJjy0Gtx5u4,4270
jax/_src/pallas/fuser/fusible_dtype.py,sha256=nbL2uEqq5xNP-ciRYYdm2SADiaQNzBYP2eaWPFKx-Zg,18103
jax/_src/pallas/fuser/fusion.py,sha256=AvfY5PSPzoAT5Vu6NpMHPqobSgNfvv_600VbWIlcslI,1515
jax/_src/pallas/fuser/jaxpr_fusion.py,sha256=nLN1pzdDG0trODXScQnRABL_EgpT-CQGHXH84skJWTE,10994
jax/_src/pallas/mosaic/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/_src/pallas/mosaic/core.py,sha256=olm3uj-9mcDiL9LxIxTMG8JEmGEBvf2E5rk9dujDn7A,14810
jax/_src/pallas/mosaic/error_handling.py,sha256=6pkd7C8BfSZkJ-qxmOrpg6upMWGAfRmmSZNE_XpRNK4,5567
jax/_src/pallas/mosaic/helpers.py,sha256=DFBL7RuFtfBQc-aUaC4bILa5Xzsbb91Wu3VCn7b3v4M,2767
jax/_src/pallas/mosaic/lowering.py,sha256=L2fcdbS9DHcaiFnB5nWwJb7MKJkVlCSUDYuueFyXtvI,148671
jax/_src/pallas/mosaic/pallas_call_registration.py,sha256=C0PWsSmKNpc_JBs1pl5urOJnQjtjS1OBehMqYG2-kR0,12078
jax/_src/pallas/mosaic/pipeline.py,sha256=NKSmYGTukCBLfn2CcAFClWVnxNt9TAFZf6-vmvPKCyg,83447
jax/_src/pallas/mosaic/primitives.py,sha256=xr9FDEmihAGPBNsy9LbwFS-qCnvKKHeA-V_TpZ4pdDE,41261
jax/_src/pallas/mosaic/random.py,sha256=HsvxeXr31a_LxCro4Nax-kkPzjIsI5iBhqU5IKf_liM,7964
jax/_src/pallas/mosaic/sc_core.py,sha256=MPtLX-eDn8WpCfFBbXwn43jk_ob6w4ZgIN-DDFoN1e4,11347
jax/_src/pallas/mosaic/sc_lowering.py,sha256=OvJ56iNLtP_CH177r6lVEBYuu-PT2HlfgDzQJ-cmVts,35330
jax/_src/pallas/mosaic/sc_primitives.py,sha256=-2n6BMpdyVshh0j3OoUtfHDsxPPQiaM8RUHJWKHlnkE,39292
jax/_src/pallas/mosaic/tpu_info.py,sha256=vL8TqgCDsxVJZZ_WFoV97vXKEsaTayP1VkEvsjmA1Qw,12588
jax/_src/pallas/mosaic/interpret/__init__.py,sha256=4S3pYNuiw6yzPmNNm2iRi40_Jf9xlHj-oIRb9Gp4sxk,581
jax/_src/pallas/mosaic/interpret/interpret_pallas_call.py,sha256=lCNQ65QWj7k9E1cq8cdwbNcitINB47NenZUID0ZUHWc,79197
jax/_src/pallas/mosaic/interpret/race_detection_state.py,sha256=7zhjevGaCJ1ABJsSSbQBPrtZGBOWZnLnz3IkBFDUsU8,6395
jax/_src/pallas/mosaic/interpret/shared_memory.py,sha256=5BMb1g9UisSmDeI9Ku7mdejvlu7DdAIBCL6XZ7Dulp0,22629
jax/_src/pallas/mosaic/interpret/thread_map.py,sha256=Xsvu_ifyq-QLsNX6ETODokUvgevf-n-8Ipbmw8qm3wg,2492
jax/_src/pallas/mosaic/interpret/utils.py,sha256=c1GoYpa3vqjMtIbke8eT-DLZhti0jxagNkWSY96e_MI,11516
jax/_src/pallas/mosaic/interpret/vector_clock.py,sha256=ONHt41Ul8bcV5Kivs9kfzM2W1sV6y3H1nuP-BgBlD4Q,1349
jax/_src/pallas/mosaic_gpu/__init__.py,sha256=mS3YSYUZFIG68T5Zo9CFGkT7wCyWRycnWV4L36CnHYU,581
jax/_src/pallas/mosaic_gpu/core.py,sha256=tUB_2ZJWe_1hOMDHYQCCAw7sPfgciHD-uRNOTY4fqR8,55167
jax/_src/pallas/mosaic_gpu/helpers.py,sha256=_2YbqyUlVlHZM5KgU5zJMFCl5I3-3J530mLecBmWna4,13545
jax/_src/pallas/mosaic_gpu/lowering.py,sha256=zRkZ3kasqumJXxYdbqtlrQC_-ZdEsgEFPmfGX-CxKgc,143759
jax/_src/pallas/mosaic_gpu/pallas_call_registration.py,sha256=tOXBfbbsalBQNChiipUer2e1R8iU5_E3gc2MVZQxbKw,5649
jax/_src/pallas/mosaic_gpu/pipeline.py,sha256=x6-4xUp5SZ39Zo5_0CvD4T-1wvKlakBXjr2D9vdMUds,40124
jax/_src/pallas/mosaic_gpu/primitives.py,sha256=GUl-TwPPSMHKXIDDIgfZOmYWoNt0D1eDLuKg6V7QHBI,137526
jax/_src/pallas/mosaic_gpu/torch.py,sha256=kSKl9LETLDPN4XfOyBpgU6fqaoe38cdXl8tfKvCfZxQ,11119
jax/_src/pallas/mosaic_gpu/interpret/__init__.py,sha256=-aCGt5dPNdKLi-ALblRdGErT_9Or12mpWJ52yApKN0I,581
jax/_src/pallas/mosaic_gpu/interpret/gpu_callbacks.py,sha256=8-y_msULQPRRjI9vrHE86jwDhtcFn4VTlm61YZ8h4Xs,21494
jax/_src/pallas/mosaic_gpu/interpret/interpret_pallas_call.py,sha256=34Ba5_B1QaqebNJitW6-5OvCSxQ8vfmPllRfcZacrTE,17035
jax/_src/pallas/mosaic_gpu/interpret/jaxpr_interpret.py,sha256=T6mOTfZoHe4txh1TuwePsl8VVTTu-GQb5ht7rULv9x4,10892
jax/_src/pallas/pipelining/__init__.py,sha256=4S3pYNuiw6yzPmNNm2iRi40_Jf9xlHj-oIRb9Gp4sxk,581
jax/_src/pallas/pipelining/internal.py,sha256=deWZl4xUce2LyUK0lY8PlSK381wLoVvNqhhRnWCVLM4,2713
jax/_src/pallas/pipelining/pipeline_test_util.py,sha256=CHOAEW2R8qH3CAywpBIKDLk8ztvsYofEmlzPWjaExfw,2059
jax/_src/pallas/pipelining/schedule_api.py,sha256=kuXbaNTWVqWn1VQAsplEWAu-GzH81CK6Kv1gPdF57CY,11398
jax/_src/pallas/pipelining/schedulers.py,sha256=Tzf43kBuHAZ5n4cSS8MD2hFzKOs5l0FchcKLs6TVTe4,20563
jax/_src/pallas/triton/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/_src/pallas/triton/core.py,sha256=Z4Vk2LFEpkvRzhRBfZlvEGRhFHoxnjuPxC3EfgMPONs,1222
jax/_src/pallas/triton/lowering.py,sha256=A25t_QP8ltgxFkc-85XDXzmD6gnqJ7YLdt2J3XwOjmQ,97794
jax/_src/pallas/triton/pallas_call_registration.py,sha256=Tvuy1W0VgkLr0_1enmNDjxnLu5UwTG3c7imOyT9WUyU,6912
jax/_src/pallas/triton/primitives.py,sha256=R5BVdwenBSdnStf9h2EEYLPDX0FOcxMqqjCx72Rjc0c,5816
jax/_src/scipy/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/scipy/fft.py,sha256=YHiLcxEwNKeBgHPh2R6FwajmeBbNCkbOgskwhOu6BMk,15857
jax/_src/scipy/integrate.py,sha256=vHyXnhczC0l-Q7ZUWQOIakc1Knqlne4MMShb1DPIHf4,2334
jax/_src/scipy/linalg.py,sha256=UPXkUkLiEeH69nucw1ZUhsMiH3vrGFzftWgfr7fIyhk,85003
jax/_src/scipy/ndimage.py,sha256=6XBQq3MXu2F0i1UaTCAG6MOMPfbPNn4hMnRNHhoyVVY,6945
jax/_src/scipy/signal.py,sha256=ZmDTOnoDUxAufPrBBGGT4EsEoBEQ7itkoKgPxDiLaS4,47551
jax/_src/scipy/special.py,sha256=L0LrlhHO5SHynIii1Kdk0Ao92e_SRtkLzjSaS0Ak2ho,100849
jax/_src/scipy/cluster/__init__.py,sha256=S-o0CP72882Xv3l6k2PlrEx6eNvXBTk5Oaed5KF4iL8,581
jax/_src/scipy/cluster/vq.py,sha256=LPoQUfgmh50I-ooN43p7vBiucOgJ29FIji9ZFbFikOc,3039
jax/_src/scipy/optimize/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/scipy/optimize/_lbfgs.py,sha256=ge0lgeV-z7CNO-JrdjvyeBR8Ya7WEKQFXlW33LC31-Y,7906
jax/_src/scipy/optimize/bfgs.py,sha256=RmbR6n91EhFgkxJ9Oe-HB1_0RdvD5voG6ZMMOnWoFvQ,5733
jax/_src/scipy/optimize/line_search.py,sha256=hK1fIvhOLl1pgpOwWfx1N5mfIMxjPKPyhW03bMgKIwY,13351
jax/_src/scipy/optimize/minimize.py,sha256=vYoSJVP6ijGz0nyjjuwk-dHVFMlJ6MOeDiCePaFoLig,4954
jax/_src/scipy/sparse/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/scipy/sparse/linalg.py,sha256=m1mRTkyNJjlqv1aHrZPVN6IC2mGwD_4AqMljpfW4tAI,27246
jax/_src/scipy/spatial/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/_src/scipy/spatial/transform.py,sha256=a10DkF7vLkS4BMm86swDwoqdV-xDtRjCNfho965M5fk,17634
jax/_src/scipy/stats/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/scipy/stats/_core.py,sha256=mK1v73aUZymO0O6bEwMLi9wLUqAOilEu8b_PBoOC0fk,10585
jax/_src/scipy/stats/bernoulli.py,sha256=HT89cwug-OktD0KVKDWrpihz4GB_hEOU_R567Q_8whw,4578
jax/_src/scipy/stats/beta.py,sha256=jjRE4LlyBt2BO2jmmN1z5PEasjl4LykhKTNc8846etU,8026
jax/_src/scipy/stats/betabinom.py,sha256=LceEdt1jRBxYmV9UMjZBJs7gR1GAcGcU4JbOcF2iEiU,3480
jax/_src/scipy/stats/binom.py,sha256=KIydYZOcwN109i6rodNR-Lda-cog16r2jrQcDarUeeo,2859
jax/_src/scipy/stats/cauchy.py,sha256=BnKgESULqAHQZydTKMxT4oRexDWbMyHWj37CFROaar4,8901
jax/_src/scipy/stats/chi2.py,sha256=p3TVmUKKRscb-7-uBUB2aorLFdh2USag5iRt1P7yr7U,8045
jax/_src/scipy/stats/dirichlet.py,sha256=G9FAdSoODRw-lZkZlGsg1sq2bbqMLZ47wG8P4o_oaow,3237
jax/_src/scipy/stats/expon.py,sha256=NcqNEDneSnAtba3Yol0XyfNZ8_OIwcvTih2NZLE8OoY,7845
jax/_src/scipy/stats/gamma.py,sha256=876SXGzsazCTNxGXL38_PqF0MbFZ9k1XOTcXDhUJdS4,7076
jax/_src/scipy/stats/gennorm.py,sha256=s-Wo42aMKpE1EzeqC-2Vx_uH-zDg8qWjdUAe2d5azr0,2994
jax/_src/scipy/stats/geom.py,sha256=A5XokLcEPUlAjhBh-0BF9PoqGGZmE5EkkT3xT3R61Fc,2282
jax/_src/scipy/stats/gumbel_l.py,sha256=PR7XwEY_RAxKLCLJIEq3QTqzdvAv-ufQl0QgRkFh65Q,8345
jax/_src/scipy/stats/gumbel_r.py,sha256=_1nDUgHNtYlNTILqyZq9jxT4ZDdElwltJ6I5gYVmMUY,8361
jax/_src/scipy/stats/kde.py,sha256=r-c_7eXAbl4NatN0Lr3vKe0aCfcfWg5X_iJDYKobaEI,10240
jax/_src/scipy/stats/laplace.py,sha256=KRoMp7S0Gn-xTiY9A-1_Iv1lnwswGhjfSg6WDf9Oaio,3274
jax/_src/scipy/stats/logistic.py,sha256=508gMdwcsWI6ynyrD4w7VZMw9HB8K-Xyzux6oVEdKkk,6291
jax/_src/scipy/stats/multinomial.py,sha256=W22uKDVNPoF8JkQwI0TmnXPJ_7SFa3mgzUYLTJKG2ks,2532
jax/_src/scipy/stats/multivariate_normal.py,sha256=LqDX6XJxocbMlqM3HrnUHEw2ZOGjfIvXowhJmmHD1mo,3569
jax/_src/scipy/stats/nbinom.py,sha256=pF8EOulsrjJhQmq_NjMhfeQaXf5qh7wC9A6mNk8oiUU,2637
jax/_src/scipy/stats/norm.py,sha256=8YfKWloDLhnfpfcuamJlPZuqgLtwmsNKVMYT53VKKsY,8421
jax/_src/scipy/stats/pareto.py,sha256=zfIQvzbhCHx2LdLN5dOZjEFa55FV_YLnE4xfy1mFIw8,8908
jax/_src/scipy/stats/poisson.py,sha256=oBS4VROKeGFCNHCHPKiVXtxgixlFOzFXnlElQpcwcpQ,7289
jax/_src/scipy/stats/t.py,sha256=-POac0PdQa3tNpgksvZaVRHYrrbQQrlyXj6l8iP15gk,3226
jax/_src/scipy/stats/truncnorm.py,sha256=LQaZJwBZ9Y3pWOABOU0gsbHD0nq9zQEQCKQzxLJx_Aw,9270
jax/_src/scipy/stats/uniform.py,sha256=e0INco1aevZnIfWfVXVXzFbt3rEOUz0UDGscr81SU94,4372
jax/_src/scipy/stats/vonmises.py,sha256=S5GpPFeRpE5BH5-MEcyenno1zxbbvOpL2Lvn23vTG-M,2484
jax/_src/scipy/stats/wrapcauchy.py,sha256=4vJw9eZn5iMgzpR3iJmuFhFq0GTCqQm3P___4UB-czQ,2401
jax/_src/state/__init__.py,sha256=e2UVK7vSG8FSS1C0heFQyWt8UG93ypk0hQcFwIkVKPQ,1038
jax/_src/state/discharge.py,sha256=iYTu5aGY3FuM3luAl-RJTlPucyAmjG4FrYuiKawlENs,37019
jax/_src/state/indexing.py,sha256=_tpM_yibLkmiobeBIL3llJNT49mLYUrGTLeyWL0wrw0,13929
jax/_src/state/primitives.py,sha256=zRbUNIF3cAZgZS4CBEcVKzhAfA7jL2qADr7baEJCGsk,44668
jax/_src/state/types.py,sha256=nXpJnNfTu2xOuvt9PBQGL-U8QL1jOjZInfM3pfV8O_8,19763
jax/_src/state/utils.py,sha256=Vw3m1-ab_D798MvWjOY4b0jlyzqLMVEbUjAn2wktutA,4031
jax/_src/third_party/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
jax/_src/third_party/scipy/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
jax/_src/third_party/scipy/betaln.py,sha256=EW6FFB1myCHMGwDXqQXPIRaEL-UvVZNahZu_wdsixSc,2388
jax/_src/third_party/scipy/interpolate.py,sha256=XJA3BNVCwIowXBH5g3M4dsKwHpMVyHKBxL2Oy6RkbME,6360
jax/_src/third_party/scipy/linalg.py,sha256=jgDDGByArWWjetmCdX6LCdzPT2Ao43TCe56syYpBxFw,3878
jax/_src/third_party/scipy/signal_helper.py,sha256=ihye51hrdpDnsJuZ5z9scJhl1a-pEN44rCFgLQdTM_0,3569
jax/_src/third_party/scipy/special.py,sha256=Q8kobJMwgE5SiOYDLyliaHGz8Sy7RLCzFrJoCIw9eT4,9118
jax/_src/tpu/__init__.py,sha256=4S3pYNuiw6yzPmNNm2iRi40_Jf9xlHj-oIRb9Gp4sxk,581
jax/_src/tpu/linalg/__init__.py,sha256=RrUPd-WnsNpraGGzWYI-IYTpNcbtQeGqaotRh0Bwwuw,779
jax/_src/tpu/linalg/eigh.py,sha256=YVAeL2Plggn_0Cop0vXaI1JX6U9DWJfwOdljvea_Y6k,26537
jax/_src/tpu/linalg/qdwh.py,sha256=tX9TiLEe0Flr7CJiRW7UNOy1g1hGkmQIiowmzCQ2Xw8,9553
jax/_src/tpu/linalg/stack.py,sha256=XPaFJtuPGfry1ssSqTNKdHcUs7yRUyzeD6aeYz0SrU8,2570
jax/_src/tpu/linalg/svd.py,sha256=WdmOjLDpZoCKErDXef01QfMn7ZOmdb3O3_FVoDyU88U,10581
jax/example_libraries/__init__.py,sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw,581
jax/example_libraries/optimizers.py,sha256=T2D3qDPWZ50kwTyDe2IXPPLbOUu04cQTdf51oHdXVvU,20426
jax/example_libraries/stax.py,sha256=eOLn-UowNy-bEXngwaNUMXbUka5qTqmbQXQUYS8Ou00,13834
jax/experimental/__init__.py,sha256=V0WZhTy7yIYVX0BkO5Tpk5FU3OS1aqfXoIHPoxOZSfk,2047
jax/experimental/buffer_callback.py,sha256=ocKnbOO_9n62E2olM0c2a4ep8AHTan4I_AL-eR5VbrQ,765
jax/experimental/checkify.py,sha256=V84hTTNrwnckaRzgvWDmpcKKA95kAakjZwwlfnRaXt0,1213
jax/experimental/compute_on.py,sha256=Mdg2n8JtAsrq1ap_PwUPwoqpoCNGO5rjf7T47e_iaAE,644
jax/experimental/custom_dce.py,sha256=ORRk_kBjbseQ06T_LmLPxYCO7-Rby5U8Erro5U5Gi4A,682
jax/experimental/custom_partitioning.py,sha256=RlaIvfdM8VqlEJC3FajqCmB_EOHyk6iyfGUgaUyR61g,1054
jax/experimental/fused.py,sha256=wDd-XYnPgHojubWmb00tJR23dW7XnZe07AxjaVK39Bc,6803
jax/experimental/hijax.py,sha256=ZXsAkisauzalzJP0FVOGeJ4FB8UuzUH2-uSVnEQppr0,1444
jax/experimental/jet.py,sha256=jFclY6Ka0yPRGIOu3mIRhcqotpcdqs-eCq8hPbrMGpk,27645
jax/experimental/layout.py,sha256=fR0lxc08SqRZ0jv9bh6QACBFt_rFP_4naq5y43pU5GQ,738
jax/experimental/mesh_utils.py,sha256=Sf5ivjPuZ_u0ufQ38MLmhbEFnlIMs4SHd6lGXWQJgQY,923
jax/experimental/multihost_utils.py,sha256=EdWGV3qVcKLPIa_aw7hsGCDDIcKKNzi44jmGjsUtTbY,25865
jax/experimental/ode.py,sha256=a8ZsN4gVrHXM_LPJF3T5SeC-RHjFTGUd62eA1WDdkLM,10903
jax/experimental/pjit.py,sha256=iy2zCGu1eqrbbYTHm4UT9QCcHXzgAn0mcnxL_jukvaA,1195
jax/experimental/profiler.py,sha256=Rxlgbb4TPJsfeWXwLYG58gW0PCqBjDVWswlCCLr7fBg,1413
jax/experimental/random.py,sha256=srL15IIcOo8A_DX_4WaGpDggdALBWPJBxkQV_f1esVc,721
jax/experimental/rnn.py,sha256=betLPzLUoJIU919CJPaFgGub_t6fHkGY8WYFeI_5jXs,20096
jax/experimental/scheduling_groups.py,sha256=WAg1sEdkIH3_RWJeKwufDASSVPVq_JX0s76Va-2n-wo,8088
jax/experimental/serialize_executable.py,sha256=T_hxV_VhKZy--jmRdDCN1ezF2_YsEcwESTMqAMFYLJI,4632
jax/experimental/shard_alike.py,sha256=dbrhVCnWEX39TsQYRyGoClFByDmqtmDlKGtOmlT-m2M,646
jax/experimental/shard_map.py,sha256=daJk2CGjpmspNgkUMVCBvEosmbJ8BxgcGvRMMFwAPrM,1472
jax/experimental/topologies.py,sha256=AKCCES0n_D3WKyluII_vVRhV72Fy2dHamjAgSKVtyYw,2078
jax/experimental/transfer.py,sha256=jGPl9y0Qb31h_4LPJmi2FTZVghIN0bmk1sqH02tVRm0,2535
jax/experimental/x64_context.py,sha256=ediyKqtdUW15sPTLTlxB8K52BKSgrMAC0AUnFbNuUT4,1192
jax/experimental/xla_metadata.py,sha256=YoNAzPbdARYfGzGTUVoxTfHG9dDrobUvWaPI4uvHRJk,658
jax/experimental/_private_mm/__init__.py,sha256=VIjw6l_CQ5-xaicoK8-2QgM5W7GYrBvv4xASfTB0ehU,775
jax/experimental/_private_mm/mini_dime.py,sha256=NdkS5N-2RVcfObOkROOX8cWN2jPWkC0NeNGSzNSbDHo,9738
jax/experimental/_private_mm/mm.py,sha256=C19wJQmkdSmUeMNpYTSDm90DRlHm41NoymTEvBX0Xno,11435
jax/experimental/_private_mm/profile_utils.py,sha256=dZ1wvrhpaRNbEyiU7FpIuIi6qUfOvBHdMOcXqsYNRrg,2169
jax/experimental/array_serialization/__init__.py,sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw,581
jax/experimental/array_serialization/pytree_serialization.py,sha256=Wf4atVPHTLhpXFuZS38cSvQIfyQGr9YqIPH156bE7pU,21623
jax/experimental/array_serialization/pytree_serialization_utils.py,sha256=agMetED8ST3oeyk2yCgw9nJ08VcNvP9zWuzQEPfZpMI,2798
jax/experimental/array_serialization/serialization.py,sha256=nw09uNmya2DlJA_2DEAfB0OgMUwwbv7qVywvcIyzmog,13454
jax/experimental/array_serialization/serialization_test.py,sha256=ugXal_ujJMdUR8uwMQDNFlwfh-v9bUgr2DZutOl_ncw,42380
jax/experimental/array_serialization/tensorstore_impl.py,sha256=Gft1py-bs-6Z1HzJRtuI2aCAxTnyvZht8G060_DLxco,25258
jax/experimental/colocated_python/__init__.py,sha256=Cp1CgJT_Sn5qLkiO7jW5S0HPWieBmOdpEE-AaYpKs9E,980
jax/experimental/colocated_python/api.py,sha256=IhXs8bj18Lx921Hb9khYCHpA7x9olBUB77CKYdNrESg,6737
jax/experimental/colocated_python/func.py,sha256=EM2aXQ_IHB_wcjRJSjO7C-FNzElGdYczk1zr_f4J3uM,26061
jax/experimental/colocated_python/func_backend.py,sha256=JQdiIUulodBQWSazJqSXKMC11UT9l3BbHVr_SveUcaI,1382
jax/experimental/colocated_python/obj.py,sha256=ZjG6HezLco6NH51UVglXsUJedr_gclmK75N12fknruU,8045
jax/experimental/colocated_python/obj_backend.py,sha256=OUNUTcSzDHepePd7BuU1R0nBIoN9yInJ4_UK4f6FkTM,2476
jax/experimental/colocated_python/serialization.py,sha256=cJu71M01Y69UEeUJRI7Fd5PFtNF4cFsmPWJxdK1Kjeg,12301
jax/experimental/compilation_cache/__init__.py,sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw,581
jax/experimental/compilation_cache/compilation_cache.py,sha256=0r1UkLBOJBsXql-Zbvwg3SPg7hl4RLt3dWBG0UzE1xI,689
jax/experimental/jax2tf/__init__.py,sha256=ZqsPlkalGAkzdTwRIutfW1k8NmOHzJn6kfIB1A98pdw,944
jax/experimental/jax2tf/call_tf.py,sha256=4JmMVY5Y1oNDzuVKt6HdoTYuwWBLnCs-bo6lFrJsNRQ,28754
jax/experimental/jax2tf/jax2tf.py,sha256=HuTLz9sQmXTgMnix7Ody58QUr5sNTzKCcQJ_2D21eVo,38656
jax/experimental/jax2tf/examples/serving/__init__.py,sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw,581
jax/experimental/jax2tf/examples/serving/model_server_request.py,sha256=De9TWLSyC5EbkQRg00S9bvVOo17HJuS_M4cztRFjOa4,4912
jax/experimental/key_reuse/__init__.py,sha256=5CjqLFSLnAhm_YyrowsWA48ENwxcvcD5KQrNQJTgLpQ,1688
jax/experimental/key_reuse/_core.py,sha256=-YeNKbFpc_WDCC6vZiReai7R4DHQWFG7RTxtcjw41i8,23241
jax/experimental/mosaic/__init__.py,sha256=-MnUIUUmkSaNkO5bOpPdXrcTnKVWt9ModYX4cnlq9LY,883
jax/experimental/mosaic/dialects.py,sha256=eYxSlw_cu_8usVXcEA3f-cqmlxDUwcTiqLiqNAgWWfs,769
jax/experimental/mosaic/gpu/__init__.py,sha256=_Ujy7nb1Diwr2egsD6tg8W_S9uukzSb9K9fX9RtulVM,4206
jax/experimental/mosaic/gpu/constraints.py,sha256=sqGwlvOxy9QYFz_BaSS92MBnsq-WZs1ObdtourAv62M,30661
jax/experimental/mosaic/gpu/core.py,sha256=3WFpAMvH12eCkguPcM3H2AkDKioXNnGjotST4XygfWU,39357
jax/experimental/mosaic/gpu/dialect_lowering.py,sha256=NVjx1UivC03s1HhxF1Bw0_OgBVoJAXojr_NSBcMp1jE,85255
jax/experimental/mosaic/gpu/fragmented_array.py,sha256=UDsycPnSvIJVpA1-oR0otxZgG_4jIIJ5m4_hHSrXcOU,159616
jax/experimental/mosaic/gpu/inference_utils.py,sha256=vI-2K7qPlOK4yXFqgY3wpRGnr38XKUT83ZtR0PWGatM,9876
jax/experimental/mosaic/gpu/launch_context.py,sha256=5R6GDVIPT23_bfWfdsJbG2hfWb4xEI870jfyw_NQ7pM,64766
jax/experimental/mosaic/gpu/layout_inference.py,sha256=tDdqk_e8UX2kcRbwyANpmsSDFbb89ofW5sh9YNDG8Wg,78351
jax/experimental/mosaic/gpu/layouts.py,sha256=FvoBURj9YOyDN8Yuvi6vWfLRkhU5XzxqsbaQfjPq9mg,9018
jax/experimental/mosaic/gpu/mma.py,sha256=rbvwPWp4i5KouLAczxWMMP8-CM0hgekYLmxxoDvtrvU,7550
jax/experimental/mosaic/gpu/mma_utils.py,sha256=AF7QKWnJua5ukJmNtL7KStGmaZaCSaEBtu6wudhgl-4,10707
jax/experimental/mosaic/gpu/profiler.py,sha256=fbT7kHwnWtmSt3rV3NoZwz0coBtL7ybRZWeyUVOD6RQ,14668
jax/experimental/mosaic/gpu/tcgen05.py,sha256=kCliEt5EUhBHG8J_ZLB1hNulwIb-tSx250nTvmnrXH4,66113
jax/experimental/mosaic/gpu/test_util.py,sha256=-LMoPbjM0ROnMlttzf7OwR1CGCyRtB8S4LFUXKAyT3I,2688
jax/experimental/mosaic/gpu/utils.py,sha256=hTlLo2I9d4Ga4kn6jyAlt2le-2URrQQgGYqWIO3Wzj4,69999
jax/experimental/mosaic/gpu/wgmma.py,sha256=6Mo64a8U3IbCuLzZAbVMK0IdAxLajAnZ7mWHBk0QJAM,18749
jax/experimental/pallas/__init__.py,sha256=pNVS_U-FPqXRxuQGw5qESq-iabCeZOmD4Xm3hVwKDOk,3831
jax/experimental/pallas/fuser.py,sha256=FmH7hj89DZq_Ijhkx41IFSWhhZs-aR1_RWTHWrtUQA8,1294
jax/experimental/pallas/mosaic_gpu.py,sha256=IEK-ijzRC8PTkXvTjhQiMcmM52oNz55cK2lZt465w7Q,6967
jax/experimental/pallas/tpu.py,sha256=xETpCOIv0Z284BwIjW-dwl9lRlDUMRLsHEmNTHpZQSo,6407
jax/experimental/pallas/tpu_sc.py,sha256=BDwGwqlKozDB7BQw1wjFX6KraDu8mJN5vxt_74-a0qc,2493
jax/experimental/pallas/triton.py,sha256=iiHsE4yPCAY_XL_PAOLt2aCVYz2cNh4BaBoczOyqhQY,1640
jax/experimental/pallas/ops/__init__.py,sha256=vC8qlznuwgbYeFDJqP8pDUIf2gEKFrCHg3EZYdTkiwM,764
jax/experimental/pallas/ops/gpu/__init__.py,sha256=mS3YSYUZFIG68T5Zo9CFGkT7wCyWRycnWV4L36CnHYU,581
jax/experimental/pallas/ops/gpu/all_gather_mgpu.py,sha256=q8odTiv1lSLhEzdCdKDbScGDXGYiwf6wkQJW3q96O5Q,8750
jax/experimental/pallas/ops/gpu/attention.py,sha256=AkTo-_jOemqjz2iY6AyfaB8JoCpoENJYVzrwFgRP5Q4,22855
jax/experimental/pallas/ops/gpu/attention_mgpu.py,sha256=TpiJwoJQ0EVa5Tlkn_7s-1pBijbnVP7SbCCJbm__fmM,36425
jax/experimental/pallas/ops/gpu/blackwell_matmul_mgpu.py,sha256=6Vsko0l62JFmeEd48Vq_bQ0KsVi-k6_kGXBzY95_9rs,12105
jax/experimental/pallas/ops/gpu/blackwell_ragged_dot_mgpu.py,sha256=fXNmtSmYUcXwDs7w2al6uS_PzdLIauhWIOtM1VoGjeA,15803
jax/experimental/pallas/ops/gpu/collective_matmul_mgpu.py,sha256=3Jm6-JTKNY4f6odXkVXL6IXuH1EI2KpAdxNeaUx2B1U,9929
jax/experimental/pallas/ops/gpu/decode_attention.py,sha256=_topNUYIEEg3OP9RgC3Qv5zKbJuB7bSWy6MPCScqry0,17128
jax/experimental/pallas/ops/gpu/hopper_matmul_mgpu.py,sha256=btmi_BOKLpuClFyfNrsB-eIcwXbTicPAU6LOAPIDzeI,11353
jax/experimental/pallas/ops/gpu/hopper_mixed_type_matmul_mgpu.py,sha256=qkftSh32aiD7WcKqo9a4-QFJule-Rx0CDMXnlmAI-dg,12605
jax/experimental/pallas/ops/gpu/layer_norm.py,sha256=ifcUo1WIazMEXijjhGbFVQJ6_jdrh218CtVq69RoU3Y,11271
jax/experimental/pallas/ops/gpu/paged_attention.py,sha256=MKaQ0GKMrhf2xYGvzvHQdM1AxeyLHc8uFGnCZHNmU3k,16191
jax/experimental/pallas/ops/gpu/ragged_dot_mgpu.py,sha256=FK-Ue906S1WQMiKVh1a48ydF_O4QcrLOdcazZ9K9ROI,12351
jax/experimental/pallas/ops/gpu/reduce_scatter_mgpu.py,sha256=eeG21RS-I0SxjSXBcrBIBIDtXylCAV47fOxryt1s1rs,9821
jax/experimental/pallas/ops/gpu/rms_norm.py,sha256=1pxIRygmfsWZJRuGAnEOIFdQJD2QtT17mX2fTOLnDE4,10091
jax/experimental/pallas/ops/gpu/softmax.py,sha256=3E-coLRFVOYWyAcC6uLhIodrhfbl8wa68rIAEYz9mBM,2766
jax/experimental/pallas/ops/gpu/transposed_ragged_dot_mgpu.py,sha256=gRws5pC09r3H7w010JpQrsDHkTtk4Q4WFn53KKqfSiQ,9841
jax/experimental/pallas/ops/tpu/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/experimental/pallas/ops/tpu/all_gather.py,sha256=9XXcBpb_pXZGn8EOa-daiO_EJecOLdBtx9QCxCowEL8,5573
jax/experimental/pallas/ops/tpu/example_kernel.py,sha256=jzsyvU-Lk0PIorE9XCw2NDdqdp8Cn4bqsSe76DBxeEM,802
jax/experimental/pallas/ops/tpu/flash_attention.py,sha256=ZcttPmkBLcKOQfpxhJSWA2RFtY-6Dek0L2z2wehbALA,49781
jax/experimental/pallas/ops/tpu/matmul.py,sha256=x-2MCTChhr6BwQMQXzKcRT8PDVQ0ZvPjGxYOCaAgb7g,2717
jax/experimental/pallas/ops/tpu/megablox/__init__.py,sha256=HRWx7FtM9_17OLitbxwIZp-A9RM61uSL6QAcBGJyw_w,650
jax/experimental/pallas/ops/tpu/megablox/common.py,sha256=-hkn5qlRxNqi9noij1SI0FAl3lodvHwDRYAt5Da_zDw,2100
jax/experimental/pallas/ops/tpu/megablox/gmm.py,sha256=PnH6tp1yv31d5QPtedi7oI7sBgqu0iQ2Pm2lJ4EzqTA,27243
jax/experimental/pallas/ops/tpu/megablox/ops.py,sha256=4Xg3O0bnP5HTWsawS0hAm8mF9HTvRHVK65RlopkA1rw,2903
jax/experimental/pallas/ops/tpu/paged_attention/__init__.py,sha256=Q6gFLakYVRJDRxxvqq6vmtDYup-edUHZy54j9194JfY,700
jax/experimental/pallas/ops/tpu/paged_attention/paged_attention_kernel.py,sha256=_BEerBHpykoijwcqpagwKf--pXIWtt479Yz0qzzGlC8,22009
jax/experimental/pallas/ops/tpu/paged_attention/quantization_utils.py,sha256=5y0QHJo2yDsafF_lgeSO_izChspLHpVOvbFWZ-2d0ss,2556
jax/experimental/pallas/ops/tpu/paged_attention/util.py,sha256=t1_5K06BjSyQKaa-zqQnJDpW4dX1iQ-URfb_hEUJ53g,3254
jax/experimental/pallas/ops/tpu/ragged_paged_attention/__init__.py,sha256=XBTKcqySDW4Bl6lDQvhpu0PAJK0o_L3fWuRE359qy9s,1036
jax/experimental/pallas/ops/tpu/ragged_paged_attention/kernel.py,sha256=KQnu-8EJJDOVqDpK4ZpisE4dtV7gW2TtoJMWru28ij4,31880
jax/experimental/pallas/ops/tpu/ragged_paged_attention/tuned_block_sizes.py,sha256=vCdHiRElQ59pwcdbN9M8IztGbsfdyKka4kk0nGwrqrs,97758
jax/experimental/pallas/ops/tpu/random/__init__.py,sha256=gEMCMS8LKTvoFm0t7khiAnQg78EDUuMI-fvTvw250eo,682
jax/experimental/pallas/ops/tpu/random/philox.py,sha256=Sbrnpu1_b4VcC-gZysO6VTqZ3RqwLUMl7rcEv3FBfRk,7588
jax/experimental/pallas/ops/tpu/random/prng_utils.py,sha256=5r3pl293zG_WWLzWkY5WKP-lxqFmLxwGx9hWu9OHiX4,1780
jax/experimental/pallas/ops/tpu/random/threefry.py,sha256=lEDoysDSPHvGHJIwmChwx0oDgFO-R_2vPEFzN3Llqqc,4358
jax/experimental/pallas/ops/tpu/splash_attention/__init__.py,sha256=8y556G1Vpmglvw8FbNzeB4x6QK6PLkKUBnMsL4vleNs,2738
jax/experimental/pallas/ops/tpu/splash_attention/splash_attention_kernel.py,sha256=X5_oXcOj1Legomgv0yjTvE_MA7GZzRnLu7yF_hmw8XE,78523
jax/experimental/pallas/ops/tpu/splash_attention/splash_attention_mask.py,sha256=ncsRlbb-oHneOkZqkSpOB6r7V0p_lZY60zBEsar364c,16307
jax/experimental/pallas/ops/tpu/splash_attention/splash_attention_mask_info.py,sha256=6h4bS38tKAB1rKbcuFK1XBH6R14HSzGRtr3lOHrD3j4,40291
jax/experimental/roofline/__init__.py,sha256=ftyNH95ANQHFYSq9hsweKfJNP47sfipMm-3XFsr32yw,1260
jax/experimental/roofline/roofline.py,sha256=_Jg5-T3eymNJbw_UObKmZVBiza-7ucrqiPqmtXeuu6g,11821
jax/experimental/roofline/rooflines.py,sha256=C4g-_7iV1VilK257r6plGTcMHNDImAgosZFI2HWVdNw,28455
jax/experimental/source_mapper/__init__.py,sha256=g5b_71QFD84RscRTmCDm7l_TfUYkcW6gZuRd8rgFg9Q,1698
jax/experimental/source_mapper/common.py,sha256=s58vJuszNzoWy-YmpfnuL-noc9zXsEu8_6H7iYwC3pk,2508
jax/experimental/source_mapper/generate_map.py,sha256=k0SrNKPQMAG3fgc_Pw30Vhn2u92nTkt-BbjyhVDs52Q,2129
jax/experimental/source_mapper/hlo.py,sha256=QEDIXY1F9fwzgACNIe8KwxqH3ICXHL8ROthsa4ANFt0,7375
jax/experimental/source_mapper/jaxpr.py,sha256=PXa9zMzP2_fD_0vxo8Za-mw21pXp2Aao6IQkQIqmoJU,2595
jax/experimental/source_mapper/mlir.py,sha256=U2YSWC9tp_DlnsrRzuzsAU7W2PY6ZwtWedHiuMSTQ5M,4686
jax/experimental/sparse/__init__.py,sha256=8m4B-xYwUfXp9McGxgMMDiyWagueWwekQz1bRLXUzJU,11419
jax/experimental/sparse/_base.py,sha256=qyiBgPa12uwS1nLpSDw8dTD1VEq0v9P2_CGTNyPCA_g,3239
jax/experimental/sparse/_lowerings.py,sha256=1xWv8jIdl8x-7tIHLB8CQ96V5jH7IIPjnhCUYsxKOmo,13006
jax/experimental/sparse/ad.py,sha256=zJLXiDufoAVcLTHdFitEXdbQUM_XtDiJoIktwDVuNZU,7789
jax/experimental/sparse/api.py,sha256=DrqEZmlIr6n6kpJjNli9J_rqMX3UO3sPPW7E8_t-u4g,6596
jax/experimental/sparse/bcoo.py,sha256=Ahu8Wv20MMa7Ud2BDLOplV2F2hZvbamt3g5B3_i4eNE,129441
jax/experimental/sparse/bcsr.py,sha256=JUUJbTKO77gjLBD7B1AhGEXjLkFzV4cpzNpOMIo95dY,41196
jax/experimental/sparse/coo.py,sha256=8DLqCT3_HLr0_3WbDY8HzH06OY3YJOW5VYN9eaDACPw,23801
jax/experimental/sparse/csr.py,sha256=DuqeKs--XQVZEjCVWPSlYlvn9fN5omDRB9oe_97wDGk,23933
jax/experimental/sparse/linalg.py,sha256=y849Xo9yo_GNDKpCd2s10xgGY_z1_EABXDRNyLRR0mI,23446
jax/experimental/sparse/nm.py,sha256=Kt9p6YeLP3etlcMOJxN43ETflPNXwHH8DKWjqL9wLOI,8451
jax/experimental/sparse/random.py,sha256=VNhvzYvrFzk5KmK2FGEnD3KuImtUyyFIYAl_jefQUlc,3883
jax/experimental/sparse/test_util.py,sha256=hBw1SDgjVL2KkadjSdvIs5DcHqmdrtQHTHUCwj-MBfw,9433
jax/experimental/sparse/transform.py,sha256=psLE3DLJ1-4ZOPAtAdqgBZ7uqtoQIlH5k8rfNG-_VJI,37575
jax/experimental/sparse/util.py,sha256=zb8BrK9aG5DC9ubS1ZmjSKfLhnnGVKA5ydy9i0hKzA8,4337
jax/extend/__init__.py,sha256=ED6qQKaAf27YT4aaYK5vuZj3-qU5ez0QmiCj263dIWU,1645
jax/extend/backend.py,sha256=c5Q_B6MAdyfZw-fjQP6tz57iKeAVzkXue1adGvE0Itg,1311
jax/extend/ifrt_programs.py,sha256=TCWvR-K8nHQ9PuUNoMWVAfEuYsutTNrefKwLg4AIRao,787
jax/extend/linear_util.py,sha256=02ixe_vuJbwdBvRNcpQGG4DtE_6cPbIpFCzpus8l1oE,1601
jax/extend/random.py,sha256=w1OM6RCcGwmzIr9bdGJJTNKIcMO4ypXbzmeEORlBHjQ,1083
jax/extend/sharding.py,sha256=GRpNqvTo7-_NhNk7j9JW3HXE8_WkDocmnj07hfFp60U,1327
jax/extend/source_info_util.py,sha256=IxZYjLDlwOH1uE3A_vc25dy25QZ6FABMb4RqrkiA5hU,1215
jax/extend/core/__init__.py,sha256=DR0r2Q3t1CSkK4rGV8F8681fbdPc0jqlJzLsgxZHNLI,1100
jax/extend/core/primitives.py,sha256=BQd8FTrHnEFY4daqeBJnpVsV6ALMgxM6mz1-N7hjwA0,6321
jax/extend/mlir/__init__.py,sha256=KNPC1NUExtJ6tuxXjORksUCo_oqJ_jxGS-VASRDfXg4,889
jax/extend/mlir/ir.py,sha256=1IirtLROpBL2nVDNSlP_3c6u09xXaKLqBkzF1JtmGz4,631
jax/extend/mlir/passmanager.py,sha256=1rfuCocMdFlx6It95ynBCP-8JC5xEAEG7rBdLJ7SpFI,640
jax/extend/mlir/dialects/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/extend/mlir/dialects/arith.py,sha256=Au6mT9C4OrdpaGGwJ4_bauTFjGFQKF7wilv62mxzP1E,643
jax/extend/mlir/dialects/builtin.py,sha256=Y1QDRes-Arh7uBLJpySSqygBwUovalHyBq71m3uPbiQ,645
jax/extend/mlir/dialects/chlo.py,sha256=sfxvYEEBgGVd8fDXOlyfGtOLCCAW0xquLrSlTJFA81A,642
jax/extend/mlir/dialects/func.py,sha256=sCnRWz5gKtZ8Cf5NpeCIQ0zVh-nM6H95ttjzksTRrXk,642
jax/extend/mlir/dialects/math.py,sha256=G2gFkarEChMUEjv-vjDc4O09e-1orPhh_zHJhjbPPRQ,642
jax/extend/mlir/dialects/memref.py,sha256=nPR3oOrsENZ-mHjHKFSD48PpQMZHsVLfUVCwomBhv_w,644
jax/extend/mlir/dialects/mpmd.py,sha256=TzdnqMzVUtGHW3pFmnUa16PDC0xPgOCqFi8hUV6tbKM,642
jax/extend/mlir/dialects/scf.py,sha256=hmcJIF7RUqzeeP_NIOampnvCB5Jb6T8Xex4T03IeKrI,641
jax/extend/mlir/dialects/sdy.py,sha256=4dpcOKEHHkDh6cXdyEm2VGE9W-8701MD9egZmWDlkRM,641
jax/extend/mlir/dialects/sparse_tensor.py,sha256=TaFY58y273rRITr8WxSM7LB1-sTAa03dCAjH4F8ZWaE,651
jax/extend/mlir/dialects/stablehlo.py,sha256=AbJQc7zghgrJh-BgRB31L7iHrNLVaEk4AELejcq7QiE,647
jax/extend/mlir/dialects/vector.py,sha256=xP03nScJg-t7olD8_P8SiFGVnV6Qi0Fwn9uGK0MHOXI,644
jax/image/__init__.py,sha256=cthOdx-HsZ6SgvW19tdNCDHmC4Proc8da1hY1hRztTI,1027
jax/interpreters/__init__.py,sha256=HNgKpgKpz4svFyTlHE9wMrDp5iQtg6XeeUshYBOnRko,690
jax/interpreters/ad.py,sha256=AQhDO4X0q4OUteW82lbo0vJvwdb7M1JNFyilqX1Noak,4850
jax/interpreters/batching.py,sha256=_JqcgCOGwIjnNn_C8tMfq3cQGmAQnJJ-rAC52GoBYyc,5850
jax/interpreters/mlir.py,sha256=HJhvrZDSOhpt2xh0qulx9qq41VhIH7ERldcaR3T62xk,3310
jax/interpreters/partial_eval.py,sha256=DQTJb6lRalIkppjv5-8L6iA-hS6VutlPHvw1XR2RtTE,6780
jax/interpreters/pxla.py,sha256=JE83tuceZ8FCjLCvmix885DUPVWCPfoFfipu-YZFVpg,7082
jax/interpreters/xla.py,sha256=TYm_YJ1oYol39-yei642UjRy2W-tHaxgJ34bw1oD9W0,899
jax/lax/__init__.py,sha256=5GydyjK77gROwaq6kEr7oYdrT5F6L123V6gx4vDjz7Q,12113
jax/lax/linalg.py,sha256=-ApN-HR0U_5z4VJdw6IAoq96II92n3OJ6CR_1KyzoNo,1717
jax/lib/__init__.py,sha256=i2G6ZoULU2uH8HBmxui_CTzii_5bW5ibXbHPlIvkONs,660
jax/nn/__init__.py,sha256=G9v8E63aJYY0rgGMX_bTRB52uBR8a0qEkDQn92zA_z8,1782
jax/nn/__init__.pyi,sha256=aqHPaDoGq0xSkCrUPqBc9N3zoWTTBRFNKeL9qn8t0T4,5042
jax/nn/initializers.py,sha256=finfAJmOEZdPIzY2roSIBTOmC4rKEW-rchpH9d6Dc2g,1469
jax/numpy/__init__.py,sha256=ED4QneCjb_DZo3trzWARqRD7HrnsLWdiYNZp14s_kJg,12759
jax/numpy/__init__.pyi,sha256=RJ5vLM7Q_8xS3ujX6rVqD3ioVtKBjaUNJNxxQoZ9nwo,45009
jax/numpy/fft.py,sha256=iNatBIYRSwpXquPSgtNRMe6-dhbKkv2yoH6UJ6PFoGM,1084
jax/numpy/linalg.py,sha256=bnzLmLpLYSDRIICkjCEsH1WmA7wNOdSKhmO25gPvTqI,1419
jax/ops/__init__.py,sha256=VIAczIKYAFWawHdNmLmptUYnFQiBwHwlsa4LpgGEbIc,870
jax/scipy/__init__.py,sha256=DXszF0t3ZbfDeoFEP_GyBb0OOr6F-4yqsZzmen7pY58,1474
jax/scipy/fft.py,sha256=PVKQ4r-vl4J9Ct6LrhZWZ7jMF7frIXggiO98Bh0oVIE,809
jax/scipy/integrate.py,sha256=H9iq7YKgsZQ1d0SUvGcgPnEgD1TBdWJXHYjfUOGzT8Y,777
jax/scipy/linalg.py,sha256=igI76mnoJVRwE2rQe2mXpn4kcYH2F4I8x7qi4NH1mDc,1408
jax/scipy/ndimage.py,sha256=ei-ZfRiDXrlY4nYC1SFL7nnGHFrC_emD_NAicGhH4vU,788
jax/scipy/signal.py,sha256=nW4h017jShZBLL3J2NMhLwlOv46wK1MhEPeKlPTj710,975
jax/scipy/special.py,sha256=jwos1-Kn5wS2I_XYMuy0tfmVb-KKCRuW9WVrqv3I3yc,2621
jax/scipy/cluster/__init__.py,sha256=sgIbEEx62OubzDO36E3EkqjbYY_3qsErPOTQE6lsj5E,768
jax/scipy/cluster/vq.py,sha256=lkLSlkWFi6TLjwuzTX7hVSiC1IkCYd3OWgn8YtIAJd0,776
jax/scipy/interpolate/__init__.py,sha256=RI5oUXPC-8tKmYURnnrnA-e1LDbnNVOVVqkGSpagEBM,760
jax/scipy/optimize/__init__.py,sha256=cAw1rT1e0Mu-22vBUuJb6Q-CQF8fbIUBlKYpynB7e2s,822
jax/scipy/sparse/__init__.py,sha256=ugltlihEjwgorQxgfziHC-03UeYy4JzdQmTxrt1cnTo,757
jax/scipy/sparse/linalg.py,sha256=wpjjS1G1UNISyhZVFoJMMgIkqs-L8ASdvtQAbvEBsuk,810
jax/scipy/spatial/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/scipy/spatial/transform.py,sha256=cm7lKi8aUepI3cOwTnjrldhFh7hBv89CR_m3pFj3r2o,802
jax/scipy/stats/__init__.py,sha256=_rg_z3SzFW5UNdLOwvm1RgTHyRWiu7LulOePjNqE0Mo,2093
jax/scipy/stats/bernoulli.py,sha256=llkwAlvFH0qNU1BHdxn9CXXOBAPg0ffnbW_WQQ4LKVg,819
jax/scipy/stats/beta.py,sha256=5UIf6_JtqVZfAl1dDc2N3K8SkG0OV1bBkC3vWpY8-Hs,851
jax/scipy/stats/betabinom.py,sha256=5T6965_jHQxG4oR2YocPQRKcDo5AGxkhIyrk2iz-Kk0,792
jax/scipy/stats/binom.py,sha256=XGO5jDk3nRRgGwHQU6KEl9_kQ5X2qF-2TLgvm3ZUvvs,659
jax/scipy/stats/cauchy.py,sha256=MfAMnK9eG2wcutA5FVAyhU6iVYSSNgzI1XFT1xtNIfg,881
jax/scipy/stats/chi2.py,sha256=f-6i-0twBiOrak26tc7bEnWgOODA3iFwEFgi_An9udo,851
jax/scipy/stats/dirichlet.py,sha256=FUHSEmXKmVUneXuxgaIZM28s6GxlusbymdRYEPnDiUE,792
jax/scipy/stats/expon.py,sha256=YClw5PcqAXgq3VD4ukZ19HfTmRCMtrGk6E2BpXb6U9E,866
jax/scipy/stats/gamma.py,sha256=TvP4BE8MROMxQL6YNDQIHSaZCfxumHkjGbCXw9O2iUo,852
jax/scipy/stats/gennorm.py,sha256=hqwz7ueAgdtzs1v0LHTTro2PLKRDjWxTFuu9jXwXOyI,804
jax/scipy/stats/geom.py,sha256=pdp2tCdXOa3CUWyKJCyrFrMR7TBX0La5tVI4ALAETI4,787
jax/scipy/stats/gumbel_l.py,sha256=uGI-yIa7ymknKuVsuol49H-a8ylid993coiWq6mnNuM,868
jax/scipy/stats/gumbel_r.py,sha256=fcbp-XJbmb3DlIDS8BTfKmXhf0CDh3EwMFzmWQ0VMo4,868
jax/scipy/stats/laplace.py,sha256=7Lap3E3_neHYpG5mUFlYxZnwAT1J8mTrNPnn8BSEGrA,804
jax/scipy/stats/logistic.py,sha256=IoHe_TMB0CH5EMAXjmqRMP8LjR4lTS_x9Zf4K7K5e2g,845
jax/scipy/stats/multinomial.py,sha256=hfRYQL_GeszF4sJXx5Y416JgXoXVw5vxtQIt1NmkCDU,794
jax/scipy/stats/multivariate_normal.py,sha256=YYqtLLZmjsLWeCIoDpxUk-rHde-g9unfo6ofg44nSBQ,802
jax/scipy/stats/nbinom.py,sha256=hWMMt7D_rLRtiGjlpHcbOXJF03BfA3v9Cu972twYKCY,660
jax/scipy/stats/norm.py,sha256=qNMz53OSiWOZGt6TZkJZvWAFocdUMUu-DsysM9kg2K4,879
jax/scipy/stats/pareto.py,sha256=6lR5cOlMmlSosez0hLa2BbKA4h74HWHL8Dugqj9PPIs,867
jax/scipy/stats/poisson.py,sha256=OBwWU15Qbjv6HBrob8YXteB9f4StLmYAuEYyu_h14Nc,825
jax/scipy/stats/t.py,sha256=kPmOWYVodYyD6cTxzX98qNxiFeWpDHBM1WwrdK5Ztoc,784
jax/scipy/stats/truncnorm.py,sha256=q5aSLcidwhZqFi9sNjqvFF9RxT_YLcYBMW2DofDoE5M,855
jax/scipy/stats/uniform.py,sha256=KCmUJttnFPbr-_ECG7O5xd-fy8CqvysIhepltRj_6SU,818
jax/scipy/stats/vonmises.py,sha256=NYGQbIKES93vjVumCkF6IihydEhrhH7JMaJFmGDfCwM,791
jax/scipy/stats/wrapcauchy.py,sha256=pcs6O0cdJTyhsN6LGTOxBm4TyuKkm3HRIdxfZVoz8r4,793
jax/tools/__init__.py,sha256=BTyMF0yh0M75AsNwF_0NSd5gHnIVpxhsLznsWkxu3XQ,581
jax/tools/colab_tpu.py,sha256=IPEQ1HZB4xW0MRSPGDeFAw9GoMFaphNQ0kLRIwGypmo,890
jax/tools/jax_to_ir.py,sha256=fFvvhbWBsHJel1cBJJOBhLeacpk-OJVJzRE-hZHxK90,8711
jax/tools/pgo_nsys_converter.py,sha256=C2gz5kMqZRqI8t11HOswanwcGDlrAHF_PTh-p4hkVrI,3407
jax-0.9.0.dist-info/licenses/AUTHORS,sha256=OOBQygrI1Zc94AYJycH1r_BSCwP38Hyv3sludltJ7hY,313
jax-0.9.0.dist-info/licenses/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
jax-0.9.0.dist-info/METADATA,sha256=8FDC7E8dr7mxOLbN4yhbaa5WAJGOU2NWff0hrA4yBsc,13409
jax-0.9.0.dist-info/WHEEL,sha256=YLJXdYXQ2FQ0Uqn2J-6iEIC-3iOey8lH3xCtvFLkd8Q,91
jax-0.9.0.dist-info/top_level.txt,sha256=ISwsLPm6sa03z9QjweSkzTYBq-YpzESArafqWboNrrQ,9
jax-0.9.0.dist-info/RECORD,,
