jax/__init__.py,sha256=Blxt8TPIRlIP9Hge1Znr0p6YwEMJ0msLr3z19E2s4io,10474
jax/ad_checkpoint.py,sha256=N3-hWts5BdssIjnoNm1NnGzACdpQmtjol7fd_priwKo,802
jax/api_util.py,sha256=JnZFSuqh5WBglJwerrWxcIDslBaXYZQrCv6j0jPfu7M,769
jax/cloud_tpu_init.py,sha256=nNdPdZZffdRGb38rEx1YOLXNpymSNt05ZVWSm8lhZ8c,633
jax/collect_profile.py,sha256=44nqGZ4fjcCEEqVGpn6Z8_nDuGFLRtTfUyar9YfGKHk,4730
jax/core.py,sha256=PgDSVfJID35sKK53fpCgC-fNr4o_3s2OrhOIwyoJbSI,8077
jax/custom_batching.py,sha256=MEoypZfGgYqvZrBKtnvL5sQjl8YrjVm11hiFWoTCtV8,691
jax/custom_derivatives.py,sha256=3p2yyjQgpyzB2AUSsH_0vsh04-_GZZ6J44NHLAOrVso,1411
jax/custom_transpose.py,sha256=vGj3GjRJQpxl-MhvYztIedauNBiWMj9nD_aJHQiNaYc,664
jax/debug.py,sha256=HXdNZlaM5LxtEYS6fsH3J4LRgiDVL-5W6AXcimbFS5o,1083
jax/distributed.py,sha256=D9-SeP4MWzq9jw2itQrlm10cn5BtR-9Zjn2YGzEXBDU,673
jax/dlpack.py,sha256=TWzMqAIedhARYUrnreJ35tEYt0SepzrpvTQtmnSdhEw,710
jax/dtypes.py,sha256=7e7GFGTP6YAT5EEl6SjFxkm3MsdozqkfFUzI4lfkD3I,1146
jax/errors.py,sha256=MaZbeSxeRyWRMfq4cj6P0GA9UFykLgX3aURPAKhqbts,1363
jax/export.py,sha256=3GnVQzmZviW7c2Rmih_VKsH7F0Pf-wv7RuyalU7FKvo,1365
jax/flatten_util.py,sha256=jGwQUB80s2jr78XuEaT1_OP2OUyVm8Pk-yAQq-C1SsY,645
jax/monitoring.py,sha256=9JRDF2xgMh8NcGYdk79ytPhRGQ_CCQTJHX7g72jjxWs,1191
jax/profiler.py,sha256=w7VxxnWqxWNV5LRSvIofTb11I4ndaUQKptxjaUq3P4c,1119
jax/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
jax/random.py,sha256=5ONY8K1aaL7gymnR-KLhE_bl6i6rLjgu1KgvpWOqjws,9858
jax/sharding.py,sha256=KiCemR1RSuSFI3K-0mawH62ihEUkTwtrOtjecglcJU4,1672
jax/stages.py,sha256=lcGVsE8R-t3t5Pb6BWLzNelU-_USwzQ8Qj8bgMdM218,1314
jax/test_util.py,sha256=DVJAPxzZMKdM01Nhx7s2kPe5mcdv8ZL8a_37sgyZPUs,835
jax/tree.py,sha256=PTjPWygVYFjdLA7CLayik2QMP8K6EIbNsd4AuyhEpBA,956
jax/tree_util.py,sha256=ZCuOkjx2V6A0EVV3e9sp7JtvhI5sxHF7u1-Vqpak-es,3125
jax/typing.py,sha256=CrQpi9UTB9MYfg-zEsqP3n2qe28hlY_QEsgY-9mrtAs,3331
jax/util.py,sha256=OXU0bu3n2-y901BwEhxd1CvOeGoEBaXPdvYaVNHOT3E,1135
jax/version.py,sha256=sUUdz8imakjJr_c7oXQsLoO0FuFtbBSARLAx5sCU4R8,5877
jax/_src/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/abstract_arrays.py,sha256=pTl0x1Q7nQCrKEk4dmoMINW6hQuJ-Bezwp5Dd6pw1lI,2577
jax/_src/ad_checkpoint.py,sha256=_Z2l7tDvv2FMYG6qh-ISh4aNaOvPTjIFqXJVDIYN9yI,38741
jax/_src/ad_util.py,sha256=asumAmqWn7RJXOxzo8WeHMDwIk9znu2-AV9v7W7SYcY,4212
jax/_src/api.py,sha256=TX0ckdFV0CL0NJrmDtC9sZ5UAd4QrWxfY_l5TG26jdE,120909
jax/_src/api_util.py,sha256=ezjkpBmwteUPF47FNqATvRXAjbg6OVV4hAlPcivDlfM,28107
jax/_src/array.py,sha256=mgDquAEHNmGi3RZ7fDa06NGcAc0tEI1Yn7qTqiqF1H8,46402
jax/_src/basearray.py,sha256=8rNil89gqk9_q3ZcwmvERdMHAqvLPx7A8SVVAgpRMYQ,5524
jax/_src/basearray.pyi,sha256=J0xiRDHLGV3QCpI6LPrhZeI9Hj33ATFG4xUKsp1y6Xw,12820
jax/_src/blocked_sampler.py,sha256=_Yow0h3WsUkChkB-ySxQ2EMEYnmcIs2lsD5EtpUmUQw,5686
jax/_src/cache_key.py,sha256=Wqk9HBjiIgilCXVoe7NEaWKRD8y2bniO6eqIKqwCS_Q,11359
jax/_src/callback.py,sha256=OTPOdx1g_PJmn5P4ZNY78jFEHsXY-1ckGWJReRTl3i4,18539
jax/_src/checkify.py,sha256=C4NFCW7uSLOrfO-UqB8BOoyKxj75oWjfrZjSSpNOowA,55872
jax/_src/cloud_tpu_init.py,sha256=WMPDU3sXejhYFnykJA_hiClC_zMsLzRujZu9FuVFPnI,3405
jax/_src/compilation_cache.py,sha256=rhBv2tXVyuK9BjZmdwzBIzQ7DwwINkxzKcvTekoBFKs,11129
jax/_src/compilation_cache_interface.py,sha256=T-39MQ5ygMI96IHmJIyBwUobcQwMTXkANIGpCyr0eoo,887
jax/_src/compiler.py,sha256=u2op_3F8F8JU3ewf3Xhlt5tBA0FPqgC9r8_WephcQxk,27439
jax/_src/compute_on.py,sha256=JaX9OWWs3KguOeMyNgLvaakYjucngSi4EvG720RUwhs,2091
jax/_src/config.py,sha256=GS1TqRbWyMNNmpZOSKUJjbaO5Kx8T1gWTtK1238p7R8,68340
jax/_src/core.py,sha256=dtzOnp5pi1EpOnNCDE1eK-8Ngy_yqtPIJ7U4zhI9SKg,122581
jax/_src/custom_api_util.py,sha256=uv6uyrXU1BHGzKuAe498W2PD9bcWt1PEa-8UAqr9Zaw,876
jax/_src/custom_batching.py,sha256=PPhjqnm1hJQKmoMr-OYwtt7BlJIUOaDSZ-lb249BqAw,10746
jax/_src/custom_derivatives.py,sha256=Hi9WTt_zYza8xQsp9jLCZ8ZbBQZL_QslF0SDqF5pdco,72650
jax/_src/custom_partitioning.py,sha256=gfzfn2hB6sxr6Uu99CfJrza_cVNR-UiglgUDjh9J7pI,22622
jax/_src/custom_transpose.py,sha256=QD6CdGixm163Vy0yUWtXFQ-jJ67Jq8oQFtdbKAoGAgQ,8283
jax/_src/debugging.py,sha256=_XY-rTEjo_-dARgWAj6R_iE4WPM6wlzYlfEyOwQDpZc,25904
jax/_src/deprecations.py,sha256=HfOjbKMgyKOc2qC2QcDXRyfRtEIbC9K_rucJzRxoP3s,4685
jax/_src/dispatch.py,sha256=ZVE_PlGyeWWAJz9wsXD5ddAfdbKr-mCS4y1nOxdwTsA,23656
jax/_src/distributed.py,sha256=dEO9cQZD8GibtXesCPfw9Q3USyxI5nKjnjWwLAMoKNo,10855
jax/_src/dlpack.py,sha256=pzQdxNalm0JLOOZJyHaLYSAT5DfP_UTPHFl5lBSX9SU,11544
jax/_src/dtypes.py,sha256=eGFE5gs2fpc_rlaI9EJKYSOxhhhUZJQthArUwpSU0xM,31764
jax/_src/earray.py,sha256=zZ_hg_akEIM1LMArtYuFrwKv_TYRXaw_MpIJCG05PIA,4400
jax/_src/effects.py,sha256=ZqHKVxNurnN2g3-PJNdvYeut8_XEjNul8h_Yjojx18Q,4682
jax/_src/environment_info.py,sha256=9Gwd-KeYyMlORpCWQR6a8ruZgnJkom8qT_1RMsjBacw,2010
jax/_src/errors.py,sha256=2yIUoXYA4XLYehScnEoerQIu6geOQskNqQV-_6jerG0,24766
jax/_src/flatten_util.py,sha256=T8p7oExtun3ddhh8fn4XL5_TzZEh6ChM_qiZhmcPaOk,3554
jax/_src/hardware_utils.py,sha256=L03Dapnam8MdnSyBGq2d8nio5gPF51xd_Ufxw2pVjCE,1728
jax/_src/jaxpr_util.py,sha256=Jv3fcRlWFNUt-WQwHYfmLeXOSKAYka9RKEwrkYl-BkQ,6439
jax/_src/lax_reference.py,sha256=SyNb2xHEY_1sa7KEnLkQ9PxFHwKKh3853dlLBksGpkM,18163
jax/_src/layout.py,sha256=HRqNA8RLCEmpxZxnz9Rj5l8G3n4ekelJPA0mDe_bLAw,5367
jax/_src/lazy_loader.py,sha256=G6czW2Uq6WRC1EjDiX6rslMu02cZUGRkKNrvIhXksxw,1842
jax/_src/linear_util.py,sha256=v3P-SRcFVJ9FQDUOT6jRFP7Q7i3HM78BqhZ51E5u76Q,13858
jax/_src/logging_config.py,sha256=usQjXadUzBGKOSNtwPsXDPcZeQDaoT06Gzn7ON46yIY,2040
jax/_src/lru_cache.py,sha256=4-A3WLeb-UhUxH7D5VrDUdjUl_BxahLXs8TvS4upO7s,6661
jax/_src/mesh.py,sha256=Lale59WeSqYCSQLbwAAWBlUtuU1ioiQwL_qjHRBeQzg,13566
jax/_src/mesh_utils.py,sha256=__UdF2brPMxs3St0MItS--ORzG2j1Pzqn1r6EH0MccE,30905
jax/_src/monitoring.py,sha256=ro92p2vr1PU4f0b7Ghk_kkJQBLoxmO9MQ1n3APyBfqk,3888
jax/_src/op_shardings.py,sha256=ZYx-ASvqGPXifwEBC1utdbAJA-3ZCgWM037y6l4Wp4I,3865
jax/_src/partition_spec.py,sha256=Rlw94T11ETt8EGpQgagi4QJ3sBCuQ4uvg7FLTMl8SzY,1689
jax/_src/path.py,sha256=ot2lEjzaYVYCCGoOI8YhZYT01HgME-6dtGpmckXd41A,1452
jax/_src/pickle_util.py,sha256=gE8IrNWtg4hfu84T5rQ2DVAdpozh5xwSzAlRi7UIWcQ,2312
jax/_src/pjit.py,sha256=1LDaldUiK70yoRduIyBJrDhx2E0vdvZQ1j8XgMSougc,118320
jax/_src/pretty_printer.py,sha256=mnGHSMkb8EaXh_1_9dsKOGh2XI_9dSaaAUvFJP-XdMM,15500
jax/_src/prng.py,sha256=csL8MtemeRltL9EboBi-KFjoktGr1Jr1pwgpsJcDT78,43741
jax/_src/profiler.py,sha256=278hQdUi8B7FVXo-YKhY9jf1qXSaBNem50ZAeB6F4DI,16195
jax/_src/public_test_util.py,sha256=8HSKZbFsZXbSsb5d_PBZDcMyEbTh9-DK-CtULw8MdfQ,10573
jax/_src/random.py,sha256=L118dTz6kw1LnRB0W6An3ko9Qf0XnmrqLK8RO7dFCms,100010
jax/_src/shard_alike.py,sha256=BBiCvSws_KhVzh8TruuAUe7hx78NR5ueZKR-Kvl95gE,3827
jax/_src/sharding.py,sha256=cWdTuiyilWxlBEUDxr8P72GPtWwS9miGqObQ9AL4cd4,8998
jax/_src/sharding_impls.py,sha256=9EYEAbUQxnWhY1Zrp4G13xfFLRpaRyHanF5peWVmp80,62312
jax/_src/sharding_specs.py,sha256=UCYUIcQfS6lcA993W33zZI5tfAg5zcefP9cZUyW9NoI,8617
jax/_src/source_info_util.py,sha256=EMqYD1WALWmyEZ1BQu1lpmlHQdVZkWH2_kjNHcqnLTE,10140
jax/_src/sourcemap.py,sha256=5s4zTThhX6PZwgQ4v4UbYL4norg4uHipqgUGdHBn76Y,6640
jax/_src/stages.py,sha256=2OA3Q3WTksq0mw3BnajHXqmxceQL9EaW7Ixcn_RBbvw,28781
jax/_src/test_util.py,sha256=FrwR4m1TeA6w22EiNlYsUxPxkGYwfP1u8Cu8UpHOVXw,75596
jax/_src/tpu_custom_call.py,sha256=pS457mRYf7m_cDYnlL4FpV6mmASITi8k0wEHhLuFcc8,23391
jax/_src/traceback_util.py,sha256=5DgCqqOxQ9JNrreoZx7u3ZglimkfIFxHT2tVrlnOcKo,9103
jax/_src/tree.py,sha256=pQY3bwJ4puT4TZqPcAmmeAdoeHsoG4EkMwFPN-0iwqQ,9310
jax/_src/tree_util.py,sha256=u0B0VFQSv253y-ma6IboriYL_s7lq_yTlZjvkD-J78E,47588
jax/_src/typing.py,sha256=_yNwYCbn30QE4Ordkc5d4Y2bF2M78C5LhK42EggAetw,3287
jax/_src/util.py,sha256=CGCfqhXavF6h-ZFzML16cAY8r1cJQvAFO3Ygua7XWj8,19655
jax/_src/xla_bridge.py,sha256=9Kg0SGUfLc6vFN7E7L4LEjX4IwEmdHL-_9gwiMlFKW8,44921
jax/_src/xla_metadata.py,sha256=8I7y4gHQPVPksnC8T_jEcH2jHB-wJZLET6WtLhSoe5k,1589
jax/_src/clusters/__init__.py,sha256=y2_ZijfznsIKEjc6cUqDLOrJkGYdUz9bnfqt4Th9mQQ,1173
jax/_src/clusters/cloud_tpu_cluster.py,sha256=qwH3okD1Zv6ozm9WC8GOxWPvPwB_g13XCnmbxlmiFyU,8232
jax/_src/clusters/cluster.py,sha256=3cxnC1kNeMWfWt0JiMKRlyRq1YrHD199ifPAg4tSdZU,6061
jax/_src/clusters/k8s_cluster.py,sha256=TzUXU4ksY11X2f6iYjhb5LEGZfr6r5WZM5tc9DQhvBQ,4188
jax/_src/clusters/mpi4py_cluster.py,sha256=ugDhirp00Hh3knxX3qVb5yxSz1d0_O1DuQUcVfyO8rg,2608
jax/_src/clusters/ompi_cluster.py,sha256=Vfk3AimGa2jtDEwjpJJXuYP9p5uv1tBwEKtpGn1f4IQ,2317
jax/_src/clusters/slurm_cluster.py,sha256=hsZQJXznkCXGDTfKgMDgjVf4bmbAtSne6pa0wW4iJT0,2321
jax/_src/cudnn/__init__.py,sha256=GAqDerxlHGBQ6DBTA0PKaJfn2GviIqCG0kpnVRHOJAM,615
jax/_src/cudnn/fused_attention_stablehlo.py,sha256=lp7io8tsE1PDXZvhlb30WVAu5j7v2OUNRU-wVA3Xl_8,41096
jax/_src/cudnn/fusion.py,sha256=y3uCu2pohDXpND3RoycHtHVv0agyKvlSJPkPjsbydeQ,2803
jax/_src/debugger/__init__.py,sha256=VPrkm_4GZ-dm4SN7L4xqXCwGSFiD5K4BkXGy0MvWnSw,885
jax/_src/debugger/cli_debugger.py,sha256=C9FQoMwTZOesOTG0X5dVDT_3nGTExqKZixdPPzTDkew,4780
jax/_src/debugger/colab_debugger.py,sha256=PHmiL-QES4t97iICJt34a5FXSNTm-AaSUERPGCZ3QHA,7823
jax/_src/debugger/colab_lib.py,sha256=jCbwj2Q4f8SEo7CBr6AeiUtzlsIv4BfKa5CVbq4Bn2U,4298
jax/_src/debugger/core.py,sha256=WilcRMqpzssKa5YL_P7SFtKaUVUaXqDutxO_N-ZRnp0,8593
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=FsUQF5RXxGXNCmuRsFGhCR9c1GrOYy1Pqryj2l1EEAQ,63078
jax/_src/export/serialization.py,sha256=GDeQuoYirJqQNDMOBZK7w3aI1MTbA4jbQAjxHGPMpYY,16973
jax/_src/export/serialization_generated.py,sha256=MkKvLYnFffFcwcNfhkodReB33jl1A9_vSPgVxwqQrFQ,26039
jax/_src/export/shape_poly.py,sha256=zjbgDZDjlFpV9zadsFVx8Aop3Ea7q4HZRX2qtZOcJXQ,86089
jax/_src/export/shape_poly_decision.py,sha256=7cEAydAto4UiXlOYdq7oHqp4eXwp4lc3OAY-KKgy9Bo,20904
jax/_src/extend/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/_src/extend/ffi.py,sha256=uYeYQLPAoYNhtWsL8PKJSimeIkbpczt55gtcmfdxrQA,13888
jax/_src/extend/random.py,sha256=1KmQCCYxFMJO8KB0qJz9ZVQhnK-05Sv4Jn0jXLPpWQw,1258
jax/_src/image/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/image/scale.py,sha256=_Hezs_YE3kCj6_cTn6XvMuU26sPiedsiJR8-VVhKEDA,13740
jax/_src/internal_test_util/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/internal_test_util/deprecation_module.py,sha256=iZ6fRYNaUcOwrftwpDXMnlxrLrrQbdMrbybmmQFwql4,851
jax/_src/internal_test_util/export_back_compat_test_util.py,sha256=Ebr7V03FZtqonF3zIHXImgHljXqzChDV6ONbDktEtRQ,15402
jax/_src/internal_test_util/lax_test_util.py,sha256=TCvYcOZfaAJ6ktX952MtC3vE1f0Ce0iIE_FhEdRkJIo,12140
jax/_src/internal_test_util/test_harnesses.py,sha256=uyZ7T9vs9L3csziypR0xpcn2R8KZPlRpT91dk7NoXCI,124454
jax/_src/internal_test_util/lazy_loader_module/__init__.py,sha256=UP_c2VfWHRhfEi6n4F6qLHZuageXhPiVSAcUKHrrxFs,705
jax/_src/internal_test_util/lazy_loader_module/lazy_test_submodule.py,sha256=luZv4DGyWwdcBX0_NBVwu81IE4jFYVbRkKo9gQPmxPk,610
jax/_src/interpreters/__init__.py,sha256=hnuhN7h1jY5__Kt2rBvTrk0o8nX4fQl244fDCIMh0qk,690
jax/_src/interpreters/ad.py,sha256=eYOMEHtp-fCfbfH0d_OpSqvxfrhWmkJY_A6ZiuT85Do,32367
jax/_src/interpreters/batching.py,sha256=vuQe-HMxCY1ROFwNCJaVOICbZZiBZOWLMmasKytnbos,50898
jax/_src/interpreters/mlir.py,sha256=48A6szki6zuP0He080s9E2tAfmQVK4Q6I40n9EtSEJQ,129203
jax/_src/interpreters/partial_eval.py,sha256=D-aQN1wlouDH9nFsM6zCJmJjFKxEUoLVuwNKKrXmgks,123236
jax/_src/interpreters/pxla.py,sha256=UaAhWuckXvzM0QMN_zD5bk4RogIHp9j2Wh0n1n-dIes,129210
jax/_src/interpreters/xla.py,sha256=Ok2_SEAS1wNYa_ofTPekBZdhbiPRp2-pRmWlHvvPtyM,12187
jax/_src/lax/__init__.py,sha256=jRL3qQoUco8BVUAh8lV-KuY38h_4GxXcb41asljW2yI,690
jax/_src/lax/ann.py,sha256=53ap20vxBD-hV6L1JZsLNfzyACtK8qkE2fmPqsJivvk,16761
jax/_src/lax/convolution.py,sha256=G5neAsqoqEgN7Tz7QmHIf3zSlhtOi58c3KnOUL6T4RA,44638
jax/_src/lax/eigh.py,sha256=GhrwPlBnfUlCa8e6XNq40kERV9-knvD4j5rnntxdvdQ,22235
jax/_src/lax/fft.py,sha256=HDXCtJWpzmDezq3FpJqxj0mmlcmGJMLRxfBXUoe8lFM,7173
jax/_src/lax/lax.py,sha256=m8efmiT7JT3mszc2hDN8DvbCcNHFspRwCFYYvqZxyFg,240105
jax/_src/lax/linalg.py,sha256=Om5HBBN8vwjjWJwa_ye7tiIMTNgKAPbpejTMVSBouTQ,98528
jax/_src/lax/other.py,sha256=K_TGgJ3KYRQeG-nX0txlCyk3u33O75iK8233wv9o-NQ,12374
jax/_src/lax/parallel.py,sha256=87i-E8kSuakf2sLfMcmiizN6H4ZrXkJpb-hEAhEWa_c,65712
jax/_src/lax/qdwh.py,sha256=dmslOVnzg-DhxjSIHV9jHCOKopVosYD-xe1FPmSgnCo,9391
jax/_src/lax/slicing.py,sha256=kWkildXd-l_LZCK2cx_3Xhb1eZAIYT4viiviUPmsQpI,112018
jax/_src/lax/special.py,sha256=djqaB1CuQX53mqxeHQhz5hV6kZGN-HVhtzAwqezU5dY,26940
jax/_src/lax/stack.py,sha256=7hxK1zlVa_Q9gO48Ymj8S1nA8DN6keYgcOY0DAtbjOY,2560
jax/_src/lax/svd.py,sha256=iVY3iFwt0ug1ow_SkFIJd0lNYdn9eXrCKhNbZ0Dh4Bc,8794
jax/_src/lax/utils.py,sha256=DTHX1f_PQ0Wxi7pdiizQppJ-lXkPDajp_yrJPs2u3AE,4767
jax/_src/lax/windowed_reductions.py,sha256=C9wqD6qF71C-zcNdNdNzRqvam138wfFBM03kttacM_A,42356
jax/_src/lax/control_flow/__init__.py,sha256=wtJfgmITbPhBhnPASAkvCIKVE5mHFU9DSkCthwggdks,2097
jax/_src/lax/control_flow/common.py,sha256=Wpr7T_Mm9I3WHLCO_aH-LxjyVWK6h0DtuO_BpMzLlMc,11041
jax/_src/lax/control_flow/conditionals.py,sha256=KInU4EGZALjkx1mhqrvCsWlobCIL_uLEaUw2jGBnJ68,43996
jax/_src/lax/control_flow/for_loop.py,sha256=10MAG6zkwTPRADA57HuFXL0QiyFIh52Na5GAB_D5tWo,36058
jax/_src/lax/control_flow/loops.py,sha256=u5ctTANfViakStOMZn9Z4OvuvFRfV5KhSXXMDPjP6MM,110528
jax/_src/lax/control_flow/solves.py,sha256=s_mk-cxA63Sg1gn_I3v1T8PWxj-YL1tkp-7MxDPXVFU,18959
jax/_src/lib/__init__.py,sha256=YcWdnxi0AXAixsvK_WzFfX0-gAQBzcDFyT5ovazarwY,5392
jax/_src/lib/mosaic_gpu.py,sha256=lHnM852MFg7K5UUwry1zN_r5nrPSWl6ZVwOxVnsWAz4,898
jax/_src/lib/triton.py,sha256=6zzcpqutI1cZW4sG9tLwPFyPDw35xhpPoNH8wkG-VbE,662
jax/_src/lib/mlir/__init__.py,sha256=UoirBDcNhqLa8Gb1Qrxd4wYxScz_p-1tb-ewbTsK1jM,733
jax/_src/lib/mlir/dialects/__init__.py,sha256=6KUUW3fBmKZn0eM7VlIFh5yiSLRQ9PNC1Cl9tAxoF2o,2032
jax/_src/nn/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/nn/functions.py,sha256=EBMQ207jRk_E4M13BxOA1P7umsse5O3cvFyrk6hpIV8,33478
jax/_src/nn/initializers.py,sha256=DzkIkSdy2_5s6iwNGzNo9gut0nu-Nwhuij4K9hkoDDk,25503
jax/_src/numpy/__init__.py,sha256=-QhmzWj2qFbKVIT9TDAIvQWaKQibiRXy3r1zQkI3X4M,581
jax/_src/numpy/array_api_metadata.py,sha256=je15NAJj_CFlgqeDqo5iSg9EzmN9IwCkZ_E8TtStKcg,3347
jax/_src/numpy/array_methods.py,sha256=rJTkTzw7XRGMYGmp_LnBmd_VAoi2EWjNtkC18OJVpQc,45056
jax/_src/numpy/fft.py,sha256=zErL_2pI-ed2Sb-Lljh3F9D1T7bMQesbjYpLAmNkjOQ,49879
jax/_src/numpy/index_tricks.py,sha256=y5xMJNntEu86_18ALVKc5sB_xNE83XqLpVNLYSsT10g,9888
jax/_src/numpy/lax_numpy.py,sha256=0QsUmDWlZJHK9un_ZJzC5D_Zyw7G3n_72ULopxa325s,439830
jax/_src/numpy/linalg.py,sha256=tFjsnJLvsCf5rDH5w-ZC2zN3bykTrg_met2ASKMbaNI,77349
jax/_src/numpy/polynomial.py,sha256=qrnmH3kreK3tESf6I5kKW1KVT0yjwscDwE0Pe3edKIs,27754
jax/_src/numpy/reductions.py,sha256=EP1a37zS9V89ctd7H4NFGQv2PgBYCHMgokSyitajw3Y,98514
jax/_src/numpy/setops.py,sha256=TAk4vQ6N9-P533Mhl0CQzEBFIQHCmYLTP5Px9SFeLiw,47279
jax/_src/numpy/ufunc_api.py,sha256=ik6r9MIGuPA5Q8CneGrTWTt-NVVaai-bgd4W6KAygoI,23508
jax/_src/numpy/ufuncs.py,sha256=ohQnQgM6pmTyrxrcGW4Qb0q8mUa54MwcqA6TLULTcHU,96067
jax/_src/numpy/util.py,sha256=WZNRoFGqvhs23Z9y9-nNbMIG8nH7CxkR576WqoyY7AA,17831
jax/_src/numpy/vectorize.py,sha256=kkpGhVAIzikmzmc3BxbHC-WwsTpYvOy9bt26UrldlOo,13964
jax/_src/ops/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/ops/scatter.py,sha256=ORkZSW5StWhAnE_ZWHSyvficC304XVYQSV18KTTTR7Q,17484
jax/_src/ops/special.py,sha256=Ypg7maHS3_-cklEXoTSwrxReDVaEsUHx15D3Zk9BeYA,4151
jax/_src/pallas/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/_src/pallas/core.py,sha256=CoR9iPwAhZrAqh2YTIw5RkQ3gbPdvRntKXKadUrD3Qc,35910
jax/_src/pallas/pallas_call.py,sha256=oF_fIHM7ZiZjIlytIr-uU_rIyVJjNJY2l9nLOlXP_kc,64031
jax/_src/pallas/primitives.py,sha256=bR04y743bFyQwoNe11LZzK6RGcKgjWGwxFoPoCJCnrY,31047
jax/_src/pallas/utils.py,sha256=nT2Kxs4KIhPVana1v6tvm3HAEnWO6y_y5GWx-kPc-AI,11060
jax/_src/pallas/mosaic/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/_src/pallas/mosaic/core.py,sha256=gkA5ViWbFeRMGElkTKnC4i8W0HiwnY5dQNVKR8habi4,6774
jax/_src/pallas/mosaic/error_handling.py,sha256=ABMM-DgazL1QrL08LALIXiozl-aCA5Kv7m6QAecoEY8,5529
jax/_src/pallas/mosaic/lowering.py,sha256=pt1lyU6Fl6G7_V1FXQW_y6WFu_yQZV93-1dLXIhyDZ4,104158
jax/_src/pallas/mosaic/pallas_call_registration.py,sha256=5WDfWcK_ZNMTXBwYQO053rjAmiesN-F9DLCr67bcYu0,9592
jax/_src/pallas/mosaic/pipeline.py,sha256=t16wtBrCRztvT49LbgBlaSoBp3VTKoch04fNLSA9JVo,42919
jax/_src/pallas/mosaic/primitives.py,sha256=dm6_aoERl5SwUxR8J8nIyf1hmoqKoYD74A7BzWCPHp8,30454
jax/_src/pallas/mosaic/random.py,sha256=HGnHHWRrTPm4JSC6hKv8ss7SBwzTIX0R23UKk4o05gE,7791
jax/_src/pallas/mosaic/verification.py,sha256=7uKOMn-UohXETxuS9OPH33I_RBx-PQNQklNEE8qyDA4,21825
jax/_src/pallas/mosaic_gpu/__init__.py,sha256=lgXwIUt_WCtu8ExBLbJk7dd6SGhoHZ_Y30TLLy3qhxw,1527
jax/_src/pallas/mosaic_gpu/core.py,sha256=Ytjxz1EDJp16H5rj_Beaz0f5cg3SyyCwE92QIDoiawg,11793
jax/_src/pallas/mosaic_gpu/lowering.py,sha256=C7ieIiJ_TDbnMU2DMUmqQSx9MnjLU2ZpCmkhfdMXYpE,40572
jax/_src/pallas/mosaic_gpu/pallas_call_registration.py,sha256=iVRKIrWxEMYScZ9K-5ASabpeIOpgqg3vfjL6hayIgtU,2303
jax/_src/pallas/mosaic_gpu/primitives.py,sha256=28BEGqO3_j2ffLXoscanGkEXJDz9SyZtOmD41M1p3Lw,9036
jax/_src/pallas/triton/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/_src/pallas/triton/core.py,sha256=O9OecyoUxyRJXmHHN4Yh4HcqjCDA4L9BrvMaYLZnIFY,1372
jax/_src/pallas/triton/lowering.py,sha256=TE_p9FFxu5PiyZPq-RwZzzTPVe1UNoBxjxCpEKuWkH4,88291
jax/_src/pallas/triton/pallas_call_registration.py,sha256=lX-Z9cj2H6imyOYMaxA2UmWZeLybltVNBOTKv_LfK6c,4395
jax/_src/pallas/triton/primitives.py,sha256=on8o0zgU7H19hChzji_twx3QAORFLwOEDxj4Zgd52V4,4223
jax/_src/scipy/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/scipy/fft.py,sha256=rEOuNrkRnMOofq0i8XunDVONpgMyUws_vn7HgGCjHAs,15266
jax/_src/scipy/integrate.py,sha256=4VtR2wYVO037Z0yiOdQbniKtpxL4boZQNIppy4J9qy0,2346
jax/_src/scipy/linalg.py,sha256=cAf3vBZgDFu71-VVbQa1WBgbbeB29HASp_OKS7wXN84,75010
jax/_src/scipy/ndimage.py,sha256=IuE4PMzY5HuxST9D0G0On1SVFRpNF367Y2cIQv8duTc,7092
jax/_src/scipy/signal.py,sha256=E01k5pOeAtg2has0eb8GRoaogdfBWNYFzhcCdHt7dAs,47453
jax/_src/scipy/special.py,sha256=PncIKVBzp3ixbQRUWP0uQhChVLvii-kYgvf9ejZgt8U,80205
jax/_src/scipy/cluster/__init__.py,sha256=S-o0CP72882Xv3l6k2PlrEx6eNvXBTk5Oaed5KF4iL8,581
jax/_src/scipy/cluster/vq.py,sha256=3Xfr7esTCzSawrDGZ37hu_MXJ-EqnZtff6hYmXhz0Fc,3043
jax/_src/scipy/interpolate/__init__.py,sha256=S-o0CP72882Xv3l6k2PlrEx6eNvXBTk5Oaed5KF4iL8,581
jax/_src/scipy/optimize/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/scipy/optimize/_lbfgs.py,sha256=kqKkdDeRYiH5prawdntOXfCs-g311NYwnY57jV8DDbw,7769
jax/_src/scipy/optimize/bfgs.py,sha256=JRdDBxnfNrVxK5UDyehFGF3eQTJenWfdZVmKOMheenE,5609
jax/_src/scipy/optimize/line_search.py,sha256=E51ROBUJYmanL1byuPpIa28HmN1vCf_BAaBC_dQZgoc,13235
jax/_src/scipy/optimize/minimize.py,sha256=mCojZkL0TXQhix6w0-GroM7ITU2rodYdDwmvynLsqWs,4979
jax/_src/scipy/sparse/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/scipy/sparse/linalg.py,sha256=IGX8lZ2il77S3WA50Lz7MdBiw7lqdD8mHgWRaCUHX6k,27245
jax/_src/scipy/spatial/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/_src/scipy/spatial/transform.py,sha256=hz2kXBC7VlrJfs_phRbioCiO2xWDt6N0Fo-KtiLfST0,17513
jax/_src/scipy/stats/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/scipy/stats/_core.py,sha256=dUhvAaJQoGBeTEdmlZdfd8JPI69M-duDxa3ITV6ax34,10620
jax/_src/scipy/stats/bernoulli.py,sha256=zOvo5ZLr8y3O0byR0cp3JRPKPPHoz8oVyG3-C_5-NJo,4541
jax/_src/scipy/stats/beta.py,sha256=h2LF8lIveFCa-qNZ_XLD1hQIK9_dLSmlCjrqeSdz_ME,7988
jax/_src/scipy/stats/betabinom.py,sha256=dmhOAUk6py5Fnu-jR95KnwJ6suvvfwNF9CXNHz73CSc,3447
jax/_src/scipy/stats/binom.py,sha256=E7a3go8bCM5bSiC72l1pDQEuuzSpXLmgfG40iPF90wA,2825
jax/_src/scipy/stats/cauchy.py,sha256=26eHaNIfxwh1og3kTqnPYgivH0GTIuWVwElai4VJwv8,8884
jax/_src/scipy/stats/chi2.py,sha256=vPFrhpbLRPDheYenmHQ7QvZrRt8cAyRtOpU9VmemNIY,8008
jax/_src/scipy/stats/dirichlet.py,sha256=_wWyx07z62_EH25i3J4jS5q53oj0UH5A3-XgQzapwbY,3198
jax/_src/scipy/stats/expon.py,sha256=8q5nUo_N8Rl8BjMVfFKv2ydB-x7-TkKtIdZi5FVrbl4,2268
jax/_src/scipy/stats/gamma.py,sha256=fzZUpOFkFrxqjbx3fzRnQNd9zp_6RhQ-m-my2MCjsds,6957
jax/_src/scipy/stats/gennorm.py,sha256=kp22L5H-rR5n7sn9rpeWsWta-F76yZaXkO0Ovvf9zc8,2989
jax/_src/scipy/stats/geom.py,sha256=2hQ4LoaqNHFBahdRTGS9oyWMYdFQufJb2ByqOw9709s,2243
jax/_src/scipy/stats/kde.py,sha256=Mi9nMB6q3eueSbyNMfs3DokLhsjte0lYsSIdAZ9Birc,10141
jax/_src/scipy/stats/laplace.py,sha256=9PlK08r9wRHelqSIuKkv8KGO0oPV2AArI0Fs6rICFj0,3269
jax/_src/scipy/stats/logistic.py,sha256=HB3Brjwwl-yyGB_9K1lpdiQlmitsHwNLfgl0g4owFSM,6271
jax/_src/scipy/stats/multinomial.py,sha256=CVgpOPCVcr_wTNuF1vXcUjgsENe2mVO7gYBYTSdCOwI,2469
jax/_src/scipy/stats/multivariate_normal.py,sha256=Dei7nhj3j5gB1lBi9q_ZpdMHi1UY7XCr3GTzNuaWF1E,3433
jax/_src/scipy/stats/nbinom.py,sha256=kr6FLGYROqMMtfLfWLI1fumHBl_tSX1_Ys2icp59Hdw,2603
jax/_src/scipy/stats/norm.py,sha256=pUS-JhrjrFtdbfO178IeOJizZBU_WjXU6aUr8W67tQg,8401
jax/_src/scipy/stats/pareto.py,sha256=WcRAchdj198RF_W3egR8Bteok1dXE0PHv9ygw-pQYtM,2596
jax/_src/scipy/stats/poisson.py,sha256=Y43e2hY6L89go0Vc1HJCDrf61-vDZcQVIC6zVJdaYyQ,3348
jax/_src/scipy/stats/t.py,sha256=I32sQ9TVaes4keK5AIK6I-vX--nvl4qtZctuc0zIhG0,3221
jax/_src/scipy/stats/truncnorm.py,sha256=V4_Y2OEMYPepPyWVqMZKO0hRV8HwwnI6_djK9xYwdDA,9239
jax/_src/scipy/stats/uniform.py,sha256=pR3NYr6ZHTs6xhw-LECsRJqp5l_iBbsw689QHxr9Puc,4361
jax/_src/scipy/stats/vonmises.py,sha256=yxvmgBI_HWQn8U3u7wtLv9PjqOk7hRX8I4gWKfqRZKg,2451
jax/_src/scipy/stats/wrapcauchy.py,sha256=SpHIjzS9j3TqgAMTHXxtDz73aBfezqYUaNSsGYCky5I,2371
jax/_src/state/__init__.py,sha256=y81PmPpa4deaKYs3Soh68PIrEr1FsRqzipn92O6K5OA,825
jax/_src/state/discharge.py,sha256=_Jrv78QtbethKGzPPCmGj7PC9Aq5yX-LcyChAU7e4kY,40429
jax/_src/state/indexing.py,sha256=9rtibPMDHrTan3EoDTQCbvLtc339H8Nhpa11dZ26hJM,9400
jax/_src/state/primitives.py,sha256=4JKHrIez8oKzNYh9WmMWkxVvLin_efBkojmwMkBHmoU,24163
jax/_src/state/types.py,sha256=Mk9LrBMLRhIz7hm8pDblNSVObiD32_bZC9aFl6qy-MY,10470
jax/_src/state/utils.py,sha256=fv7oalhAmGZhOmhdvGcZo8Z5vbtlda-fx8mPfXx07p8,4100
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=TT77MW5HcPt-C6m-jHT43mGPLuAKO9ANcoEsXjl8JDE,2373
jax/_src/third_party/scipy/interpolate.py,sha256=5C9-SwIUrwlpBUgAKYskaSLgTbOtpwESJRPv-6cjj0Y,6307
jax/_src/third_party/scipy/linalg.py,sha256=SzmpCoi4kSnLTflDsRa5QdoBcijKHuwmtN9xac2TWlw,3786
jax/_src/third_party/scipy/signal_helper.py,sha256=Ygxut_RD0Tixc4Gt0AcW32QhNTmV5EBxLYn6RjTUipU,3495
jax/_src/third_party/scipy/special.py,sha256=ojE5A5mClaexr_qca69NYpBTgvBEdvVaVMRmptVh0yE,9155
jax/example_libraries/__init__.py,sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw,581
jax/example_libraries/optimizers.py,sha256=T2D3qDPWZ50kwTyDe2IXPPLbOUu04cQTdf51oHdXVvU,20426
jax/example_libraries/stax.py,sha256=jNCoL5RlFv9JnLssUHthGRaNcA9xY57BKPxF76Nz7u8,13862
jax/experimental/__init__.py,sha256=OQS_hX2SykwFBXHl4t0VHv5WKOsqpLjuVL5wXg0CZik,1012
jax/experimental/attrs.py,sha256=q4iyf45X5uCnPiRSYLYIQokxA3Z2aqg9pAZWCgQZtQY,9463
jax/experimental/checkify.py,sha256=V84hTTNrwnckaRzgvWDmpcKKA95kAakjZwwlfnRaXt0,1213
jax/experimental/compute_on.py,sha256=Mdg2n8JtAsrq1ap_PwUPwoqpoCNGO5rjf7T47e_iaAE,644
jax/experimental/custom_partitioning.py,sha256=E0IrgU4LX1P-khUlG69nf9d4SPi9wbaBmyTiyvE6PbU,856
jax/experimental/host_callback.py,sha256=DYHqsf4u9SoTn4aDpTv9qhjIhJcn6gYJ9qvvv7CblvY,90327
jax/experimental/jet.py,sha256=IUxcSJxmHvshrQk0KFA2El5ipEOjlVfeD1exwaAE5pM,26855
jax/experimental/layout.py,sha256=q2i-_DZq6t3TaTmtkRSwh5CiseuwbH-VMqIMBUL4ItY,675
jax/experimental/mesh_utils.py,sha256=X4EYn9ZS-cEOBncDPtqJZA7A4FDEOWCAw2YP7qoJMNY,930
jax/experimental/multihost_utils.py,sha256=64DX0s7MVCw57_uW8zRtSyCwGE79_cEvDC0k78kBXGY,18584
jax/experimental/ode.py,sha256=5MJfwT04Jgk0_J72aGLuMVPJ9J_B5gEtOMs8g_F6qnI,10651
jax/experimental/pjit.py,sha256=Yq-bII0QH18i7VYcphiiQmvx3xRcr8Ktsr1ijPhVepg,808
jax/experimental/profiler.py,sha256=DJ0ETyXEZ56c6LYWRY_ankgPOs5oWZejGtGXS-Ttbus,1424
jax/experimental/rnn.py,sha256=JfyOKcTOYm-78pDjuVhXb4qIJs4LF1f6cOExHQy19bY,18666
jax/experimental/serialize_executable.py,sha256=-ytQl3LPU31nJnxHC-eZEJcHx724XiCm9-ThYj2qaUk,3041
jax/experimental/shard_alike.py,sha256=dbrhVCnWEX39TsQYRyGoClFByDmqtmDlKGtOmlT-m2M,646
jax/experimental/shard_map.py,sha256=dBLLGdVa7SOtRCMDJ003rbAL3vk4vn1assL1Bg2A4Ao,95423
jax/experimental/topologies.py,sha256=DxNh3ErC2h9eUrG8UzN007aE-r_EiZtU7R2NeyWkbus,2140
jax/experimental/x64_context.py,sha256=lUrmtrYhu0ZGwd_j8vrzFqQnwfDptgpHxZbqV94Hb30,1710
jax/experimental/xla_metadata.py,sha256=YoNAzPbdARYfGzGTUVoxTfHG9dDrobUvWaPI4uvHRJk,658
jax/experimental/array_api/__init__.py,sha256=JYhDscKw62F_3Y0uaTh2SrJADrKBOOQwwavlvXo0KKA,1090
jax/experimental/array_serialization/__init__.py,sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw,581
jax/experimental/array_serialization/serialization.py,sha256=jD5zks3QyWCqd05PfPVwKGt5jNwPu3vnMQ-kY2MfEIA,26145
jax/experimental/array_serialization/serialization_test.py,sha256=lzAZ_sa9uf-tB4STu1rP4LJBBWhSpIqUNcutCpgMdFw,22984
jax/experimental/compilation_cache/__init__.py,sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw,581
jax/experimental/compilation_cache/compilation_cache.py,sha256=OIZLvS1c3Yw_EbbqlV28pC40g5dPdBB-XDrXnHcT6As,820
jax/experimental/export/__init__.py,sha256=voUU40tcps7sW-o3oVUbT1yaf_jtbsSRdBDFjUFFQGs,3344
jax/experimental/jax2tf/__init__.py,sha256=ZqsPlkalGAkzdTwRIutfW1k8NmOHzJn6kfIB1A98pdw,944
jax/experimental/jax2tf/call_tf.py,sha256=sG0sTtog95aDvRMHsy5vH1cJ-yhZC-ICLeyDedVw9-M,27754
jax/experimental/jax2tf/impl_no_xla.py,sha256=-tSLHRewsQZUAgD9YqdR76yMxTk4sOQN-oJanjgPsSA,54543
jax/experimental/jax2tf/jax2tf.py,sha256=vLkTrVo-mQQxbhw7MNompvV4y788lj3hh8pE7DexpjM,144362
jax/experimental/jax2tf/examples/__init__.py,sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw,581
jax/experimental/jax2tf/examples/keras_reuse_main.py,sha256=AN1R_wzL4NyCgMYEb6ab8fos2vruCL0PGj_GYAEWNEs,2936
jax/experimental/jax2tf/examples/keras_reuse_main_test.py,sha256=0nnk1mDlssNRhOIG4u1_WSx6KXF9C7F5Eih6f5hFWjw,1670
jax/experimental/jax2tf/examples/mnist_lib.py,sha256=mW6LDrwf14wtTBj67clOQbuQz4E_sPINlxs6PnDuyHE,11524
jax/experimental/jax2tf/examples/saved_model_lib.py,sha256=czppig2onwKslN38JDp6qx3z7iHzFqL_gAcIVmfrAN8,7031
jax/experimental/jax2tf/examples/saved_model_main.py,sha256=wuf8oOJ5QNr3R71ltZyAci6o7oKKMpvyn_l96ISMm6U,8003
jax/experimental/jax2tf/examples/saved_model_main_test.py,sha256=Ij1lZWIgrePGW8aElclPa2F9-ZW_msbAsJSTMQL799g,2467
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/jax2tf/tests/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/experimental/jax2tf/tests/back_compat_tf_test.py,sha256=xOP6CFs9sD3yKiI2vkk0W4rPKdB5T5C_4IA1E5-H09g,5360
jax/experimental/jax2tf/tests/call_tf_test.py,sha256=SDD02UKUgG8O-G-cLWcrDml2BaNy2fGmuu2TvqeylnE,66083
jax/experimental/jax2tf/tests/control_flow_ops_test.py,sha256=sqYQJWhcY-awqj6u6IF26qRBF7xPzodkN2XnZpRMleg,10108
jax/experimental/jax2tf/tests/converters.py,sha256=6VtplsM_xFdZxfIPzSkDDypkkBZCXPWWwoaBm-IPcM4,3557
jax/experimental/jax2tf/tests/cross_compilation_check.py,sha256=JaozDkwQO-Aaf1fuooGumLG686s2nnE3OB9fHd6aWwc,7372
jax/experimental/jax2tf/tests/jax2tf_limitations.py,sha256=FWkCsZeRHZq93XP2No2CtiO4uLq24AcLbh6kLns1Frw,60812
jax/experimental/jax2tf/tests/jax2tf_test.py,sha256=SuA8baS1t_jC5jVP2Y2l111Lc_znntdz6wxFbpa9cto,66458
jax/experimental/jax2tf/tests/jax_primitives_coverage_test.py,sha256=h6G_o2ctliPnuSF8a0qB-4Qz3XxXWZx_RXak6mlR454,6656
jax/experimental/jax2tf/tests/model_harness.py,sha256=zhPlq2VnostLmvDfkPdE74A6UPec7wLRS6c_0oFwaLA,13934
jax/experimental/jax2tf/tests/models_test_main.py,sha256=qk_lozA90084XzxnrkmjzLd7RvEGfUIMWNM3v2FG-ac,8928
jax/experimental/jax2tf/tests/primitives_test.py,sha256=Lbs2VymWPjAqBluaHbCHsRWdbX8r_9qoLnXrZDHjLWA,14520
jax/experimental/jax2tf/tests/savedmodel_test.py,sha256=Ch1luggaR0F9_jRSLAovuqRSCnVw3EIUmYBYfKKJRE0,13067
jax/experimental/jax2tf/tests/shape_poly_test.py,sha256=h7qldAUkuRT_y6yMMy_oS1L7yU97vU2Ji0cbiwOmvLA,127641
jax/experimental/jax2tf/tests/sharding_test.py,sha256=kvR-ioVdXdps5OKmaqu2LtM25QRuTNi43_AbkcTPMks,25905
jax/experimental/jax2tf/tests/tf_test_util.py,sha256=YIli6L4hltJraWYyPbK4hOTZZ_64UafOn45MbMNpkGg,19962
jax/experimental/key_reuse/__init__.py,sha256=5CjqLFSLnAhm_YyrowsWA48ENwxcvcD5KQrNQJTgLpQ,1688
jax/experimental/key_reuse/_core.py,sha256=RydzFNDWPeUXcSdkPnrkW4WXOVtuWODVX2N8OOxzJdY,22816
jax/experimental/mosaic/__init__.py,sha256=jurBAcW7zN1OrDKqkEsEEfWNzMYpvurRg9CFEnIFrJw,835
jax/experimental/mosaic/dialects.py,sha256=seTU0P-foAZipjdeIjopV59L9G1uTV6VkxFX4qFaP7Y,762
jax/experimental/mosaic/gpu/__init__.py,sha256=sJDJJO-XDjd15Ex_TeKKPeiv3oHOYJZsSTpzrBmbBjI,1573
jax/experimental/mosaic/gpu/core.py,sha256=prz2P_NwpWO5QrMwNjqxfMN5umfNXeobFIPgLu_8F88,35116
jax/experimental/mosaic/gpu/fragmented_array.py,sha256=00f5QDJEZbHEMh_cqng98APK5NEEN49TK7gkHD8G2aI,36281
jax/experimental/mosaic/gpu/profiler.py,sha256=9fy3sPtNADYPoRDEUGYNcf3lvnWHjEvqDSRj13U1wv8,10892
jax/experimental/mosaic/gpu/utils.py,sha256=bkJNdzAbTxVsHTH5D6xtBV2_D347P_58gJn2bC71X3s,32011
jax/experimental/mosaic/gpu/wgmma.py,sha256=sbHVdhp957LVl-iVtQPLoq0Kv4zd2ap5Yoni8rjaa84,18025
jax/experimental/pallas/__init__.py,sha256=hjTzF9dC7PP7ZfBExaTo8A_IuQgskboTZoO-wF5bUFA,2686
jax/experimental/pallas/gpu.py,sha256=iZ-lwEKNvbcNJJ_q891XCIlUGGpFbra-CLvw6BJ10D8,866
jax/experimental/pallas/tpu.py,sha256=EcsOuRfQcxZyEg_1WKRV-luhbvkDSHLW7_0iHq4cQl4,3382
jax/experimental/pallas/ops/__init__.py,sha256=vC8qlznuwgbYeFDJqP8pDUIf2gEKFrCHg3EZYdTkiwM,764
jax/experimental/pallas/ops/gpu/__init__.py,sha256=mS3YSYUZFIG68T5Zo9CFGkT7wCyWRycnWV4L36CnHYU,581
jax/experimental/pallas/ops/gpu/attention.py,sha256=XDPHeeeII6FZ21w-Eb0vF8UKCn-2szBWsGcvw3vnm-w,19429
jax/experimental/pallas/ops/gpu/decode_attention.py,sha256=L3IScS2HeoWYBgPUbcYihdQjjpx91kHAPB-9c48FHlg,10235
jax/experimental/pallas/ops/gpu/layer_norm.py,sha256=Ha44Tbe_5S6EzFbwmT7wi3ggW4wwcg5uweHuGBUzsAc,11121
jax/experimental/pallas/ops/gpu/rms_norm.py,sha256=znuldZjC7cr7fHWOdDErI4LeBI8i_iVusu2VhHoIX8M,10050
jax/experimental/pallas/ops/gpu/softmax.py,sha256=TaydT29d3PReZ_fNdweLVosZ29zAhRp7ZZIdEIULOtU,2771
jax/experimental/pallas/ops/tpu/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/experimental/pallas/ops/tpu/all_gather.py,sha256=oLZefmIUQ1Ag_C3kz5h114E2l4OiW9-Nn6DkpB4AC8I,5581
jax/experimental/pallas/ops/tpu/example_kernel.py,sha256=jzsyvU-Lk0PIorE9XCw2NDdqdp8Cn4bqsSe76DBxeEM,802
jax/experimental/pallas/ops/tpu/flash_attention.py,sha256=HDWf7scXNZg4HyJrcFdbmuiWbqtiVCJFFu0y40pTPMA,50881
jax/experimental/pallas/ops/tpu/matmul.py,sha256=jfpCbL3yQ37So4yOtb9pF6K7PtA0WmV65yrzQ8yp4cs,2726
jax/experimental/pallas/ops/tpu/megablox/__init__.py,sha256=Qutkom55JBjOTRrTwsmPrjkZPUx3om4iqXxp5IUZ5c8,643
jax/experimental/pallas/ops/tpu/megablox/common.py,sha256=9zg98VzHJh5l4Sf_SKly1nrXFWeWeyKoe7ZSPn6Ht38,1969
jax/experimental/pallas/ops/tpu/megablox/gmm.py,sha256=vNNgQIirDK7pyDViP2foec7GC8EhJuxNjR1mp3kHwjU,27249
jax/experimental/pallas/ops/tpu/megablox/ops.py,sha256=4Xg3O0bnP5HTWsawS0hAm8mF9HTvRHVK65RlopkA1rw,2903
jax/experimental/pallas/ops/tpu/paged_attention/__init__.py,sha256=bCNpNWR-hJeynDYamiq0BfIsRtvvBoGu2SopmYG48UY,681
jax/experimental/pallas/ops/tpu/paged_attention/paged_attention_kernel.py,sha256=f6z0ukhHjObUOYoWsLc8Wxvhv8g3UVpjGxUixdnf_UE,21960
jax/experimental/pallas/ops/tpu/paged_attention/quantization_utils.py,sha256=5y0QHJo2yDsafF_lgeSO_izChspLHpVOvbFWZ-2d0ss,2556
jax/experimental/pallas/ops/tpu/splash_attention/__init__.py,sha256=TpPvw5zBBqvleRSZQear7qZrvlZKI_PFwFyc45IR5q8,2389
jax/experimental/pallas/ops/tpu/splash_attention/splash_attention_kernel.py,sha256=Zcdj49rrxiNpTWSX1sCXRQunW1zPFrDrb5XQe230xNk,73432
jax/experimental/pallas/ops/tpu/splash_attention/splash_attention_mask.py,sha256=o-9hXrMgy6vq7nq8SiUFPceTGUdvvAsXNHR1bnjPa5A,14554
jax/experimental/pallas/ops/tpu/splash_attention/splash_attention_mask_info.py,sha256=smfmy5BKcRUofB-hpca39y63IZQkz01TfhwlzP5jkbA,33409
jax/experimental/sparse/__init__.py,sha256=iYfDK0sdq3ErijcE7Va7cRjEHjzuJ3ykCwnHSYxNMPs,11186
jax/experimental/sparse/_base.py,sha256=qyiBgPa12uwS1nLpSDw8dTD1VEq0v9P2_CGTNyPCA_g,3239
jax/experimental/sparse/_lowerings.py,sha256=NW3WFfrvsf5O9sxB4vN6fMj6b7F0NxtQEuj3L1oO-NA,7661
jax/experimental/sparse/ad.py,sha256=YT1jU0dsB3VoX5d0QUx9ihNUx2L_HfAh6ouJuEdhmXE,7784
jax/experimental/sparse/api.py,sha256=5MsOHBMgSnDn8X6024oDJFBUxY1BuU9ldrXy89rHQho,6631
jax/experimental/sparse/bcoo.py,sha256=mayAz3xUjvqNjbALIEUs_GH2Dhlp20nYNLzzyKH4qNg,129097
jax/experimental/sparse/bcsr.py,sha256=H5MudpWldZJu2-Bw464JxPis3GPiOkjrhtqeSiJUBdk,38423
jax/experimental/sparse/coo.py,sha256=2bBCeU2NXmlp7rsGsOQvn--yTrYCbya_ss0K7IjpCsg,24143
jax/experimental/sparse/csr.py,sha256=W3OmrmwtJBKy1Qp-vPHwamOGureWHi5oZZeIjuqxzJc,24435
jax/experimental/sparse/linalg.py,sha256=G7ndX6rm2nds_e3DglFPogdcNq4vHOc1kmpZd3RHl0A,23435
jax/experimental/sparse/nm.py,sha256=_yfZ2HKjZmZUicFG5BOC_sqlKfo79K7nCvw9sU4Ih3M,8563
jax/experimental/sparse/random.py,sha256=702kXl0gkbcSBthBBbyD3JAUinP_PlRDPCLMZHCdgcA,3882
jax/experimental/sparse/test_util.py,sha256=bCe34sMR4-ak0sc3vU0CqXlPR9NkUY42FBKmM0sGvxE,9464
jax/experimental/sparse/transform.py,sha256=i6fzxSub8SlogK2eOiaD8PcP9stEKD0WcXv-wGPjkII,37152
jax/experimental/sparse/util.py,sha256=cu1qAHQ2L-4GrhbxWJuuotUC0OMRB2NEpN0oEhsoduA,4427
jax/extend/__init__.py,sha256=rzXCgE1QFpnc0HPRj52vJOxD9_a5LzALYh_bQrYCHIU,1265
jax/extend/backend.py,sha256=kn0ghtRuNckWaiLKIAonFzcaBL0_Gm7GiRmsxrRI7L0,968
jax/extend/ffi.py,sha256=Bo8zJigDbyClf75aMNeDPKKJoWQTNdJsp2fvctaVH2g,915
jax/extend/ifrt_programs.py,sha256=oCXLRZL4g3Syl8tQsTnjQNw8AL0zfKBlhUIUQ6Fh9gc,801
jax/extend/linear_util.py,sha256=Fi4NWtIl3vvh-6GonAS1PmPIseris_w-ZG_lJsV01oU,986
jax/extend/random.py,sha256=w1OM6RCcGwmzIr9bdGJJTNKIcMO4ypXbzmeEORlBHjQ,1083
jax/extend/source_info_util.py,sha256=IxZYjLDlwOH1uE3A_vc25dy25QZ6FABMb4RqrkiA5hU,1215
jax/extend/core/__init__.py,sha256=o_q4S8jW_Tc0986SMqpjcifigZq-Ez2OhFZo6aBLwzM,1036
jax/extend/core/primitives.py,sha256=aFeGezH4qE5D7ZhPJZFGGkx1Ahv3-drh-1UeGOcf6kI,6352
jax/extend/mlir/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
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/scf.py,sha256=hmcJIF7RUqzeeP_NIOampnvCB5Jb6T8Xex4T03IeKrI,641
jax/extend/mlir/dialects/sdy.py,sha256=rtDlK5Z2OMDjBduAaRn08XBH-yGdZkvZogSB7RoFxBc,746
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=QYDYW-0BjdeWUd_6bf1e-a7Ku-UfgRl7yHRmqLIjF90,3224
jax/interpreters/batching.py,sha256=qyVCJDVfj2AC7lCBsb50RXnaqvZyQ_sdddEBPZL4PmQ,2582
jax/interpreters/mlir.py,sha256=OLDnNDROIGVMCC7-nrsCRBGdzd6lAUXGm7gO_GZYj14,2896
jax/interpreters/partial_eval.py,sha256=Q_igvKqlBbumnFmxG0fSYlhXA7UGc_o8H55Tgp2eehU,4471
jax/interpreters/pxla.py,sha256=B-WvovzyId65My-eOizYtBI5mFvn8GXWIdhqowDRDrk,1824
jax/interpreters/xla.py,sha256=jeBirvMzH6DZQZpY-gxH7EABKsePx_v_7ZF0CJY2hDI,3003
jax/lax/__init__.py,sha256=n5HlWFG3MuPOahm_RSHnpb06YL6FuZw78os0R71JxiA,11036
jax/lax/linalg.py,sha256=w66gTsb4Z1X5_PjFUQbSP2FJgN0nNEZ9wMaBS9li6xY,1016
jax/lib/__init__.py,sha256=wy3yxhlsb9k2th9Dfsr6eS59L2qLIxH_IoTM4cPfx3M,774
jax/lib/xla_bridge.py,sha256=RlgLzihbRKJNPKVumdx1QojpZ-xf9XScRmJO708ZjRc,1767
jax/lib/xla_client.py,sha256=hZaRLYD0DVMEhICzoqZOd_luG8aWtfB5S2HD2dBzzbY,2270
jax/lib/xla_extension.py,sha256=bTVt2zrN5alcfoTpH6M7Oo4TeajM2QYNlR-9ZLfK9bw,1358
jax/nn/__init__.py,sha256=ucg5g4JQjNZnTsFEGi4CtIyGS7E2m9htEehaF6nRBCg,1930
jax/nn/initializers.py,sha256=finfAJmOEZdPIzY2roSIBTOmC4rKEW-rchpH9d6Dc2g,1469
jax/numpy/__init__.py,sha256=11UH_2Yb_R0NGgWPF8R4lVpQdCdRvA3LZkM8OC9wOEI,12161
jax/numpy/__init__.pyi,sha256=9cKJzHl0DCcs5yVj9Uy6yRBqdjaO3iZctVQJ7G419m8,40785
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=jLoDaaxGZTcRu94dbVgk7qa4Nm-CHu_L8rMsvEIwzWI,1350
jax/scipy/ndimage.py,sha256=ei-ZfRiDXrlY4nYC1SFL7nnGHFrC_emD_NAicGhH4vU,788
jax/scipy/signal.py,sha256=nW4h017jShZBLL3J2NMhLwlOv46wK1MhEPeKlPTj710,975
jax/scipy/special.py,sha256=hT-5s-5ki7yXeQG3ZO-GMTjSrx3JHla41UNnQ0Njwsk,1718
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=ssiwAm2AdZ81MPMM11015yMNogj4DGsNfWfq507yhlo,1995
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=D4a9KT9p2u0vRjpBS6sQgZ95vuzyZLCh-Mon8HiNq4U,788
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/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=eMTqIxq9njSnXtEInWHGJBfINEo6N4yVrwji4jsyc58,643
jax/scipy/stats/norm.py,sha256=qNMz53OSiWOZGt6TZkJZvWAFocdUMUu-DsysM9kg2K4,879
jax/scipy/stats/pareto.py,sha256=xOqP9JMLAYdVXMFpCfCoRqxJNXVeWfJXkEqpOsrUjHc,789
jax/scipy/stats/poisson.py,sha256=Guu0WOaGAwtujTTWVlBZiBg2fLu4JOjx0fumX4m4Ems,804
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/build_utils.py,sha256=zqgkBIwuwOlCGd-aMwRPdsx9JbykpLYMvfzwEHiMDA8,4194
jax/tools/colab_tpu.py,sha256=IPEQ1HZB4xW0MRSPGDeFAw9GoMFaphNQ0kLRIwGypmo,890
jax/tools/jax_to_ir.py,sha256=GaLgxfDS4Y37yErqOL5BpoicRsnwMHTeA5ucaUR5qBU,8804
jax/tools/pgo_nsys_converter.py,sha256=TvJG79B8NFz__UaE0Z4U1jnuclgH5wcJ2swybggmYVY,2934
jax-0.4.34.dist-info/AUTHORS,sha256=OOBQygrI1Zc94AYJycH1r_BSCwP38Hyv3sludltJ7hY,313
jax-0.4.34.dist-info/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
jax-0.4.34.dist-info/METADATA,sha256=0tH6M28p8hTiw_zuROddq0-6esdeiH2HkbGLP3vnmug,22500
jax-0.4.34.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
jax-0.4.34.dist-info/top_level.txt,sha256=DabAh8MXQ-HM0EtjKD-Wx_D6tTIW-_de73pdG_f_KTE,4
jax-0.4.34.dist-info/RECORD,,
