1158 lines
88 KiB
Plaintext
1158 lines
88 KiB
Plaintext
jax-0.7.1.dist-info/INSTALLER,sha256=zuuue4knoyJ-UwPPXg8fezS7VCrXJQrAP7zeNuwvFQg,4
|
|
jax-0.7.1.dist-info/METADATA,sha256=ugn38ec-_tC_C1_YsB-UuVmr2Izt6YGg4hPyuUclo3U,13412
|
|
jax-0.7.1.dist-info/RECORD,,
|
|
jax-0.7.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
|
|
jax-0.7.1.dist-info/licenses/AUTHORS,sha256=OOBQygrI1Zc94AYJycH1r_BSCwP38Hyv3sludltJ7hY,313
|
|
jax-0.7.1.dist-info/licenses/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
|
|
jax-0.7.1.dist-info/top_level.txt,sha256=DabAh8MXQ-HM0EtjKD-Wx_D6tTIW-_de73pdG_f_KTE,4
|
|
jax/__init__.py,sha256=MMxq5H9vvfoEEbj8Dzn85cKtTienUtOjjUfwLeIMq5s,9751
|
|
jax/__pycache__/__init__.cpython-312.pyc,,
|
|
jax/__pycache__/ad_checkpoint.cpython-312.pyc,,
|
|
jax/__pycache__/api_util.cpython-312.pyc,,
|
|
jax/__pycache__/cloud_tpu_init.cpython-312.pyc,,
|
|
jax/__pycache__/collect_profile.cpython-312.pyc,,
|
|
jax/__pycache__/core.cpython-312.pyc,,
|
|
jax/__pycache__/custom_batching.cpython-312.pyc,,
|
|
jax/__pycache__/custom_derivatives.cpython-312.pyc,,
|
|
jax/__pycache__/custom_transpose.cpython-312.pyc,,
|
|
jax/__pycache__/debug.cpython-312.pyc,,
|
|
jax/__pycache__/distributed.cpython-312.pyc,,
|
|
jax/__pycache__/dlpack.cpython-312.pyc,,
|
|
jax/__pycache__/dtypes.cpython-312.pyc,,
|
|
jax/__pycache__/errors.cpython-312.pyc,,
|
|
jax/__pycache__/export.cpython-312.pyc,,
|
|
jax/__pycache__/ffi.cpython-312.pyc,,
|
|
jax/__pycache__/flatten_util.cpython-312.pyc,,
|
|
jax/__pycache__/memory.cpython-312.pyc,,
|
|
jax/__pycache__/monitoring.cpython-312.pyc,,
|
|
jax/__pycache__/profiler.cpython-312.pyc,,
|
|
jax/__pycache__/random.cpython-312.pyc,,
|
|
jax/__pycache__/ref.cpython-312.pyc,,
|
|
jax/__pycache__/sharding.cpython-312.pyc,,
|
|
jax/__pycache__/stages.cpython-312.pyc,,
|
|
jax/__pycache__/test_util.cpython-312.pyc,,
|
|
jax/__pycache__/tree.cpython-312.pyc,,
|
|
jax/__pycache__/tree_util.cpython-312.pyc,,
|
|
jax/__pycache__/typing.cpython-312.pyc,,
|
|
jax/__pycache__/util.cpython-312.pyc,,
|
|
jax/__pycache__/version.cpython-312.pyc,,
|
|
jax/_src/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
|
|
jax/_src/__pycache__/__init__.cpython-312.pyc,,
|
|
jax/_src/__pycache__/abstract_arrays.cpython-312.pyc,,
|
|
jax/_src/__pycache__/ad_checkpoint.cpython-312.pyc,,
|
|
jax/_src/__pycache__/ad_util.cpython-312.pyc,,
|
|
jax/_src/__pycache__/api.cpython-312.pyc,,
|
|
jax/_src/__pycache__/api_util.cpython-312.pyc,,
|
|
jax/_src/__pycache__/array.cpython-312.pyc,,
|
|
jax/_src/__pycache__/basearray.cpython-312.pyc,,
|
|
jax/_src/__pycache__/blocked_sampler.cpython-312.pyc,,
|
|
jax/_src/__pycache__/buffer_callback.cpython-312.pyc,,
|
|
jax/_src/__pycache__/cache_key.cpython-312.pyc,,
|
|
jax/_src/__pycache__/callback.cpython-312.pyc,,
|
|
jax/_src/__pycache__/checkify.cpython-312.pyc,,
|
|
jax/_src/__pycache__/cloud_tpu_init.cpython-312.pyc,,
|
|
jax/_src/__pycache__/compilation_cache.cpython-312.pyc,,
|
|
jax/_src/__pycache__/compilation_cache_interface.cpython-312.pyc,,
|
|
jax/_src/__pycache__/compiler.cpython-312.pyc,,
|
|
jax/_src/__pycache__/compute_on.cpython-312.pyc,,
|
|
jax/_src/__pycache__/config.cpython-312.pyc,,
|
|
jax/_src/__pycache__/core.cpython-312.pyc,,
|
|
jax/_src/__pycache__/custom_api_util.cpython-312.pyc,,
|
|
jax/_src/__pycache__/custom_batching.cpython-312.pyc,,
|
|
jax/_src/__pycache__/custom_dce.cpython-312.pyc,,
|
|
jax/_src/__pycache__/custom_derivatives.cpython-312.pyc,,
|
|
jax/_src/__pycache__/custom_partitioning.cpython-312.pyc,,
|
|
jax/_src/__pycache__/custom_partitioning_sharding_rule.cpython-312.pyc,,
|
|
jax/_src/__pycache__/custom_transpose.cpython-312.pyc,,
|
|
jax/_src/__pycache__/debugging.cpython-312.pyc,,
|
|
jax/_src/__pycache__/deprecations.cpython-312.pyc,,
|
|
jax/_src/__pycache__/dispatch.cpython-312.pyc,,
|
|
jax/_src/__pycache__/distributed.cpython-312.pyc,,
|
|
jax/_src/__pycache__/dlpack.cpython-312.pyc,,
|
|
jax/_src/__pycache__/dtypes.cpython-312.pyc,,
|
|
jax/_src/__pycache__/earray.cpython-312.pyc,,
|
|
jax/_src/__pycache__/effects.cpython-312.pyc,,
|
|
jax/_src/__pycache__/environment_info.cpython-312.pyc,,
|
|
jax/_src/__pycache__/error_check.cpython-312.pyc,,
|
|
jax/_src/__pycache__/errors.cpython-312.pyc,,
|
|
jax/_src/__pycache__/ffi.cpython-312.pyc,,
|
|
jax/_src/__pycache__/flatten_util.cpython-312.pyc,,
|
|
jax/_src/__pycache__/frozen_dict.cpython-312.pyc,,
|
|
jax/_src/__pycache__/hardware_utils.cpython-312.pyc,,
|
|
jax/_src/__pycache__/hashable_array.cpython-312.pyc,,
|
|
jax/_src/__pycache__/hijax.cpython-312.pyc,,
|
|
jax/_src/__pycache__/jaxpr_util.cpython-312.pyc,,
|
|
jax/_src/__pycache__/lax_reference.cpython-312.pyc,,
|
|
jax/_src/__pycache__/layout.cpython-312.pyc,,
|
|
jax/_src/__pycache__/lazy_loader.cpython-312.pyc,,
|
|
jax/_src/__pycache__/linear_util.cpython-312.pyc,,
|
|
jax/_src/__pycache__/logging_config.cpython-312.pyc,,
|
|
jax/_src/__pycache__/lru_cache.cpython-312.pyc,,
|
|
jax/_src/__pycache__/memory.cpython-312.pyc,,
|
|
jax/_src/__pycache__/mesh.cpython-312.pyc,,
|
|
jax/_src/__pycache__/mesh_utils.cpython-312.pyc,,
|
|
jax/_src/__pycache__/monitoring.cpython-312.pyc,,
|
|
jax/_src/__pycache__/named_sharding.cpython-312.pyc,,
|
|
jax/_src/__pycache__/op_shardings.cpython-312.pyc,,
|
|
jax/_src/__pycache__/partition_spec.cpython-312.pyc,,
|
|
jax/_src/__pycache__/path.cpython-312.pyc,,
|
|
jax/_src/__pycache__/pickle_util.cpython-312.pyc,,
|
|
jax/_src/__pycache__/pjit.cpython-312.pyc,,
|
|
jax/_src/__pycache__/pretty_printer.cpython-312.pyc,,
|
|
jax/_src/__pycache__/prng.cpython-312.pyc,,
|
|
jax/_src/__pycache__/profiler.cpython-312.pyc,,
|
|
jax/_src/__pycache__/public_test_util.cpython-312.pyc,,
|
|
jax/_src/__pycache__/random.cpython-312.pyc,,
|
|
jax/_src/__pycache__/ref.cpython-312.pyc,,
|
|
jax/_src/__pycache__/shard_alike.cpython-312.pyc,,
|
|
jax/_src/__pycache__/shard_map.cpython-312.pyc,,
|
|
jax/_src/__pycache__/sharding.cpython-312.pyc,,
|
|
jax/_src/__pycache__/sharding_impls.cpython-312.pyc,,
|
|
jax/_src/__pycache__/sharding_specs.cpython-312.pyc,,
|
|
jax/_src/__pycache__/source_info_util.cpython-312.pyc,,
|
|
jax/_src/__pycache__/sourcemap.cpython-312.pyc,,
|
|
jax/_src/__pycache__/stages.cpython-312.pyc,,
|
|
jax/_src/__pycache__/test_loader.cpython-312.pyc,,
|
|
jax/_src/__pycache__/test_multiprocess.cpython-312.pyc,,
|
|
jax/_src/__pycache__/test_util.cpython-312.pyc,,
|
|
jax/_src/__pycache__/test_warning_util.cpython-312.pyc,,
|
|
jax/_src/__pycache__/tpu_custom_call.cpython-312.pyc,,
|
|
jax/_src/__pycache__/traceback_util.cpython-312.pyc,,
|
|
jax/_src/__pycache__/tree.cpython-312.pyc,,
|
|
jax/_src/__pycache__/tree_util.cpython-312.pyc,,
|
|
jax/_src/__pycache__/typing.cpython-312.pyc,,
|
|
jax/_src/__pycache__/util.cpython-312.pyc,,
|
|
jax/_src/__pycache__/xla_bridge.cpython-312.pyc,,
|
|
jax/_src/__pycache__/xla_metadata.cpython-312.pyc,,
|
|
jax/_src/__pycache__/xla_metadata_lib.cpython-312.pyc,,
|
|
jax/_src/abstract_arrays.py,sha256=s-9suhJYHqQrjp7Zk8WIuB4OHQ0Um3EkUdF9pzTtcuk,3778
|
|
jax/_src/ad_checkpoint.py,sha256=kPHR0mjIcKWnrCAjirdLe4XH7Evb3g9Qgi8m5OBAl9A,38397
|
|
jax/_src/ad_util.py,sha256=i1Oj_yGWZXcPb33GVukSYkLskVN3L4Kt0AUZ2F-i8wc,4446
|
|
jax/_src/api.py,sha256=IROjnF9XCBZXZ9jfn7H7Ccx9OTpc8JJiji4yz4UHjdI,133907
|
|
jax/_src/api_util.py,sha256=AnOPgETxh47DgwF1_8IDt6ODA2bucs5yPnP5lyCTdpY,29979
|
|
jax/_src/array.py,sha256=ku-DIz8JxzgUjwXZQLIGxwT5gamjb-1GTOY-74ayAHo,50639
|
|
jax/_src/basearray.py,sha256=FQOcRwUhh1w4byHijEzCZAVqa8Tukr3bSr4fEqZz1q4,6904
|
|
jax/_src/basearray.pyi,sha256=IG5GkrrMFSIphXGJCngtgP_gCHf6boXBUIoZy5rDTxo,13980
|
|
jax/_src/blocked_sampler.py,sha256=mxHv2Eec-AXzPuFe4YK7EN0AVTzmNrWFSD4Ph01ocUs,5890
|
|
jax/_src/buffer_callback.py,sha256=1qpEUw7Lku1JrNIU-vKW36oYO47aPkAxmJPyt7YmJog,10533
|
|
jax/_src/cache_key.py,sha256=Yon7NspNxWKIm0yBJso5j6I4Z6mnKFLmAIg_g0rc228,12923
|
|
jax/_src/callback.py,sha256=oxjTnGgO_bCWpukB3ZTGq-Mr8rr6Go4Z0wEZcePXKkw,33571
|
|
jax/_src/checkify.py,sha256=r41dRdsjz4cFjeNHMseL71alrgNHr9A0Oiyy9WEuXDI,57899
|
|
jax/_src/cloud_tpu_init.py,sha256=zGPXTkqQkvNFyNoke9WGJp1MRe7OpHe44Q7NffoS1Lg,4911
|
|
jax/_src/clusters/__init__.py,sha256=blIgNT92mYE9Bst7obP5_xhZ2fE2hKqwxH6R0cAQZ94,1283
|
|
jax/_src/clusters/__pycache__/__init__.cpython-312.pyc,,
|
|
jax/_src/clusters/__pycache__/cloud_tpu_cluster.cpython-312.pyc,,
|
|
jax/_src/clusters/__pycache__/cluster.cpython-312.pyc,,
|
|
jax/_src/clusters/__pycache__/k8s_cluster.cpython-312.pyc,,
|
|
jax/_src/clusters/__pycache__/mpi4py_cluster.cpython-312.pyc,,
|
|
jax/_src/clusters/__pycache__/ompi_cluster.cpython-312.pyc,,
|
|
jax/_src/clusters/__pycache__/slurm_cluster.cpython-312.pyc,,
|
|
jax/_src/clusters/cloud_tpu_cluster.py,sha256=mlbAUcQjIdZ54du59mMq7kLDTcAOfW_CO56za0zXJNc,8488
|
|
jax/_src/clusters/cluster.py,sha256=raW-A1TKgHDyqZeMfv3lgdm4ImDHxrzoMnFAOfgiK4M,5858
|
|
jax/_src/clusters/k8s_cluster.py,sha256=uo4QCCwpqixEB3Mo_28fPmPpCkze-fQb1PK8I_WIGy4,9279
|
|
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/compilation_cache.py,sha256=k3qxlaakStn1ToP84Z9i8J0cKQ0HQ6uvsBhF-3D9AsI,12154
|
|
jax/_src/compilation_cache_interface.py,sha256=ONHvsIC3Kzoda0TcQWnhhjxtOWYixjkO9Or__BAW3_0,865
|
|
jax/_src/compiler.py,sha256=bkoqSUInR1LCuq9dQjNYeAdJSwljbQkrSud6WOspHFU,32904
|
|
jax/_src/compute_on.py,sha256=0b74TSV1R31kJTR53thafdtB0sYiFGqmRoAbZwAFcps,1911
|
|
jax/_src/config.py,sha256=6kQRd1AmQiDnZhqtWqFpxCH_XtvQy0ZtLAfbAd31zTI,73059
|
|
jax/_src/core.py,sha256=dy4Efz4P5OUI7kv9yFyF-B4SUKloGzz0x2R_dZipIGY,144178
|
|
jax/_src/cudnn/__init__.py,sha256=03MR0G6UMqHXi-JBeyj6tipV6ioP9a-iSN1mW-8Ws5A,631
|
|
jax/_src/cudnn/__pycache__/__init__.cpython-312.pyc,,
|
|
jax/_src/cudnn/__pycache__/fused_attention_stablehlo.cpython-312.pyc,,
|
|
jax/_src/cudnn/__pycache__/fusion.cpython-312.pyc,,
|
|
jax/_src/cudnn/__pycache__/scaled_matmul_stablehlo.cpython-312.pyc,,
|
|
jax/_src/cudnn/fused_attention_stablehlo.py,sha256=x6-GSybcouy9OspcD5AWGTQiuneqhNnTnUbS9aPPf9I,82388
|
|
jax/_src/cudnn/fusion.py,sha256=Aob-4WLRssNMbS3vdDJuhDjKPDshEEfu5VAYjL6b97w,2877
|
|
jax/_src/cudnn/scaled_matmul_stablehlo.py,sha256=jiyz3XFyN_7biMRh7903i6omK4zrB0PLaoaD3_ekjVI,25343
|
|
jax/_src/custom_api_util.py,sha256=uv6uyrXU1BHGzKuAe498W2PD9bcWt1PEa-8UAqr9Zaw,876
|
|
jax/_src/custom_batching.py,sha256=gU1T11Uq_WStXvn1dcGUXneZhd6FfJp6XHxo0syENY8,15238
|
|
jax/_src/custom_dce.py,sha256=UHUTZ9UZM4EyeTBrHDw3xp7RgPqeeoSb5d1lRLKIoUY,17470
|
|
jax/_src/custom_derivatives.py,sha256=MO3Hx85sB4t55p892DMt8J62-KneZfdK4RCTvu3NiXg,80666
|
|
jax/_src/custom_partitioning.py,sha256=hEZutF-EN8tA6jwXATMosJg12duXiBQThteGzTWAM-E,26178
|
|
jax/_src/custom_partitioning_sharding_rule.py,sha256=R-vcj-qF_FDPktXAme_cxFXorga5UAdQh_3XpuT7QaM,18193
|
|
jax/_src/custom_transpose.py,sha256=6-lfrre7KJKZ7BrXUl6QKR-_b3QZ4Gsuf_OwwGjsLuE,9204
|
|
jax/_src/debugger/__init__.py,sha256=iZU1X5nkpSBK5pwLSuWhZsPZgtLUovBF3tyWPmQoh1Q,899
|
|
jax/_src/debugger/__pycache__/__init__.cpython-312.pyc,,
|
|
jax/_src/debugger/__pycache__/cli_debugger.cpython-312.pyc,,
|
|
jax/_src/debugger/__pycache__/colab_debugger.cpython-312.pyc,,
|
|
jax/_src/debugger/__pycache__/colab_lib.cpython-312.pyc,,
|
|
jax/_src/debugger/__pycache__/core.cpython-312.pyc,,
|
|
jax/_src/debugger/__pycache__/web_debugger.cpython-312.pyc,,
|
|
jax/_src/debugger/cli_debugger.py,sha256=bzn-5Rd4C9WwGsVret8gbFNaLpXAf0SzvgltXNuZrcU,4778
|
|
jax/_src/debugger/colab_debugger.py,sha256=z49K89L1bTbGN-kRyCr1ElOhTuSxgpoEBpEI0L8YSyQ,7854
|
|
jax/_src/debugger/colab_lib.py,sha256=jCbwj2Q4f8SEo7CBr6AeiUtzlsIv4BfKa5CVbq4Bn2U,4298
|
|
jax/_src/debugger/core.py,sha256=sRN6dv-Vhuz6j5eenDq8s1ZTUbxYiOv-JxDWx9519ew,8682
|
|
jax/_src/debugger/web_debugger.py,sha256=Or-gJL40oBUfClu8mg1YUBoKA9cQytZgJAeGGrPPK_w,3433
|
|
jax/_src/debugging.py,sha256=CWLUmDtnjn1o9xywJJX-0rQ8MEzhSAZtAODL9VOcOfQ,29150
|
|
jax/_src/deprecations.py,sha256=uGEmw6EZHOUfKVXqNpxTPJWrmfnmlbxTUlwGlY2IxXE,5009
|
|
jax/_src/dispatch.py,sha256=0IBRPdjO-Yt2ygLCvx7ZEol10Pa1y2wacNxH-op60UU,26299
|
|
jax/_src/distributed.py,sha256=ib_SIFkIYLiQzEBxmvVeA9tpJRuaCm1V97_MvQBU2Lc,15163
|
|
jax/_src/dlpack.py,sha256=bxRkfVUf6ppc4TULw6SorGDvPVnaeRhrkxd6pXKjEp8,11978
|
|
jax/_src/dtypes.py,sha256=8TqBoM4mMd-cGl3F6H5mNWmMtbM_yvbZMh8IxfdNbMo,37296
|
|
jax/_src/earray.py,sha256=WS8XoNTCP6zQTC5DahYR0S6g_V9YHj3wKW3L9B8RIhU,4405
|
|
jax/_src/effects.py,sha256=qQCTznjR8_76HVCp_MAAKKXbs1HvIspt1Z3w3UYS5CA,5041
|
|
jax/_src/environment_info.py,sha256=Xc9-Y5_DvVjW3iEhen51ATJvIESluEIrWJIF_e-oqA8,2031
|
|
jax/_src/error_check.py,sha256=h7k-P11vRPV5HoqKeKx5mYoe9jqXCF6a3_2gpyJhRwk,12975
|
|
jax/_src/errors.py,sha256=Vp2gp5kQxtoss9aby-vpGwO5wBxRndvddVD_3-UWyZ4,24710
|
|
jax/_src/export/__init__.py,sha256=mS3YSYUZFIG68T5Zo9CFGkT7wCyWRycnWV4L36CnHYU,581
|
|
jax/_src/export/__pycache__/__init__.cpython-312.pyc,,
|
|
jax/_src/export/__pycache__/_export.cpython-312.pyc,,
|
|
jax/_src/export/__pycache__/serialization.cpython-312.pyc,,
|
|
jax/_src/export/__pycache__/serialization_generated.cpython-312.pyc,,
|
|
jax/_src/export/__pycache__/shape_poly.cpython-312.pyc,,
|
|
jax/_src/export/__pycache__/shape_poly_decision.cpython-312.pyc,,
|
|
jax/_src/export/_export.py,sha256=nCFV89UFGXZKt2iEF8OOuXVlTme20Xl3kCQ733RmMFI,71677
|
|
jax/_src/export/serialization.py,sha256=0fYNO50OjU8TWRC1_wHJm_z1pFbLnaGQai3Tif5KluU,19466
|
|
jax/_src/export/serialization_generated.py,sha256=Bl505_cl_U9oGOUNN0rF6sL7hlZduTzY_nc9-ekd0J0,27781
|
|
jax/_src/export/shape_poly.py,sha256=X5d2DINA9cvefSrWmpOkc9EeSQaWfP8GYZz3Zhvfpdg,85962
|
|
jax/_src/export/shape_poly_decision.py,sha256=x24hHa1CGTskS83TMeo7xGILKsrgufYsGReiKLQAJtc,20476
|
|
jax/_src/extend/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
|
|
jax/_src/extend/__pycache__/__init__.cpython-312.pyc,,
|
|
jax/_src/extend/__pycache__/random.cpython-312.pyc,,
|
|
jax/_src/extend/random.py,sha256=Rc68o6EA7zPLY3nK-cyF8Tq0WPdvnk81rg9sLb22QRg,1269
|
|
jax/_src/ffi.py,sha256=vdFKS9H74o0TBC3qZc7swzRsgUvE1qyYdW-clYYjWMk,29274
|
|
jax/_src/flatten_util.py,sha256=YS73oEhjhdaL7CArKOkhYYx-GtwPpYM5JDd1ExDAkN4,3483
|
|
jax/_src/frozen_dict.py,sha256=V9UhNofzdhy7pTM16OOf7i1_fIfmRqLAUH-GSWcU8R4,1457
|
|
jax/_src/hardware_utils.py,sha256=YQcgkyPgQN94grS6VvBIVVFL2R-LDCsdY8Tqj43jnVA,2349
|
|
jax/_src/hashable_array.py,sha256=4ff9eVDYR7DFK3l9nNF0btGecguXonn1AzeWviiJQcQ,1036
|
|
jax/_src/hijax.py,sha256=010yVh5IvSN_fMjRiNUpr_k7-I_yqwDPPp2TBMFsHH8,9181
|
|
jax/_src/image/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
|
|
jax/_src/image/__pycache__/__init__.cpython-312.pyc,,
|
|
jax/_src/image/__pycache__/scale.cpython-312.pyc,,
|
|
jax/_src/image/scale.py,sha256=C1q4vSC4jODLT44jl6xJcbvAeTY4wDMGwkNeTE_2rAE,13880
|
|
jax/_src/internal_test_util/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
|
|
jax/_src/internal_test_util/__pycache__/__init__.cpython-312.pyc,,
|
|
jax/_src/internal_test_util/__pycache__/deprecation_module.cpython-312.pyc,,
|
|
jax/_src/internal_test_util/__pycache__/export_back_compat_test_util.cpython-312.pyc,,
|
|
jax/_src/internal_test_util/__pycache__/lax_test_util.cpython-312.pyc,,
|
|
jax/_src/internal_test_util/__pycache__/test_harnesses.cpython-312.pyc,,
|
|
jax/_src/internal_test_util/deprecation_module.py,sha256=iZ6fRYNaUcOwrftwpDXMnlxrLrrQbdMrbybmmQFwql4,851
|
|
jax/_src/internal_test_util/export_back_compat_test_data/__init__.py,sha256=gEMCMS8LKTvoFm0t7khiAnQg78EDUuMI-fvTvw250eo,682
|
|
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/__init__.cpython-312.pyc,,
|
|
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/annotate_data_placement.cpython-312.pyc,,
|
|
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/cpu_cholesky_lapack_potrf.cpython-312.pyc,,
|
|
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/cpu_eig_lapack_geev.cpython-312.pyc,,
|
|
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/cpu_eigh_lapack_syev.cpython-312.pyc,,
|
|
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/cpu_hessenberg_lapack_gehrd.cpython-312.pyc,,
|
|
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/cpu_lu_lapack_getrf.cpython-312.pyc,,
|
|
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/cpu_qr_lapack_geqrf.cpython-312.pyc,,
|
|
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/cpu_schur_lapack_gees.cpython-312.pyc,,
|
|
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/cpu_svd_lapack_gesdd.cpython-312.pyc,,
|
|
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/cpu_triangular_solve_blas_trsm.cpython-312.pyc,,
|
|
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/cpu_tridiagonal_lapack_sytrd_hetrd.cpython-312.pyc,,
|
|
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/cpu_tridiagonal_solve_lapack_gtsv.cpython-312.pyc,,
|
|
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/cuda_eigh_cusolver_syev.cpython-312.pyc,,
|
|
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/cuda_lu_cusolver_getrf.cpython-312.pyc,,
|
|
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/cuda_lu_pivots_to_permutation.cpython-312.pyc,,
|
|
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/cuda_qr_cusolver_geqrf.cpython-312.pyc,,
|
|
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/cuda_svd_cusolver_gesvd.cpython-312.pyc,,
|
|
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/cuda_threefry2x32.cpython-312.pyc,,
|
|
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/cuda_tridiagonal_cusolver_sytrd.cpython-312.pyc,,
|
|
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/cuda_tridiagonal_solve.cpython-312.pyc,,
|
|
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/rocm_eigh_hipsolver_syev.cpython-312.pyc,,
|
|
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/shardy_sharding_ops_with_different_meshes.cpython-312.pyc,,
|
|
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/stablehlo_dynamic_approx_top_k.cpython-312.pyc,,
|
|
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/stablehlo_dynamic_rng_bit_generator.cpython-312.pyc,,
|
|
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/stablehlo_dynamic_top_k.cpython-312.pyc,,
|
|
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/tpu_ApproxTopK.cpython-312.pyc,,
|
|
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/tpu_Eigh.cpython-312.pyc,,
|
|
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/tpu_Lu.cpython-312.pyc,,
|
|
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/tpu_Qr.cpython-312.pyc,,
|
|
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/tpu_Sharding.cpython-312.pyc,,
|
|
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/tpu_stablehlo_dynamic_reduce_window.cpython-312.pyc,,
|
|
jax/_src/internal_test_util/export_back_compat_test_data/annotate_data_placement.py,sha256=EpoTASRU-bjg8-0oLRNZ_Ir80p2frmvTFY1tKodKMsY,27102
|
|
jax/_src/internal_test_util/export_back_compat_test_data/cpu_cholesky_lapack_potrf.py,sha256=liVPn-gYqIyzFzi4y3yAeURsUH4l71cC4Ci7rs2EKwU,45965
|
|
jax/_src/internal_test_util/export_back_compat_test_data/cpu_eig_lapack_geev.py,sha256=6SvL_1_UGaUkKcqXX5O5EU-D_KJxckHz5q_5sFmLDAk,35800
|
|
jax/_src/internal_test_util/export_back_compat_test_data/cpu_eigh_lapack_syev.py,sha256=4OHoOpk2znnttnKIwfdVVnPXloC0jdJcDDZRzzkjFKE,58741
|
|
jax/_src/internal_test_util/export_back_compat_test_data/cpu_hessenberg_lapack_gehrd.py,sha256=cMbJDAZ_vdGK5g4hgdfBBmyxf3Dzk4jyOnHZGC0lZ1E,37276
|
|
jax/_src/internal_test_util/export_back_compat_test_data/cpu_lu_lapack_getrf.py,sha256=W7sTmxVhYK3bPUY2OQVyz03O-5tNKztupdaCPz0GGcs,80518
|
|
jax/_src/internal_test_util/export_back_compat_test_data/cpu_qr_lapack_geqrf.py,sha256=gCjyU4yeampoNYPTwvXLBeTaK99gkCNaj4tIfOC5X8w,28588
|
|
jax/_src/internal_test_util/export_back_compat_test_data/cpu_schur_lapack_gees.py,sha256=yQYdzlYdLC4saeV4pPspeyLI91YKzbhFWG83s2txLEA,25050
|
|
jax/_src/internal_test_util/export_back_compat_test_data/cpu_svd_lapack_gesdd.py,sha256=oFLAqsyRQQ5dYvn3qYSWAYkr7sVdxaGUtV21x5SSUMI,44276
|
|
jax/_src/internal_test_util/export_back_compat_test_data/cpu_triangular_solve_blas_trsm.py,sha256=LvDY70AYYAWa4ZlFkv1M3vLBVcrb5vHzh-Hw9JaSlYg,19086
|
|
jax/_src/internal_test_util/export_back_compat_test_data/cpu_tridiagonal_lapack_sytrd_hetrd.py,sha256=m-93hKd0NonVaA-0fomLu5kiWsbC_Ijfu9K1PKh5dxU,58875
|
|
jax/_src/internal_test_util/export_back_compat_test_data/cpu_tridiagonal_solve_lapack_gtsv.py,sha256=oQI9RA2tVCe6iKkF-HwQvgK1i2Z7vJdtdRneDfxcYlw,31786
|
|
jax/_src/internal_test_util/export_back_compat_test_data/cuda_eigh_cusolver_syev.py,sha256=i37_O3sNPXdJhfziBGIXSefjxGY8O49BBgUsriD8l4c,43043
|
|
jax/_src/internal_test_util/export_back_compat_test_data/cuda_lu_cusolver_getrf.py,sha256=rTViiEOXgqRovwdxMUNTIMvrC4VnEJuBsFqnBf34boc,28728
|
|
jax/_src/internal_test_util/export_back_compat_test_data/cuda_lu_pivots_to_permutation.py,sha256=80gav7QLSABoktOIImerdCeOfMx6H4PQnnkMrEqFovg,4159
|
|
jax/_src/internal_test_util/export_back_compat_test_data/cuda_qr_cusolver_geqrf.py,sha256=MxhvzJkTFtRxSiSexJ37XOqdypQsCa4s2WVIasNrGR8,31584
|
|
jax/_src/internal_test_util/export_back_compat_test_data/cuda_svd_cusolver_gesvd.py,sha256=uowHFqyj-fXAfNcJXz9TvUyXHXr8IZDLHJv-SffSfSE,80048
|
|
jax/_src/internal_test_util/export_back_compat_test_data/cuda_threefry2x32.py,sha256=jlXolWJmf_pjGdr8M_N6q6D3aL2NL12x4huNqCYyiyM,14920
|
|
jax/_src/internal_test_util/export_back_compat_test_data/cuda_tridiagonal_cusolver_sytrd.py,sha256=0nv_mDOGjjbwhrQUxeCMTsjLn5pAzPuyYOwSRQPZrz8,39049
|
|
jax/_src/internal_test_util/export_back_compat_test_data/cuda_tridiagonal_solve.py,sha256=jO7yNzENuubOaabB0441suDSOeU7pLKQmynfmpJ1ueQ,7756
|
|
jax/_src/internal_test_util/export_back_compat_test_data/pallas/__init__.py,sha256=gEMCMS8LKTvoFm0t7khiAnQg78EDUuMI-fvTvw250eo,682
|
|
jax/_src/internal_test_util/export_back_compat_test_data/pallas/__pycache__/__init__.cpython-312.pyc,,
|
|
jax/_src/internal_test_util/export_back_compat_test_data/pallas/__pycache__/mosaic_gpu_add_one.cpython-312.pyc,,
|
|
jax/_src/internal_test_util/export_back_compat_test_data/pallas/__pycache__/mosaic_matmul.cpython-312.pyc,,
|
|
jax/_src/internal_test_util/export_back_compat_test_data/pallas/__pycache__/mosaic_semaphore_dma.cpython-312.pyc,,
|
|
jax/_src/internal_test_util/export_back_compat_test_data/pallas/__pycache__/triton_add_one.cpython-312.pyc,,
|
|
jax/_src/internal_test_util/export_back_compat_test_data/pallas/mosaic_gpu_add_one.py,sha256=gFgTn32nAZGHaatj606pjLgIv5i6p9vn9yqQkSunf70,102068
|
|
jax/_src/internal_test_util/export_back_compat_test_data/pallas/mosaic_matmul.py,sha256=3zMrunLj4r2sskF-UJCezccCMR-j8vrfGeg-aFzlsXQ,31203
|
|
jax/_src/internal_test_util/export_back_compat_test_data/pallas/mosaic_semaphore_dma.py,sha256=7U13cnQrTs33Bc0TA-I5yRKlHXYN8CEkEedBBe4w0ms,16908
|
|
jax/_src/internal_test_util/export_back_compat_test_data/pallas/triton_add_one.py,sha256=sgk-uKJO5c2-2KO_mO5mrZHWzBhCruM5MIRbbnhva-w,10043
|
|
jax/_src/internal_test_util/export_back_compat_test_data/rocm_eigh_hipsolver_syev.py,sha256=Ei1LQm0kkFU_zbsN_lsHr65zv7k62PsEWXYSQKq7Xf4,115014
|
|
jax/_src/internal_test_util/export_back_compat_test_data/shardy_sharding_ops_with_different_meshes.py,sha256=mKPsDNmGRMCdbkeMQRo59H6aDbBPtmugY1KJiFl7aU8,22351
|
|
jax/_src/internal_test_util/export_back_compat_test_data/stablehlo_dynamic_approx_top_k.py,sha256=2IgM2d_n2RRgg7dfUmhryKScHmUPArE05atL9_HKw0E,13062
|
|
jax/_src/internal_test_util/export_back_compat_test_data/stablehlo_dynamic_rng_bit_generator.py,sha256=gtPimKyp7-w-_Zt2xxKbPMy7j4ZFfoRERF22PSphU7k,11422
|
|
jax/_src/internal_test_util/export_back_compat_test_data/stablehlo_dynamic_top_k.py,sha256=Zac1QqMUUsnUHgWTqD3a2Djoh4-oRA5H6o0AbQU69A0,14167
|
|
jax/_src/internal_test_util/export_back_compat_test_data/tpu_ApproxTopK.py,sha256=MzL1cZYrKGODHjmPRYlc5iKfa52Z0pLsxSOl5jtTVQ4,12559
|
|
jax/_src/internal_test_util/export_back_compat_test_data/tpu_Eigh.py,sha256=1D7yFSMn4-_OiO7RE0bHjSBQ8_0uGBmc49NLqy4rlbk,10146
|
|
jax/_src/internal_test_util/export_back_compat_test_data/tpu_Lu.py,sha256=hPRVhBpIWrqs4lWdG-5Jhax021wrA3QlcOv0_Mwh9HA,6185
|
|
jax/_src/internal_test_util/export_back_compat_test_data/tpu_Qr.py,sha256=L4zVrUCoSpOEqYo21litnykOdd07tbBA4EpfrSV_mdE,10228
|
|
jax/_src/internal_test_util/export_back_compat_test_data/tpu_Sharding.py,sha256=8o3QlFx2o1ynfxI_v0BvNOCzPk6nXXB5apQlR4bc9ok,20023
|
|
jax/_src/internal_test_util/export_back_compat_test_data/tpu_stablehlo_dynamic_reduce_window.py,sha256=r3tKmpt6kpO0N5xJ0iDS61JiAFn5NV6KXXCTz8HTVIw,22996
|
|
jax/_src/internal_test_util/export_back_compat_test_util.py,sha256=OIMasP4ZX9USvDrJRrJyVnJs0RCoQuIuHoxlhPVkhFk,15892
|
|
jax/_src/internal_test_util/lax_test_util.py,sha256=tyMSRQ_AGkmMTKSmYFONpToMKOcRtT6E8uF3ilbvbW0,13016
|
|
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-312.pyc,,
|
|
jax/_src/internal_test_util/lazy_loader_module/__pycache__/lazy_test_submodule.cpython-312.pyc,,
|
|
jax/_src/internal_test_util/lazy_loader_module/lazy_test_submodule.py,sha256=luZv4DGyWwdcBX0_NBVwu81IE4jFYVbRkKo9gQPmxPk,610
|
|
jax/_src/internal_test_util/test_harnesses.py,sha256=DzpTf-4kmuKhyTAJlcFHmZYGopuqlWH8RxKQPq10Awo,126435
|
|
jax/_src/interpreters/__init__.py,sha256=hnuhN7h1jY5__Kt2rBvTrk0o8nX4fQl244fDCIMh0qk,690
|
|
jax/_src/interpreters/__pycache__/__init__.cpython-312.pyc,,
|
|
jax/_src/interpreters/__pycache__/ad.cpython-312.pyc,,
|
|
jax/_src/interpreters/__pycache__/batching.cpython-312.pyc,,
|
|
jax/_src/interpreters/__pycache__/mlir.cpython-312.pyc,,
|
|
jax/_src/interpreters/__pycache__/partial_eval.cpython-312.pyc,,
|
|
jax/_src/interpreters/__pycache__/pxla.cpython-312.pyc,,
|
|
jax/_src/interpreters/ad.py,sha256=31q6FPWXRW8z3aOfKQEk8NO02TjImJ0RGLdQ0TQ4ozY,65682
|
|
jax/_src/interpreters/batching.py,sha256=ABj9xoJq74TE9EC3fa0ioPh2lTod_jrUbdygn273eaA,52544
|
|
jax/_src/interpreters/mlir.py,sha256=9gyxzMDqbisoZMJN60A7QBojlxUFvqUPUwMN6IvtnWc,140682
|
|
jax/_src/interpreters/partial_eval.py,sha256=NVqXcp6AEjb6rtssyEYsyld2yNY1C5U-dhFGdIVVjj0,122604
|
|
jax/_src/interpreters/pxla.py,sha256=B3-75aynPSaj6v0j2uASi-alPXqdZOIIapTb5tRjZq8,141073
|
|
jax/_src/jaxpr_util.py,sha256=TOdYMOrOMa5Ei4WnawyfuH7nDdZ25Dt6TkpBSzUIXOI,10475
|
|
jax/_src/lax/__init__.py,sha256=AO7JATuTQEKMM120JB3L6k27E69T40TNy-PSafuMk6U,8388
|
|
jax/_src/lax/__pycache__/__init__.cpython-312.pyc,,
|
|
jax/_src/lax/__pycache__/ann.cpython-312.pyc,,
|
|
jax/_src/lax/__pycache__/convolution.cpython-312.pyc,,
|
|
jax/_src/lax/__pycache__/fft.cpython-312.pyc,,
|
|
jax/_src/lax/__pycache__/lax.cpython-312.pyc,,
|
|
jax/_src/lax/__pycache__/linalg.cpython-312.pyc,,
|
|
jax/_src/lax/__pycache__/other.cpython-312.pyc,,
|
|
jax/_src/lax/__pycache__/parallel.cpython-312.pyc,,
|
|
jax/_src/lax/__pycache__/slicing.cpython-312.pyc,,
|
|
jax/_src/lax/__pycache__/special.cpython-312.pyc,,
|
|
jax/_src/lax/__pycache__/utils.cpython-312.pyc,,
|
|
jax/_src/lax/__pycache__/windowed_reductions.cpython-312.pyc,,
|
|
jax/_src/lax/ann.py,sha256=kZBr1f99HjbpzX58JCxwWhGTHFOBZSxyQx6FnjH4OPI,17043
|
|
jax/_src/lax/control_flow/__init__.py,sha256=WRnqzy4ec591tqipY4GcyXFzniF0njoIVkP0HhKGbiI,2244
|
|
jax/_src/lax/control_flow/__pycache__/__init__.cpython-312.pyc,,
|
|
jax/_src/lax/control_flow/__pycache__/common.cpython-312.pyc,,
|
|
jax/_src/lax/control_flow/__pycache__/conditionals.cpython-312.pyc,,
|
|
jax/_src/lax/control_flow/__pycache__/for_loop.cpython-312.pyc,,
|
|
jax/_src/lax/control_flow/__pycache__/loops.cpython-312.pyc,,
|
|
jax/_src/lax/control_flow/__pycache__/solves.cpython-312.pyc,,
|
|
jax/_src/lax/control_flow/common.py,sha256=WjX6gD_Rv76SGVz2yHNEvUWO2teYeSImk4wJBiX92BY,7427
|
|
jax/_src/lax/control_flow/conditionals.py,sha256=jsTvQFHHFO5dMZHFuj_kQYAcicADocJA1_U4-BBKHNc,54872
|
|
jax/_src/lax/control_flow/for_loop.py,sha256=81SNzdYYpg7-OvFlSvMlOGSJGxV3VvRDSq356O1k0sA,15984
|
|
jax/_src/lax/control_flow/loops.py,sha256=T5dqhhSGBqv8NdElD8D_MHaroNjrrjoZ5NS1OwsjhhI,135549
|
|
jax/_src/lax/control_flow/solves.py,sha256=H74Ef8FWE1cFrIc3BrH5n7CowiRhmr5Wjw9LVEqIxVc,20491
|
|
jax/_src/lax/convolution.py,sha256=oWHCmaWAMiMnuIGSSVm9-tJlmV6ulG9qBcNAQTOLcNE,45955
|
|
jax/_src/lax/fft.py,sha256=3jloTWfLPoPvHbUfHde9Dqt-E96D5UXDmqgBxPEAsOg,7277
|
|
jax/_src/lax/lax.py,sha256=RzGQ8UDy0kNLYqf9VNwAZ-XgKLSjm0cX0Xj-z_gFj-0,361313
|
|
jax/_src/lax/linalg.py,sha256=caQawwxxzPaSBkHUhul5vsqg3Em1cZ7eCUOtfsSYxd0,101192
|
|
jax/_src/lax/other.py,sha256=B11uj0fddkstQoi6lA3-gK2k7y4Bf01r3BOkZRB_eps,13874
|
|
jax/_src/lax/parallel.py,sha256=sGoIBGu1HxkXSI5NF7du6loKPFEHpuQtM1AcL24nPKc,98872
|
|
jax/_src/lax/slicing.py,sha256=7Ta9XQgGCV0x_5_E3HwkQCp6_yjTmltCtd3rB0ahxkM,150635
|
|
jax/_src/lax/special.py,sha256=9V-lSV5LJhy-thwO8-mfcWTT8FA_7QpkBhj63d6UYJc,28872
|
|
jax/_src/lax/utils.py,sha256=_pnxh04eIN7qRRpDHLIt4JETO8CITYDWZpoCyYEEZWI,9502
|
|
jax/_src/lax/windowed_reductions.py,sha256=J_SdxEU0MbtWOOgKsv6gOkCa7Uwl2_pPJz15OFjoB4g,47330
|
|
jax/_src/lax_reference.py,sha256=Wd7LhL6SRNKWxRsqDnKZWy0liM95w7q4g4SmVzWN9_k,19551
|
|
jax/_src/layout.py,sha256=rNJTOUmy2b-dBxRr4ofx2cHGb0jmxmC7swkRzmenmMw,5797
|
|
jax/_src/lazy_loader.py,sha256=G6czW2Uq6WRC1EjDiX6rslMu02cZUGRkKNrvIhXksxw,1842
|
|
jax/_src/lib/__init__.py,sha256=kYStL8sThFLneY-r7kq3wz1htaEA2UPRPwa8g4v6YyU,7757
|
|
jax/_src/lib/__pycache__/__init__.cpython-312.pyc,,
|
|
jax/_src/lib/__pycache__/mosaic_gpu.cpython-312.pyc,,
|
|
jax/_src/lib/__pycache__/triton.cpython-312.pyc,,
|
|
jax/_src/lib/mlir/__init__.py,sha256=KyKFigc6V_REcOMXB95auEAFQaOuFbfnxlxeyLcqfJk,676
|
|
jax/_src/lib/mlir/__pycache__/__init__.cpython-312.pyc,,
|
|
jax/_src/lib/mlir/dialects/__init__.py,sha256=AmPoWPMMC7cOKxh68U9nEoIbL7FuiDh-5jOsf6_w57o,1965
|
|
jax/_src/lib/mlir/dialects/__pycache__/__init__.cpython-312.pyc,,
|
|
jax/_src/lib/mosaic_gpu.py,sha256=andYvo1TC8Eru9Pd_rM76L66Ozami3Q5awjdTkdIE10,1017
|
|
jax/_src/lib/triton.py,sha256=lMwAguUTQxYs9HkdLqeracg8sz5RL7gm1W6PpzSrzdk,2202
|
|
jax/_src/linear_util.py,sha256=36c5Fk-oVwTFxFqgBjbNc0aNUGIZs3cc1ZzH5hroT5c,18971
|
|
jax/_src/logging_config.py,sha256=zKijn3gjyw21ljvWjTsBX1bE6AyIa3zHRQ15YvaDDmk,4452
|
|
jax/_src/lru_cache.py,sha256=NejEgo8BmO7xy8uORQrAHsN0HpCnY8qsEUnuEhDCkPs,6741
|
|
jax/_src/memory.py,sha256=xku_pvnyIR2XpG3brybs0u1MjQ75xRh97AlXgf2w-ro,745
|
|
jax/_src/mesh.py,sha256=KVkohEcDJwD_V2WFE5zkBq-6qiLC5jhUjD0XaOJXC84,20472
|
|
jax/_src/mesh_utils.py,sha256=x69vZykWOT-SlFWE0TeIcc-VLoDzfwT32GRXhGDKRBE,32963
|
|
jax/_src/monitoring.py,sha256=PSOb1js-21Hb4INsIdiJpm0s483xeljnoRr0Lha4nzM,6241
|
|
jax/_src/named_sharding.py,sha256=LRS0I8lEbtUlODpCxk2nCJFvFjEIVu6kAiAzTqgPJ-0,20994
|
|
jax/_src/nn/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
|
|
jax/_src/nn/__pycache__/__init__.cpython-312.pyc,,
|
|
jax/_src/nn/__pycache__/functions.cpython-312.pyc,,
|
|
jax/_src/nn/__pycache__/initializers.cpython-312.pyc,,
|
|
jax/_src/nn/functions.py,sha256=Z0I4NDekTFylUAqOnNElvwbtWMyybKmmQC0dQ0aZPcc,48674
|
|
jax/_src/nn/initializers.py,sha256=ek71wlA5QDGbalynnwXDvNEMyiJFdw9dY1AMgX2MBiU,27013
|
|
jax/_src/numpy/__init__.py,sha256=meOGLN490Uchi49wwU8Fqsx5zstPBqVUqzG5IUeDiN0,4072
|
|
jax/_src/numpy/__pycache__/__init__.cpython-312.pyc,,
|
|
jax/_src/numpy/__pycache__/array_api_metadata.cpython-312.pyc,,
|
|
jax/_src/numpy/__pycache__/array_constructors.cpython-312.pyc,,
|
|
jax/_src/numpy/__pycache__/array_creation.cpython-312.pyc,,
|
|
jax/_src/numpy/__pycache__/array_methods.cpython-312.pyc,,
|
|
jax/_src/numpy/__pycache__/einsum.cpython-312.pyc,,
|
|
jax/_src/numpy/__pycache__/error.cpython-312.pyc,,
|
|
jax/_src/numpy/__pycache__/fft.cpython-312.pyc,,
|
|
jax/_src/numpy/__pycache__/index_tricks.cpython-312.pyc,,
|
|
jax/_src/numpy/__pycache__/indexing.cpython-312.pyc,,
|
|
jax/_src/numpy/__pycache__/lax_numpy.cpython-312.pyc,,
|
|
jax/_src/numpy/__pycache__/linalg.cpython-312.pyc,,
|
|
jax/_src/numpy/__pycache__/polynomial.cpython-312.pyc,,
|
|
jax/_src/numpy/__pycache__/reductions.cpython-312.pyc,,
|
|
jax/_src/numpy/__pycache__/scalar_types.cpython-312.pyc,,
|
|
jax/_src/numpy/__pycache__/setops.cpython-312.pyc,,
|
|
jax/_src/numpy/__pycache__/sorting.cpython-312.pyc,,
|
|
jax/_src/numpy/__pycache__/tensor_contractions.cpython-312.pyc,,
|
|
jax/_src/numpy/__pycache__/ufunc_api.cpython-312.pyc,,
|
|
jax/_src/numpy/__pycache__/ufuncs.cpython-312.pyc,,
|
|
jax/_src/numpy/__pycache__/util.cpython-312.pyc,,
|
|
jax/_src/numpy/__pycache__/vectorize.cpython-312.pyc,,
|
|
jax/_src/numpy/__pycache__/window_functions.cpython-312.pyc,,
|
|
jax/_src/numpy/array_api_metadata.py,sha256=MVvoctyTPUK3Ac66PmSMWkcHZI3Km7LsAL-2sC6N3tc,3686
|
|
jax/_src/numpy/array_constructors.py,sha256=JXpPAGsiObYpb1wTWZbl_Sd-SPvxCFVR3PK5bgklp9M,14823
|
|
jax/_src/numpy/array_creation.py,sha256=G39BOP6BjSiCUC_gKTxq_1HIGBSgEZRcKaRX6dNBY6E,30527
|
|
jax/_src/numpy/array_methods.py,sha256=YYAEOzp7ml47rIpUAw25ZIuBQ4dIqCJZJfLKFU-6zs0,50176
|
|
jax/_src/numpy/einsum.py,sha256=zsjNJlY9cv47zyY42VpVrex4FFPL_ahIBLrMHnA6YUU,24396
|
|
jax/_src/numpy/error.py,sha256=WwhoOFBsaCETiEzJClflyD-MIm0edzQnPUbW_iDecgI,6392
|
|
jax/_src/numpy/fft.py,sha256=tXbcj-tsfVo6UFj5LIjES39-gs12ylsHwPJXFGR-8sk,52321
|
|
jax/_src/numpy/index_tricks.py,sha256=ko4vrQ8VxIm6QP97ppiGxKF4rsxq3DCh703a0DMa-yo,10101
|
|
jax/_src/numpy/indexing.py,sha256=d_rudeLBEBu129eSGan_Ij-_X3fj8ApCu9jBaTPCvck,52226
|
|
jax/_src/numpy/lax_numpy.py,sha256=MGQKHUZqIqQeMzmWHKg8jWDkl4m043B8WWqZZDGbbGg,336248
|
|
jax/_src/numpy/linalg.py,sha256=XS4z_mDXK0nZwtlXITfjXA2o8fCW2oiBrQtDoOquFyk,79425
|
|
jax/_src/numpy/polynomial.py,sha256=0ksPTmOu63XtFTJKRHg2XdT5BKD0KkpuchTUbgtK9ho,28500
|
|
jax/_src/numpy/reductions.py,sha256=m9DuttToWqBpDVybVSU97Wn-gZHJkX3Z7xXsmEY8um0,109327
|
|
jax/_src/numpy/scalar_types.py,sha256=92ODM21nG1eeTvmWpMikTU1dCFC-DTcV_gZfPMX-b04,3836
|
|
jax/_src/numpy/setops.py,sha256=SIh5WY6o6AaKB2VarCu_LudqT1tDwjzzmSe3ztbvjTM,47926
|
|
jax/_src/numpy/sorting.py,sha256=-O8u0ir18gQYzTf8dx8KeA6FFdt6eC1XCVqFWhazhlY,16237
|
|
jax/_src/numpy/tensor_contractions.py,sha256=ErJWjXRGlTGTNGGHukfAwszA27UEYCKVlwpJCczXEUg,24522
|
|
jax/_src/numpy/ufunc_api.py,sha256=IMO2pOj9CoFTOoUvHECIASm18V_opsSpEp-GffTZQOk,24376
|
|
jax/_src/numpy/ufuncs.py,sha256=LabJGcOGYMAtl5H3vt5VR-14pMlFzX6UmEAbh61plUU,118752
|
|
jax/_src/numpy/util.py,sha256=QqmKGhV7lm7AZP0cPk2S4t6_uEPcLgLbV-ZwpbQ9nXA,18156
|
|
jax/_src/numpy/vectorize.py,sha256=thLB2fBHVQBas7oaVWSsaOhtLgVWRwQekq_L5F7IjAA,14077
|
|
jax/_src/numpy/window_functions.py,sha256=jZsnPyv4nF6IXNHMCiX_9-gBGDdJfyb7V74fLN8TaJ0,5473
|
|
jax/_src/op_shardings.py,sha256=nRnGgfaKz7mIB7sHxh_QkiCa1oPMA9VsDpPyQkivtlU,3857
|
|
jax/_src/ops/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
|
|
jax/_src/ops/__pycache__/__init__.cpython-312.pyc,,
|
|
jax/_src/ops/__pycache__/scatter.cpython-312.pyc,,
|
|
jax/_src/ops/__pycache__/special.cpython-312.pyc,,
|
|
jax/_src/ops/scatter.py,sha256=s8upXyNVGeBShjwHxS6YLpYAbIMgDzBTOKJym27-bkA,18300
|
|
jax/_src/ops/special.py,sha256=zYKBMe29eMqKXDeaZ5BK8YehQdpAxAHPFKTmNjwKBa4,4380
|
|
jax/_src/pallas/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
|
|
jax/_src/pallas/__pycache__/__init__.cpython-312.pyc,,
|
|
jax/_src/pallas/__pycache__/core.cpython-312.pyc,,
|
|
jax/_src/pallas/__pycache__/cost_estimate.cpython-312.pyc,,
|
|
jax/_src/pallas/__pycache__/helpers.cpython-312.pyc,,
|
|
jax/_src/pallas/__pycache__/hlo_interpreter.cpython-312.pyc,,
|
|
jax/_src/pallas/__pycache__/pallas_call.cpython-312.pyc,,
|
|
jax/_src/pallas/__pycache__/primitives.cpython-312.pyc,,
|
|
jax/_src/pallas/__pycache__/utils.cpython-312.pyc,,
|
|
jax/_src/pallas/core.py,sha256=508UCDqaVi6id4q6g_nfZQ2JD4z3iWwIqy8p659AYpU,50338
|
|
jax/_src/pallas/cost_estimate.py,sha256=DJnl0aNbAtHErpMRcyojeHsCxlu5DthB33bah6T3BI0,8338
|
|
jax/_src/pallas/fuser/__init__.py,sha256=cHEWioEa-43e-714tP3fFN9hdyZg_ebBFbFJur7bXwg,1182
|
|
jax/_src/pallas/fuser/__pycache__/__init__.cpython-312.pyc,,
|
|
jax/_src/pallas/fuser/__pycache__/block_spec.cpython-312.pyc,,
|
|
jax/_src/pallas/fuser/__pycache__/custom_evaluate.cpython-312.pyc,,
|
|
jax/_src/pallas/fuser/__pycache__/fuser_utils.cpython-312.pyc,,
|
|
jax/_src/pallas/fuser/__pycache__/fusible.cpython-312.pyc,,
|
|
jax/_src/pallas/fuser/__pycache__/fusible_dtype.cpython-312.pyc,,
|
|
jax/_src/pallas/fuser/__pycache__/fusion.cpython-312.pyc,,
|
|
jax/_src/pallas/fuser/__pycache__/jaxpr_fusion.cpython-312.pyc,,
|
|
jax/_src/pallas/fuser/block_spec.py,sha256=1LjreSLXjiNNa1l0Y52fq2E-HHVIwL2q2QVM6N9ud_I,72090
|
|
jax/_src/pallas/fuser/custom_evaluate.py,sha256=jv5JjL5BlUUK89ZKx9sufff0bVyqJA3s5ag6Bpus5ys,2823
|
|
jax/_src/pallas/fuser/fuser_utils.py,sha256=2crsWld47NLWESj0R4sQnxjTB6yaYO-fCfamR8NEbCE,1270
|
|
jax/_src/pallas/fuser/fusible.py,sha256=k0CaPyV321MFuP8St6H1bB-fOiZvemByzEXbDD5lbi4,2689
|
|
jax/_src/pallas/fuser/fusible_dtype.py,sha256=06IUZU0b2_N5_Cmp3TojymjtOUBEmxX-B1jpQsPAKUU,16465
|
|
jax/_src/pallas/fuser/fusion.py,sha256=AvfY5PSPzoAT5Vu6NpMHPqobSgNfvv_600VbWIlcslI,1515
|
|
jax/_src/pallas/fuser/jaxpr_fusion.py,sha256=vB_rWH0XA1L-45p3Q22be2K40qiIA9jd6tQfTinA4gk,11061
|
|
jax/_src/pallas/helpers.py,sha256=x4e7sGaZPSU6noFmP3n0Eq0zYeRokQvaFKt5MfUQUl4,4018
|
|
jax/_src/pallas/hlo_interpreter.py,sha256=B71ICSEt2dpqA5uMMJvHmBpPAeN_VeeneiIRFdHVH18,19185
|
|
jax/_src/pallas/mosaic/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
|
|
jax/_src/pallas/mosaic/__pycache__/__init__.cpython-312.pyc,,
|
|
jax/_src/pallas/mosaic/__pycache__/core.cpython-312.pyc,,
|
|
jax/_src/pallas/mosaic/__pycache__/error_handling.cpython-312.pyc,,
|
|
jax/_src/pallas/mosaic/__pycache__/helpers.cpython-312.pyc,,
|
|
jax/_src/pallas/mosaic/__pycache__/interpret.cpython-312.pyc,,
|
|
jax/_src/pallas/mosaic/__pycache__/lowering.cpython-312.pyc,,
|
|
jax/_src/pallas/mosaic/__pycache__/pallas_call_registration.cpython-312.pyc,,
|
|
jax/_src/pallas/mosaic/__pycache__/pipeline.cpython-312.pyc,,
|
|
jax/_src/pallas/mosaic/__pycache__/primitives.cpython-312.pyc,,
|
|
jax/_src/pallas/mosaic/__pycache__/random.cpython-312.pyc,,
|
|
jax/_src/pallas/mosaic/__pycache__/verification.cpython-312.pyc,,
|
|
jax/_src/pallas/mosaic/core.py,sha256=nkZwiGkhFfJxN7NYyxqRiXZ_e1rfE6tSv8EN5win3EY,10590
|
|
jax/_src/pallas/mosaic/error_handling.py,sha256=FCbB-b5Dl75erMmr-FMvc4QJwCXqn2z7ioNC2Eh0vQc,5542
|
|
jax/_src/pallas/mosaic/helpers.py,sha256=DFBL7RuFtfBQc-aUaC4bILa5Xzsbb91Wu3VCn7b3v4M,2767
|
|
jax/_src/pallas/mosaic/interpret.py,sha256=4jgG-1XYyhjhfWWrgW9Wqk9SUvbKKcYglljfVSl7YQo,98240
|
|
jax/_src/pallas/mosaic/lowering.py,sha256=bl17IDZ5cv8vxiYE7mLDLdM7IRPxY71q_eBXCGvtbrs,143269
|
|
jax/_src/pallas/mosaic/pallas_call_registration.py,sha256=VwBVcxAVmsRbEGrlbOq_TQmmfEiOkf7UFxd2b4hQHBs,10565
|
|
jax/_src/pallas/mosaic/pipeline.py,sha256=VSdDd-94HPkhuAsAfhqJ1198a4G9WAABN9bcNRJcLKU,82197
|
|
jax/_src/pallas/mosaic/primitives.py,sha256=TflxJ7Z_zld5LjbiUyUF2wKSJJyEAOY4RGzGsQ3xdMw,30054
|
|
jax/_src/pallas/mosaic/random.py,sha256=doT0M7IF_-ix6eU56fJMEdgwDM7QgvEwCc3Knn1U1nI,7952
|
|
jax/_src/pallas/mosaic/verification.py,sha256=3k5rJTC5n26T3EUj_R7a5A9ppgvaDKMknTDymA-tABc,24512
|
|
jax/_src/pallas/mosaic_gpu/__init__.py,sha256=mS3YSYUZFIG68T5Zo9CFGkT7wCyWRycnWV4L36CnHYU,581
|
|
jax/_src/pallas/mosaic_gpu/__pycache__/__init__.cpython-312.pyc,,
|
|
jax/_src/pallas/mosaic_gpu/__pycache__/core.cpython-312.pyc,,
|
|
jax/_src/pallas/mosaic_gpu/__pycache__/helpers.cpython-312.pyc,,
|
|
jax/_src/pallas/mosaic_gpu/__pycache__/lowering.cpython-312.pyc,,
|
|
jax/_src/pallas/mosaic_gpu/__pycache__/pallas_call_registration.cpython-312.pyc,,
|
|
jax/_src/pallas/mosaic_gpu/__pycache__/pipeline.cpython-312.pyc,,
|
|
jax/_src/pallas/mosaic_gpu/__pycache__/primitives.cpython-312.pyc,,
|
|
jax/_src/pallas/mosaic_gpu/core.py,sha256=qDmJCLL5lLJOY2LWcShfspzqzjcj9U2mb9xqCcqsRQs,47324
|
|
jax/_src/pallas/mosaic_gpu/helpers.py,sha256=e2ECfoe2kbd9CgKjDNOCNjbX9BGnxLvTw5vWtIxHw_8,7108
|
|
jax/_src/pallas/mosaic_gpu/lowering.py,sha256=ED6etSqVTt1bdDZZjtLuuccR6DMtQm_fbfvgMZYKKK4,124547
|
|
jax/_src/pallas/mosaic_gpu/pallas_call_registration.py,sha256=xOYbx1YePen1uADDo8pjJFEdPGTbr3gcGavitAeoBV0,5730
|
|
jax/_src/pallas/mosaic_gpu/pipeline.py,sha256=4jwoB-GQaaTQUoQ7540A7IRrPoXKTRz-3oXkbeUS9bI,32248
|
|
jax/_src/pallas/mosaic_gpu/primitives.py,sha256=10Ta2H7IVYbDSm2rhO1WdVU3iNcj0-aphv0zyKU3jkU,109145
|
|
jax/_src/pallas/pallas_call.py,sha256=S7Q9hOTNsSHhxIByz-0m2_CHLcOc71mKW8Q8xy73UpM,70029
|
|
jax/_src/pallas/primitives.py,sha256=g8lqqYNdURKer4Gx2s5RVS1EP4j7weh92xAz2NcUKVg,47055
|
|
jax/_src/pallas/triton/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
|
|
jax/_src/pallas/triton/__pycache__/__init__.cpython-312.pyc,,
|
|
jax/_src/pallas/triton/__pycache__/core.cpython-312.pyc,,
|
|
jax/_src/pallas/triton/__pycache__/lowering.cpython-312.pyc,,
|
|
jax/_src/pallas/triton/__pycache__/pallas_call_registration.cpython-312.pyc,,
|
|
jax/_src/pallas/triton/__pycache__/primitives.cpython-312.pyc,,
|
|
jax/_src/pallas/triton/core.py,sha256=Z4Vk2LFEpkvRzhRBfZlvEGRhFHoxnjuPxC3EfgMPONs,1222
|
|
jax/_src/pallas/triton/lowering.py,sha256=oXLHubIXzFIhK7VT8nOpLpt6QY1Xz3RSNeH-W6ctVA4,98772
|
|
jax/_src/pallas/triton/pallas_call_registration.py,sha256=SHD6MMwZ_45UwJcYqwVnYzC-Of2reeNLXVF1TxDNe2I,6892
|
|
jax/_src/pallas/triton/primitives.py,sha256=R5BVdwenBSdnStf9h2EEYLPDX0FOcxMqqjCx72Rjc0c,5816
|
|
jax/_src/pallas/utils.py,sha256=zPdw0wnqXhDrp12l4KfMHuivTNHC9rc-W8UW3lSly7s,14342
|
|
jax/_src/partition_spec.py,sha256=7dBKnj-ONLW7kVIWjnMag-_8fIUIXPUGZ4pnF8f_Slk,6220
|
|
jax/_src/path.py,sha256=d9R4bFgk3cHS37ZUO8up_dHW10Vz3oR2NZTaRM28gdc,2162
|
|
jax/_src/pickle_util.py,sha256=gE8IrNWtg4hfu84T5rQ2DVAdpozh5xwSzAlRi7UIWcQ,2312
|
|
jax/_src/pjit.py,sha256=2LaCRZtbOfk0NRa1gVhBZbULuJzTVwUnJjZx9vvsGws,146600
|
|
jax/_src/pretty_printer.py,sha256=_UZ4oTCAFc55iKvstsMhquEW4AmuhfimYHXYchMnM50,3419
|
|
jax/_src/prng.py,sha256=ztET0aPt5zc84xXAbIu8NmYvdpzUxP36Ar_QOLB9P4s,46683
|
|
jax/_src/profiler.py,sha256=RC6eraXJVIZOFVl2J2sqGefkWTHzP5x-9CIsdgOojSU,17592
|
|
jax/_src/public_test_util.py,sha256=iuO-3tHEx9M5XEfULXiMdy1KM8ro31Idz96YSS3EQnY,13087
|
|
jax/_src/random.py,sha256=UQvebSXnTTZDemh9p4N6yMqPM8yK2c8h-zg5giTdIno,114055
|
|
jax/_src/ref.py,sha256=gdbMvj7VVOeQRl6wDvNf4zntzpWrF4GRCkOvc5ELW8g,1135
|
|
jax/_src/scipy/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
|
|
jax/_src/scipy/__pycache__/__init__.cpython-312.pyc,,
|
|
jax/_src/scipy/__pycache__/fft.cpython-312.pyc,,
|
|
jax/_src/scipy/__pycache__/integrate.cpython-312.pyc,,
|
|
jax/_src/scipy/__pycache__/linalg.cpython-312.pyc,,
|
|
jax/_src/scipy/__pycache__/ndimage.cpython-312.pyc,,
|
|
jax/_src/scipy/__pycache__/signal.cpython-312.pyc,,
|
|
jax/_src/scipy/__pycache__/special.cpython-312.pyc,,
|
|
jax/_src/scipy/cluster/__init__.py,sha256=S-o0CP72882Xv3l6k2PlrEx6eNvXBTk5Oaed5KF4iL8,581
|
|
jax/_src/scipy/cluster/__pycache__/__init__.cpython-312.pyc,,
|
|
jax/_src/scipy/cluster/__pycache__/vq.cpython-312.pyc,,
|
|
jax/_src/scipy/cluster/vq.py,sha256=LPoQUfgmh50I-ooN43p7vBiucOgJ29FIji9ZFbFikOc,3039
|
|
jax/_src/scipy/fft.py,sha256=HQhZQPNr1lfUbfQPA5mWnA7mQHZBeh-h2YClf0zbqJo,15357
|
|
jax/_src/scipy/integrate.py,sha256=2eo0ZUM3qdh8w9W4PJ1cKqZNCcN_fhFancTKjqu7gq8,2374
|
|
jax/_src/scipy/linalg.py,sha256=7ETV6uxshB29fculn1Wm3zygP-yJD05aUfRqDiV1e7w,79654
|
|
jax/_src/scipy/ndimage.py,sha256=6XBQq3MXu2F0i1UaTCAG6MOMPfbPNn4hMnRNHhoyVVY,6945
|
|
jax/_src/scipy/optimize/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
|
|
jax/_src/scipy/optimize/__pycache__/__init__.cpython-312.pyc,,
|
|
jax/_src/scipy/optimize/__pycache__/_lbfgs.cpython-312.pyc,,
|
|
jax/_src/scipy/optimize/__pycache__/bfgs.cpython-312.pyc,,
|
|
jax/_src/scipy/optimize/__pycache__/line_search.cpython-312.pyc,,
|
|
jax/_src/scipy/optimize/__pycache__/minimize.cpython-312.pyc,,
|
|
jax/_src/scipy/optimize/_lbfgs.py,sha256=UlpYZmDI950InTMDGdfaPtMcplN9uaYpqmjeYaG_bA8,7874
|
|
jax/_src/scipy/optimize/bfgs.py,sha256=RmbR6n91EhFgkxJ9Oe-HB1_0RdvD5voG6ZMMOnWoFvQ,5733
|
|
jax/_src/scipy/optimize/line_search.py,sha256=1sEzt6U8YWoFeOmGy6tn4ZdpI8I-D6b42h5_0-QZHEs,13184
|
|
jax/_src/scipy/optimize/minimize.py,sha256=vYoSJVP6ijGz0nyjjuwk-dHVFMlJ6MOeDiCePaFoLig,4954
|
|
jax/_src/scipy/signal.py,sha256=XAAx6rK9Qsv7sYt1u51kVhdH-NE7vhMfhdX14Ch4ByE,47429
|
|
jax/_src/scipy/sparse/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
|
|
jax/_src/scipy/sparse/__pycache__/__init__.cpython-312.pyc,,
|
|
jax/_src/scipy/sparse/__pycache__/linalg.cpython-312.pyc,,
|
|
jax/_src/scipy/sparse/linalg.py,sha256=a2ENNIyq4SnTS5wkGwu4M9p-6LnEBqqeScmaw25fXs0,27379
|
|
jax/_src/scipy/spatial/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
|
|
jax/_src/scipy/spatial/__pycache__/__init__.cpython-312.pyc,,
|
|
jax/_src/scipy/spatial/__pycache__/transform.cpython-312.pyc,,
|
|
jax/_src/scipy/spatial/transform.py,sha256=a10DkF7vLkS4BMm86swDwoqdV-xDtRjCNfho965M5fk,17634
|
|
jax/_src/scipy/special.py,sha256=xhmnPiJaj44X_gvzbzcUoLRbxYFtNUL82z2Nsfr3n68,90627
|
|
jax/_src/scipy/stats/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
|
|
jax/_src/scipy/stats/__pycache__/__init__.cpython-312.pyc,,
|
|
jax/_src/scipy/stats/__pycache__/_core.cpython-312.pyc,,
|
|
jax/_src/scipy/stats/__pycache__/bernoulli.cpython-312.pyc,,
|
|
jax/_src/scipy/stats/__pycache__/beta.cpython-312.pyc,,
|
|
jax/_src/scipy/stats/__pycache__/betabinom.cpython-312.pyc,,
|
|
jax/_src/scipy/stats/__pycache__/binom.cpython-312.pyc,,
|
|
jax/_src/scipy/stats/__pycache__/cauchy.cpython-312.pyc,,
|
|
jax/_src/scipy/stats/__pycache__/chi2.cpython-312.pyc,,
|
|
jax/_src/scipy/stats/__pycache__/dirichlet.cpython-312.pyc,,
|
|
jax/_src/scipy/stats/__pycache__/expon.cpython-312.pyc,,
|
|
jax/_src/scipy/stats/__pycache__/gamma.cpython-312.pyc,,
|
|
jax/_src/scipy/stats/__pycache__/gennorm.cpython-312.pyc,,
|
|
jax/_src/scipy/stats/__pycache__/geom.cpython-312.pyc,,
|
|
jax/_src/scipy/stats/__pycache__/gumbel_l.cpython-312.pyc,,
|
|
jax/_src/scipy/stats/__pycache__/gumbel_r.cpython-312.pyc,,
|
|
jax/_src/scipy/stats/__pycache__/kde.cpython-312.pyc,,
|
|
jax/_src/scipy/stats/__pycache__/laplace.cpython-312.pyc,,
|
|
jax/_src/scipy/stats/__pycache__/logistic.cpython-312.pyc,,
|
|
jax/_src/scipy/stats/__pycache__/multinomial.cpython-312.pyc,,
|
|
jax/_src/scipy/stats/__pycache__/multivariate_normal.cpython-312.pyc,,
|
|
jax/_src/scipy/stats/__pycache__/nbinom.cpython-312.pyc,,
|
|
jax/_src/scipy/stats/__pycache__/norm.cpython-312.pyc,,
|
|
jax/_src/scipy/stats/__pycache__/pareto.cpython-312.pyc,,
|
|
jax/_src/scipy/stats/__pycache__/poisson.cpython-312.pyc,,
|
|
jax/_src/scipy/stats/__pycache__/t.cpython-312.pyc,,
|
|
jax/_src/scipy/stats/__pycache__/truncnorm.cpython-312.pyc,,
|
|
jax/_src/scipy/stats/__pycache__/uniform.cpython-312.pyc,,
|
|
jax/_src/scipy/stats/__pycache__/vonmises.cpython-312.pyc,,
|
|
jax/_src/scipy/stats/__pycache__/wrapcauchy.cpython-312.pyc,,
|
|
jax/_src/scipy/stats/_core.py,sha256=X05I6c5EVsc37Lw7LKp4AdO9PveexkkCS7ow5_zNMj0,10642
|
|
jax/_src/scipy/stats/bernoulli.py,sha256=HT89cwug-OktD0KVKDWrpihz4GB_hEOU_R567Q_8whw,4578
|
|
jax/_src/scipy/stats/beta.py,sha256=jjRE4LlyBt2BO2jmmN1z5PEasjl4LykhKTNc8846etU,8026
|
|
jax/_src/scipy/stats/betabinom.py,sha256=LceEdt1jRBxYmV9UMjZBJs7gR1GAcGcU4JbOcF2iEiU,3480
|
|
jax/_src/scipy/stats/binom.py,sha256=KIydYZOcwN109i6rodNR-Lda-cog16r2jrQcDarUeeo,2859
|
|
jax/_src/scipy/stats/cauchy.py,sha256=BnKgESULqAHQZydTKMxT4oRexDWbMyHWj37CFROaar4,8901
|
|
jax/_src/scipy/stats/chi2.py,sha256=p3TVmUKKRscb-7-uBUB2aorLFdh2USag5iRt1P7yr7U,8045
|
|
jax/_src/scipy/stats/dirichlet.py,sha256=G9FAdSoODRw-lZkZlGsg1sq2bbqMLZ47wG8P4o_oaow,3237
|
|
jax/_src/scipy/stats/expon.py,sha256=NcqNEDneSnAtba3Yol0XyfNZ8_OIwcvTih2NZLE8OoY,7845
|
|
jax/_src/scipy/stats/gamma.py,sha256=876SXGzsazCTNxGXL38_PqF0MbFZ9k1XOTcXDhUJdS4,7076
|
|
jax/_src/scipy/stats/gennorm.py,sha256=s-Wo42aMKpE1EzeqC-2Vx_uH-zDg8qWjdUAe2d5azr0,2994
|
|
jax/_src/scipy/stats/geom.py,sha256=A5XokLcEPUlAjhBh-0BF9PoqGGZmE5EkkT3xT3R61Fc,2282
|
|
jax/_src/scipy/stats/gumbel_l.py,sha256=PR7XwEY_RAxKLCLJIEq3QTqzdvAv-ufQl0QgRkFh65Q,8345
|
|
jax/_src/scipy/stats/gumbel_r.py,sha256=_1nDUgHNtYlNTILqyZq9jxT4ZDdElwltJ6I5gYVmMUY,8361
|
|
jax/_src/scipy/stats/kde.py,sha256=wgCbK8d6FFKuN6Tk3nPu7L8PZ7EYHdQ-h860QIvqtQ8,10249
|
|
jax/_src/scipy/stats/laplace.py,sha256=KRoMp7S0Gn-xTiY9A-1_Iv1lnwswGhjfSg6WDf9Oaio,3274
|
|
jax/_src/scipy/stats/logistic.py,sha256=508gMdwcsWI6ynyrD4w7VZMw9HB8K-Xyzux6oVEdKkk,6291
|
|
jax/_src/scipy/stats/multinomial.py,sha256=W22uKDVNPoF8JkQwI0TmnXPJ_7SFa3mgzUYLTJKG2ks,2532
|
|
jax/_src/scipy/stats/multivariate_normal.py,sha256=LqDX6XJxocbMlqM3HrnUHEw2ZOGjfIvXowhJmmHD1mo,3569
|
|
jax/_src/scipy/stats/nbinom.py,sha256=pF8EOulsrjJhQmq_NjMhfeQaXf5qh7wC9A6mNk8oiUU,2637
|
|
jax/_src/scipy/stats/norm.py,sha256=8YfKWloDLhnfpfcuamJlPZuqgLtwmsNKVMYT53VKKsY,8421
|
|
jax/_src/scipy/stats/pareto.py,sha256=uN0kgylc46jwnTWbyZJf6ZWxgTEzyN-R7rA7OPiHqLQ,2630
|
|
jax/_src/scipy/stats/poisson.py,sha256=EPcPC-ThmNFxw5GUDUTybSZ2TRnEYIn7yAcGb8XXZoE,3387
|
|
jax/_src/scipy/stats/t.py,sha256=-POac0PdQa3tNpgksvZaVRHYrrbQQrlyXj6l8iP15gk,3226
|
|
jax/_src/scipy/stats/truncnorm.py,sha256=LQaZJwBZ9Y3pWOABOU0gsbHD0nq9zQEQCKQzxLJx_Aw,9270
|
|
jax/_src/scipy/stats/uniform.py,sha256=e0INco1aevZnIfWfVXVXzFbt3rEOUz0UDGscr81SU94,4372
|
|
jax/_src/scipy/stats/vonmises.py,sha256=S5GpPFeRpE5BH5-MEcyenno1zxbbvOpL2Lvn23vTG-M,2484
|
|
jax/_src/scipy/stats/wrapcauchy.py,sha256=4vJw9eZn5iMgzpR3iJmuFhFq0GTCqQm3P___4UB-czQ,2401
|
|
jax/_src/shard_alike.py,sha256=wG_kUvyH6aNRjgViwVffEAz1tbfhBF66kf3vw0XexIg,3989
|
|
jax/_src/shard_map.py,sha256=XYUh8obdfjooFSFkxNKjJZsQ1-mPpw0mJMAn9NWdSSY,89086
|
|
jax/_src/sharding.py,sha256=mtNHNhrrbQdmHEARTzFBtdeE0WXESc-4Ru0WcXXxPXE,7843
|
|
jax/_src/sharding_impls.py,sha256=jUdMk005aVMQ0sgJKeOtKI-gDVuOBVCajnughRN74QU,46885
|
|
jax/_src/sharding_specs.py,sha256=UCYUIcQfS6lcA993W33zZI5tfAg5zcefP9cZUyW9NoI,8617
|
|
jax/_src/source_info_util.py,sha256=M7NfK6Kk0FuMAjMY-nXp3hPkgyAeScQYYkfwvG57dSM,10457
|
|
jax/_src/sourcemap.py,sha256=5s4zTThhX6PZwgQ4v4UbYL4norg4uHipqgUGdHBn76Y,6640
|
|
jax/_src/stages.py,sha256=WmtlVKLHP0tQlzud8GyBtNxwagqgxNgORSgtJuItmC8,33069
|
|
jax/_src/state/__init__.py,sha256=TU0qQETOXKzVAd1oVaIEtOJTC2OMj1PdFhsExMHFx2g,988
|
|
jax/_src/state/__pycache__/__init__.cpython-312.pyc,,
|
|
jax/_src/state/__pycache__/discharge.cpython-312.pyc,,
|
|
jax/_src/state/__pycache__/indexing.cpython-312.pyc,,
|
|
jax/_src/state/__pycache__/primitives.cpython-312.pyc,,
|
|
jax/_src/state/__pycache__/types.cpython-312.pyc,,
|
|
jax/_src/state/__pycache__/utils.cpython-312.pyc,,
|
|
jax/_src/state/discharge.py,sha256=BAhdyBduBiWaxv3NyBgvp-eVEN3l0Td94iRNMstAfV8,33293
|
|
jax/_src/state/indexing.py,sha256=UlPKL39NIhlnsKbizF_L_8PmItR97KvfPSwjT09ArhY,12040
|
|
jax/_src/state/primitives.py,sha256=SySNpMi1hF6x4ATWXCAsclUOcBXFTuAgvIRsPz5L2Vk,36568
|
|
jax/_src/state/types.py,sha256=sunywstixGNWjQURcSElbWpZ45-vFIfQAnzZLnY67YY,16093
|
|
jax/_src/state/utils.py,sha256=6q9G1IwIEvNFoZ806I8n1jVjAhJ8_9cfpwWn4yPIvgs,3996
|
|
jax/_src/test_loader.py,sha256=l7LDHdnqD5g44nUo4Hdwtc8IplXjCcn-L9-RO4Q_qhs,7260
|
|
jax/_src/test_multiprocess.py,sha256=rfZsYMKU9hrPkhusULYp0OkrpHNjQYUD1wCxNbKTxrk,9384
|
|
jax/_src/test_util.py,sha256=O8G4fV7vaqe7oEK9w7TgKVmC83g3mxLfABT5eyNM5BI,85065
|
|
jax/_src/test_warning_util.py,sha256=9CvuFdeyGcmTH5xfz3Wqs0qOUGpOVcOwh4SnJsp6FbE,4074
|
|
jax/_src/third_party/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
jax/_src/third_party/__pycache__/__init__.cpython-312.pyc,,
|
|
jax/_src/third_party/scipy/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
jax/_src/third_party/scipy/__pycache__/__init__.cpython-312.pyc,,
|
|
jax/_src/third_party/scipy/__pycache__/betaln.cpython-312.pyc,,
|
|
jax/_src/third_party/scipy/__pycache__/interpolate.cpython-312.pyc,,
|
|
jax/_src/third_party/scipy/__pycache__/linalg.cpython-312.pyc,,
|
|
jax/_src/third_party/scipy/__pycache__/signal_helper.cpython-312.pyc,,
|
|
jax/_src/third_party/scipy/__pycache__/special.cpython-312.pyc,,
|
|
jax/_src/third_party/scipy/betaln.py,sha256=EW6FFB1myCHMGwDXqQXPIRaEL-UvVZNahZu_wdsixSc,2388
|
|
jax/_src/third_party/scipy/interpolate.py,sha256=XJA3BNVCwIowXBH5g3M4dsKwHpMVyHKBxL2Oy6RkbME,6360
|
|
jax/_src/third_party/scipy/linalg.py,sha256=jgDDGByArWWjetmCdX6LCdzPT2Ao43TCe56syYpBxFw,3878
|
|
jax/_src/third_party/scipy/signal_helper.py,sha256=ihye51hrdpDnsJuZ5z9scJhl1a-pEN44rCFgLQdTM_0,3569
|
|
jax/_src/third_party/scipy/special.py,sha256=Q8kobJMwgE5SiOYDLyliaHGz8Sy7RLCzFrJoCIw9eT4,9118
|
|
jax/_src/tpu/__init__.py,sha256=4S3pYNuiw6yzPmNNm2iRi40_Jf9xlHj-oIRb9Gp4sxk,581
|
|
jax/_src/tpu/__pycache__/__init__.cpython-312.pyc,,
|
|
jax/_src/tpu/linalg/__init__.py,sha256=RrUPd-WnsNpraGGzWYI-IYTpNcbtQeGqaotRh0Bwwuw,779
|
|
jax/_src/tpu/linalg/__pycache__/__init__.cpython-312.pyc,,
|
|
jax/_src/tpu/linalg/__pycache__/eigh.cpython-312.pyc,,
|
|
jax/_src/tpu/linalg/__pycache__/qdwh.cpython-312.pyc,,
|
|
jax/_src/tpu/linalg/__pycache__/stack.cpython-312.pyc,,
|
|
jax/_src/tpu/linalg/__pycache__/svd.cpython-312.pyc,,
|
|
jax/_src/tpu/linalg/eigh.py,sha256=ixc6cNXn0T0aqcx2AGraGGgKWKfZqbwlL8L6xks2m-8,24665
|
|
jax/_src/tpu/linalg/qdwh.py,sha256=tX9TiLEe0Flr7CJiRW7UNOy1g1hGkmQIiowmzCQ2Xw8,9553
|
|
jax/_src/tpu/linalg/stack.py,sha256=XPaFJtuPGfry1ssSqTNKdHcUs7yRUyzeD6aeYz0SrU8,2570
|
|
jax/_src/tpu/linalg/svd.py,sha256=vNffW6Q6CqSvNX19WEJDuo9Gu1YNdUmgQPHH3xFwIGQ,10316
|
|
jax/_src/tpu_custom_call.py,sha256=rni_PmVXHYwVAE52hjdH4rQU5_Fwc87hUcZD-MeTLTc,27080
|
|
jax/_src/traceback_util.py,sha256=ywejARaYxtWnK-Oxsi-MdgZiUYflvU3UtujpBq8MGbo,8135
|
|
jax/_src/tree.py,sha256=jPEIRgsIpo0E05GFMZLg-AelZnsPC2vdGjbhkfXhYS4,15703
|
|
jax/_src/tree_util.py,sha256=J1CUBsfegSt4T9rI0F3b6NQkDomPMVzc-xzqIWnkaBY,48088
|
|
jax/_src/typing.py,sha256=Ed5ujNHsKktTk0Z2nhZojS_77K1INnsPvj4S7o0KQ5Q,3529
|
|
jax/_src/util.py,sha256=S4whnIzZbhN5YZJyAGMVQ1UP1pE9tLxs3YMNt1ZtdEs,26940
|
|
jax/_src/xla_bridge.py,sha256=4wqo55iE8dhfqU9oT28ulFuhH4xfQqCDPLIpkzwWiSo,45052
|
|
jax/_src/xla_metadata.py,sha256=48mF2jQYDW4lf5muzE3he5y-mM1sV24HQ0Kq9O143wc,3834
|
|
jax/_src/xla_metadata_lib.py,sha256=gyRBCCpMfUG-o51PZQ9dO_rafReE7Py_yZOs6cHSY_Q,1474
|
|
jax/ad_checkpoint.py,sha256=jnNhXa-FiezkCFP9BlCul194Pz11olQPdZEeoCJQi-U,932
|
|
jax/api_util.py,sha256=avYenJ6674XMZHYtWzplyMPO0j2NXOk13BYb1o3vnCc,979
|
|
jax/cloud_tpu_init.py,sha256=r_UB06-qStzlo7BWtH7eZ7OMl3ayrG_CQXWQJJwWsYs,651
|
|
jax/collect_profile.py,sha256=nCmC3aNbZgJ-djAkRdRSG10JInh6tLZX5dHCjkVn0VI,4860
|
|
jax/core.py,sha256=H40vR3n3MY6_FH1Qr85qH-DoiAZtk9B02LfSfe7oJtc,4796
|
|
jax/custom_batching.py,sha256=MEoypZfGgYqvZrBKtnvL5sQjl8YrjVm11hiFWoTCtV8,691
|
|
jax/custom_derivatives.py,sha256=CVhRiRgnWk5hBsc3ZX7mFD0Yst2S9VDAirtwF8rSZcY,1967
|
|
jax/custom_transpose.py,sha256=vGj3GjRJQpxl-MhvYztIedauNBiWMj9nD_aJHQiNaYc,664
|
|
jax/debug.py,sha256=HXdNZlaM5LxtEYS6fsH3J4LRgiDVL-5W6AXcimbFS5o,1083
|
|
jax/distributed.py,sha256=CP5IDXwA4QzxXilcTCM1F6435U3dYsdWkDzO4hr_evE,710
|
|
jax/dlpack.py,sha256=7YqIO8PDBqaW5vUtr0FYgpt6LO3IR8uDAeG5vyg0nqM,1776
|
|
jax/dtypes.py,sha256=iaMlUUaiUH8zFsy5IJEff4s5QTu0EzD5msV-JYAyD6I,1188
|
|
jax/errors.py,sha256=LCZJqNAUhGcqPQYd1yA600R97ayA7lunuB_lu4zf0iY,1794
|
|
jax/example_libraries/__init__.py,sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw,581
|
|
jax/example_libraries/__pycache__/__init__.cpython-312.pyc,,
|
|
jax/example_libraries/__pycache__/optimizers.cpython-312.pyc,,
|
|
jax/example_libraries/__pycache__/stax.cpython-312.pyc,,
|
|
jax/example_libraries/optimizers.py,sha256=T2D3qDPWZ50kwTyDe2IXPPLbOUu04cQTdf51oHdXVvU,20426
|
|
jax/example_libraries/stax.py,sha256=eOLn-UowNy-bEXngwaNUMXbUka5qTqmbQXQUYS8Ou00,13834
|
|
jax/experimental/__init__.py,sha256=PIvLtzE4mJ7QO7vrfXDCpeKpefgVtJ_l5xlSMTvJLac,1529
|
|
jax/experimental/__pycache__/__init__.cpython-312.pyc,,
|
|
jax/experimental/__pycache__/buffer_callback.cpython-312.pyc,,
|
|
jax/experimental/__pycache__/checkify.cpython-312.pyc,,
|
|
jax/experimental/__pycache__/compute_on.cpython-312.pyc,,
|
|
jax/experimental/__pycache__/custom_dce.cpython-312.pyc,,
|
|
jax/experimental/__pycache__/custom_partitioning.cpython-312.pyc,,
|
|
jax/experimental/__pycache__/host_callback.cpython-312.pyc,,
|
|
jax/experimental/__pycache__/jet.cpython-312.pyc,,
|
|
jax/experimental/__pycache__/layout.cpython-312.pyc,,
|
|
jax/experimental/__pycache__/mesh_utils.cpython-312.pyc,,
|
|
jax/experimental/__pycache__/multihost_utils.cpython-312.pyc,,
|
|
jax/experimental/__pycache__/ode.cpython-312.pyc,,
|
|
jax/experimental/__pycache__/pjit.cpython-312.pyc,,
|
|
jax/experimental/__pycache__/profiler.cpython-312.pyc,,
|
|
jax/experimental/__pycache__/rnn.cpython-312.pyc,,
|
|
jax/experimental/__pycache__/scheduling_groups.cpython-312.pyc,,
|
|
jax/experimental/__pycache__/serialize_executable.cpython-312.pyc,,
|
|
jax/experimental/__pycache__/shard_alike.cpython-312.pyc,,
|
|
jax/experimental/__pycache__/shard_map.cpython-312.pyc,,
|
|
jax/experimental/__pycache__/topologies.cpython-312.pyc,,
|
|
jax/experimental/__pycache__/transfer.cpython-312.pyc,,
|
|
jax/experimental/__pycache__/x64_context.cpython-312.pyc,,
|
|
jax/experimental/__pycache__/xla_metadata.cpython-312.pyc,,
|
|
jax/experimental/array_serialization/__init__.py,sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw,581
|
|
jax/experimental/array_serialization/__pycache__/__init__.cpython-312.pyc,,
|
|
jax/experimental/array_serialization/__pycache__/pytree_serialization.cpython-312.pyc,,
|
|
jax/experimental/array_serialization/__pycache__/pytree_serialization_utils.cpython-312.pyc,,
|
|
jax/experimental/array_serialization/__pycache__/serialization.cpython-312.pyc,,
|
|
jax/experimental/array_serialization/__pycache__/tensorstore_impl.cpython-312.pyc,,
|
|
jax/experimental/array_serialization/pytree_serialization.py,sha256=Wf4atVPHTLhpXFuZS38cSvQIfyQGr9YqIPH156bE7pU,21623
|
|
jax/experimental/array_serialization/pytree_serialization_utils.py,sha256=6WBo_JYB2I2Iqk16ceIA4G4n35dJk5FWxckgm7DZsSQ,2868
|
|
jax/experimental/array_serialization/serialization.py,sha256=3wiRVOMdD3odJbxFPwK3gj1ttUhkrWWTEutDF_XXPBI,13434
|
|
jax/experimental/array_serialization/tensorstore_impl.py,sha256=1ocq6sZTUpW9VMJNfTQvpWVOQsm9bfFGlqngK6GkBAk,25126
|
|
jax/experimental/buffer_callback.py,sha256=ocKnbOO_9n62E2olM0c2a4ep8AHTan4I_AL-eR5VbrQ,765
|
|
jax/experimental/checkify.py,sha256=V84hTTNrwnckaRzgvWDmpcKKA95kAakjZwwlfnRaXt0,1213
|
|
jax/experimental/colocated_python/__init__.py,sha256=Cp1CgJT_Sn5qLkiO7jW5S0HPWieBmOdpEE-AaYpKs9E,980
|
|
jax/experimental/colocated_python/__pycache__/__init__.cpython-312.pyc,,
|
|
jax/experimental/colocated_python/__pycache__/api.cpython-312.pyc,,
|
|
jax/experimental/colocated_python/__pycache__/func.cpython-312.pyc,,
|
|
jax/experimental/colocated_python/__pycache__/func_backend.cpython-312.pyc,,
|
|
jax/experimental/colocated_python/__pycache__/obj.cpython-312.pyc,,
|
|
jax/experimental/colocated_python/__pycache__/obj_backend.cpython-312.pyc,,
|
|
jax/experimental/colocated_python/__pycache__/serialization.cpython-312.pyc,,
|
|
jax/experimental/colocated_python/api.py,sha256=ie1z3UnsCJZsMt_RDTOjSi-sUZ-HLM5jvZEctAD1C0I,4476
|
|
jax/experimental/colocated_python/func.py,sha256=ET2am9LdOVR7lRTyow4k8zu1g5sUbfY085pUVNBFVZ8,17380
|
|
jax/experimental/colocated_python/func_backend.py,sha256=JQdiIUulodBQWSazJqSXKMC11UT9l3BbHVr_SveUcaI,1382
|
|
jax/experimental/colocated_python/obj.py,sha256=EsENLLgnS8ZYR6ubgYFUGHfSa7Hhq_MTbotbaZua-wQ,5657
|
|
jax/experimental/colocated_python/obj_backend.py,sha256=4TDWWp2p_xtv02R-I0rrRsW3kl0lWdiWKbpSR18Cg1I,2310
|
|
jax/experimental/colocated_python/serialization.py,sha256=1KbFDpQHCBa-hf8lEbOK8nqTEGW26hxfnetrY02cqAU,9186
|
|
jax/experimental/compilation_cache/__init__.py,sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw,581
|
|
jax/experimental/compilation_cache/__pycache__/__init__.cpython-312.pyc,,
|
|
jax/experimental/compilation_cache/__pycache__/compilation_cache.cpython-312.pyc,,
|
|
jax/experimental/compilation_cache/compilation_cache.py,sha256=WdtnZcUszZBWpVpoVhuR2DFMNK77UBNNZHb2P3sMaOs,1674
|
|
jax/experimental/compute_on.py,sha256=Mdg2n8JtAsrq1ap_PwUPwoqpoCNGO5rjf7T47e_iaAE,644
|
|
jax/experimental/custom_dce.py,sha256=ORRk_kBjbseQ06T_LmLPxYCO7-Rby5U8Erro5U5Gi4A,682
|
|
jax/experimental/custom_partitioning.py,sha256=RlaIvfdM8VqlEJC3FajqCmB_EOHyk6iyfGUgaUyR61g,1054
|
|
jax/experimental/host_callback.py,sha256=vKbDE9f-jiaNcNTDitYhgNhv2-ZPL3AQ_hlaR__ADX8,1601
|
|
jax/experimental/jax2tf/__init__.py,sha256=ZqsPlkalGAkzdTwRIutfW1k8NmOHzJn6kfIB1A98pdw,944
|
|
jax/experimental/jax2tf/__pycache__/__init__.cpython-312.pyc,,
|
|
jax/experimental/jax2tf/__pycache__/call_tf.cpython-312.pyc,,
|
|
jax/experimental/jax2tf/__pycache__/impl_no_xla.cpython-312.pyc,,
|
|
jax/experimental/jax2tf/__pycache__/jax2tf.cpython-312.pyc,,
|
|
jax/experimental/jax2tf/call_tf.py,sha256=027WlrLz68JD98hvGxkHMUt3Ss8wYw3sdscEso642i8,28929
|
|
jax/experimental/jax2tf/impl_no_xla.py,sha256=zi-U4UaZ7xVJ_Y9gmhHn8evKRX4_6SKJM33INEDucLY,54479
|
|
jax/experimental/jax2tf/jax2tf.py,sha256=HMGnF3Padip3aPB51Iejp_YamSq6etFXO_ELDAGvs_A,146836
|
|
jax/experimental/jet.py,sha256=wGNQmyCUOqi-4y1aObnWt0HfkOCcMnHdkmFw25fYxeQ,27605
|
|
jax/experimental/key_reuse/__init__.py,sha256=5CjqLFSLnAhm_YyrowsWA48ENwxcvcD5KQrNQJTgLpQ,1688
|
|
jax/experimental/key_reuse/__pycache__/__init__.cpython-312.pyc,,
|
|
jax/experimental/key_reuse/__pycache__/_core.cpython-312.pyc,,
|
|
jax/experimental/key_reuse/_core.py,sha256=-YeNKbFpc_WDCC6vZiReai7R4DHQWFG7RTxtcjw41i8,23241
|
|
jax/experimental/layout.py,sha256=fR0lxc08SqRZ0jv9bh6QACBFt_rFP_4naq5y43pU5GQ,738
|
|
jax/experimental/mesh_utils.py,sha256=Sf5ivjPuZ_u0ufQ38MLmhbEFnlIMs4SHd6lGXWQJgQY,923
|
|
jax/experimental/mosaic/__init__.py,sha256=-MnUIUUmkSaNkO5bOpPdXrcTnKVWt9ModYX4cnlq9LY,883
|
|
jax/experimental/mosaic/__pycache__/__init__.cpython-312.pyc,,
|
|
jax/experimental/mosaic/__pycache__/dialects.cpython-312.pyc,,
|
|
jax/experimental/mosaic/dialects.py,sha256=eYxSlw_cu_8usVXcEA3f-cqmlxDUwcTiqLiqNAgWWfs,769
|
|
jax/experimental/mosaic/gpu/__init__.py,sha256=pVzThaQ_m5OdP-5eRel6YH6TZLxpAwLGyHBREkrF6Ro,3984
|
|
jax/experimental/mosaic/gpu/__pycache__/__init__.cpython-312.pyc,,
|
|
jax/experimental/mosaic/gpu/__pycache__/core.cpython-312.pyc,,
|
|
jax/experimental/mosaic/gpu/__pycache__/dialect_lowering.cpython-312.pyc,,
|
|
jax/experimental/mosaic/gpu/__pycache__/equations.cpython-312.pyc,,
|
|
jax/experimental/mosaic/gpu/__pycache__/fragmented_array.cpython-312.pyc,,
|
|
jax/experimental/mosaic/gpu/__pycache__/inference_utils.cpython-312.pyc,,
|
|
jax/experimental/mosaic/gpu/__pycache__/launch_context.cpython-312.pyc,,
|
|
jax/experimental/mosaic/gpu/__pycache__/layout_inference.cpython-312.pyc,,
|
|
jax/experimental/mosaic/gpu/__pycache__/layout_inference2.cpython-312.pyc,,
|
|
jax/experimental/mosaic/gpu/__pycache__/layouts.cpython-312.pyc,,
|
|
jax/experimental/mosaic/gpu/__pycache__/mma_utils.cpython-312.pyc,,
|
|
jax/experimental/mosaic/gpu/__pycache__/profiler.cpython-312.pyc,,
|
|
jax/experimental/mosaic/gpu/__pycache__/tcgen05.cpython-312.pyc,,
|
|
jax/experimental/mosaic/gpu/__pycache__/transform_inference.cpython-312.pyc,,
|
|
jax/experimental/mosaic/gpu/__pycache__/utils.cpython-312.pyc,,
|
|
jax/experimental/mosaic/gpu/__pycache__/wgmma.cpython-312.pyc,,
|
|
jax/experimental/mosaic/gpu/core.py,sha256=uDsADvgog7deJr6CuteiCohJ4BA-tOfO6azsDBQvClw,36941
|
|
jax/experimental/mosaic/gpu/dialect_lowering.py,sha256=ED6Ysc4DdKfI9nwC7LiWgTVhqtJilNph_fjWwnKHnpo,72867
|
|
jax/experimental/mosaic/gpu/equations.py,sha256=qzA6diVIfu18tSuAJISTP-KckSfCogFe5R2bau2OJyM,18734
|
|
jax/experimental/mosaic/gpu/fragmented_array.py,sha256=7sppBx2Xf1hknPy--aFCHw-Qb0FmnTRyi-eVisHJl7M,138004
|
|
jax/experimental/mosaic/gpu/inference_utils.py,sha256=sfWZlqu1lz8jvlj021jRNj-u7bhuP5JjIymC8pTn18Q,8874
|
|
jax/experimental/mosaic/gpu/launch_context.py,sha256=qc6LPhcWHxuyDjRbInx7JGzz6tSLWMp1PfbwfJ7CqhE,55026
|
|
jax/experimental/mosaic/gpu/layout_inference.py,sha256=_kWHfwQJ9hRapLedxK12RPlb_pWsnx9y-skJkERu_po,26923
|
|
jax/experimental/mosaic/gpu/layout_inference2.py,sha256=5QuXpojJY4M9jzIBAmmqQDp-QUk7kJAulkFHhBt3o8s,34621
|
|
jax/experimental/mosaic/gpu/layouts.py,sha256=oKM66-O-TV_a_1szU-vZht31R6Mm3kTzWcfItNTswkE,11358
|
|
jax/experimental/mosaic/gpu/mma_utils.py,sha256=paLgZRYaHcJpqRcljrzcUts6tt0jehxkNyypudcMJoA,9687
|
|
jax/experimental/mosaic/gpu/profiler.py,sha256=S8vaJ-0up8QNqpqaWCTFgwOa2I96W-Q8_LbPWs9qxqo,15553
|
|
jax/experimental/mosaic/gpu/tcgen05.py,sha256=3aIWEQKOvAMVrUU1PKKY-ahrR540D4ao7EM4q8p9VEg,58086
|
|
jax/experimental/mosaic/gpu/transform_inference.py,sha256=KBnTpRUaaOysZdCTrVSSV4JnWZZ8uOlh1VRaht36AjU,18323
|
|
jax/experimental/mosaic/gpu/utils.py,sha256=vVew_GtC5tXlmSDTb3Id6bQkqP_cUhh4MTomx0ure3w,56253
|
|
jax/experimental/mosaic/gpu/wgmma.py,sha256=l5-3MJvasuJbbCLQ-WXpsG-mqMpsClNccgLgblq37LU,17511
|
|
jax/experimental/multihost_utils.py,sha256=UjuVSAT47VlOExyDZfSUh6WQDsXHv_Ej3-vWkRi1b-U,21887
|
|
jax/experimental/ode.py,sha256=a8ZsN4gVrHXM_LPJF3T5SeC-RHjFTGUd62eA1WDdkLM,10903
|
|
jax/experimental/pallas/__init__.py,sha256=B-brrvwln4LLTzrWUGBuCGEQw1bplNjwDzehm6aO29I,6762
|
|
jax/experimental/pallas/__pycache__/__init__.cpython-312.pyc,,
|
|
jax/experimental/pallas/__pycache__/fuser.cpython-312.pyc,,
|
|
jax/experimental/pallas/__pycache__/mosaic_gpu.cpython-312.pyc,,
|
|
jax/experimental/pallas/__pycache__/tpu.cpython-312.pyc,,
|
|
jax/experimental/pallas/__pycache__/triton.cpython-312.pyc,,
|
|
jax/experimental/pallas/fuser.py,sha256=27RRgL3R9gmnJeomX9xOy4HsctvtaTwYKOoqGAIKs3I,1211
|
|
jax/experimental/pallas/mosaic_gpu.py,sha256=-38_Kc5TCA4grx550IZmWf4clLhIigrdB1DSSCd3s98,5593
|
|
jax/experimental/pallas/ops/__init__.py,sha256=vC8qlznuwgbYeFDJqP8pDUIf2gEKFrCHg3EZYdTkiwM,764
|
|
jax/experimental/pallas/ops/__pycache__/__init__.cpython-312.pyc,,
|
|
jax/experimental/pallas/ops/gpu/__init__.py,sha256=mS3YSYUZFIG68T5Zo9CFGkT7wCyWRycnWV4L36CnHYU,581
|
|
jax/experimental/pallas/ops/gpu/__pycache__/__init__.cpython-312.pyc,,
|
|
jax/experimental/pallas/ops/gpu/__pycache__/attention.cpython-312.pyc,,
|
|
jax/experimental/pallas/ops/gpu/__pycache__/attention_mgpu.cpython-312.pyc,,
|
|
jax/experimental/pallas/ops/gpu/__pycache__/blackwell_matmul_mgpu.cpython-312.pyc,,
|
|
jax/experimental/pallas/ops/gpu/__pycache__/collective_matmul_mgpu.cpython-312.pyc,,
|
|
jax/experimental/pallas/ops/gpu/__pycache__/decode_attention.cpython-312.pyc,,
|
|
jax/experimental/pallas/ops/gpu/__pycache__/hopper_matmul_mgpu.cpython-312.pyc,,
|
|
jax/experimental/pallas/ops/gpu/__pycache__/layer_norm.cpython-312.pyc,,
|
|
jax/experimental/pallas/ops/gpu/__pycache__/paged_attention.cpython-312.pyc,,
|
|
jax/experimental/pallas/ops/gpu/__pycache__/ragged_dot_mgpu.cpython-312.pyc,,
|
|
jax/experimental/pallas/ops/gpu/__pycache__/rms_norm.cpython-312.pyc,,
|
|
jax/experimental/pallas/ops/gpu/__pycache__/softmax.cpython-312.pyc,,
|
|
jax/experimental/pallas/ops/gpu/attention.py,sha256=AkTo-_jOemqjz2iY6AyfaB8JoCpoENJYVzrwFgRP5Q4,22855
|
|
jax/experimental/pallas/ops/gpu/attention_mgpu.py,sha256=0ortsKohWQuUUvVD6XRWAqEnF3-dP1mJ7dHgcGiSI0A,36368
|
|
jax/experimental/pallas/ops/gpu/blackwell_matmul_mgpu.py,sha256=RpOf0o6s0eO_yc0nZ3iOMGsiUxCVLMubZqgGQy_QnRQ,11390
|
|
jax/experimental/pallas/ops/gpu/collective_matmul_mgpu.py,sha256=bbYkuebsO-jreAuNQ2DGQOtq8BizsTCF8fzcXiynGfk,8452
|
|
jax/experimental/pallas/ops/gpu/decode_attention.py,sha256=CKEX1sGsMcpcUmB7f7xP85Z7I0xYIzxmWgO4lrdIFNQ,17020
|
|
jax/experimental/pallas/ops/gpu/hopper_matmul_mgpu.py,sha256=OXdIonDXLm44UMgBzokuARo_rLe5vJN6fgwWLyPB4MM,6401
|
|
jax/experimental/pallas/ops/gpu/layer_norm.py,sha256=60rrgVs0gTYSRTS-bllzQBvC75eSAEF0R1fSdiwIDuY,11257
|
|
jax/experimental/pallas/ops/gpu/paged_attention.py,sha256=MKaQ0GKMrhf2xYGvzvHQdM1AxeyLHc8uFGnCZHNmU3k,16191
|
|
jax/experimental/pallas/ops/gpu/ragged_dot_mgpu.py,sha256=QJ-IxLnuPf8QSDQVtNMIEDqHT6z5FZRtY5ypyYueBNE,11083
|
|
jax/experimental/pallas/ops/gpu/rms_norm.py,sha256=sOi5t-CmvkA18v6dGGr27IbbYmqroUEH6NliEcERalg,10129
|
|
jax/experimental/pallas/ops/gpu/softmax.py,sha256=3E-coLRFVOYWyAcC6uLhIodrhfbl8wa68rIAEYz9mBM,2766
|
|
jax/experimental/pallas/ops/tpu/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
|
|
jax/experimental/pallas/ops/tpu/__pycache__/__init__.cpython-312.pyc,,
|
|
jax/experimental/pallas/ops/tpu/__pycache__/all_gather.cpython-312.pyc,,
|
|
jax/experimental/pallas/ops/tpu/__pycache__/example_kernel.cpython-312.pyc,,
|
|
jax/experimental/pallas/ops/tpu/__pycache__/flash_attention.cpython-312.pyc,,
|
|
jax/experimental/pallas/ops/tpu/__pycache__/matmul.cpython-312.pyc,,
|
|
jax/experimental/pallas/ops/tpu/all_gather.py,sha256=9XXcBpb_pXZGn8EOa-daiO_EJecOLdBtx9QCxCowEL8,5573
|
|
jax/experimental/pallas/ops/tpu/example_kernel.py,sha256=jzsyvU-Lk0PIorE9XCw2NDdqdp8Cn4bqsSe76DBxeEM,802
|
|
jax/experimental/pallas/ops/tpu/flash_attention.py,sha256=Zc2UjAlPz1LwugHz_O_3xZzBx3gG-zi0bIH3qo3kz3w,49855
|
|
jax/experimental/pallas/ops/tpu/matmul.py,sha256=x-2MCTChhr6BwQMQXzKcRT8PDVQ0ZvPjGxYOCaAgb7g,2717
|
|
jax/experimental/pallas/ops/tpu/megablox/__init__.py,sha256=HRWx7FtM9_17OLitbxwIZp-A9RM61uSL6QAcBGJyw_w,650
|
|
jax/experimental/pallas/ops/tpu/megablox/__pycache__/__init__.cpython-312.pyc,,
|
|
jax/experimental/pallas/ops/tpu/megablox/__pycache__/common.cpython-312.pyc,,
|
|
jax/experimental/pallas/ops/tpu/megablox/__pycache__/gmm.cpython-312.pyc,,
|
|
jax/experimental/pallas/ops/tpu/megablox/__pycache__/ops.cpython-312.pyc,,
|
|
jax/experimental/pallas/ops/tpu/megablox/common.py,sha256=9zg98VzHJh5l4Sf_SKly1nrXFWeWeyKoe7ZSPn6Ht38,1969
|
|
jax/experimental/pallas/ops/tpu/megablox/gmm.py,sha256=PnH6tp1yv31d5QPtedi7oI7sBgqu0iQ2Pm2lJ4EzqTA,27243
|
|
jax/experimental/pallas/ops/tpu/megablox/ops.py,sha256=4Xg3O0bnP5HTWsawS0hAm8mF9HTvRHVK65RlopkA1rw,2903
|
|
jax/experimental/pallas/ops/tpu/paged_attention/__init__.py,sha256=Q6gFLakYVRJDRxxvqq6vmtDYup-edUHZy54j9194JfY,700
|
|
jax/experimental/pallas/ops/tpu/paged_attention/__pycache__/__init__.cpython-312.pyc,,
|
|
jax/experimental/pallas/ops/tpu/paged_attention/__pycache__/paged_attention_kernel.cpython-312.pyc,,
|
|
jax/experimental/pallas/ops/tpu/paged_attention/__pycache__/quantization_utils.cpython-312.pyc,,
|
|
jax/experimental/pallas/ops/tpu/paged_attention/__pycache__/util.cpython-312.pyc,,
|
|
jax/experimental/pallas/ops/tpu/paged_attention/paged_attention_kernel.py,sha256=MRVFNdCdETdHNS1SAAXWV_fufPJHpMsMfSQhiEHKxFQ,22026
|
|
jax/experimental/pallas/ops/tpu/paged_attention/quantization_utils.py,sha256=5y0QHJo2yDsafF_lgeSO_izChspLHpVOvbFWZ-2d0ss,2556
|
|
jax/experimental/pallas/ops/tpu/paged_attention/util.py,sha256=t1_5K06BjSyQKaa-zqQnJDpW4dX1iQ-URfb_hEUJ53g,3254
|
|
jax/experimental/pallas/ops/tpu/ragged_paged_attention/__init__.py,sha256=XBTKcqySDW4Bl6lDQvhpu0PAJK0o_L3fWuRE359qy9s,1036
|
|
jax/experimental/pallas/ops/tpu/ragged_paged_attention/__pycache__/__init__.cpython-312.pyc,,
|
|
jax/experimental/pallas/ops/tpu/ragged_paged_attention/__pycache__/kernel.cpython-312.pyc,,
|
|
jax/experimental/pallas/ops/tpu/ragged_paged_attention/__pycache__/tuned_block_sizes.cpython-312.pyc,,
|
|
jax/experimental/pallas/ops/tpu/ragged_paged_attention/kernel.py,sha256=-yANCoLsDtpi1xh9nOGdOxRaxQwxi6heUdc7k3lMHgc,31879
|
|
jax/experimental/pallas/ops/tpu/ragged_paged_attention/tuned_block_sizes.py,sha256=vCdHiRElQ59pwcdbN9M8IztGbsfdyKka4kk0nGwrqrs,97758
|
|
jax/experimental/pallas/ops/tpu/random/__init__.py,sha256=gEMCMS8LKTvoFm0t7khiAnQg78EDUuMI-fvTvw250eo,682
|
|
jax/experimental/pallas/ops/tpu/random/__pycache__/__init__.cpython-312.pyc,,
|
|
jax/experimental/pallas/ops/tpu/random/__pycache__/philox.cpython-312.pyc,,
|
|
jax/experimental/pallas/ops/tpu/random/__pycache__/prng_utils.cpython-312.pyc,,
|
|
jax/experimental/pallas/ops/tpu/random/__pycache__/threefry.cpython-312.pyc,,
|
|
jax/experimental/pallas/ops/tpu/random/philox.py,sha256=Zi9svyA6bUoSKNiAjdyYTqj1ALDgcLR4_fxiPeW1PvI,7494
|
|
jax/experimental/pallas/ops/tpu/random/prng_utils.py,sha256=5r3pl293zG_WWLzWkY5WKP-lxqFmLxwGx9hWu9OHiX4,1780
|
|
jax/experimental/pallas/ops/tpu/random/threefry.py,sha256=bSCSKbj8BCKnHjTeedOCBPfEDURYl1KeD9f4tsznPDs,4339
|
|
jax/experimental/pallas/ops/tpu/splash_attention/__init__.py,sha256=8y556G1Vpmglvw8FbNzeB4x6QK6PLkKUBnMsL4vleNs,2738
|
|
jax/experimental/pallas/ops/tpu/splash_attention/__pycache__/__init__.cpython-312.pyc,,
|
|
jax/experimental/pallas/ops/tpu/splash_attention/__pycache__/splash_attention_kernel.cpython-312.pyc,,
|
|
jax/experimental/pallas/ops/tpu/splash_attention/__pycache__/splash_attention_mask.cpython-312.pyc,,
|
|
jax/experimental/pallas/ops/tpu/splash_attention/__pycache__/splash_attention_mask_info.cpython-312.pyc,,
|
|
jax/experimental/pallas/ops/tpu/splash_attention/splash_attention_kernel.py,sha256=xwro3bPN1U4-h04UgIFH0KpuoTv3thYAsSHDOz3oW_U,78075
|
|
jax/experimental/pallas/ops/tpu/splash_attention/splash_attention_mask.py,sha256=ncsRlbb-oHneOkZqkSpOB6r7V0p_lZY60zBEsar364c,16307
|
|
jax/experimental/pallas/ops/tpu/splash_attention/splash_attention_mask_info.py,sha256=6h4bS38tKAB1rKbcuFK1XBH6R14HSzGRtr3lOHrD3j4,40291
|
|
jax/experimental/pallas/tpu.py,sha256=uocrR4XaR2kyoXS6UzmuoQ9_aEejcTRWQl2BNlblN6A,6295
|
|
jax/experimental/pallas/triton.py,sha256=fl9lVt8THVyqTW25XlQXJQg3ev-Hjb7k1MpLRwL7W4M,2330
|
|
jax/experimental/pjit.py,sha256=HC4rP1t3qDoM8H6OqKrA5uZiujVaAFOtNrYU0stLpLM,729
|
|
jax/experimental/profiler.py,sha256=Rxlgbb4TPJsfeWXwLYG58gW0PCqBjDVWswlCCLr7fBg,1413
|
|
jax/experimental/rnn.py,sha256=betLPzLUoJIU919CJPaFgGub_t6fHkGY8WYFeI_5jXs,20096
|
|
jax/experimental/roofline/__init__.py,sha256=ftyNH95ANQHFYSq9hsweKfJNP47sfipMm-3XFsr32yw,1260
|
|
jax/experimental/roofline/__pycache__/__init__.cpython-312.pyc,,
|
|
jax/experimental/roofline/__pycache__/roofline.cpython-312.pyc,,
|
|
jax/experimental/roofline/__pycache__/rooflines.cpython-312.pyc,,
|
|
jax/experimental/roofline/roofline.py,sha256=_kybhMF2X_GBLAZqTNmnrBRyRTuQxQWteBTlRoEon0c,11804
|
|
jax/experimental/roofline/rooflines.py,sha256=xeTOa2AqYqIJoAgBcZ_DI3f4LeREXYcdJMmQeY6zBE4,27210
|
|
jax/experimental/scheduling_groups.py,sha256=zquCRQmjxa8unAh0TczCpM5mXfBGPlz7PncbEU82YuQ,2258
|
|
jax/experimental/serialize_executable.py,sha256=T_hxV_VhKZy--jmRdDCN1ezF2_YsEcwESTMqAMFYLJI,4632
|
|
jax/experimental/shard_alike.py,sha256=dbrhVCnWEX39TsQYRyGoClFByDmqtmDlKGtOmlT-m2M,646
|
|
jax/experimental/shard_map.py,sha256=zkBPC3rKW787yjzwqz0vjc_2GHpRI18CuK_ocmo5rlg,4427
|
|
jax/experimental/source_mapper/__init__.py,sha256=g5b_71QFD84RscRTmCDm7l_TfUYkcW6gZuRd8rgFg9Q,1698
|
|
jax/experimental/source_mapper/__pycache__/__init__.cpython-312.pyc,,
|
|
jax/experimental/source_mapper/__pycache__/common.cpython-312.pyc,,
|
|
jax/experimental/source_mapper/__pycache__/generate_map.cpython-312.pyc,,
|
|
jax/experimental/source_mapper/__pycache__/hlo.cpython-312.pyc,,
|
|
jax/experimental/source_mapper/__pycache__/jaxpr.cpython-312.pyc,,
|
|
jax/experimental/source_mapper/__pycache__/mlir.cpython-312.pyc,,
|
|
jax/experimental/source_mapper/common.py,sha256=s58vJuszNzoWy-YmpfnuL-noc9zXsEu8_6H7iYwC3pk,2508
|
|
jax/experimental/source_mapper/generate_map.py,sha256=k0SrNKPQMAG3fgc_Pw30Vhn2u92nTkt-BbjyhVDs52Q,2129
|
|
jax/experimental/source_mapper/hlo.py,sha256=LbDbFaXXyqI4ouMR8ur9ywIFk5t4-K-eUMIIldVYm4w,4124
|
|
jax/experimental/source_mapper/jaxpr.py,sha256=PXa9zMzP2_fD_0vxo8Za-mw21pXp2Aao6IQkQIqmoJU,2595
|
|
jax/experimental/source_mapper/mlir.py,sha256=U2YSWC9tp_DlnsrRzuzsAU7W2PY6ZwtWedHiuMSTQ5M,4686
|
|
jax/experimental/sparse/__init__.py,sha256=iYfDK0sdq3ErijcE7Va7cRjEHjzuJ3ykCwnHSYxNMPs,11186
|
|
jax/experimental/sparse/__pycache__/__init__.cpython-312.pyc,,
|
|
jax/experimental/sparse/__pycache__/_base.cpython-312.pyc,,
|
|
jax/experimental/sparse/__pycache__/_lowerings.cpython-312.pyc,,
|
|
jax/experimental/sparse/__pycache__/ad.cpython-312.pyc,,
|
|
jax/experimental/sparse/__pycache__/api.cpython-312.pyc,,
|
|
jax/experimental/sparse/__pycache__/bcoo.cpython-312.pyc,,
|
|
jax/experimental/sparse/__pycache__/bcsr.cpython-312.pyc,,
|
|
jax/experimental/sparse/__pycache__/coo.cpython-312.pyc,,
|
|
jax/experimental/sparse/__pycache__/csr.cpython-312.pyc,,
|
|
jax/experimental/sparse/__pycache__/linalg.cpython-312.pyc,,
|
|
jax/experimental/sparse/__pycache__/nm.cpython-312.pyc,,
|
|
jax/experimental/sparse/__pycache__/random.cpython-312.pyc,,
|
|
jax/experimental/sparse/__pycache__/test_util.cpython-312.pyc,,
|
|
jax/experimental/sparse/__pycache__/transform.cpython-312.pyc,,
|
|
jax/experimental/sparse/__pycache__/util.cpython-312.pyc,,
|
|
jax/experimental/sparse/_base.py,sha256=qyiBgPa12uwS1nLpSDw8dTD1VEq0v9P2_CGTNyPCA_g,3239
|
|
jax/experimental/sparse/_lowerings.py,sha256=nNviA0Ed0Dq-J_mRF1us4DQ4GUP69Gu3WBFzuSbLlWI,13334
|
|
jax/experimental/sparse/ad.py,sha256=zJLXiDufoAVcLTHdFitEXdbQUM_XtDiJoIktwDVuNZU,7789
|
|
jax/experimental/sparse/api.py,sha256=DrqEZmlIr6n6kpJjNli9J_rqMX3UO3sPPW7E8_t-u4g,6596
|
|
jax/experimental/sparse/bcoo.py,sha256=9c-NP6_P6BtkodeXKh6Jwu5X7lzkEPbInAYqk9s6Wxg,129568
|
|
jax/experimental/sparse/bcsr.py,sha256=qSvCId7ZjWHNZNDVpEPdc_dTnB9446V5xsMiIFdpAxs,41314
|
|
jax/experimental/sparse/coo.py,sha256=_R22aml359eVXXBA_3BP4mQfrJAfmdCtD_BzCDLy2vw,24165
|
|
jax/experimental/sparse/csr.py,sha256=dhOQNlU57XwMZom4n8VVfhc8t4TP1OxPcwtDurK0il0,24332
|
|
jax/experimental/sparse/linalg.py,sha256=tp_aBpb4Ffxdhp9o9ZT5ThVuCWDRLVLoraoRKRUG_EQ,23425
|
|
jax/experimental/sparse/nm.py,sha256=ypXA8WaigylNbaRqJRlfXyurJC00f41VF1jxqUAiI3I,8568
|
|
jax/experimental/sparse/random.py,sha256=VNhvzYvrFzk5KmK2FGEnD3KuImtUyyFIYAl_jefQUlc,3883
|
|
jax/experimental/sparse/test_util.py,sha256=hBw1SDgjVL2KkadjSdvIs5DcHqmdrtQHTHUCwj-MBfw,9433
|
|
jax/experimental/sparse/transform.py,sha256=VF9ddMVtfVfsJMREIGSjDS41BDWwSVVw3zuwHO8bmL8,37486
|
|
jax/experimental/sparse/util.py,sha256=zb8BrK9aG5DC9ubS1ZmjSKfLhnnGVKA5ydy9i0hKzA8,4337
|
|
jax/experimental/topologies.py,sha256=GmMJ3y4gXhgtBMUwLBWMR6oPtQvQmVkXqFpcE6LgiH4,2118
|
|
jax/experimental/transfer.py,sha256=E8dFkOjF0jnpEYk6--VEdnozMEcC-ntdSG-GkKln9eI,2327
|
|
jax/experimental/x64_context.py,sha256=XXTsGH7SavrMuT1hXzThT8f04SAFWdrhjZiyfEAaqVU,2394
|
|
jax/experimental/xla_metadata.py,sha256=YoNAzPbdARYfGzGTUVoxTfHG9dDrobUvWaPI4uvHRJk,658
|
|
jax/export.py,sha256=OzLiZq2zwjC4gDUs-3VcHVFF5IUZtIhMaaTdnV89qIA,1537
|
|
jax/extend/__init__.py,sha256=POZ0JwUu9B2AtJDi58kt2P3jjV7M2fRv5303qj5UNno,2057
|
|
jax/extend/__pycache__/__init__.cpython-312.pyc,,
|
|
jax/extend/__pycache__/backend.cpython-312.pyc,,
|
|
jax/extend/__pycache__/ffi.cpython-312.pyc,,
|
|
jax/extend/__pycache__/ifrt_programs.cpython-312.pyc,,
|
|
jax/extend/__pycache__/linear_util.cpython-312.pyc,,
|
|
jax/extend/__pycache__/random.cpython-312.pyc,,
|
|
jax/extend/__pycache__/sharding.cpython-312.pyc,,
|
|
jax/extend/__pycache__/source_info_util.cpython-312.pyc,,
|
|
jax/extend/backend.py,sha256=c5Q_B6MAdyfZw-fjQP6tz57iKeAVzkXue1adGvE0Itg,1311
|
|
jax/extend/core/__init__.py,sha256=o_q4S8jW_Tc0986SMqpjcifigZq-Ez2OhFZo6aBLwzM,1036
|
|
jax/extend/core/__pycache__/__init__.cpython-312.pyc,,
|
|
jax/extend/core/__pycache__/primitives.cpython-312.pyc,,
|
|
jax/extend/core/primitives.py,sha256=QC5OylzYu3orhWTEV05sD19fTfX3N2TBKSWI4P1Rhno,6215
|
|
jax/extend/ffi.py,sha256=9HTo5pYGummGwbjAikbQntYjkbcPnvqtm-l_XpG9PAE,1979
|
|
jax/extend/ifrt_programs.py,sha256=TCWvR-K8nHQ9PuUNoMWVAfEuYsutTNrefKwLg4AIRao,787
|
|
jax/extend/linear_util.py,sha256=02ixe_vuJbwdBvRNcpQGG4DtE_6cPbIpFCzpus8l1oE,1601
|
|
jax/extend/mlir/__init__.py,sha256=KNPC1NUExtJ6tuxXjORksUCo_oqJ_jxGS-VASRDfXg4,889
|
|
jax/extend/mlir/__pycache__/__init__.cpython-312.pyc,,
|
|
jax/extend/mlir/__pycache__/ir.cpython-312.pyc,,
|
|
jax/extend/mlir/__pycache__/passmanager.cpython-312.pyc,,
|
|
jax/extend/mlir/dialects/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
|
|
jax/extend/mlir/dialects/__pycache__/__init__.cpython-312.pyc,,
|
|
jax/extend/mlir/dialects/__pycache__/arith.cpython-312.pyc,,
|
|
jax/extend/mlir/dialects/__pycache__/builtin.cpython-312.pyc,,
|
|
jax/extend/mlir/dialects/__pycache__/chlo.cpython-312.pyc,,
|
|
jax/extend/mlir/dialects/__pycache__/func.cpython-312.pyc,,
|
|
jax/extend/mlir/dialects/__pycache__/math.cpython-312.pyc,,
|
|
jax/extend/mlir/dialects/__pycache__/memref.cpython-312.pyc,,
|
|
jax/extend/mlir/dialects/__pycache__/scf.cpython-312.pyc,,
|
|
jax/extend/mlir/dialects/__pycache__/sdy.cpython-312.pyc,,
|
|
jax/extend/mlir/dialects/__pycache__/sparse_tensor.cpython-312.pyc,,
|
|
jax/extend/mlir/dialects/__pycache__/stablehlo.cpython-312.pyc,,
|
|
jax/extend/mlir/dialects/__pycache__/vector.cpython-312.pyc,,
|
|
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=4dpcOKEHHkDh6cXdyEm2VGE9W-8701MD9egZmWDlkRM,641
|
|
jax/extend/mlir/dialects/sparse_tensor.py,sha256=TaFY58y273rRITr8WxSM7LB1-sTAa03dCAjH4F8ZWaE,651
|
|
jax/extend/mlir/dialects/stablehlo.py,sha256=AbJQc7zghgrJh-BgRB31L7iHrNLVaEk4AELejcq7QiE,647
|
|
jax/extend/mlir/dialects/vector.py,sha256=xP03nScJg-t7olD8_P8SiFGVnV6Qi0Fwn9uGK0MHOXI,644
|
|
jax/extend/mlir/ir.py,sha256=1IirtLROpBL2nVDNSlP_3c6u09xXaKLqBkzF1JtmGz4,631
|
|
jax/extend/mlir/passmanager.py,sha256=1rfuCocMdFlx6It95ynBCP-8JC5xEAEG7rBdLJ7SpFI,640
|
|
jax/extend/random.py,sha256=w1OM6RCcGwmzIr9bdGJJTNKIcMO4ypXbzmeEORlBHjQ,1083
|
|
jax/extend/sharding.py,sha256=GRpNqvTo7-_NhNk7j9JW3HXE8_WkDocmnj07hfFp60U,1327
|
|
jax/extend/source_info_util.py,sha256=IxZYjLDlwOH1uE3A_vc25dy25QZ6FABMb4RqrkiA5hU,1215
|
|
jax/ffi.py,sha256=mCokfYI69_WdtY4J1As0QT81e6afaV5RsszxrX8ZCbQ,1116
|
|
jax/flatten_util.py,sha256=jGwQUB80s2jr78XuEaT1_OP2OUyVm8Pk-yAQq-C1SsY,645
|
|
jax/image/__init__.py,sha256=cthOdx-HsZ6SgvW19tdNCDHmC4Proc8da1hY1hRztTI,1027
|
|
jax/image/__pycache__/__init__.cpython-312.pyc,,
|
|
jax/interpreters/__init__.py,sha256=HNgKpgKpz4svFyTlHE9wMrDp5iQtg6XeeUshYBOnRko,690
|
|
jax/interpreters/__pycache__/__init__.cpython-312.pyc,,
|
|
jax/interpreters/__pycache__/ad.cpython-312.pyc,,
|
|
jax/interpreters/__pycache__/batching.cpython-312.pyc,,
|
|
jax/interpreters/__pycache__/mlir.cpython-312.pyc,,
|
|
jax/interpreters/__pycache__/partial_eval.cpython-312.pyc,,
|
|
jax/interpreters/__pycache__/pxla.cpython-312.pyc,,
|
|
jax/interpreters/__pycache__/xla.cpython-312.pyc,,
|
|
jax/interpreters/ad.py,sha256=swDiOsDq7BEOTAU5_xoR_6tEA-AAn_9z9bLiVuAN3BA,6484
|
|
jax/interpreters/batching.py,sha256=PcEI3xdZc5SZDV233utGwfOzOWtTU6SNg8kSYvmDtRU,9370
|
|
jax/interpreters/mlir.py,sha256=Me25JYU_OfooxaNJvNM5hfUaUvhdRousUPIbRHkuzOM,3424
|
|
jax/interpreters/partial_eval.py,sha256=LwWxdnS8YvAyfioa3Nh1Q7j368y99eW1TM-AmT1UWfg,3629
|
|
jax/interpreters/pxla.py,sha256=VN0Mlb4IreJ9Tqez8i-QWn1EXVcrDWTLPeioTkeL-qE,1801
|
|
jax/interpreters/xla.py,sha256=osmN-q8KKeFqxjR8PAUYsClcmzzr5ihI_sTU85Zxupo,2282
|
|
jax/lax/__init__.py,sha256=KrBqJmCghEdv4tuLcbewdNYVw3NT_y7uVIeGaIQNVwQ,11814
|
|
jax/lax/__pycache__/__init__.cpython-312.pyc,,
|
|
jax/lax/__pycache__/linalg.cpython-312.pyc,,
|
|
jax/lax/linalg.py,sha256=I9ZzEhenoEzcBnfnU-jCjIspi-aOXHc5KJY3t_45uWs,1631
|
|
jax/lib/__init__.py,sha256=wy3yxhlsb9k2th9Dfsr6eS59L2qLIxH_IoTM4cPfx3M,774
|
|
jax/lib/__pycache__/__init__.cpython-312.pyc,,
|
|
jax/lib/__pycache__/xla_bridge.cpython-312.pyc,,
|
|
jax/lib/__pycache__/xla_client.cpython-312.pyc,,
|
|
jax/lib/__pycache__/xla_extension.cpython-312.pyc,,
|
|
jax/lib/xla_bridge.py,sha256=mvpOuk7R6zDCbduu6Nbj6Yqj7yNkQzo5sffAQ7_HAGM,1736
|
|
jax/lib/xla_client.py,sha256=J4LN9iZEgmUDgTU4QOEyjyqcj-WCYTvJGzHnKY0UpGc,3030
|
|
jax/lib/xla_extension.py,sha256=avF7i1oEmJCtFn3g965Us_NUUToLqYfPV3F8S-m46yk,4159
|
|
jax/memory.py,sha256=uONVdGEZa7XodzKxzpCjwZ17cX3a1eRcrGnltSq7w40,625
|
|
jax/monitoring.py,sha256=xWZF_CyFg7Vv7MYmIyb3JWNA-AwsvIm9oc-MbGX4Bbw,1415
|
|
jax/nn/__init__.py,sha256=G9v8E63aJYY0rgGMX_bTRB52uBR8a0qEkDQn92zA_z8,1782
|
|
jax/nn/__init__.pyi,sha256=wsxbZkggbZaUHXNLSw4K-plsmY_fGUvcd1JCTp-FDe4,4329
|
|
jax/nn/__pycache__/__init__.cpython-312.pyc,,
|
|
jax/nn/__pycache__/initializers.cpython-312.pyc,,
|
|
jax/nn/initializers.py,sha256=finfAJmOEZdPIzY2roSIBTOmC4rKEW-rchpH9d6Dc2g,1469
|
|
jax/numpy/__init__.py,sha256=K4eMTRRaAEhlikBlOVl2eEJCkzCqflx-akbJBGAYFmc,12241
|
|
jax/numpy/__init__.pyi,sha256=8yru5ftyPcBBei0eLgqLBmFmiqTzXzhbpI8ms6J3rSA,44632
|
|
jax/numpy/__pycache__/__init__.cpython-312.pyc,,
|
|
jax/numpy/__pycache__/fft.cpython-312.pyc,,
|
|
jax/numpy/__pycache__/linalg.cpython-312.pyc,,
|
|
jax/numpy/fft.py,sha256=iNatBIYRSwpXquPSgtNRMe6-dhbKkv2yoH6UJ6PFoGM,1084
|
|
jax/numpy/linalg.py,sha256=bnzLmLpLYSDRIICkjCEsH1WmA7wNOdSKhmO25gPvTqI,1419
|
|
jax/ops/__init__.py,sha256=VIAczIKYAFWawHdNmLmptUYnFQiBwHwlsa4LpgGEbIc,870
|
|
jax/ops/__pycache__/__init__.cpython-312.pyc,,
|
|
jax/profiler.py,sha256=h7sMKCFMzMPSVlnmbXSj7w6bDt7Rf90S5U1x3WkMPCE,1276
|
|
jax/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
jax/random.py,sha256=-1oFQ0v1HFzwvFTipdl1JQVqsk8dXRJe2ZH4xTyU51g,9434
|
|
jax/ref.py,sha256=B7Mo2bbYYlOvcOtwzTnZH_dsaoxUUhcGdSu_Xh_f-CE,938
|
|
jax/scipy/__init__.py,sha256=DXszF0t3ZbfDeoFEP_GyBb0OOr6F-4yqsZzmen7pY58,1474
|
|
jax/scipy/__pycache__/__init__.cpython-312.pyc,,
|
|
jax/scipy/__pycache__/fft.cpython-312.pyc,,
|
|
jax/scipy/__pycache__/integrate.cpython-312.pyc,,
|
|
jax/scipy/__pycache__/linalg.cpython-312.pyc,,
|
|
jax/scipy/__pycache__/ndimage.cpython-312.pyc,,
|
|
jax/scipy/__pycache__/signal.cpython-312.pyc,,
|
|
jax/scipy/__pycache__/special.cpython-312.pyc,,
|
|
jax/scipy/cluster/__init__.py,sha256=sgIbEEx62OubzDO36E3EkqjbYY_3qsErPOTQE6lsj5E,768
|
|
jax/scipy/cluster/__pycache__/__init__.cpython-312.pyc,,
|
|
jax/scipy/cluster/__pycache__/vq.cpython-312.pyc,,
|
|
jax/scipy/cluster/vq.py,sha256=lkLSlkWFi6TLjwuzTX7hVSiC1IkCYd3OWgn8YtIAJd0,776
|
|
jax/scipy/fft.py,sha256=PVKQ4r-vl4J9Ct6LrhZWZ7jMF7frIXggiO98Bh0oVIE,809
|
|
jax/scipy/integrate.py,sha256=H9iq7YKgsZQ1d0SUvGcgPnEgD1TBdWJXHYjfUOGzT8Y,777
|
|
jax/scipy/interpolate/__init__.py,sha256=RI5oUXPC-8tKmYURnnrnA-e1LDbnNVOVVqkGSpagEBM,760
|
|
jax/scipy/interpolate/__pycache__/__init__.cpython-312.pyc,,
|
|
jax/scipy/linalg.py,sha256=GXC1BgLMztvwRMxddJhvJS3obG81HLQ0OJFVt_Iv4V8,1370
|
|
jax/scipy/ndimage.py,sha256=ei-ZfRiDXrlY4nYC1SFL7nnGHFrC_emD_NAicGhH4vU,788
|
|
jax/scipy/optimize/__init__.py,sha256=cAw1rT1e0Mu-22vBUuJb6Q-CQF8fbIUBlKYpynB7e2s,822
|
|
jax/scipy/optimize/__pycache__/__init__.cpython-312.pyc,,
|
|
jax/scipy/signal.py,sha256=nW4h017jShZBLL3J2NMhLwlOv46wK1MhEPeKlPTj710,975
|
|
jax/scipy/sparse/__init__.py,sha256=ugltlihEjwgorQxgfziHC-03UeYy4JzdQmTxrt1cnTo,757
|
|
jax/scipy/sparse/__pycache__/__init__.cpython-312.pyc,,
|
|
jax/scipy/sparse/__pycache__/linalg.cpython-312.pyc,,
|
|
jax/scipy/sparse/linalg.py,sha256=wpjjS1G1UNISyhZVFoJMMgIkqs-L8ASdvtQAbvEBsuk,810
|
|
jax/scipy/spatial/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
|
|
jax/scipy/spatial/__pycache__/__init__.cpython-312.pyc,,
|
|
jax/scipy/spatial/__pycache__/transform.cpython-312.pyc,,
|
|
jax/scipy/spatial/transform.py,sha256=cm7lKi8aUepI3cOwTnjrldhFh7hBv89CR_m3pFj3r2o,802
|
|
jax/scipy/special.py,sha256=aTtD4vxM5PkdpbYZ_aFDkwX4aIHovWQw4WJkFlzlcxU,2605
|
|
jax/scipy/stats/__init__.py,sha256=_rg_z3SzFW5UNdLOwvm1RgTHyRWiu7LulOePjNqE0Mo,2093
|
|
jax/scipy/stats/__pycache__/__init__.cpython-312.pyc,,
|
|
jax/scipy/stats/__pycache__/bernoulli.cpython-312.pyc,,
|
|
jax/scipy/stats/__pycache__/beta.cpython-312.pyc,,
|
|
jax/scipy/stats/__pycache__/betabinom.cpython-312.pyc,,
|
|
jax/scipy/stats/__pycache__/binom.cpython-312.pyc,,
|
|
jax/scipy/stats/__pycache__/cauchy.cpython-312.pyc,,
|
|
jax/scipy/stats/__pycache__/chi2.cpython-312.pyc,,
|
|
jax/scipy/stats/__pycache__/dirichlet.cpython-312.pyc,,
|
|
jax/scipy/stats/__pycache__/expon.cpython-312.pyc,,
|
|
jax/scipy/stats/__pycache__/gamma.cpython-312.pyc,,
|
|
jax/scipy/stats/__pycache__/gennorm.cpython-312.pyc,,
|
|
jax/scipy/stats/__pycache__/geom.cpython-312.pyc,,
|
|
jax/scipy/stats/__pycache__/gumbel_l.cpython-312.pyc,,
|
|
jax/scipy/stats/__pycache__/gumbel_r.cpython-312.pyc,,
|
|
jax/scipy/stats/__pycache__/laplace.cpython-312.pyc,,
|
|
jax/scipy/stats/__pycache__/logistic.cpython-312.pyc,,
|
|
jax/scipy/stats/__pycache__/multinomial.cpython-312.pyc,,
|
|
jax/scipy/stats/__pycache__/multivariate_normal.cpython-312.pyc,,
|
|
jax/scipy/stats/__pycache__/nbinom.cpython-312.pyc,,
|
|
jax/scipy/stats/__pycache__/norm.cpython-312.pyc,,
|
|
jax/scipy/stats/__pycache__/pareto.cpython-312.pyc,,
|
|
jax/scipy/stats/__pycache__/poisson.cpython-312.pyc,,
|
|
jax/scipy/stats/__pycache__/t.cpython-312.pyc,,
|
|
jax/scipy/stats/__pycache__/truncnorm.cpython-312.pyc,,
|
|
jax/scipy/stats/__pycache__/uniform.cpython-312.pyc,,
|
|
jax/scipy/stats/__pycache__/vonmises.cpython-312.pyc,,
|
|
jax/scipy/stats/__pycache__/wrapcauchy.cpython-312.pyc,,
|
|
jax/scipy/stats/bernoulli.py,sha256=llkwAlvFH0qNU1BHdxn9CXXOBAPg0ffnbW_WQQ4LKVg,819
|
|
jax/scipy/stats/beta.py,sha256=5UIf6_JtqVZfAl1dDc2N3K8SkG0OV1bBkC3vWpY8-Hs,851
|
|
jax/scipy/stats/betabinom.py,sha256=5T6965_jHQxG4oR2YocPQRKcDo5AGxkhIyrk2iz-Kk0,792
|
|
jax/scipy/stats/binom.py,sha256=XGO5jDk3nRRgGwHQU6KEl9_kQ5X2qF-2TLgvm3ZUvvs,659
|
|
jax/scipy/stats/cauchy.py,sha256=MfAMnK9eG2wcutA5FVAyhU6iVYSSNgzI1XFT1xtNIfg,881
|
|
jax/scipy/stats/chi2.py,sha256=f-6i-0twBiOrak26tc7bEnWgOODA3iFwEFgi_An9udo,851
|
|
jax/scipy/stats/dirichlet.py,sha256=FUHSEmXKmVUneXuxgaIZM28s6GxlusbymdRYEPnDiUE,792
|
|
jax/scipy/stats/expon.py,sha256=YClw5PcqAXgq3VD4ukZ19HfTmRCMtrGk6E2BpXb6U9E,866
|
|
jax/scipy/stats/gamma.py,sha256=TvP4BE8MROMxQL6YNDQIHSaZCfxumHkjGbCXw9O2iUo,852
|
|
jax/scipy/stats/gennorm.py,sha256=hqwz7ueAgdtzs1v0LHTTro2PLKRDjWxTFuu9jXwXOyI,804
|
|
jax/scipy/stats/geom.py,sha256=pdp2tCdXOa3CUWyKJCyrFrMR7TBX0La5tVI4ALAETI4,787
|
|
jax/scipy/stats/gumbel_l.py,sha256=uGI-yIa7ymknKuVsuol49H-a8ylid993coiWq6mnNuM,868
|
|
jax/scipy/stats/gumbel_r.py,sha256=fcbp-XJbmb3DlIDS8BTfKmXhf0CDh3EwMFzmWQ0VMo4,868
|
|
jax/scipy/stats/laplace.py,sha256=7Lap3E3_neHYpG5mUFlYxZnwAT1J8mTrNPnn8BSEGrA,804
|
|
jax/scipy/stats/logistic.py,sha256=IoHe_TMB0CH5EMAXjmqRMP8LjR4lTS_x9Zf4K7K5e2g,845
|
|
jax/scipy/stats/multinomial.py,sha256=hfRYQL_GeszF4sJXx5Y416JgXoXVw5vxtQIt1NmkCDU,794
|
|
jax/scipy/stats/multivariate_normal.py,sha256=YYqtLLZmjsLWeCIoDpxUk-rHde-g9unfo6ofg44nSBQ,802
|
|
jax/scipy/stats/nbinom.py,sha256=hWMMt7D_rLRtiGjlpHcbOXJF03BfA3v9Cu972twYKCY,660
|
|
jax/scipy/stats/norm.py,sha256=qNMz53OSiWOZGt6TZkJZvWAFocdUMUu-DsysM9kg2K4,879
|
|
jax/scipy/stats/pareto.py,sha256=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/sharding.py,sha256=TTVlZKOdFNWr3hArggbHPyv7g_zS8RMn0_tmujWI318,1295
|
|
jax/stages.py,sha256=KOQqCPRCnLKryQUrTRPVRiNuZqyV4chGi_oY2t3HpQ4,1286
|
|
jax/test_util.py,sha256=DVJAPxzZMKdM01Nhx7s2kPe5mcdv8ZL8a_37sgyZPUs,835
|
|
jax/tools/__init__.py,sha256=BTyMF0yh0M75AsNwF_0NSd5gHnIVpxhsLznsWkxu3XQ,581
|
|
jax/tools/__pycache__/__init__.cpython-312.pyc,,
|
|
jax/tools/__pycache__/colab_tpu.cpython-312.pyc,,
|
|
jax/tools/__pycache__/jax_to_ir.cpython-312.pyc,,
|
|
jax/tools/__pycache__/pgo_nsys_converter.cpython-312.pyc,,
|
|
jax/tools/colab_tpu.py,sha256=IPEQ1HZB4xW0MRSPGDeFAw9GoMFaphNQ0kLRIwGypmo,890
|
|
jax/tools/jax_to_ir.py,sha256=fFvvhbWBsHJel1cBJJOBhLeacpk-OJVJzRE-hZHxK90,8711
|
|
jax/tools/pgo_nsys_converter.py,sha256=zA_yA3Yzw4BJRWxPrPjkfLiG8uGn2JFbsraSkvnINOc,3411
|
|
jax/tree.py,sha256=UMACxhh3xfcgrO6h9QE00cqAl4gMzyRkweFfBxPnlAQ,1152
|
|
jax/tree_util.py,sha256=iRmZsGa44S8xWIzqv8KXvdUfX04FgnU2rBV5SnQ6pac,3661
|
|
jax/typing.py,sha256=zijMsLWzFOqn-u8__DbGxxI7MyjePxpUwU5nFE_P78Y,3319
|
|
jax/util.py,sha256=HGjj7010lZQHbIpx-qvh4RnJFXgH_Ds3PF-h2jVtyt8,3221
|
|
jax/version.py,sha256=SSv1oZZxNfLIY4UL-esLUtg_8gBGfvjtL-6IyrFdDg4,6725
|