jax-0.4.16.dist-info/AUTHORS,sha256=OOBQygrI1Zc94AYJycH1r_BSCwP38Hyv3sludltJ7hY,313
jax-0.4.16.dist-info/INSTALLER,sha256=zuuue4knoyJ-UwPPXg8fezS7VCrXJQrAP7zeNuwvFQg,4
jax-0.4.16.dist-info/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
jax-0.4.16.dist-info/METADATA,sha256=EWq-qHnj5OSQy9hM0Xad9qJ2B7qwieFtzFWFhxAHlUM,29328
jax-0.4.16.dist-info/RECORD,,
jax-0.4.16.dist-info/REQUESTED,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
jax-0.4.16.dist-info/WHEEL,sha256=yQN5g4mg4AybRjkgi-9yy4iQEFibGQmlz78Pik5Or-A,92
jax-0.4.16.dist-info/direct_url.json,sha256=CzTSUCT4f_pvl6pgnj3fXckPug16C9m6E4J7ntuh3gc,134
jax-0.4.16.dist-info/top_level.txt,sha256=DabAh8MXQ-HM0EtjKD-Wx_D6tTIW-_de73pdG_f_KTE,4
jax/__init__.py,sha256=oh3IzCFhDzbfrBYY0DQSUzH-eLvCIWB_2pdAe3uqxFI,9701
jax/__pycache__/__init__.cpython-311.pyc,,
jax/__pycache__/abstract_arrays.cpython-311.pyc,,
jax/__pycache__/ad_checkpoint.cpython-311.pyc,,
jax/__pycache__/api_util.cpython-311.pyc,,
jax/__pycache__/cloud_tpu_init.cpython-311.pyc,,
jax/__pycache__/collect_profile.cpython-311.pyc,,
jax/__pycache__/config.cpython-311.pyc,,
jax/__pycache__/core.cpython-311.pyc,,
jax/__pycache__/custom_batching.cpython-311.pyc,,
jax/__pycache__/custom_derivatives.cpython-311.pyc,,
jax/__pycache__/custom_transpose.cpython-311.pyc,,
jax/__pycache__/debug.cpython-311.pyc,,
jax/__pycache__/distributed.cpython-311.pyc,,
jax/__pycache__/dlpack.cpython-311.pyc,,
jax/__pycache__/dtypes.cpython-311.pyc,,
jax/__pycache__/errors.cpython-311.pyc,,
jax/__pycache__/flatten_util.cpython-311.pyc,,
jax/__pycache__/linear_util.cpython-311.pyc,,
jax/__pycache__/monitoring.cpython-311.pyc,,
jax/__pycache__/prng.cpython-311.pyc,,
jax/__pycache__/profiler.cpython-311.pyc,,
jax/__pycache__/random.cpython-311.pyc,,
jax/__pycache__/sharding.cpython-311.pyc,,
jax/__pycache__/stages.cpython-311.pyc,,
jax/__pycache__/test_util.cpython-311.pyc,,
jax/__pycache__/tree_util.cpython-311.pyc,,
jax/__pycache__/typing.cpython-311.pyc,,
jax/__pycache__/util.cpython-311.pyc,,
jax/__pycache__/version.cpython-311.pyc,,
jax/_src/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/__pycache__/__init__.cpython-311.pyc,,
jax/_src/__pycache__/abstract_arrays.cpython-311.pyc,,
jax/_src/__pycache__/ad_checkpoint.cpython-311.pyc,,
jax/_src/__pycache__/ad_util.cpython-311.pyc,,
jax/_src/__pycache__/api.cpython-311.pyc,,
jax/_src/__pycache__/api_util.cpython-311.pyc,,
jax/_src/__pycache__/array.cpython-311.pyc,,
jax/_src/__pycache__/basearray.cpython-311.pyc,,
jax/_src/__pycache__/cache_key.cpython-311.pyc,,
jax/_src/__pycache__/callback.cpython-311.pyc,,
jax/_src/__pycache__/checkify.cpython-311.pyc,,
jax/_src/__pycache__/cloud_tpu_init.cpython-311.pyc,,
jax/_src/__pycache__/compilation_cache.cpython-311.pyc,,
jax/_src/__pycache__/compilation_cache_interface.cpython-311.pyc,,
jax/_src/__pycache__/compiler.cpython-311.pyc,,
jax/_src/__pycache__/config.cpython-311.pyc,,
jax/_src/__pycache__/core.cpython-311.pyc,,
jax/_src/__pycache__/custom_api_util.cpython-311.pyc,,
jax/_src/__pycache__/custom_batching.cpython-311.pyc,,
jax/_src/__pycache__/custom_derivatives.cpython-311.pyc,,
jax/_src/__pycache__/custom_transpose.cpython-311.pyc,,
jax/_src/__pycache__/debugging.cpython-311.pyc,,
jax/_src/__pycache__/deprecations.cpython-311.pyc,,
jax/_src/__pycache__/dispatch.cpython-311.pyc,,
jax/_src/__pycache__/distributed.cpython-311.pyc,,
jax/_src/__pycache__/dlpack.cpython-311.pyc,,
jax/_src/__pycache__/dtypes.cpython-311.pyc,,
jax/_src/__pycache__/effects.cpython-311.pyc,,
jax/_src/__pycache__/environment_info.cpython-311.pyc,,
jax/_src/__pycache__/errors.cpython-311.pyc,,
jax/_src/__pycache__/flatten_util.cpython-311.pyc,,
jax/_src/__pycache__/gfile_cache.cpython-311.pyc,,
jax/_src/__pycache__/jaxpr_util.cpython-311.pyc,,
jax/_src/__pycache__/lax_reference.cpython-311.pyc,,
jax/_src/__pycache__/lazy_loader.cpython-311.pyc,,
jax/_src/__pycache__/linear_util.cpython-311.pyc,,
jax/_src/__pycache__/logging_config.cpython-311.pyc,,
jax/_src/__pycache__/maps.cpython-311.pyc,,
jax/_src/__pycache__/mesh.cpython-311.pyc,,
jax/_src/__pycache__/monitoring.cpython-311.pyc,,
jax/_src/__pycache__/op_shardings.cpython-311.pyc,,
jax/_src/__pycache__/partition_spec.cpython-311.pyc,,
jax/_src/__pycache__/path.cpython-311.pyc,,
jax/_src/__pycache__/pickle_util.cpython-311.pyc,,
jax/_src/__pycache__/pjit.cpython-311.pyc,,
jax/_src/__pycache__/pretty_printer.cpython-311.pyc,,
jax/_src/__pycache__/prng.cpython-311.pyc,,
jax/_src/__pycache__/profiler.cpython-311.pyc,,
jax/_src/__pycache__/public_test_util.cpython-311.pyc,,
jax/_src/__pycache__/random.cpython-311.pyc,,
jax/_src/__pycache__/sharding.cpython-311.pyc,,
jax/_src/__pycache__/sharding_impls.cpython-311.pyc,,
jax/_src/__pycache__/sharding_specs.cpython-311.pyc,,
jax/_src/__pycache__/source_info_util.cpython-311.pyc,,
jax/_src/__pycache__/stages.cpython-311.pyc,,
jax/_src/__pycache__/test_util.cpython-311.pyc,,
jax/_src/__pycache__/tpu_custom_call.cpython-311.pyc,,
jax/_src/__pycache__/traceback_util.cpython-311.pyc,,
jax/_src/__pycache__/tree_util.cpython-311.pyc,,
jax/_src/__pycache__/typing.cpython-311.pyc,,
jax/_src/__pycache__/util.cpython-311.pyc,,
jax/_src/__pycache__/xla_bridge.cpython-311.pyc,,
jax/_src/abstract_arrays.py,sha256=ZpKoD3prFnp0yAECVMMAA99F4v9MJxnPOyC7Exkr_gY,3203
jax/_src/ad_checkpoint.py,sha256=QGT-3WVa6B6-2MWCIqaTl4_DAnp8iAbzJZnAbUkh-V4,34700
jax/_src/ad_util.py,sha256=BBNIshxskhvZydeapb6UCNhnkmWBnbJmPF88Y5T4jMQ,3966
jax/_src/api.py,sha256=Ut1gIlLdMp8xpwqMwmZTpeIb9s7xr8E15JUNUxxAC0s,131547
jax/_src/api_util.py,sha256=6B1lRAVwlioU7jjeWgs7XoNj-L6b3qzxbMZHYeRRvIQ,26503
jax/_src/array.py,sha256=T0sp5IjoROjM_BAeiJ6wXtoXY-2gOJk1DcZSvy4FE_w,31146
jax/_src/basearray.py,sha256=dhWhN_7PfXHomkAo3NdOTQ42bNB_96y9t5Q-XuqJ8Sc,2665
jax/_src/basearray.pyi,sha256=_b1-7At-gMEIAE2CDacw_f5NrY4HbM-KMkUo-w0f8kA,10608
jax/_src/cache_key.py,sha256=NFguPpVfITRCCV9HCBF9a4dflKNJkl-2t6O4DyW3c9g,14502
jax/_src/callback.py,sha256=UgYyNKz3e5IF_awcpZJofshUB1cbp1cg-R_nAbzJ2TU,17957
jax/_src/checkify.py,sha256=_JHK95tAX5WVz4LjLMrC0B2f2Pt80IEM4g4LPCCkKVM,52919
jax/_src/cloud_tpu_init.py,sha256=mSt_RTDt9Qnxa7v6yHLEFHAFaIGBmytgYQdrXiN6iAY,2698
jax/_src/clusters/__init__.py,sha256=NPJVuAPVR1oyikfhx5dIMfhTybhoyEkhx2pzWyowNQI,1047
jax/_src/clusters/__pycache__/__init__.cpython-311.pyc,,
jax/_src/clusters/__pycache__/cloud_tpu_cluster.cpython-311.pyc,,
jax/_src/clusters/__pycache__/cluster.cpython-311.pyc,,
jax/_src/clusters/__pycache__/ompi_cluster.cpython-311.pyc,,
jax/_src/clusters/__pycache__/slurm_cluster.cpython-311.pyc,,
jax/_src/clusters/cloud_tpu_cluster.py,sha256=i4_JhJOvkzR4GTejlvxmAquWe0U4dLchthfJPZCNFcc,2456
jax/_src/clusters/cluster.py,sha256=nd_tOifc_YLyylGljPW_d926DkCz_5HygFXRXAH6jG8,4782
jax/_src/clusters/ompi_cluster.py,sha256=T9UIKSuWvJWHj5otL5apd17ivnjzn1aweTTaUJgXRBw,2263
jax/_src/clusters/slurm_cluster.py,sha256=MZ4YyNkVLGXLLT7QITOWimK_ZH5nobsBHb5zxTs76eI,2266
jax/_src/compilation_cache.py,sha256=OxUQHFA_2zcWarAx3oKchPHpUxE5OFKc2YPP31CzSLY,5080
jax/_src/compilation_cache_interface.py,sha256=xRXZ8RjI3kLnCDAxkBh8jvr_cN3e9Sk9E9CR1xkbrpo,831
jax/_src/compiler.py,sha256=Ac8MaxqJphp58uZnbsxkrbGhIEQy6x13PoJ1OBcrxwk,15708
jax/_src/config.py,sha256=iTWDXC8hH2dnSJi-f-I4ORslw19jdePou-guFftwBqg,52745
jax/_src/core.py,sha256=VUk-AR4NbaLGcUgfD0yjfZmIvdNfIUR0SsXofKqT5I0,111897
jax/_src/custom_api_util.py,sha256=uv6uyrXU1BHGzKuAe498W2PD9bcWt1PEa-8UAqr9Zaw,876
jax/_src/custom_batching.py,sha256=N690CQU4mr1AzJMh1aFjkS0Qjh3AFj6_rzaZPwn9yCI,10475
jax/_src/custom_derivatives.py,sha256=h2NKWnK3gfVPfbTJKN6nAUPo1-m0_mXaz3i7wf94iSQ,58834
jax/_src/custom_transpose.py,sha256=ET_Nq_yDITag-JYLbR2eFzs-xVMs5LEN1J8suy5GglI,8287
jax/_src/debugger/__init__.py,sha256=VPrkm_4GZ-dm4SN7L4xqXCwGSFiD5K4BkXGy0MvWnSw,885
jax/_src/debugger/__pycache__/__init__.cpython-311.pyc,,
jax/_src/debugger/__pycache__/cli_debugger.cpython-311.pyc,,
jax/_src/debugger/__pycache__/colab_debugger.cpython-311.pyc,,
jax/_src/debugger/__pycache__/colab_lib.cpython-311.pyc,,
jax/_src/debugger/__pycache__/core.cpython-311.pyc,,
jax/_src/debugger/__pycache__/web_debugger.cpython-311.pyc,,
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=0rEPhWjcCaiMey4sAuDivG1H5RFqqlfesAxlZZ1rlRU,8369
jax/_src/debugger/web_debugger.py,sha256=WJzG5FubQxA9EiuyXzHUb_v2_2GqvovyQiy6EaBomY8,3282
jax/_src/debugging.py,sha256=pXYJOHhouT8kqjtRmFZhK3pibiVe-vJMVDQKKZbEODQ,24415
jax/_src/deprecations.py,sha256=Y2ID9gpK__WiYbwkM7buxpOaC3Yc1Ynls37vJr1wOn0,1763
jax/_src/dispatch.py,sha256=ZIoYwsB0tvyD0T87VwDBYmbrgybLfNnWzZt7ntaCEgw,20431
jax/_src/distributed.py,sha256=Ch5VHb4zOG2XDiJit23OscvOtoSMsnQ_fBi-gBsvZcY,7929
jax/_src/dlpack.py,sha256=qpZ45B2c_HZvf_NwOi20iQ1c7WMgAi2GQXJE462kbMk,4702
jax/_src/dtypes.py,sha256=lSmYtFj-OeYEzjSsZH2IR2ilKVkj7XF-GZ2o1eoX08E,25940
jax/_src/effects.py,sha256=fPkC1_SzsPleP2CWiVeOYsPCdElT4SkfUASwdtvhulU,4671
jax/_src/environment_info.py,sha256=F5p2xq60p9BT36TStxLxLZoRPojHmEFuDur4zj0zl00,1970
jax/_src/errors.py,sha256=FBWohg8A0Y2r3m_6CHdA5NR8ENyUjHWBD9II500KgZ4,23776
jax/_src/extend/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/_src/extend/__pycache__/__init__.cpython-311.pyc,,
jax/_src/extend/__pycache__/random.cpython-311.pyc,,
jax/_src/extend/random.py,sha256=-E-O2w9u0eEKJINdsp5Ik8qm5Bin06crk_2rd0HIvoo,874
jax/_src/flatten_util.py,sha256=jzKbWVjG04I3Xwltjw7rOxg80FjuZLhZoSD7QftE18Q,3557
jax/_src/gfile_cache.py,sha256=yfHAo5Jl-Knt-P1GWRaaB4Ucxw3t7g66jD-rGti9KwY,1923
jax/_src/image/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/image/__pycache__/__init__.cpython-311.pyc,,
jax/_src/image/__pycache__/scale.cpython-311.pyc,,
jax/_src/image/scale.py,sha256=jROfJ5tpA6ikcBhR6zbwlLeatvUpt_TKvKm45tujUQ0,13747
jax/_src/internal_test_util/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/internal_test_util/__pycache__/__init__.cpython-311.pyc,,
jax/_src/internal_test_util/__pycache__/deprecation_module.cpython-311.pyc,,
jax/_src/internal_test_util/__pycache__/lax_test_util.cpython-311.pyc,,
jax/_src/internal_test_util/deprecation_module.py,sha256=iZ6fRYNaUcOwrftwpDXMnlxrLrrQbdMrbybmmQFwql4,851
jax/_src/internal_test_util/lax_test_util.py,sha256=nVht0i3zRFOUsLI4wr7zrEgnNnydn252gsqmGOxl9ls,12142
jax/_src/internal_test_util/lazy_loader_module/__init__.py,sha256=UP_c2VfWHRhfEi6n4F6qLHZuageXhPiVSAcUKHrrxFs,705
jax/_src/internal_test_util/lazy_loader_module/__pycache__/__init__.cpython-311.pyc,,
jax/_src/internal_test_util/lazy_loader_module/__pycache__/lazy_test_submodule.cpython-311.pyc,,
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/__pycache__/__init__.cpython-311.pyc,,
jax/_src/interpreters/__pycache__/ad.cpython-311.pyc,,
jax/_src/interpreters/__pycache__/batching.cpython-311.pyc,,
jax/_src/interpreters/__pycache__/mlir.cpython-311.pyc,,
jax/_src/interpreters/__pycache__/partial_eval.cpython-311.pyc,,
jax/_src/interpreters/__pycache__/pxla.cpython-311.pyc,,
jax/_src/interpreters/__pycache__/xla.cpython-311.pyc,,
jax/_src/interpreters/ad.py,sha256=OGvNxvIUMK5TA0UGmOPRGNsjZLBb_IU4gZCQbmkOx5o,33498
jax/_src/interpreters/batching.py,sha256=19QF6lxxcJOAZxFc55I2DtvcRn4y1ks9RaYpbTzkcJU,50670
jax/_src/interpreters/mlir.py,sha256=3ZHxcG7vvmcWRhGyRpGKVrDlXHi5jNkEwGrT4NpvCGE,99634
jax/_src/interpreters/partial_eval.py,sha256=REWtSFpL7zMyrs2mPPfKpnjiSxpGoPos5qJ-vOe6RXQ,118318
jax/_src/interpreters/pxla.py,sha256=DLwRb9hx9x7YbPnThhTwTY3uvrrMyTwBb1u1-GV8WLM,127160
jax/_src/interpreters/xla.py,sha256=U0JkJytgmkx0W6APAc5bXx6XWuvdzt1OM2CvyjKkwYg,12099
jax/_src/jaxpr_util.py,sha256=XziwIzp02nBzpsWS4tGnBMHR3syP5d0QEzLhcqOdjkA,6528
jax/_src/lax/__init__.py,sha256=jRL3qQoUco8BVUAh8lV-KuY38h_4GxXcb41asljW2yI,690
jax/_src/lax/__pycache__/__init__.cpython-311.pyc,,
jax/_src/lax/__pycache__/ann.cpython-311.pyc,,
jax/_src/lax/__pycache__/convolution.cpython-311.pyc,,
jax/_src/lax/__pycache__/eigh.cpython-311.pyc,,
jax/_src/lax/__pycache__/fft.cpython-311.pyc,,
jax/_src/lax/__pycache__/lax.cpython-311.pyc,,
jax/_src/lax/__pycache__/linalg.cpython-311.pyc,,
jax/_src/lax/__pycache__/other.cpython-311.pyc,,
jax/_src/lax/__pycache__/parallel.cpython-311.pyc,,
jax/_src/lax/__pycache__/qdwh.cpython-311.pyc,,
jax/_src/lax/__pycache__/slicing.cpython-311.pyc,,
jax/_src/lax/__pycache__/special.cpython-311.pyc,,
jax/_src/lax/__pycache__/stack.cpython-311.pyc,,
jax/_src/lax/__pycache__/svd.cpython-311.pyc,,
jax/_src/lax/__pycache__/utils.cpython-311.pyc,,
jax/_src/lax/__pycache__/windowed_reductions.cpython-311.pyc,,
jax/_src/lax/ann.py,sha256=Toj0fK82DGC75sjXmuIKwdxYrxp32908KF2b3X9DcIk,17637
jax/_src/lax/control_flow/__init__.py,sha256=DqqlTNYq_SVE-LIAKK8I13gAmKi4OdnsA7FMihqhz0c,2029
jax/_src/lax/control_flow/__pycache__/__init__.cpython-311.pyc,,
jax/_src/lax/control_flow/__pycache__/common.cpython-311.pyc,,
jax/_src/lax/control_flow/__pycache__/conditionals.cpython-311.pyc,,
jax/_src/lax/control_flow/__pycache__/for_loop.cpython-311.pyc,,
jax/_src/lax/control_flow/__pycache__/loops.cpython-311.pyc,,
jax/_src/lax/control_flow/__pycache__/solves.cpython-311.pyc,,
jax/_src/lax/control_flow/common.py,sha256=qxOTlGoIJfo0qpgCDsXWsNwL_8loTFlU03y8UW168t8,10272
jax/_src/lax/control_flow/conditionals.py,sha256=AjobAP49yBVkN1ERhzvCStVUUbQ3hDZ2Wv6Ok5H5OoA,37996
jax/_src/lax/control_flow/for_loop.py,sha256=YR65X84ztN4h-4E_7jjazx042qFGKxNEmbNXMqGZsjg,37010
jax/_src/lax/control_flow/loops.py,sha256=5qZOmwWT94RYoYn7hgeRo51kx5zu7TPnUSNQ98uexoY,102216
jax/_src/lax/control_flow/solves.py,sha256=D3PEw6MsfWehqEV8bvW2Ol3zyFO-funjbMzaJR3ArJ4,18898
jax/_src/lax/convolution.py,sha256=cYLfX-mTWACsyTfeWpvgTEuKdgW3AawuB53CC1c0f4M,44762
jax/_src/lax/eigh.py,sha256=tZCLdbrArpe8HskzEPhyEV_qSewLEbPJqc3ebdzCSqI,19853
jax/_src/lax/fft.py,sha256=PjB474NiWnbrlSQtiSS0NnbXlK9-JMjs9MGECL1pUWA,9900
jax/_src/lax/lax.py,sha256=rdhECrqEKXUHaAsprY0n409aOpapWYXs7pkO3tlIC6M,206832
jax/_src/lax/linalg.py,sha256=XxiUd22_Amld2uDke0BsLfZ8nShyFBRR-uDW5mnuNqw,90965
jax/_src/lax/other.py,sha256=xiqAD3ZNXkIs0IzslMdhCoEOi_9xvsRzRRkUg0TAr7g,10554
jax/_src/lax/parallel.py,sha256=RWfksWwXrQ2zLr3AUXPYN8slEr6Xqo3GPkvnPotomnU,73078
jax/_src/lax/qdwh.py,sha256=dtvODeExB-lk69iL3qr_55Xlku5nNVQjFqYz-1EDa4c,8023
jax/_src/lax/slicing.py,sha256=uYunf05gjvSQ_PM097tznJQ7sLV3OZIVZGX6oUTp4j8,111760
jax/_src/lax/special.py,sha256=9Q-D5HYTPOK4A1mnW5VPRR0Q7ouQ5cVJUwTWvOuAZRk,26953
jax/_src/lax/stack.py,sha256=7hxK1zlVa_Q9gO48Ymj8S1nA8DN6keYgcOY0DAtbjOY,2560
jax/_src/lax/svd.py,sha256=7qpha_UrB-6aJjwYY2JFf-U8GqTqa05flWipbWSrSBc,9833
jax/_src/lax/utils.py,sha256=nbLjW7RfPRdBW3AXOMk2h7fza5Yx5wDynDzhnavA0lA,5006
jax/_src/lax/windowed_reductions.py,sha256=l8QhYQPkISG1HeD-IobHtAqHZe7Tib10Jvk4AL7Wqsg,40168
jax/_src/lax_reference.py,sha256=jAUWKbRz4u2JkMgkUgFHwykAeKBt_Y1ZNWmJrSpfjf4,17312
jax/_src/lazy_loader.py,sha256=pXw4tMX_AEQhDzSG-5BOiTljnpV1xVtGl0qAwmD9_Zo,1333
jax/_src/lib/__init__.py,sha256=rAxQ-CARC4xVbYktpdHKzgx4UB8I4Fgrfzkfd1SiYUQ,5448
jax/_src/lib/__pycache__/__init__.cpython-311.pyc,,
jax/_src/lib/mlir/__init__.py,sha256=BiNbgNL5EDNE2U2PR1awdC8Hg-_G9FGVcAVmFTgyKoE,678
jax/_src/lib/mlir/__pycache__/__init__.cpython-311.pyc,,
jax/_src/lib/mlir/dialects/__init__.py,sha256=5_3i2XisZ6FpQXmxTUP9zTrTo9McUch44_DLM_EdUyk,1358
jax/_src/lib/mlir/dialects/__pycache__/__init__.cpython-311.pyc,,
jax/_src/linear_util.py,sha256=cjdGvOJ0cyE8wITxEkB2CP5vg2DiLuF4TsOWqmf25_Y,13769
jax/_src/logging_config.py,sha256=usQjXadUzBGKOSNtwPsXDPcZeQDaoT06Gzn7ON46yIY,2040
jax/_src/maps.py,sha256=D0AHjhj1i02yJdQTBiJzI2yE8g_cg_xjtqber3elvOk,86605
jax/_src/mesh.py,sha256=xVEkDZB_UT9_YK-3SIfLSssHZAFrlLDXqmhm4nBqQ8Y,10991
jax/_src/monitoring.py,sha256=rHaEDlfW6iwXJtdkrRhKnc_nnxGSJzeCzqR6mjJuJ7s,3120
jax/_src/nn/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/nn/__pycache__/__init__.cpython-311.pyc,,
jax/_src/nn/__pycache__/functions.cpython-311.pyc,,
jax/_src/nn/__pycache__/initializers.cpython-311.pyc,,
jax/_src/nn/functions.py,sha256=Lk-EARmgAhq03j5aWEbP9T3XTmSYNgC_wY4y7qzBcIc,16779
jax/_src/nn/initializers.py,sha256=xCLNSxQ3tyCPxtXY-OEftUMkfL6vtkw3ixHFlrUsils,25550
jax/_src/numpy/__init__.py,sha256=-QhmzWj2qFbKVIT9TDAIvQWaKQibiRXy3r1zQkI3X4M,581
jax/_src/numpy/__pycache__/__init__.cpython-311.pyc,,
jax/_src/numpy/__pycache__/array_methods.cpython-311.pyc,,
jax/_src/numpy/__pycache__/fft.cpython-311.pyc,,
jax/_src/numpy/__pycache__/index_tricks.cpython-311.pyc,,
jax/_src/numpy/__pycache__/lax_numpy.cpython-311.pyc,,
jax/_src/numpy/__pycache__/linalg.cpython-311.pyc,,
jax/_src/numpy/__pycache__/polynomial.cpython-311.pyc,,
jax/_src/numpy/__pycache__/reductions.cpython-311.pyc,,
jax/_src/numpy/__pycache__/setops.cpython-311.pyc,,
jax/_src/numpy/__pycache__/ufunc_api.cpython-311.pyc,,
jax/_src/numpy/__pycache__/ufuncs.cpython-311.pyc,,
jax/_src/numpy/__pycache__/util.cpython-311.pyc,,
jax/_src/numpy/__pycache__/vectorize.cpython-311.pyc,,
jax/_src/numpy/array_methods.py,sha256=tmj0nC9r5kzgveZZqTrAnB1vK0nBDsSKZ2XhTkNAzII,31850
jax/_src/numpy/fft.py,sha256=QW2yjWa_UM8YdDgdC3VA9Weic2ziYmmkA8dKvf_IqYU,11347
jax/_src/numpy/index_tricks.py,sha256=MtTMoLMQl4g6G2IM3S2sFMYLcXNnVUIgXPsth-giBMw,9896
jax/_src/numpy/lax_numpy.py,sha256=3yGjTIqRT1YGf9sxhMBcVB3vgk8hxQguKvOKVLNerEw,202656
jax/_src/numpy/linalg.py,sha256=r9K2hxT5kzK5SZcmXgC-BC0PxtyAOTKEtshXaWx_Wn8,26570
jax/_src/numpy/polynomial.py,sha256=lQIcmiTjlC3XGVcaFbglX5zjsunI93PeUP3wxT6bfnc,12126
jax/_src/numpy/reductions.py,sha256=yTM5_bamG8HF9fd1vVLjBlhZqdi6xLmM95gx19PmbrU,40627
jax/_src/numpy/setops.py,sha256=tPFFHXWt7fxGlNcq_SjWqPrZ8TvpSXrDNULIoxe_j-k,13736
jax/_src/numpy/ufunc_api.py,sha256=8CVhPnrjUClW1-47HrnUa35d5oefxq6MauE9e2eobW8,13688
jax/_src/numpy/ufuncs.py,sha256=rHkisu1QgKinjIn9MJFbcddbubDURZvlB1fOLKaStW0,27027
jax/_src/numpy/util.py,sha256=ZNz5PXJhoBeFs3EMdYtpulapOJ3KnkcDPO86ZdQ309s,17677
jax/_src/numpy/vectorize.py,sha256=SDNE1cW6Mn0h9_Q-PkDFKqhO4OcTEceoCkLH0u4OxLY,12120
jax/_src/op_shardings.py,sha256=QYjptRAT8mGKgCbLQB2hmgq9G4yotIUBr7C3k3WvuUY,3877
jax/_src/ops/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/ops/__pycache__/__init__.cpython-311.pyc,,
jax/_src/ops/__pycache__/scatter.cpython-311.pyc,,
jax/_src/ops/__pycache__/special.cpython-311.pyc,,
jax/_src/ops/scatter.py,sha256=qhIfpR0W2_8GZI7vyOq4hsaPaRtfT34qTfjp2L1A5jY,17428
jax/_src/ops/special.py,sha256=owD6DDN7gaBOK6JviavWyaiow9UFd0_ywrIha22ovGU,4372
jax/_src/pallas/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/_src/pallas/__pycache__/__init__.cpython-311.pyc,,
jax/_src/pallas/__pycache__/core.cpython-311.pyc,,
jax/_src/pallas/__pycache__/indexing.cpython-311.pyc,,
jax/_src/pallas/__pycache__/pallas_call.cpython-311.pyc,,
jax/_src/pallas/__pycache__/primitives.cpython-311.pyc,,
jax/_src/pallas/__pycache__/utils.cpython-311.pyc,,
jax/_src/pallas/core.py,sha256=FYg1YZOMYqtnnAO-SX3TtpAncGZqz8pfUbXyIbTe5sY,8792
jax/_src/pallas/indexing.py,sha256=kY-suAXUPm-sr0itySG1V4O00jvdk6h-0iME1tnbHcA,5774
jax/_src/pallas/mosaic/__init__.py,sha256=819-ASTHy1hx7i_6ETw7G8nfnCzeZoSc8DeX4XPenjI,1697
jax/_src/pallas/mosaic/__pycache__/__init__.cpython-311.pyc,,
jax/_src/pallas/mosaic/__pycache__/core.cpython-311.pyc,,
jax/_src/pallas/mosaic/__pycache__/kernel_regeneration_util.cpython-311.pyc,,
jax/_src/pallas/mosaic/__pycache__/lowering.cpython-311.pyc,,
jax/_src/pallas/mosaic/__pycache__/pallas_call_registration.cpython-311.pyc,,
jax/_src/pallas/mosaic/__pycache__/primitives.cpython-311.pyc,,
jax/_src/pallas/mosaic/core.py,sha256=cSEFKcEj3bLuQ_ONCS9AcUv880-wGjOUBfeMMICVBwo,6265
jax/_src/pallas/mosaic/kernel_regeneration_util.py,sha256=e3FjR_zYDDuUixKOk-W59WeDopm6Nh-n_VfLKGiKEm8,1995
jax/_src/pallas/mosaic/lowering.py,sha256=U2Vj9gHN7uaQilc0KaWalM7t1GddfXnlY6HY0bJgPIE,51770
jax/_src/pallas/mosaic/pallas_call_registration.py,sha256=DKTPkmipXTHjqgDc_aUYLTfhxF8fSv9A-uko9XGiDE0,3115
jax/_src/pallas/mosaic/primitives.py,sha256=-SLCr3ru1RT-9uucbcfZdIPJgNzhcXefkakbeU-jQno,8112
jax/_src/pallas/pallas_call.py,sha256=hooIHEC9T6aKPQF7_6dy3NJhDodJZKNdrQxdOIOagUQ,17569
jax/_src/pallas/primitives.py,sha256=k75qaFh1eDrP_ay3UTUsXjvZGDzzButRbloKQDTstTk,17106
jax/_src/pallas/triton/__init__.py,sha256=orEv3BdlkA4qlE2bmoIAcnOm85p_x4ZjfpYJDGUwtdA,822
jax/_src/pallas/triton/__pycache__/__init__.cpython-311.pyc,,
jax/_src/pallas/triton/__pycache__/lowering.cpython-311.pyc,,
jax/_src/pallas/triton/lowering.py,sha256=K1bmjkM2Gv6aGo0afTK9z2dUwfn0qkUAo_YZJjNOZIU,54398
jax/_src/pallas/utils.py,sha256=lC5OPHzZRgHfeV_FQvepg66kXQeKf8FzgkDMX3Ordk4,3140
jax/_src/partition_spec.py,sha256=apayEnjD-HJ0NykjgoUquNhQ9W7PwsEKZOPqGCx6GcA,1688
jax/_src/path.py,sha256=Zd6Q4jJEK8m4MOVUsiT2ALoUDHvoZjP37SZFYJzacVI,1121
jax/_src/pickle_util.py,sha256=UUNYUdIIcN9G8NW5EwRiVuxJo7zP8-p4v66rwq7jt9w,2302
jax/_src/pjit.py,sha256=WtgJRDhuxAh_Xzz59GYGS3lnNeGaHrkVDkBzK7sXA0Q,89310
jax/_src/pretty_printer.py,sha256=6cHfrLAvpyz1O26zfEJWJaUV8U9EqNwGiAoPFPQ0epM,13088
jax/_src/prng.py,sha256=0CMpJNwCYb22g9wGw91qep-v0Eb6X8oOMOf2ZeQuXoU,50407
jax/_src/profiler.py,sha256=PqeW4btTNHs5yHKHYmu-KzH3kt6FUBtlo5udmX6BoYM,13920
jax/_src/public_test_util.py,sha256=9cWntWtU3Io5-JKOo7p9dZ1nts4myeutDjoRl-8-k5w,9917
jax/_src/random.py,sha256=D_ly44pMxLE0YufzYZgutR7ZwCQQ8LPF6V1facs6El8,90674
jax/_src/scipy/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/scipy/__pycache__/__init__.cpython-311.pyc,,
jax/_src/scipy/__pycache__/fft.cpython-311.pyc,,
jax/_src/scipy/__pycache__/integrate.cpython-311.pyc,,
jax/_src/scipy/__pycache__/linalg.cpython-311.pyc,,
jax/_src/scipy/__pycache__/ndimage.cpython-311.pyc,,
jax/_src/scipy/__pycache__/signal.cpython-311.pyc,,
jax/_src/scipy/__pycache__/special.cpython-311.pyc,,
jax/_src/scipy/cluster/__init__.py,sha256=S-o0CP72882Xv3l6k2PlrEx6eNvXBTk5Oaed5KF4iL8,581
jax/_src/scipy/cluster/__pycache__/__init__.cpython-311.pyc,,
jax/_src/scipy/cluster/__pycache__/vq.cpython-311.pyc,,
jax/_src/scipy/cluster/vq.py,sha256=OxDujGx3RvogMoqXKWWfpxsKigzBwxG2AYPCAsqLh4g,1771
jax/_src/scipy/fft.py,sha256=mIQYSxXYjf4liU8Dk9c5ze_Z9cHKKdd9Ua7Yz1AKAII,6247
jax/_src/scipy/integrate.py,sha256=kHJe7Bmki7VIg5iauSwQtreNyoDt45eIqhOoK9hy4rY,1553
jax/_src/scipy/interpolate/__init__.py,sha256=S-o0CP72882Xv3l6k2PlrEx6eNvXBTk5Oaed5KF4iL8,581
jax/_src/scipy/interpolate/__pycache__/__init__.cpython-311.pyc,,
jax/_src/scipy/linalg.py,sha256=i4RGluKqJdlC-steRkEg3NpSpxf8kEuxQtsjImdpROU,42254
jax/_src/scipy/ndimage.py,sha256=uydh28hKMyyxjHQC_J5nEHDC-lYRJ8wNMHU_fBDHxDY,5354
jax/_src/scipy/optimize/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/scipy/optimize/__pycache__/__init__.cpython-311.pyc,,
jax/_src/scipy/optimize/__pycache__/_lbfgs.cpython-311.pyc,,
jax/_src/scipy/optimize/__pycache__/bfgs.cpython-311.pyc,,
jax/_src/scipy/optimize/__pycache__/line_search.cpython-311.pyc,,
jax/_src/scipy/optimize/__pycache__/minimize.cpython-311.pyc,,
jax/_src/scipy/optimize/_lbfgs.py,sha256=p_IMzCth237gXPrilQXye7HT8xCgU6QWiS3PClomGgk,7809
jax/_src/scipy/optimize/bfgs.py,sha256=MskxYeLfZVfJLEemLHxckhrQmE0tRdd4D5e8Ksw8Kkg,5613
jax/_src/scipy/optimize/line_search.py,sha256=iE5Ak9OYven2izsJ-a78AED7ReGOfsX7384IK1qR0yw,13430
jax/_src/scipy/optimize/minimize.py,sha256=GAnbB0g-meC0HoPBhDPtXP5T1dp5iE6vranXkCpjGJo,4999
jax/_src/scipy/signal.py,sha256=Y21Mkdo-KP5IqavSWVWDOy46YrEIN2tDM-UzxmY2Ggg,27560
jax/_src/scipy/sparse/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/scipy/sparse/__pycache__/__init__.cpython-311.pyc,,
jax/_src/scipy/sparse/__pycache__/linalg.cpython-311.pyc,,
jax/_src/scipy/sparse/linalg.py,sha256=Ev_-7aIZRjnyisSZkBydXAfzZULpg148s-znErsFhpQ,27330
jax/_src/scipy/spatial/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/_src/scipy/spatial/__pycache__/__init__.cpython-311.pyc,,
jax/_src/scipy/spatial/__pycache__/transform.cpython-311.pyc,,
jax/_src/scipy/spatial/transform.py,sha256=7D5c3xula9Gp4fam-rFZ8wZW-o2BYyUCU0MDyiB1sW0,15149
jax/_src/scipy/special.py,sha256=feaiw3dteVaUPaFuyX8cX7laVkP1MJDiyBs7-chIuxg,58130
jax/_src/scipy/stats/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/scipy/stats/__pycache__/__init__.cpython-311.pyc,,
jax/_src/scipy/stats/__pycache__/_core.cpython-311.pyc,,
jax/_src/scipy/stats/__pycache__/bernoulli.cpython-311.pyc,,
jax/_src/scipy/stats/__pycache__/beta.cpython-311.pyc,,
jax/_src/scipy/stats/__pycache__/betabinom.cpython-311.pyc,,
jax/_src/scipy/stats/__pycache__/binom.cpython-311.pyc,,
jax/_src/scipy/stats/__pycache__/cauchy.cpython-311.pyc,,
jax/_src/scipy/stats/__pycache__/chi2.cpython-311.pyc,,
jax/_src/scipy/stats/__pycache__/dirichlet.cpython-311.pyc,,
jax/_src/scipy/stats/__pycache__/expon.cpython-311.pyc,,
jax/_src/scipy/stats/__pycache__/gamma.cpython-311.pyc,,
jax/_src/scipy/stats/__pycache__/gennorm.cpython-311.pyc,,
jax/_src/scipy/stats/__pycache__/geom.cpython-311.pyc,,
jax/_src/scipy/stats/__pycache__/kde.cpython-311.pyc,,
jax/_src/scipy/stats/__pycache__/laplace.cpython-311.pyc,,
jax/_src/scipy/stats/__pycache__/logistic.cpython-311.pyc,,
jax/_src/scipy/stats/__pycache__/multinomial.cpython-311.pyc,,
jax/_src/scipy/stats/__pycache__/multivariate_normal.cpython-311.pyc,,
jax/_src/scipy/stats/__pycache__/nbinom.cpython-311.pyc,,
jax/_src/scipy/stats/__pycache__/norm.cpython-311.pyc,,
jax/_src/scipy/stats/__pycache__/pareto.cpython-311.pyc,,
jax/_src/scipy/stats/__pycache__/poisson.cpython-311.pyc,,
jax/_src/scipy/stats/__pycache__/t.cpython-311.pyc,,
jax/_src/scipy/stats/__pycache__/truncnorm.cpython-311.pyc,,
jax/_src/scipy/stats/__pycache__/uniform.cpython-311.pyc,,
jax/_src/scipy/stats/__pycache__/vonmises.cpython-311.pyc,,
jax/_src/scipy/stats/_core.py,sha256=Vjnb6HVu1nuIpna6DvcpEKvpgUZ7M9pFpNBD4gSZKGM,4959
jax/_src/scipy/stats/bernoulli.py,sha256=ad0-qS___0-0PXi_JSNcnIc8-2kspvmCECwPSCW1lBE,2250
jax/_src/scipy/stats/beta.py,sha256=vSpdgyaZ4gkHKB9uxFFQr5Fx9fC5Ei_X1etBLj7UHPs,2940
jax/_src/scipy/stats/betabinom.py,sha256=kBh8yc_9QaCBmkHcgs25HfG7vpk9MkXYklw22ffbl6I,1993
jax/_src/scipy/stats/binom.py,sha256=tFSUOmAWytQVitfp4zz3cOAMivDNCjVHD3vB0i9IXEM,1634
jax/_src/scipy/stats/cauchy.py,sha256=lfA_Jhs_mfKqRV9dfE7jyj0lnnX8QU29gOoRwghsBKA,3106
jax/_src/scipy/stats/chi2.py,sha256=R5VIaCOsI4cCi1AYDlUxe_NOCZY8b7_WXib9bEd9aDw,2924
jax/_src/scipy/stats/dirichlet.py,sha256=0cp5kgUBlEvAO-B4vWRsa1MFwpHg2aOztpAWbKxLYO0,2136
jax/_src/scipy/stats/expon.py,sha256=zOROY78Y70-DQRA7hrEOcbarXLHhfb_9BU3VPISKmpc,1314
jax/_src/scipy/stats/gamma.py,sha256=EdsviA1SEfCLmsnZv-aBHLV8dZv25sf3W9BLMEYR5Tg,2539
jax/_src/scipy/stats/gennorm.py,sha256=Y70sqdWpQnqPiL577XFHZxTIW4ZC-hq30Rg1MRahxg4,1295
jax/_src/scipy/stats/geom.py,sha256=fDxS432YyqwPuvWNvHwR6CuMckopnFGMUOsNOwrryAk,1382
jax/_src/scipy/stats/kde.py,sha256=m8SFp-zXy5_rZi4Ku90rOh4CTbeTEKfDBMUSsxWDvCA,9840
jax/_src/scipy/stats/laplace.py,sha256=0UxvVoID2Fj9K-UBQI28pGKsaGv7OKLqAXBVU1VuMvM,1787
jax/_src/scipy/stats/logistic.py,sha256=aSRXac2fq0gks2MXUNJFhWwPywQiKu8a8Up8e-29cY8,2406
jax/_src/scipy/stats/multinomial.py,sha256=I7kD1jq3Kkp-EA_0VRRwegazpDKQ-jCn_Yi26_N620g,1675
jax/_src/scipy/stats/multivariate_normal.py,sha256=-FrmiStPBfnU_2zgxyOg5qILmIHzo9YvaBzOoJ3CYGU,2219
jax/_src/scipy/stats/nbinom.py,sha256=DviJZ0ATjbVeONzaBrXOians399z-uAZvRxKQj_nTZU,1698
jax/_src/scipy/stats/norm.py,sha256=7zlhjqHNIjWp0ysIkZ2avQAGm2skdSsLih8sOBq3I20,2863
jax/_src/scipy/stats/pareto.py,sha256=Yt_WwIdmC6mvf8xGwh-SpznyRHgyglYcg64HikBUBjE,1497
jax/_src/scipy/stats/poisson.py,sha256=56Y1k4sOfWz7h361KZHH-eUAx-G9ndPKQX5pMNObedg,1684
jax/_src/scipy/stats/t.py,sha256=nX8aCRT6Muy3GNg36piZ-wmaZwWjbM6N--YN7S1CKZc,1822
jax/_src/scipy/stats/truncnorm.py,sha256=zJoVLIKz1y81wcmuIkfYnTtaC5oQ_4kh7NIIONtgu3I,3965
jax/_src/scipy/stats/uniform.py,sha256=jTU0WPQwX6ohnaeLwyvdkEN5a3Zh40gRJm-k6ATCWBU,1325
jax/_src/scipy/stats/vonmises.py,sha256=NUMe-B35glSUasmZ9Qny8rCB7cgklCTVKJuu59b_suU,1272
jax/_src/sharding.py,sha256=T6r_pLVJ8mApjB0br4nNSWj7x45nWJc3FauyXOHgcd4,4921
jax/_src/sharding_impls.py,sha256=inqm8W_9VYONBYfw-RQeklZiksb_TGeBwGaKQW8lYGY,51032
jax/_src/sharding_specs.py,sha256=UVydUNzQnQbKBx5sUTMdjr5Sk81DYWwt7swmsUS6o7Y,13532
jax/_src/source_info_util.py,sha256=U_Uq9Ax2u0HGlNRo2t21Vr1aT8Ei2UI-kiiDkfANmhc,8973
jax/_src/stages.py,sha256=i-DjohooFTLuqr5VSpyXviYlVPXJ-ApszhL01icnRCY,24210
jax/_src/state/__init__.py,sha256=lC2yPw3KD8eXCSl2SDzAFlojhUfS2skbOSRFrvQ-YFU,866
jax/_src/state/__pycache__/__init__.cpython-311.pyc,,
jax/_src/state/__pycache__/discharge.cpython-311.pyc,,
jax/_src/state/__pycache__/primitives.cpython-311.pyc,,
jax/_src/state/__pycache__/types.cpython-311.pyc,,
jax/_src/state/__pycache__/utils.cpython-311.pyc,,
jax/_src/state/discharge.py,sha256=Kq5MASTFNiJctqhKWXwMO_Z83rVFVdYubzDssc_FtVE,33458
jax/_src/state/primitives.py,sha256=biv6QHn8GcyL7IuajPJi6YGFFnpriWyoSAA42TqCyHI,21661
jax/_src/state/types.py,sha256=SZ8DkRwD6PeRHpkYGtU8ZMNnnqf3okkSVz-ODRHftOc,5708
jax/_src/state/utils.py,sha256=fn8NOU_WBT04Rd21eWmzecHiPKtBTaX5OqTqgeri5V8,2340
jax/_src/test_util.py,sha256=bDXygOqFhkUQzAfYvpXFBY-ctWygVXVZ6JnmadvvCM8,47169
jax/_src/third_party/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
jax/_src/third_party/__pycache__/__init__.cpython-311.pyc,,
jax/_src/third_party/numpy/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
jax/_src/third_party/numpy/__pycache__/__init__.cpython-311.pyc,,
jax/_src/third_party/numpy/__pycache__/linalg.cpython-311.pyc,,
jax/_src/third_party/numpy/linalg.py,sha256=TZc4moT_05TmTxRqL9fzP8DyzYBYq5iBGZIgyvtoFO0,6211
jax/_src/third_party/scipy/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
jax/_src/third_party/scipy/__pycache__/__init__.cpython-311.pyc,,
jax/_src/third_party/scipy/__pycache__/betaln.cpython-311.pyc,,
jax/_src/third_party/scipy/__pycache__/interpolate.cpython-311.pyc,,
jax/_src/third_party/scipy/__pycache__/linalg.cpython-311.pyc,,
jax/_src/third_party/scipy/__pycache__/signal_helper.cpython-311.pyc,,
jax/_src/third_party/scipy/betaln.py,sha256=TT77MW5HcPt-C6m-jHT43mGPLuAKO9ANcoEsXjl8JDE,2373
jax/_src/third_party/scipy/interpolate.py,sha256=PMusO102slpONLd0mydFZ1UAHRQsSD3PIkUbgBnFoNg,5869
jax/_src/third_party/scipy/linalg.py,sha256=8epwj40m-vHg56ehkaZGWKQHnarJhQ3UoAdQEqItk-E,2690
jax/_src/third_party/scipy/signal_helper.py,sha256=3lDx1SLvQeDiQgnsCohP1_mPsNyvlT_JXeFBa9Ehpaw,3070
jax/_src/tpu_custom_call.py,sha256=MXfeMNNX_ksGArFaycJq6k6LLgTcpge7uiJ_xaJTT9Y,14079
jax/_src/traceback_util.py,sha256=z-lRcmKrAUXPBe9uw5sO8Ibbw5qulzqCJ0a5yrIdgKM,9042
jax/_src/tree_util.py,sha256=4nJlKrkEJe9-c71iceIam7H8_SHq0JIzpyvH3-6gYzg,35994
jax/_src/typing.py,sha256=zyjBa0jDgfuweBiUDdutk3x-7HIbMtNchI2ZOm97-3g,2740
jax/_src/util.py,sha256=XHtdWl33V-UZaqjnvKMMBVTLFr-8Ey98atmcKUF-B2Y,17280
jax/_src/xla_bridge.py,sha256=shdwfjlqJIiwNuJFgjQ_oCb3SLPk3eXV1P9B5gyO1R0,27631
jax/abstract_arrays.py,sha256=F1foMV5BBNaDByw_Qu3gLSwQs6QRmWMi1IKd76qK5vg,1169
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=XVCIPglqBadvnmZz0rWy13_wxOVoR1rf0qVcUOx1dqA,4716
jax/config.py,sha256=eARsQrI4Ti7e6y56r_dAj6BJfd5qS4BhpI1ccBdLf5w,696
jax/core.py,sha256=I-pmIH_D8788fqp4vcXbY1e3ZEoI81FE7Q_qAtjvIO4,6502
jax/custom_batching.py,sha256=zQmkJIxOxxgAqEgby1q-vQWHYKzyqp1__n8Hqs04OV4,657
jax/custom_derivatives.py,sha256=FWBXGy0zp70zI46ErbGucn23R6axD_0BuvZoeZJaW0U,1341
jax/custom_transpose.py,sha256=BmKD3o61GFyTE68GoNX_3Z8NBJl5HItzBCi8RehA9ks,644
jax/debug.py,sha256=HXdNZlaM5LxtEYS6fsH3J4LRgiDVL-5W6AXcimbFS5o,1083
jax/distributed.py,sha256=_-zv-zCv4BirRPOM8Uy6zXOCp-V4ej4zNDS2YTiCRSw,638
jax/dlpack.py,sha256=Xq3mQbR6R_ak_CD2itRttfJx11cVaXtdl_xLQL3clfY,653
jax/dtypes.py,sha256=Mp2QLfj2hwLdfJHz8Kpzg2FiX85hJMyBLo3rapL3iMo,1146
jax/errors.py,sha256=KvLylW0zzk5D6q1ODpe6hWjLXtFYV6WmkjVvxPaCZdY,1238
jax/example_libraries/__init__.py,sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw,581
jax/example_libraries/__pycache__/__init__.cpython-311.pyc,,
jax/example_libraries/__pycache__/optimizers.cpython-311.pyc,,
jax/example_libraries/__pycache__/stax.cpython-311.pyc,,
jax/example_libraries/optimizers.py,sha256=d7pdstUbzeu5JMI7iCg-BLZaj__OjDdGDMFPl1lLO_Q,20536
jax/example_libraries/stax.py,sha256=cMt8-39Q-o-tWwquR00RAlbbiWaitHLK4P03Xj0Wa3M,13849
jax/experimental/__init__.py,sha256=PIs2CgK8WYTCTx1EODRmbnaQuPWaJb-NDFYy9diby_4,877
jax/experimental/__pycache__/__init__.cpython-311.pyc,,
jax/experimental/__pycache__/checkify.cpython-311.pyc,,
jax/experimental/__pycache__/custom_partitioning.cpython-311.pyc,,
jax/experimental/__pycache__/host_callback.cpython-311.pyc,,
jax/experimental/__pycache__/jet.cpython-311.pyc,,
jax/experimental/__pycache__/maps.cpython-311.pyc,,
jax/experimental/__pycache__/mesh_utils.cpython-311.pyc,,
jax/experimental/__pycache__/multihost_utils.cpython-311.pyc,,
jax/experimental/__pycache__/ode.cpython-311.pyc,,
jax/experimental/__pycache__/pjit.cpython-311.pyc,,
jax/experimental/__pycache__/profiler.cpython-311.pyc,,
jax/experimental/__pycache__/rnn.cpython-311.pyc,,
jax/experimental/__pycache__/serialize_executable.cpython-311.pyc,,
jax/experimental/__pycache__/shard_map.cpython-311.pyc,,
jax/experimental/__pycache__/topologies.cpython-311.pyc,,
jax/experimental/__pycache__/x64_context.cpython-311.pyc,,
jax/experimental/array_serialization/__init__.py,sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw,581
jax/experimental/array_serialization/__pycache__/__init__.cpython-311.pyc,,
jax/experimental/array_serialization/__pycache__/serialization.cpython-311.pyc,,
jax/experimental/array_serialization/__pycache__/serialization_test.cpython-311.pyc,,
jax/experimental/array_serialization/serialization.py,sha256=wKGV4OUM85HvH-Zf1gjDJbxp9A7gombuKmSBCWzR8os,20947
jax/experimental/array_serialization/serialization_test.py,sha256=zECsEosyj2YcggxZYcZTJ_74weqZ-u48X1nkfeNgJpU,12965
jax/experimental/checkify.py,sha256=OfaBF3DRHQpmIHEex-3nFtu3D4hHGLK7GIpSVZbewYA,1213
jax/experimental/compilation_cache/__init__.py,sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw,581
jax/experimental/compilation_cache/__pycache__/__init__.cpython-311.pyc,,
jax/experimental/compilation_cache/__pycache__/compilation_cache.cpython-311.pyc,,
jax/experimental/compilation_cache/compilation_cache.py,sha256=FuNWsnrwmZgC00X2-K27s-D1bKIR7SC7gkMGLQ1WxmM,731
jax/experimental/custom_partitioning.py,sha256=g3_8sVVyxxkHUr2iOGghXq4QWUs_yw0xQQ0B2B30JrM,21392
jax/experimental/export/__init__.py,sha256=2cJ-OgrtfVES902JD29-I0UJH22lXrSoUepUwWoEYrU,682
jax/experimental/export/__pycache__/__init__.cpython-311.pyc,,
jax/experimental/export/__pycache__/export.cpython-311.pyc,,
jax/experimental/export/__pycache__/shape_poly.cpython-311.pyc,,
jax/experimental/export/export.py,sha256=vdmUIZymtaX1l8H4BAQ1_5QKuGIjOrndPXiOWnBrZAM,47071
jax/experimental/export/shape_poly.py,sha256=W_GYssslocKujFUx3oiEPtdaz2CK_D5Eq-OZsils1UY,60453
jax/experimental/host_callback.py,sha256=-W_t9CULwkvTMmGm18lcneCRePIVi1XV5BOoVdFcvb0,79437
jax/experimental/jax2tf/__init__.py,sha256=7tXoLqrwpmFJsqrRnkqCg0EqbiIFJ9J0YOnlRGSXbHI,1085
jax/experimental/jax2tf/__pycache__/__init__.cpython-311.pyc,,
jax/experimental/jax2tf/__pycache__/call_tf.cpython-311.pyc,,
jax/experimental/jax2tf/__pycache__/impl_no_xla.cpython-311.pyc,,
jax/experimental/jax2tf/__pycache__/jax2tf.cpython-311.pyc,,
jax/experimental/jax2tf/__pycache__/jax_export.cpython-311.pyc,,
jax/experimental/jax2tf/__pycache__/shape_poly.cpython-311.pyc,,
jax/experimental/jax2tf/call_tf.py,sha256=hh4HuTPr8_7TQZErGCpDV5gsBshUU2YBx4CgfVaWDMM,27224
jax/experimental/jax2tf/examples/__init__.py,sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw,581
jax/experimental/jax2tf/examples/__pycache__/__init__.cpython-311.pyc,,
jax/experimental/jax2tf/examples/__pycache__/keras_reuse_main.cpython-311.pyc,,
jax/experimental/jax2tf/examples/__pycache__/keras_reuse_main_test.cpython-311.pyc,,
jax/experimental/jax2tf/examples/__pycache__/mnist_lib.cpython-311.pyc,,
jax/experimental/jax2tf/examples/__pycache__/saved_model_lib.cpython-311.pyc,,
jax/experimental/jax2tf/examples/__pycache__/saved_model_main.cpython-311.pyc,,
jax/experimental/jax2tf/examples/__pycache__/saved_model_main_test.cpython-311.pyc,,
jax/experimental/jax2tf/examples/keras_reuse_main.py,sha256=ZwRgp2kz871Y1IBtR_m8lVQB6wZY6IwdSSD9YLVo0is,2900
jax/experimental/jax2tf/examples/keras_reuse_main_test.py,sha256=1FBpLLwCpm9v4udaL0tEN3umiqF4a1lXSr8k483R9Eg,1616
jax/experimental/jax2tf/examples/mnist_lib.py,sha256=V9_KOPwPOR6cBCAg-piiF3WGlIJ4AaYl2iNIQmnEYLM,11486
jax/experimental/jax2tf/examples/saved_model_lib.py,sha256=54tvM9aneRhhhVNyZULCAuX6sjglaZY4xQkBSRYVwW4,7095
jax/experimental/jax2tf/examples/saved_model_main.py,sha256=f4BcJt5KvsNvQrZsKhV4_J6QrxHkppRgu792dy9E2S0,8051
jax/experimental/jax2tf/examples/saved_model_main_test.py,sha256=w6QzDTG-t8f3tdu2a7qfYLoXXWw5bcU7ANJi1nuUIRc,2453
jax/experimental/jax2tf/examples/serving/__init__.py,sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw,581
jax/experimental/jax2tf/examples/serving/__pycache__/__init__.cpython-311.pyc,,
jax/experimental/jax2tf/examples/serving/__pycache__/model_server_request.cpython-311.pyc,,
jax/experimental/jax2tf/examples/serving/model_server_request.py,sha256=IyW59CMUv4y9enSKT95mzgNQEqF8_65ZUAO_p8x41r8,4952
jax/experimental/jax2tf/impl_no_xla.py,sha256=vcfKaGQxsl8IxblcFqXcbnzFipmoAvOPMv0PxmY8Jo8,53998
jax/experimental/jax2tf/jax2tf.py,sha256=vZRnK0phdQEvmPlyve5UVuKOjiUfhyshcLSxJSGQZlg,137888
jax/experimental/jax2tf/jax_export.py,sha256=nYLyGdT2HXf6ZyJ01BU1DybnvP-Jp0rnmHfaQFVT8hY,820
jax/experimental/jax2tf/shape_poly.py,sha256=AmkM9n71JwtSaKqWYBb29_q0GCDDWIGieqPmZA_w7SY,972
jax/experimental/jax2tf/tests/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/experimental/jax2tf/tests/__pycache__/__init__.cpython-311.pyc,,
jax/experimental/jax2tf/tests/__pycache__/back_compat_test.cpython-311.pyc,,
jax/experimental/jax2tf/tests/__pycache__/back_compat_test_util.cpython-311.pyc,,
jax/experimental/jax2tf/tests/__pycache__/back_compat_tf_test.cpython-311.pyc,,
jax/experimental/jax2tf/tests/__pycache__/call_tf_test.cpython-311.pyc,,
jax/experimental/jax2tf/tests/__pycache__/control_flow_ops_test.cpython-311.pyc,,
jax/experimental/jax2tf/tests/__pycache__/converters.cpython-311.pyc,,
jax/experimental/jax2tf/tests/__pycache__/cross_compilation_check.cpython-311.pyc,,
jax/experimental/jax2tf/tests/__pycache__/jax2tf_limitations.cpython-311.pyc,,
jax/experimental/jax2tf/tests/__pycache__/jax2tf_test.cpython-311.pyc,,
jax/experimental/jax2tf/tests/__pycache__/jax_primitives_coverage_test.cpython-311.pyc,,
jax/experimental/jax2tf/tests/__pycache__/model_harness.cpython-311.pyc,,
jax/experimental/jax2tf/tests/__pycache__/models_test_main.cpython-311.pyc,,
jax/experimental/jax2tf/tests/__pycache__/primitive_harness.cpython-311.pyc,,
jax/experimental/jax2tf/tests/__pycache__/primitives_test.cpython-311.pyc,,
jax/experimental/jax2tf/tests/__pycache__/savedmodel_test.cpython-311.pyc,,
jax/experimental/jax2tf/tests/__pycache__/shape_poly_test.cpython-311.pyc,,
jax/experimental/jax2tf/tests/__pycache__/sharding_test.cpython-311.pyc,,
jax/experimental/jax2tf/tests/__pycache__/tf_test_util.cpython-311.pyc,,
jax/experimental/jax2tf/tests/back_compat_test.py,sha256=LKz9z4p5F1a9nVh2XSO7euXjOz4WLibcuNOPnl706Kw,32337
jax/experimental/jax2tf/tests/back_compat_test_util.py,sha256=mF1eUdwexzQgaL-NApA30lBahwzl-vx6reC6C4okknc,14955
jax/experimental/jax2tf/tests/back_compat_tf_test.py,sha256=w5r3tl-6_YHix44mgWVuvaPLPH2f8Wv_te080B7mBwA,5346
jax/experimental/jax2tf/tests/call_tf_test.py,sha256=yOcQEdtmBmAegCI6Fruyp4a9JxSmKFfUtbAy1cKgyHE,57852
jax/experimental/jax2tf/tests/control_flow_ops_test.py,sha256=LlIDjrL29S_LCJYF42h9SBU2AmLPH0JNMkh82PfWsv4,10142
jax/experimental/jax2tf/tests/converters.py,sha256=IMOkFWZUnlA1RCq_eHoPIy6IhoLhmFKTRWNQyTjxbVE,3527
jax/experimental/jax2tf/tests/cross_compilation_check.py,sha256=l1lxZV6YvxJGy19qqGBQ9HSo5JNNROOIGl4g9EU2OxA,7387
jax/experimental/jax2tf/tests/jax2tf_limitations.py,sha256=agmXkz5QEMt_rBdp2Z8Xzw92e30nmhN8L8L-bUJ8FGI,58717
jax/experimental/jax2tf/tests/jax2tf_test.py,sha256=7HIba-LPt2bPtDF4HXUM7pqV77IPOBTGMybkOxqjE6E,65177
jax/experimental/jax2tf/tests/jax_primitives_coverage_test.py,sha256=PE7AMvKWKK2T37SyzpNUev148u-17v-LYCANeDGGhZY,6697
jax/experimental/jax2tf/tests/model_harness.py,sha256=4wbCdtEReqIbLozI12nI7ZQy2MhkTK4nJsYl--sXngE,13927
jax/experimental/jax2tf/tests/models_test_main.py,sha256=fX4d_7NWKNdx4hTLriKIFu0udhIGU66t3_Cqduueqpg,8978
jax/experimental/jax2tf/tests/primitive_harness.py,sha256=Iqc97TRXWbgQmb2x1Z_AADF1hpb98BbmJS8BygTcywA,125141
jax/experimental/jax2tf/tests/primitives_test.py,sha256=Cs9iVI-jGeQDT6fyyYZKWtW5w8XqLt6GJUDQ1bswbp8,14036
jax/experimental/jax2tf/tests/savedmodel_test.py,sha256=ZnhxROx6NRlsPuq95fTt39yNHEQGVEGWgMuKdUUyS8M,12815
jax/experimental/jax2tf/tests/shape_poly_test.py,sha256=s1e0ZBR9Itf0hDvfYlsGCLQ3Uw7HRv8gerTV1zbssgA,149118
jax/experimental/jax2tf/tests/sharding_test.py,sha256=jg75v2FhF9YaAA12tPkYjoZ1TYR06WNFyVmrDuaokuA,30517
jax/experimental/jax2tf/tests/tf_test_util.py,sha256=TJpwrDw-iF4lUDQFWZyxTNz4-Eyp81I8YCU5or-SNcI,20102
jax/experimental/jet.py,sha256=4Lk7tOMSThQDy5zoH3QJ72Qo-s8B6jvEfUUgw2eQle0,26899
jax/experimental/maps.py,sha256=yjycDuNpT_aNQJK_XUBqhDDfY7qHUKeBpfi7w_zwSAs,964
jax/experimental/mesh_utils.py,sha256=3ujgPex5675-b1YZbk_rXUzMogeo2ux1tG-1dVysheg,15605
jax/experimental/mosaic/__init__.py,sha256=Be-z9rRmax_7Ps7HzLkWBgHOMkU_T9JGvbyWkI-njt0,770
jax/experimental/mosaic/__pycache__/__init__.cpython-311.pyc,,
jax/experimental/mosaic/__pycache__/dialects.cpython-311.pyc,,
jax/experimental/mosaic/dialects.py,sha256=fSjIbjy1Ygx6EhF1XRj9EEwX4JEo8K_Ky84_KY6EbW0,958
jax/experimental/multihost_utils.py,sha256=S3GN1u4YTL0IU8mOfyYwtZeqSS74kqDR5Db2iywoD3M,16085
jax/experimental/ode.py,sha256=rYtcxYMLz5vR5aiPYiRcQKsHqLFF4JR4taypMeJr78M,10659
jax/experimental/pallas/__init__.py,sha256=R6j17GnEqAa_bLaJdpsiQXKCCaY9JxQSCTlmrz3Jz74,2174
jax/experimental/pallas/__pycache__/__init__.cpython-311.pyc,,
jax/experimental/pallas/__pycache__/gpu.cpython-311.pyc,,
jax/experimental/pallas/__pycache__/tpu.cpython-311.pyc,,
jax/experimental/pallas/gpu.py,sha256=JfWURS5-kosWDZTENWHOKolig0-Wk2iD0lSqS_kvmuU,897
jax/experimental/pallas/ops/__init__.py,sha256=OOnVMiFg3iFxhCoKLDFL8FzN1Z7ifDdzlxXUbNqem-k,781
jax/experimental/pallas/ops/__pycache__/__init__.cpython-311.pyc,,
jax/experimental/pallas/ops/__pycache__/attention.cpython-311.pyc,,
jax/experimental/pallas/ops/__pycache__/layer_norm.cpython-311.pyc,,
jax/experimental/pallas/ops/__pycache__/rms_norm.cpython-311.pyc,,
jax/experimental/pallas/ops/__pycache__/softmax.cpython-311.pyc,,
jax/experimental/pallas/ops/attention.py,sha256=-ychF4e2tClqJ27t3iGyLOtSeosZxbZs4Ob-XNzvAws,18706
jax/experimental/pallas/ops/layer_norm.py,sha256=T-zec_Kec0e_sLFg0h4sa8TUoUOvxR45CRcNJmBON7E,10949
jax/experimental/pallas/ops/rms_norm.py,sha256=0NCrrGab5ZG0-xlzUbU2ePW3Y9h7pghC6TztM8nmL7A,9842
jax/experimental/pallas/ops/softmax.py,sha256=1DN8CrgJU2KZhFKf-VrXp5pbH64Yk-ocxkZmYUOWHDE,2647
jax/experimental/pallas/ops/tpu/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/experimental/pallas/ops/tpu/__pycache__/__init__.cpython-311.pyc,,
jax/experimental/pallas/ops/tpu/__pycache__/flash_attention.cpython-311.pyc,,
jax/experimental/pallas/ops/tpu/flash_attention.py,sha256=QxcRVuOH6e80b3TVG32zpb3AcPDCOdlpoopfMIypFSM,35746
jax/experimental/pallas/tpu.py,sha256=FzfBcToGrgpjelkQKNaleF4Ig6tRNAKmayxLkSZwYnY,1515
jax/experimental/pjit.py,sha256=yy8-zVSHV9SJno7hEXmoyWmbdYqG6fF68EVoXs3CX9E,1440
jax/experimental/profiler.py,sha256=DJ0ETyXEZ56c6LYWRY_ankgPOs5oWZejGtGXS-Ttbus,1424
jax/experimental/rnn.py,sha256=ohc84MTY5FhBcxgXENzDt19rbbkuL0z5DKV5qo1nTaI,19199
jax/experimental/serialize_executable.py,sha256=8wx9DBxOd2SxDYvDUSJQ6WmRc8Bsls9xUIcCOYCU2ik,3014
jax/experimental/shard_map.py,sha256=CmzbnuO_Or9Dt4UvizlpsdDQE0wpSHM62dC902fm8_c,76100
jax/experimental/sparse/__init__.py,sha256=hRWildIHCGDjwsHx8I8YzwJk6Q5cbkPSQ2NMPd1EPUM,11186
jax/experimental/sparse/__pycache__/__init__.cpython-311.pyc,,
jax/experimental/sparse/__pycache__/_base.cpython-311.pyc,,
jax/experimental/sparse/__pycache__/_lowerings.cpython-311.pyc,,
jax/experimental/sparse/__pycache__/ad.cpython-311.pyc,,
jax/experimental/sparse/__pycache__/api.cpython-311.pyc,,
jax/experimental/sparse/__pycache__/bcoo.cpython-311.pyc,,
jax/experimental/sparse/__pycache__/bcsr.cpython-311.pyc,,
jax/experimental/sparse/__pycache__/coo.cpython-311.pyc,,
jax/experimental/sparse/__pycache__/csr.cpython-311.pyc,,
jax/experimental/sparse/__pycache__/linalg.cpython-311.pyc,,
jax/experimental/sparse/__pycache__/random.cpython-311.pyc,,
jax/experimental/sparse/__pycache__/test_util.cpython-311.pyc,,
jax/experimental/sparse/__pycache__/transform.cpython-311.pyc,,
jax/experimental/sparse/__pycache__/util.cpython-311.pyc,,
jax/experimental/sparse/_base.py,sha256=GV1aSqPsxkysLQQUQ1TU2V2sHZejnKD0SK4HVxuqWus,3188
jax/experimental/sparse/_lowerings.py,sha256=4BKVcIwJ1gJqO3gi7_X7HVTmY10HoZ-UnWzqCNj-bBk,7229
jax/experimental/sparse/ad.py,sha256=Rwg1Lz1SwE3CorrypbPOMdOrIyUViPsns3akjtbv9JA,7572
jax/experimental/sparse/api.py,sha256=jOBzLmtMafrEssVTtn6m2wSIdsRP79W7HBIz1yEgo8A,6609
jax/experimental/sparse/bcoo.py,sha256=w5jZsw17PHjmItohWJRLb5WeB9FYKF9-7Noo6ZBeUf4,128160
jax/experimental/sparse/bcsr.py,sha256=qRQk-HvbPiCvaRFzjHSxAjM_B-4qTBomvU3-eucZd0g,38191
jax/experimental/sparse/coo.py,sha256=RCDodUA9RJm2fZ33eGWfU5YAgn8PLCmcJ9gSSAdE1K0,24134
jax/experimental/sparse/csr.py,sha256=j7ib2vljbZl4gBJ8ijpLUO-cixg_4A4mK0c1emGcYgo,24422
jax/experimental/sparse/linalg.py,sha256=FnUdYNTAYp4XPjyKCx5kiu4ZxP2GDyusSB6qClhZVVw,23337
jax/experimental/sparse/random.py,sha256=eW-JQxJ1RdWwzPIlS83Zg9GdIEdE9J7uUuECMlCHZdQ,3889
jax/experimental/sparse/test_util.py,sha256=oA3ytZtN_oY1Wc-w8FSR8dMJ5UQpTW_ePS0xmxY2dq4,7709
jax/experimental/sparse/transform.py,sha256=MtDh639AIX9ksINXwzwa63jH3PO_U621yqZWakC0N1Q,36915
jax/experimental/sparse/util.py,sha256=kZWNFGXyXlkKrpy3HN9CNeB_Hf1DP_S0PkMol33mOz4,4385
jax/experimental/topologies.py,sha256=IN2LBhwTIYzFZnph7cRnJUHCH1zXREZjLha3g8uVFe0,1864
jax/experimental/x64_context.py,sha256=QXJ2sdwE_hStQ-L5QPbrNVRXws3k_FuzZnWa4iTR_VM,1736
jax/extend/__init__.py,sha256=hEGR-vzXv8qZMpGHtWhD0z7EqIuI0xAnhI88KnTAkt0,1165
jax/extend/__pycache__/__init__.cpython-311.pyc,,
jax/extend/__pycache__/linear_util.cpython-311.pyc,,
jax/extend/__pycache__/random.cpython-311.pyc,,
jax/extend/linear_util.py,sha256=HlQLycGFvu0ZbYw0qU7A5l_mRbf6w75rd4VA7s12yb8,986
jax/extend/random.py,sha256=ZlGFFsoZpRNQQEpxC_JwZYtbjQYRECDQh4zHIEv-U5M,1071
jax/flatten_util.py,sha256=jGwQUB80s2jr78XuEaT1_OP2OUyVm8Pk-yAQq-C1SsY,645
jax/image/__init__.py,sha256=pEcSFsopr4U6kry9Gy82id9Whs95vn0XfBIy7haJm6c,1027
jax/image/__pycache__/__init__.cpython-311.pyc,,
jax/interpreters/__init__.py,sha256=HNgKpgKpz4svFyTlHE9wMrDp5iQtg6XeeUshYBOnRko,690
jax/interpreters/__pycache__/__init__.cpython-311.pyc,,
jax/interpreters/__pycache__/ad.cpython-311.pyc,,
jax/interpreters/__pycache__/batching.cpython-311.pyc,,
jax/interpreters/__pycache__/mlir.cpython-311.pyc,,
jax/interpreters/__pycache__/partial_eval.cpython-311.pyc,,
jax/interpreters/__pycache__/pxla.cpython-311.pyc,,
jax/interpreters/__pycache__/xla.cpython-311.pyc,,
jax/interpreters/ad.py,sha256=u8bSc_VvMVs4DB8MMb9aO2sAMqqeJUzivFnE8JmMfuU,2748
jax/interpreters/batching.py,sha256=LaCZQ2bg8xB3NdlsxkdI9DmPH0D5KJGrzSl6OYr6E2I,2688
jax/interpreters/mlir.py,sha256=tsqxwluECVhLc7tCleKrTDLQ6yaVINSqXVKJghMa_S8,2583
jax/interpreters/partial_eval.py,sha256=KzL42jZ5EZsj6NZOp8PAigI7m9wGyKjqC9Zdv0Fshu8,4213
jax/interpreters/pxla.py,sha256=juyzYAMMosmHElRzhwuBDXxewUqedLV-ajhhEZfltJs,1896
jax/interpreters/xla.py,sha256=3pIxS72NZCAk9QlEPDp4-ZvthMsilkG5pOvL_POV3zs,4720
jax/lax/__init__.py,sha256=wEsSEWDsDVesym9Z8RIV0TeToLQYkN4k7xdx1nNxtzY,10614
jax/lax/__pycache__/__init__.cpython-311.pyc,,
jax/lax/__pycache__/linalg.cpython-311.pyc,,
jax/lax/linalg.py,sha256=w66gTsb4Z1X5_PjFUQbSP2FJgN0nNEZ9wMaBS9li6xY,1016
jax/lib/__init__.py,sha256=n6Ugts8VenX6PmSqtbE3cxWrVeZCVYevWggJFZ2sgAc,769
jax/lib/__pycache__/__init__.cpython-311.pyc,,
jax/lib/__pycache__/xla_bridge.cpython-311.pyc,,
jax/lib/xla_bridge.py,sha256=rO_3LqPwRtrZFVAh5ZtjGP8Xc2mb2GuREFGBJbm60SU,842
jax/linear_util.py,sha256=vH-b8qwOfaImoYX2kFiJlVfoLFixU9Q9W1_0b-FlR-Y,2453
jax/monitoring.py,sha256=TvXZn2SrpvtilfovXIJO_dHexznozSHncz7CEmTmbuw,1131
jax/nn/__init__.py,sha256=xEdi9Drex40slz4dJRQmndY9qm7lf0ceu67qNmkKdQo,1428
jax/nn/__pycache__/__init__.cpython-311.pyc,,
jax/nn/__pycache__/initializers.cpython-311.pyc,,
jax/nn/initializers.py,sha256=P6Ck5_RjLUUB-pwiCtCbg5BozocSxtz1UDgdfVK6Ghs,1469
jax/numpy/__init__.py,sha256=MmcfVBkzZwJK81Z1IWAomgvGNpl-DYQZcDT5h-AOTu4,12076
jax/numpy/__init__.pyi,sha256=vGqi8kH02T-fRYcgEKLfycDq1d8VQrNjZUwryRc9mxY,7324
jax/numpy/__pycache__/__init__.cpython-311.pyc,,
jax/numpy/__pycache__/fft.cpython-311.pyc,,
jax/numpy/__pycache__/linalg.cpython-311.pyc,,
jax/numpy/fft.py,sha256=uPIPoBVZmUyE1BOCtJeYVSTPwKUWDhbNV7GpzeenTTk,1084
jax/numpy/linalg.py,sha256=1-lGB4PtTA9xr5NZR4SfwFaYM0xGTEG09pZERJEzCqY,1203
jax/ops/__init__.py,sha256=3kFyxifjh_OkJqQDDYjSkQw6EH173Rf-fb7s5aPP1iQ,870
jax/ops/__pycache__/__init__.cpython-311.pyc,,
jax/prng.py,sha256=xjrTOsXwDdOk2p83GGN8ZV99gqnZffYgzYdUlYjUTXw,2395
jax/profiler.py,sha256=tMxGkMNYoug4DQGLE-OKx0z_nhlR-gBSs6vFpXd3jzY,1119
jax/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
jax/random.py,sha256=MS8f2ggzd_ffhfBEF6qEy5FufFiwiW189tbFUzTkxgM,7665
jax/scipy/__init__.py,sha256=IsnrQSBykAPMH3rSy7XltTWbqsGYVIySjaSMRk66i1c,1474
jax/scipy/__pycache__/__init__.cpython-311.pyc,,
jax/scipy/__pycache__/fft.cpython-311.pyc,,
jax/scipy/__pycache__/integrate.cpython-311.pyc,,
jax/scipy/__pycache__/linalg.cpython-311.pyc,,
jax/scipy/__pycache__/ndimage.cpython-311.pyc,,
jax/scipy/__pycache__/signal.cpython-311.pyc,,
jax/scipy/__pycache__/special.cpython-311.pyc,,
jax/scipy/cluster/__init__.py,sha256=vQWrNtPmBCHRT1-G_u6MN9byPWzUfwNawdc9HzSYGcI,768
jax/scipy/cluster/__pycache__/__init__.cpython-311.pyc,,
jax/scipy/cluster/__pycache__/vq.cpython-311.pyc,,
jax/scipy/cluster/vq.py,sha256=LrojEdKqZlhFlSZCYbuCWck7Jn2MeF9aqFf6ip_Y2DM,776
jax/scipy/fft.py,sha256=0EABRVTc4R4OlsBbYFtTV6_dSkVmUdzSnuAZyPW4_nE,809
jax/scipy/integrate.py,sha256=STEn2u4Kw5N-L36DwzH0AoSSvT34knl11TtiZT9KvYQ,777
jax/scipy/interpolate/__init__.py,sha256=RI5oUXPC-8tKmYURnnrnA-e1LDbnNVOVVqkGSpagEBM,760
jax/scipy/interpolate/__pycache__/__init__.cpython-311.pyc,,
jax/scipy/linalg.py,sha256=pd7kF8mgpRnaCxrY_DI2koCGleKYwl4tBSMIs4NMlQg,2012
jax/scipy/ndimage.py,sha256=JUcTRgkwTxZ6GwR3yMLFU31szaYuUPc83JGfW2yWxOQ,788
jax/scipy/optimize/__init__.py,sha256=sLE5C83qmDBkS7qF0IzbeStYQwrAJWAze9-pmTVPFNg,822
jax/scipy/optimize/__pycache__/__init__.cpython-311.pyc,,
jax/scipy/signal.py,sha256=ILku2cOpXj0XWAGwBVTv7cb-n2BI_O8gl1R0juJZGxo,975
jax/scipy/sparse/__init__.py,sha256=HCUon--lvEN7ep4bU1gcntvclsQS3K3C2okuFHJpWz4,757
jax/scipy/sparse/__pycache__/__init__.cpython-311.pyc,,
jax/scipy/sparse/__pycache__/linalg.cpython-311.pyc,,
jax/scipy/sparse/linalg.py,sha256=sXphTZ_Gba0IfRZ6vuGkvwcslNi0nOwvt-yQ3XrtBVc,810
jax/scipy/spatial/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/scipy/spatial/__pycache__/__init__.cpython-311.pyc,,
jax/scipy/spatial/__pycache__/transform.cpython-311.pyc,,
jax/scipy/spatial/transform.py,sha256=fKQnQ0IXpn5ri0pDjhEWQxc7Oem02FWhD2wb1Om4bro,802
jax/scipy/special.py,sha256=emQ2I1-KiU68KK2hu8DUMpY9dbmqaNSyGgwg-Tsl_zs,1490
jax/scipy/stats/__init__.py,sha256=lBmF2s1V6vqb7BiI4j5JDg5teVPHzq13FOt7b_WamcE,1930
jax/scipy/stats/__pycache__/__init__.cpython-311.pyc,,
jax/scipy/stats/__pycache__/bernoulli.cpython-311.pyc,,
jax/scipy/stats/__pycache__/beta.cpython-311.pyc,,
jax/scipy/stats/__pycache__/betabinom.cpython-311.pyc,,
jax/scipy/stats/__pycache__/binom.cpython-311.pyc,,
jax/scipy/stats/__pycache__/cauchy.cpython-311.pyc,,
jax/scipy/stats/__pycache__/chi2.cpython-311.pyc,,
jax/scipy/stats/__pycache__/dirichlet.cpython-311.pyc,,
jax/scipy/stats/__pycache__/expon.cpython-311.pyc,,
jax/scipy/stats/__pycache__/gamma.cpython-311.pyc,,
jax/scipy/stats/__pycache__/gennorm.cpython-311.pyc,,
jax/scipy/stats/__pycache__/geom.cpython-311.pyc,,
jax/scipy/stats/__pycache__/laplace.cpython-311.pyc,,
jax/scipy/stats/__pycache__/logistic.cpython-311.pyc,,
jax/scipy/stats/__pycache__/multinomial.cpython-311.pyc,,
jax/scipy/stats/__pycache__/multivariate_normal.cpython-311.pyc,,
jax/scipy/stats/__pycache__/nbinom.cpython-311.pyc,,
jax/scipy/stats/__pycache__/norm.cpython-311.pyc,,
jax/scipy/stats/__pycache__/pareto.cpython-311.pyc,,
jax/scipy/stats/__pycache__/poisson.cpython-311.pyc,,
jax/scipy/stats/__pycache__/t.cpython-311.pyc,,
jax/scipy/stats/__pycache__/truncnorm.cpython-311.pyc,,
jax/scipy/stats/__pycache__/uniform.cpython-311.pyc,,
jax/scipy/stats/__pycache__/vonmises.cpython-311.pyc,,
jax/scipy/stats/bernoulli.py,sha256=V_oR9XYdYZBKZ3Nh5Y5nH_6HicaJtsqoaRFg4GYeujs,819
jax/scipy/stats/beta.py,sha256=rEx9J6IrNfs2Ln0Q1XwKl6TkbNqD5nIu8pdsQe5I-Po,851
jax/scipy/stats/betabinom.py,sha256=-B4rZ-30Nvr4Lxg9D2DXcqxKdBI3pEItbQ-8ncNJWs0,792
jax/scipy/stats/binom.py,sha256=XGO5jDk3nRRgGwHQU6KEl9_kQ5X2qF-2TLgvm3ZUvvs,659
jax/scipy/stats/cauchy.py,sha256=mZmRyAN4umg3GfohvaArByDHuXK54FUHSTkN9ESk2xs,881
jax/scipy/stats/chi2.py,sha256=tO4FT3DBLI-jJTfiH1_f5SqEvnRadBHNQUj1vVqm-Xc,851
jax/scipy/stats/dirichlet.py,sha256=KOGl-X9h2kKS0YlfWzsbMT5xEg3ONDgn-MWvQMBmD_A,792
jax/scipy/stats/expon.py,sha256=WS-OP1ng1jUlodys956ykXfnvs18u7hfGzQv7PnKLDQ,788
jax/scipy/stats/gamma.py,sha256=x98RFaDH4iSYxLwgmqKHsIVtEQBLSGGVZvjogfF4Ssc,852
jax/scipy/stats/gennorm.py,sha256=_NdhXS5lWu6cDZgyizZfDlm1rFVDgXGAEOYIk3Cp7FU,804
jax/scipy/stats/geom.py,sha256=uyOJ8a88yGVTe5YTjg_JO0sLZzBi9OdHMAEOYxtpaaQ,787
jax/scipy/stats/laplace.py,sha256=Oi_fMUikhKflSBJCu1R7UjGkgJLij45M-_9Q5VG0yuw,804
jax/scipy/stats/logistic.py,sha256=c-_nDR9e1a8yk7Ngv7Og5G1vfqjhFLzdTMxDE5UwzUc,845
jax/scipy/stats/multinomial.py,sha256=mysALVSGzMZahxx69AeoGbS-p1Vod2RkGdHZfKJAtRw,794
jax/scipy/stats/multivariate_normal.py,sha256=OL3t4XR7qmqBuw7Ojl3Chvsbwf_UsqtkiTL-9gHQ8JU,802
jax/scipy/stats/nbinom.py,sha256=eMTqIxq9njSnXtEInWHGJBfINEo6N4yVrwji4jsyc58,643
jax/scipy/stats/norm.py,sha256=boHN9rGUO9hx51JuRTxMep-on2HmsNxIbM3YoN-c-HM,879
jax/scipy/stats/pareto.py,sha256=gzEhI-vI4gllIHFFSe1VaTvGbf6GYzPG4siMkxOKRsg,789
jax/scipy/stats/poisson.py,sha256=hMSNeBSToe3D9W6-isGDSZDB-Qyk8GBUtdT8J2PHCMU,804
jax/scipy/stats/t.py,sha256=RO4Zd0Lm2creTHIsFR2tVerxe5TdzRWMOlSSyaH3s1w,784
jax/scipy/stats/truncnorm.py,sha256=PnB_oNVAW0trLzh8ua_T9V0vPnrlBmeBpA1FkPDPoXE,855
jax/scipy/stats/uniform.py,sha256=HKppa0nu4HOg6RFGyV735zhgqX0xsMwYfM0mr_2kQ28,790
jax/scipy/stats/vonmises.py,sha256=nQvzYinMztVFMPZm0fd14ZmHxxoi1CZC_fvRgmBwAMo,791
jax/sharding.py,sha256=UgnvqjbgLSbXzUHLJUzA1d9KgFba-1Nzq1sMAecUB94,1184
jax/stages.py,sha256=fxDMZsTC4Ehzedq5af92xB0RMrU0gfLlLCKE4MWioOg,1272
jax/test_util.py,sha256=Ld3VuPa_Q4huIBWhk-ZImd3IaB-5QKZkWu6QlDYApaU,835
jax/tools/__init__.py,sha256=BTyMF0yh0M75AsNwF_0NSd5gHnIVpxhsLznsWkxu3XQ,581
jax/tools/__pycache__/__init__.cpython-311.pyc,,
jax/tools/__pycache__/build_utils.cpython-311.pyc,,
jax/tools/__pycache__/colab_tpu.cpython-311.pyc,,
jax/tools/__pycache__/jax_to_ir.cpython-311.pyc,,
jax/tools/build_utils.py,sha256=0fY0sS7QewP3mgodPDl5jVzsoKrsLtZniR1_LTw1vzo,2630
jax/tools/colab_tpu.py,sha256=M8F2u_rcvDdTKHB502zhticEZEqNGE8j5dzgPwoELgo,1553
jax/tools/jax_to_ir.py,sha256=ksDTFbKTomXcau9qapiOpD3iw-Jg5efD0lNykcQNtvU,8684
jax/tree_util.py,sha256=lHsCe98dcD2gVT5hhT8Ti2xEHIjiPFN1MlGx4h7l-pI,4127
jax/typing.py,sha256=tpCqNUt1m5JUbkLUQeF50O0DOonvrCahZuqgiiaXTxs,2996
jax/util.py,sha256=hG_kZcqiOEO2HjTehzSyEFVPctNZneOIORqfC9hwzyA,1091
jax/version.py,sha256=M6tDSnWcBwmIwOKv6VaZxNk5nnGJZavaGL9_ncnmTgc,5008
