| Path |
Digest |
Size |
| jax/__init__.py |
sha256=jTnOMnMx0fc7hQ7gp5nMzktsV13yuAGAv_PorOFv4aQ
|
10056 |
| jax/ad_checkpoint.py |
sha256=jnNhXa-FiezkCFP9BlCul194Pz11olQPdZEeoCJQi-U
|
932 |
| jax/api_util.py |
sha256=Jd8iVlf7i8crDpyvCg7OTaumZjtGvTIO9pUHXlXmQsU
|
921 |
| jax/cloud_tpu_init.py |
sha256=r_UB06-qStzlo7BWtH7eZ7OMl3ayrG_CQXWQJJwWsYs
|
651 |
| jax/collect_profile.py |
sha256=44nqGZ4fjcCEEqVGpn6Z8_nDuGFLRtTfUyar9YfGKHk
|
4730 |
| jax/core.py |
sha256=5jk3AE6GFVpacctYdAD_LyaBFUkVOpwqMNra54Rl7qc
|
12270 |
| jax/custom_batching.py |
sha256=MEoypZfGgYqvZrBKtnvL5sQjl8YrjVm11hiFWoTCtV8
|
691 |
| jax/custom_derivatives.py |
sha256=-yrA5kDE7gDErJFdartoj43ELYeogp6PI8kjP8p_TLk
|
1453 |
| jax/custom_transpose.py |
sha256=vGj3GjRJQpxl-MhvYztIedauNBiWMj9nD_aJHQiNaYc
|
664 |
| jax/debug.py |
sha256=HXdNZlaM5LxtEYS6fsH3J4LRgiDVL-5W6AXcimbFS5o
|
1083 |
| jax/distributed.py |
sha256=D9-SeP4MWzq9jw2itQrlm10cn5BtR-9Zjn2YGzEXBDU
|
673 |
| jax/dlpack.py |
sha256=TWzMqAIedhARYUrnreJ35tEYt0SepzrpvTQtmnSdhEw
|
710 |
| jax/dtypes.py |
sha256=iaMlUUaiUH8zFsy5IJEff4s5QTu0EzD5msV-JYAyD6I
|
1188 |
| jax/errors.py |
sha256=MaZbeSxeRyWRMfq4cj6P0GA9UFykLgX3aURPAKhqbts
|
1363 |
| jax/export.py |
sha256=OzLiZq2zwjC4gDUs-3VcHVFF5IUZtIhMaaTdnV89qIA
|
1537 |
| jax/flatten_util.py |
sha256=jGwQUB80s2jr78XuEaT1_OP2OUyVm8Pk-yAQq-C1SsY
|
645 |
| jax/monitoring.py |
sha256=9JRDF2xgMh8NcGYdk79ytPhRGQ_CCQTJHX7g72jjxWs
|
1191 |
| jax/profiler.py |
sha256=w7VxxnWqxWNV5LRSvIofTb11I4ndaUQKptxjaUq3P4c
|
1119 |
| jax/py.typed |
sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU
|
0 |
| jax/random.py |
sha256=N3MzWsmgxQtps5khNgCzqGY5Ix7u4jfXLfgzWCteSp4
|
9420 |
| jax/sharding.py |
sha256=2Id3LLKs6Ms3Ou8xzUZ1ZbWqkfSMv4S_bAqxfDOxz60
|
1624 |
| jax/stages.py |
sha256=lcGVsE8R-t3t5Pb6BWLzNelU-_USwzQ8Qj8bgMdM218
|
1314 |
| jax/test_util.py |
sha256=DVJAPxzZMKdM01Nhx7s2kPe5mcdv8ZL8a_37sgyZPUs
|
835 |
| jax/tree.py |
sha256=tI9yJ2PIEhAFiSBbVKnxV1Wfly6mb3b3mJXECzugJs0
|
1078 |
| jax/tree_util.py |
sha256=ZCuOkjx2V6A0EVV3e9sp7JtvhI5sxHF7u1-Vqpak-es
|
3125 |
| jax/typing.py |
sha256=CrQpi9UTB9MYfg-zEsqP3n2qe28hlY_QEsgY-9mrtAs
|
3331 |
| jax/util.py |
sha256=OXU0bu3n2-y901BwEhxd1CvOeGoEBaXPdvYaVNHOT3E
|
1135 |
| jax/version.py |
sha256=rL6BkfTHqaI2rthtfq52W_uZtu_GIcjTV9qsyLLuvlw
|
6013 |
| jax/_src/__init__.py |
sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg
|
581 |
| jax/_src/abstract_arrays.py |
sha256=7EJaMMmZxb02biU7F1vex5stEueRVCvjWdJAJigWHVk
|
3597 |
| jax/_src/ad_checkpoint.py |
sha256=1zAeh1qQVRYVHZVNWMdc63EpkVX9ecDgR-Ptua7L5R0
|
38949 |
| jax/_src/ad_util.py |
sha256=n14jlpCC3f7itcVEL8ijSM_C4ofyB8wbKKTzwWzxbr0
|
4164 |
| jax/_src/api.py |
sha256=MkZoRI0NUqREEAM_CIiwjPLtpgRrUN7FBDIUQJJ_42Y
|
120975 |
| jax/_src/api_util.py |
sha256=CMWyhLPVRn663VCot8ih93IW5MZsh0l2nPbTwJo6CeU
|
28283 |
| jax/_src/array.py |
sha256=ZJQZjl6cvfk_5lZ-PudyMZaQPX1WuexV1A00Vyhqs8s
|
46641 |
| jax/_src/basearray.py |
sha256=12hYSzFDsq0LVpPX8nQ-gH5mLE253PBWRSnYvQsuVwQ
|
6503 |
| jax/_src/basearray.pyi |
sha256=F8VxJtn9jB6iGzI-2B_kOedk1Pullmmyl21azn_k-CE
|
13022 |
| jax/_src/blocked_sampler.py |
sha256=rYk-BNwg0fIXWqgMXG5v3kjJw66Bod7vjNSQTa8atuw
|
5666 |
| jax/_src/cache_key.py |
sha256=8ytVe0xpceRekMKyEUxC0T3GaEKAv_JgreT7xPuCB5M
|
12883 |
| jax/_src/callback.py |
sha256=0sbAJzJEOBSOi_QXnjCTl2AQ7P0K07jOXx-YlsfeBPM
|
23789 |
| jax/_src/checkify.py |
sha256=f2ZHU_z35_oUvxwP2wIsriOvoA5y5t5g_0z06b6Okts
|
56181 |
| jax/_src/cloud_tpu_init.py |
sha256=SVyU-xfB2YxNfZtlv2ebpEgoSnp64WUlnyJuutz_oag
|
3519 |
| jax/_src/compilation_cache.py |
sha256=m2pr7zVMcYgCY2UsyaajGp5jTEZ1tztrh4YKCiYeZM0
|
11226 |
| jax/_src/compilation_cache_interface.py |
sha256=T-39MQ5ygMI96IHmJIyBwUobcQwMTXkANIGpCyr0eoo
|
887 |
| jax/_src/compiler.py |
sha256=-_sZy3A5jMBKXX_jRbzCMyjMoSwfVoPDS2faoDhegg0
|
27795 |
| jax/_src/compute_on.py |
sha256=GYS3AbPWjfRgw9SsCFa7R7q2dpTluQeejqTjnQ3iVTk
|
2074 |
| jax/_src/config.py |
sha256=JzB74t5HbEUOAEQjFMPLvZYjxKmhh3zEBVkorTsmZTg
|
66668 |
| jax/_src/core.py |
sha256=TVoSkLmSWWgJtuw3g4O5NbABxllEXXfdo3M6AIAtzTI
|
109795 |
| jax/_src/custom_api_util.py |
sha256=uv6uyrXU1BHGzKuAe498W2PD9bcWt1PEa-8UAqr9Zaw
|
876 |
| jax/_src/custom_batching.py |
sha256=JAq_Q4N-MOvT5F22UOHiUQgTuUMHnKqR5zi-61I1VuQ
|
14287 |
| jax/_src/custom_derivatives.py |
sha256=oMz7MVBMeFGNlrRsBlz5kztWQoi6gyQZlSTg2rlUu40
|
68827 |
| jax/_src/custom_partitioning.py |
sha256=uzggcd45rR67Qxg6hZyiVEgaNaxSEvCjEtzRjGJUUdg
|
22724 |
| jax/_src/custom_partitioning_sharding_rule.py |
sha256=tNtA2zjFAWGq_ydHPOTL_onmmb0MTJddQPRLpKmoQSA
|
15889 |
| jax/_src/custom_transpose.py |
sha256=8UOwMc39nsG90p1jDBw50_QZ8YX5a_qFP1CsFYFoV0c
|
7788 |
| jax/_src/debugging.py |
sha256=_XY-rTEjo_-dARgWAj6R_iE4WPM6wlzYlfEyOwQDpZc
|
25904 |
| jax/_src/deprecations.py |
sha256=9-P_ueCZHojeESUuEWL6mq89DIAXyCmmA_nEfRjT0nw
|
4706 |
| jax/_src/dispatch.py |
sha256=RKyj_SSODRDBzy_vI_PvORwEDEBTOO1vrz5gKVG2u1U
|
22959 |
| jax/_src/distributed.py |
sha256=MDRG-sqa2zYnxqpGo7ShBFmNgwWXVXvSgyGHiHh5bEU
|
11878 |
| jax/_src/dlpack.py |
sha256=pzQdxNalm0JLOOZJyHaLYSAT5DfP_UTPHFl5lBSX9SU
|
11544 |
| jax/_src/dtypes.py |
sha256=mYDaUESQcizEMymUrLmpNd-yghShXBYkmxeJuMlh_Tc
|
33858 |
| jax/_src/earray.py |
sha256=ziY31_lJX_TmgpC87I1pq0xFZAv2BqZsEt3Kbe6XDBc
|
4500 |
| jax/_src/effects.py |
sha256=ZqHKVxNurnN2g3-PJNdvYeut8_XEjNul8h_Yjojx18Q
|
4682 |
| jax/_src/environment_info.py |
sha256=TLwjyscnAh8rrQWePe1b14-IeSc-z-XP51Y92dMQeGI
|
1873 |
| jax/_src/errors.py |
sha256=s3zDLKcfvcCVSrJgNON0YF7bb7RB8Gp-xWinVHcvsNU
|
24741 |
| jax/_src/flatten_util.py |
sha256=T8p7oExtun3ddhh8fn4XL5_TzZEh6ChM_qiZhmcPaOk
|
3554 |
| jax/_src/hardware_utils.py |
sha256=L03Dapnam8MdnSyBGq2d8nio5gPF51xd_Ufxw2pVjCE
|
1728 |
| jax/_src/jaxpr_util.py |
sha256=Jv3fcRlWFNUt-WQwHYfmLeXOSKAYka9RKEwrkYl-BkQ
|
6439 |
| jax/_src/lax_reference.py |
sha256=SyNb2xHEY_1sa7KEnLkQ9PxFHwKKh3853dlLBksGpkM
|
18163 |
| jax/_src/layout.py |
sha256=Km3wPwPjWyBE37S4V0prIvE0FqI56Nq1u0TCVhuL3tE
|
5375 |
| jax/_src/lazy_loader.py |
sha256=G6czW2Uq6WRC1EjDiX6rslMu02cZUGRkKNrvIhXksxw
|
1842 |
| jax/_src/linear_util.py |
sha256=BE8f0Ni748OUXGS2R3kAJsK2e_pyzRW-YRgOp8ElvxQ
|
13284 |
| jax/_src/logging_config.py |
sha256=zKijn3gjyw21ljvWjTsBX1bE6AyIa3zHRQ15YvaDDmk
|
4452 |
| jax/_src/lru_cache.py |
sha256=4-A3WLeb-UhUxH7D5VrDUdjUl_BxahLXs8TvS4upO7s
|
6661 |
| jax/_src/mesh.py |
sha256=qMvyMkO0mkSJs_tnqzpRULjeH7z3sFy_iGrmyDfqZt4
|
16931 |
| jax/_src/mesh_utils.py |
sha256=AYtLuCPor_YRW3jAqS-NmiAKRCaoOmpDnOioLX7Le2U
|
32394 |
| jax/_src/monitoring.py |
sha256=ro92p2vr1PU4f0b7Ghk_kkJQBLoxmO9MQ1n3APyBfqk
|
3888 |
| jax/_src/op_shardings.py |
sha256=gsygmZDpugleC2AkgHEoq6JIDo-ACOK-O7BoJjmyWO8
|
3894 |
| jax/_src/partition_spec.py |
sha256=qnecyMIbJJjWABMiZXzjXu_3WSB3TFF5ZL5Kwitw7J0
|
2186 |
| jax/_src/path.py |
sha256=ot2lEjzaYVYCCGoOI8YhZYT01HgME-6dtGpmckXd41A
|
1452 |
| jax/_src/pickle_util.py |
sha256=gE8IrNWtg4hfu84T5rQ2DVAdpozh5xwSzAlRi7UIWcQ
|
2312 |
| jax/_src/pjit.py |
sha256=b7N1dZvcy6Dg4Yy0wfbxx9EzCChMR3QFPFkt1r2JWJw
|
125074 |
| jax/_src/pretty_printer.py |
sha256=mnGHSMkb8EaXh_1_9dsKOGh2XI_9dSaaAUvFJP-XdMM
|
15500 |
| jax/_src/prng.py |
sha256=DoEg6BFySzk4JYlPuqSHuCcy-E_hj2ENgPyNzZMroD8
|
44351 |
| jax/_src/profiler.py |
sha256=278hQdUi8B7FVXo-YKhY9jf1qXSaBNem50ZAeB6F4DI
|
16195 |
| jax/_src/public_test_util.py |
sha256=SZxqWnOEobpLnMpXA18FAFrhdU9YG0y-JvTZDTE2p_g
|
11152 |
| jax/_src/random.py |
sha256=-0fcrzyHs4w9dNgahL_hpTDou-5x6OuCA5kNhBpRsak
|
100275 |
| jax/_src/shard_alike.py |
sha256=qjvCzQLe_ef39MlT98hUv9WPtXkO5ctX8pntSYRpqfo
|
4028 |
| jax/_src/sharding.py |
sha256=frnT8rcK_erouiHIyySG20lFYlZXjfzmh3B9WtXEzMw
|
8102 |
| jax/_src/sharding_impls.py |
sha256=G5uASfu2WaOIQf2ckIlrbEk4rWcx2IYV1ieNRPX7rno
|
66232 |
| jax/_src/sharding_specs.py |
sha256=ql8VcvciCOMmrc2TXnb4khTatU_in3BOAc90QpJDMl0
|
8646 |
| jax/_src/source_info_util.py |
sha256=EMqYD1WALWmyEZ1BQu1lpmlHQdVZkWH2_kjNHcqnLTE
|
10140 |
| jax/_src/sourcemap.py |
sha256=5s4zTThhX6PZwgQ4v4UbYL4norg4uHipqgUGdHBn76Y
|
6640 |
| jax/_src/stages.py |
sha256=I1hobEQkEXwXQUexzjyE0UvqdCXm9ChdGQk5V4x0bXk
|
29132 |
| jax/_src/test_util.py |
sha256=1m7bMvo7WIwKI3hurjNk_zCQhNLaodk0S1661DhJZu0
|
75920 |
| jax/_src/tpu_custom_call.py |
sha256=PEl63Bxk7rUEdsIX_EQhRs3vAz2rhJxJS8qrDCGehik
|
26120 |
| jax/_src/traceback_util.py |
sha256=5DgCqqOxQ9JNrreoZx7u3ZglimkfIFxHT2tVrlnOcKo
|
9103 |
| jax/_src/tree.py |
sha256=SIwfh6UL_xEuEMq-ky4bmOmoq1LllYa0ZUhmh4l4-gM
|
12592 |
| jax/_src/tree_util.py |
sha256=PV5DtrAVgWa01770CSRr-D9O3BYqCzlBkc2goSdULF8
|
47329 |
| jax/_src/typing.py |
sha256=_yNwYCbn30QE4Ordkc5d4Y2bF2M78C5LhK42EggAetw
|
3287 |
| jax/_src/util.py |
sha256=CdeFeNtzV5-elqeSosT_OYbPfzk83dkZVu_QVl9uRGQ
|
19972 |
| jax/_src/xla_bridge.py |
sha256=XZ1P2wpMB455ZtNfmxKNEQiHOwhZMNFLI_UitkaWODw
|
46381 |
| jax/_src/xla_metadata.py |
sha256=IvJL8QejbnCGogGcaTCMMcseAWsKW0JouZrCRzUxV9c
|
1519 |
| jax/_src/clusters/__init__.py |
sha256=blIgNT92mYE9Bst7obP5_xhZ2fE2hKqwxH6R0cAQZ94
|
1283 |
| jax/_src/clusters/cloud_tpu_cluster.py |
sha256=xpeiJ-pVH6uNypUfIoxwNVaRuATiEtIqGVtNrvYN_lE
|
8405 |
| jax/_src/clusters/cluster.py |
sha256=4_heeOPzwH7NHumJKZspEfCReABCJC9SDfW8z3ORZp0
|
5856 |
| jax/_src/clusters/k8s_cluster.py |
sha256=TzUXU4ksY11X2f6iYjhb5LEGZfr6r5WZM5tc9DQhvBQ
|
4188 |
| jax/_src/clusters/mpi4py_cluster.py |
sha256=ugDhirp00Hh3knxX3qVb5yxSz1d0_O1DuQUcVfyO8rg
|
2608 |
| jax/_src/clusters/ompi_cluster.py |
sha256=Vfk3AimGa2jtDEwjpJJXuYP9p5uv1tBwEKtpGn1f4IQ
|
2317 |
| jax/_src/clusters/slurm_cluster.py |
sha256=hsZQJXznkCXGDTfKgMDgjVf4bmbAtSne6pa0wW4iJT0
|
2321 |
| jax/_src/cudnn/__init__.py |
sha256=03MR0G6UMqHXi-JBeyj6tipV6ioP9a-iSN1mW-8Ws5A
|
631 |
| jax/_src/cudnn/fused_attention_stablehlo.py |
sha256=w1pH1Sug4SKpPxOMlaSNLHg9Jy02WDNxgs8VdfxXnUQ
|
65310 |
| jax/_src/cudnn/fusion.py |
sha256=3ifqe61DuTyoOXQL7Nelacz2KRdcJdPRI__ub4s_vI8
|
2808 |
| jax/_src/debugger/__init__.py |
sha256=iZU1X5nkpSBK5pwLSuWhZsPZgtLUovBF3tyWPmQoh1Q
|
899 |
| jax/_src/debugger/cli_debugger.py |
sha256=C9FQoMwTZOesOTG0X5dVDT_3nGTExqKZixdPPzTDkew
|
4780 |
| jax/_src/debugger/colab_debugger.py |
sha256=PHmiL-QES4t97iICJt34a5FXSNTm-AaSUERPGCZ3QHA
|
7823 |
| jax/_src/debugger/colab_lib.py |
sha256=jCbwj2Q4f8SEo7CBr6AeiUtzlsIv4BfKa5CVbq4Bn2U
|
4298 |
| jax/_src/debugger/core.py |
sha256=WilcRMqpzssKa5YL_P7SFtKaUVUaXqDutxO_N-ZRnp0
|
8593 |
| jax/_src/debugger/web_debugger.py |
sha256=Or-gJL40oBUfClu8mg1YUBoKA9cQytZgJAeGGrPPK_w
|
3433 |
| jax/_src/export/__init__.py |
sha256=mS3YSYUZFIG68T5Zo9CFGkT7wCyWRycnWV4L36CnHYU
|
581 |
| jax/_src/export/_export.py |
sha256=LVBomrMfn6sQWQnQZiYdxQjBobpg9eSLofiD0pwNR3A
|
65979 |
| jax/_src/export/serialization.py |
sha256=WLEwoP7dnrQSBnz73KzXQXPs88Cl3jdrtjnpLiY4s9w
|
19203 |
| jax/_src/export/serialization_generated.py |
sha256=hd6uuEJUksQMgq5_SMVN06lfVaC41PPnMR0HVhngqVM
|
27772 |
| jax/_src/export/shape_poly.py |
sha256=CvCN0Gu_3rXbKYVjy8nzTMemz9daaekZTp93t32QGVA
|
86635 |
| jax/_src/export/shape_poly_decision.py |
sha256=UgKnxdKNvNEx_VNdjJW-3lMg-5Hh41o62Tij07zQUWk
|
20305 |
| jax/_src/extend/__init__.py |
sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc
|
581 |
| jax/_src/extend/ffi.py |
sha256=Qn_3K2ZOTptZtXcULwiVe2zw2zSCp-7AxwJDKZgz7Kg
|
24358 |
| jax/_src/extend/random.py |
sha256=1KmQCCYxFMJO8KB0qJz9ZVQhnK-05Sv4Jn0jXLPpWQw
|
1258 |
| jax/_src/image/__init__.py |
sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg
|
581 |
| jax/_src/image/scale.py |
sha256=_Hezs_YE3kCj6_cTn6XvMuU26sPiedsiJR8-VVhKEDA
|
13740 |
| jax/_src/internal_test_util/__init__.py |
sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg
|
581 |
| jax/_src/internal_test_util/deprecation_module.py |
sha256=iZ6fRYNaUcOwrftwpDXMnlxrLrrQbdMrbybmmQFwql4
|
851 |
| jax/_src/internal_test_util/export_back_compat_test_util.py |
sha256=S8wDJb98vYYeIq-omRo_b8jErOBa2DJAoX9mO1V-cI8
|
15336 |
| jax/_src/internal_test_util/lax_test_util.py |
sha256=TCvYcOZfaAJ6ktX952MtC3vE1f0Ce0iIE_FhEdRkJIo
|
12140 |
| jax/_src/internal_test_util/test_harnesses.py |
sha256=d3BdtWFyEKoB8v8vWTbp01UfhQkzHagQkpSptra28zc
|
126010 |
| jax/_src/internal_test_util/lazy_loader_module/__init__.py |
sha256=UP_c2VfWHRhfEi6n4F6qLHZuageXhPiVSAcUKHrrxFs
|
705 |
| jax/_src/internal_test_util/lazy_loader_module/lazy_test_submodule.py |
sha256=luZv4DGyWwdcBX0_NBVwu81IE4jFYVbRkKo9gQPmxPk
|
610 |
| jax/_src/interpreters/__init__.py |
sha256=hnuhN7h1jY5__Kt2rBvTrk0o8nX4fQl244fDCIMh0qk
|
690 |
| jax/_src/interpreters/ad.py |
sha256=XMATpKbgp0LbJRRaxpWAMYFvMXuYDZKjXDv2t6U7RZs
|
45251 |
| jax/_src/interpreters/batching.py |
sha256=esSUzj-TcBvjMSQbVxhx56fQEf8nfMoALVKR8NAUH6c
|
47526 |
| jax/_src/interpreters/mlir.py |
sha256=89I-UvfDVkZQ4ppca7CfmzRfTgP21HPZ-LjmEhn4BwY
|
131764 |
| jax/_src/interpreters/partial_eval.py |
sha256=lfIKElr9iD_j3iUi6JrMBiVuPPbgh6t1XAT00MNaT5w
|
114589 |
| jax/_src/interpreters/pxla.py |
sha256=tOuXKhHm5hlef0PrJ5aDaNa5LzBFukFAxCRzhNyAqy0
|
133449 |
| jax/_src/interpreters/xla.py |
sha256=HZffIbn2iA48aOt1cNy7V6d6DYz1SdI7S-TDz-_qQUo
|
5671 |
| jax/_src/lax/__init__.py |
sha256=jRL3qQoUco8BVUAh8lV-KuY38h_4GxXcb41asljW2yI
|
690 |
| jax/_src/lax/ann.py |
sha256=53ap20vxBD-hV6L1JZsLNfzyACtK8qkE2fmPqsJivvk
|
16761 |
| jax/_src/lax/convolution.py |
sha256=kxGQ5J18irVz3S1sxqoqlTW6otJO_5g3zPCHi09FRhY
|
44551 |
| jax/_src/lax/eigh.py |
sha256=GhrwPlBnfUlCa8e6XNq40kERV9-knvD4j5rnntxdvdQ
|
22235 |
| jax/_src/lax/fft.py |
sha256=fIrp5WrJCZM0FYZpZnmC67Y7rsWCthbOwK3GXryoEqQ
|
7258 |
| jax/_src/lax/lax.py |
sha256=zoWUePAThhZTThvt8idsW7KHzhavrxzFj7DepUZ6gUc
|
267582 |
| jax/_src/lax/linalg.py |
sha256=NUT0WJE84P_JfiD3GEDwYRjYHnXyJWENGZ4fk5NojFY
|
106576 |
| jax/_src/lax/other.py |
sha256=k5QaTGae3ps89MM2UzOG1_foHg_W3C5Ysq3BMKJWdyc
|
12497 |
| jax/_src/lax/parallel.py |
sha256=fj7RdeafVxEflCJFfTV5PxM2hEHs6ZaKImOHLh6brGI
|
71907 |
| jax/_src/lax/qdwh.py |
sha256=dmslOVnzg-DhxjSIHV9jHCOKopVosYD-xe1FPmSgnCo
|
9391 |
| jax/_src/lax/slicing.py |
sha256=3WhFIUgZOXMBkau-cogosiNLG0moPsLQxM3E4FlQNL0
|
128699 |
| jax/_src/lax/special.py |
sha256=djqaB1CuQX53mqxeHQhz5hV6kZGN-HVhtzAwqezU5dY
|
26940 |
| jax/_src/lax/stack.py |
sha256=7hxK1zlVa_Q9gO48Ymj8S1nA8DN6keYgcOY0DAtbjOY
|
2560 |
| jax/_src/lax/svd.py |
sha256=iVY3iFwt0ug1ow_SkFIJd0lNYdn9eXrCKhNbZ0Dh4Bc
|
8794 |
| jax/_src/lax/utils.py |
sha256=5sIMDI-apvUjZ4Cll-qWvhQxh9SU8ypUdAlFdS6D9_E
|
4297 |
| jax/_src/lax/windowed_reductions.py |
sha256=sRQsMHMbkNDoxt-QFEgcjssIVEZjipV-u0m8huwnFbU
|
42337 |
| jax/_src/lax/control_flow/__init__.py |
sha256=If-Hh196w69VMIk8l2iCXfcRL4C5Dpi3clMO-qdbtmI
|
2158 |
| jax/_src/lax/control_flow/common.py |
sha256=0hENKOH91IlZy7o8ey5OMJiAWeGwjl6AS4ODJcRmONE
|
10972 |
| jax/_src/lax/control_flow/conditionals.py |
sha256=5_e_l-eOZfe76KldazRPp1q12pU6ZKyHCDbqWvgDPL4
|
42542 |
| jax/_src/lax/control_flow/for_loop.py |
sha256=Ph-U0L_ih9aIzwUOuDZ4FHnjkHaoKI9zQOqh4qZRbDM
|
35763 |
| jax/_src/lax/control_flow/loops.py |
sha256=21-37LJp5Hxv6-CfK8xGE2yzk8IaXkk6raNT25MTp0g
|
110344 |
| jax/_src/lax/control_flow/solves.py |
sha256=TPPv-SJvh44F3cUI8PfM6UtR6rvLn0PncHnJdMpVXCw
|
18383 |
| jax/_src/lib/__init__.py |
sha256=bUYdjvw8d2XkBpIYLe2i31-v8yu5BEMlM7MpdY4uBa4
|
6807 |
| jax/_src/lib/mosaic_gpu.py |
sha256=lHnM852MFg7K5UUwry1zN_r5nrPSWl6ZVwOxVnsWAz4
|
898 |
| jax/_src/lib/triton.py |
sha256=6zzcpqutI1cZW4sG9tLwPFyPDw35xhpPoNH8wkG-VbE
|
662 |
| jax/_src/lib/mlir/__init__.py |
sha256=UoirBDcNhqLa8Gb1Qrxd4wYxScz_p-1tb-ewbTsK1jM
|
733 |
| jax/_src/lib/mlir/dialects/__init__.py |
sha256=6KUUW3fBmKZn0eM7VlIFh5yiSLRQ9PNC1Cl9tAxoF2o
|
2032 |
| jax/_src/nn/__init__.py |
sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg
|
581 |
| jax/_src/nn/functions.py |
sha256=oTbQXMyq126FRelcSo4oDGLZ1FWFIiQomJ9ERRcW62M
|
36336 |
| jax/_src/nn/initializers.py |
sha256=TOGKwbExnXQvMai1amFpEH2PFaZxiVz95KGBVaopnmA
|
25450 |
| jax/_src/numpy/__init__.py |
sha256=-QhmzWj2qFbKVIT9TDAIvQWaKQibiRXy3r1zQkI3X4M
|
581 |
| jax/_src/numpy/array_api_metadata.py |
sha256=je15NAJj_CFlgqeDqo5iSg9EzmN9IwCkZ_E8TtStKcg
|
3347 |
| jax/_src/numpy/array_methods.py |
sha256=56ieb3635_adZXfUvnMwJtoqt9-ulNUOlWezpyKFpYs
|
45675 |
| jax/_src/numpy/fft.py |
sha256=ZZih9b0NDQYrD82RwdR4UwqrbXvIKv-px18hyANG9wE
|
51814 |
| jax/_src/numpy/index_tricks.py |
sha256=aRFc9So43wmgzk-eVGBeN16BI4Ny7eXYFr7yEtrgDqM
|
9992 |
| jax/_src/numpy/lax_numpy.py |
sha256=AGwAmAW4KfKB0MMfqOCzp0_BHyZwTIU4jmqxmOBfy98
|
485882 |
| jax/_src/numpy/linalg.py |
sha256=FkVID2eaagG3yAzXESMORnS4iS071TMrq6yuPpv8Rgo
|
77620 |
| jax/_src/numpy/polynomial.py |
sha256=LU6iRWNBBHG2BmYztPVH4tE-5ZWTnCGVzTNNTXBHxqs
|
27913 |
| jax/_src/numpy/reductions.py |
sha256=khOHS0fiXYX4A9aLUX5_m2zd0nuJ9WA5t4QTYx4KFTw
|
109085 |
| jax/_src/numpy/setops.py |
sha256=Uptcxj3KK04zB5P4wTrr0Qoj0INwDl44d9zEQmvXl8E
|
47405 |
| jax/_src/numpy/ufunc_api.py |
sha256=VXdn5cXraEXPjBXUTTU_jhWFGrdOF5NlrDEXHBhyn0w
|
24204 |
| jax/_src/numpy/ufuncs.py |
sha256=2AYPa4CZWiX_w7KM4XusPTbMRC3FZzAcvaOR-dAV82s
|
116122 |
| jax/_src/numpy/util.py |
sha256=EYp4q_C4DsVA2pNkK-JzwFHNUVkXy2CvMIY-M1yeDP4
|
11987 |
| jax/_src/numpy/vectorize.py |
sha256=lugYJ7HrEpG-L0yNd9UD9mEa6ASQlaaACdb72PFtat4
|
14018 |
| jax/_src/ops/__init__.py |
sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg
|
581 |
| jax/_src/ops/scatter.py |
sha256=wsV-jnH9epkp-GjIy8AG9xgpVoBUrPEmaj6h0eXhBis
|
17625 |
| jax/_src/ops/special.py |
sha256=Ypg7maHS3_-cklEXoTSwrxReDVaEsUHx15D3Zk9BeYA
|
4151 |
| jax/_src/pallas/__init__.py |
sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc
|
581 |
| jax/_src/pallas/core.py |
sha256=k2rdm8pQIqbejiDQnTAZNpJ7tvrpA3KslQ-BdCXMS3Y
|
37982 |
| jax/_src/pallas/cost_estimate.py |
sha256=RtYcdNqOTQx9NQbnR_rxZY2HHXbEo2xzuQa_Uj3vGyM
|
8283 |
| jax/_src/pallas/pallas_call.py |
sha256=IXvFDKrMmXxTfXjJsRwuvd2KBAhntdjqydJ0KmjAzpc
|
74725 |
| jax/_src/pallas/primitives.py |
sha256=rB0fzchNW6qhDurO373EWcw5XXhkRDvydrFECXp-WUQ
|
32411 |
| jax/_src/pallas/utils.py |
sha256=tNjoP20oRWC4A-ns5aSUB5LjRCS11fy4E1SH4rw-8-M
|
14067 |
| jax/_src/pallas/mosaic/__init__.py |
sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc
|
581 |
| jax/_src/pallas/mosaic/core.py |
sha256=KP2vNaGAMJOiMVYicaSIvTR1Z7Mz0CbJP7NHkab9Bg8
|
8069 |
| jax/_src/pallas/mosaic/error_handling.py |
sha256=nmjLsqhnE9-foi7lwMAWdWNQMdHd2jOMqJtDxhBhr1o
|
5480 |
| jax/_src/pallas/mosaic/lowering.py |
sha256=ekUEJ6ngpLNg388mvwcCi_SXGn2f8t7Y8mg9f-YHClM
|
116206 |
| jax/_src/pallas/mosaic/pallas_call_registration.py |
sha256=ZHFbV8Xzpz4a0Za6UIIGB6N33I530zB4ZxYZh8JQxnA
|
8967 |
| jax/_src/pallas/mosaic/pipeline.py |
sha256=8ShG9SBvCySphlB-GC2glH8nDyU3euVwXt9_rH4H9Ag
|
44060 |
| jax/_src/pallas/mosaic/primitives.py |
sha256=iP7ROK4kwpcXWoNMozzYO_uZCrl8VK2igDvKI4auE4E
|
30418 |
| jax/_src/pallas/mosaic/random.py |
sha256=yBP4tvNPUodwWMSssHDUJkwW92uU3K2wtvziRg00CpM
|
7989 |
| jax/_src/pallas/mosaic/verification.py |
sha256=fQAVzbETThqBWA_yp2jLbeX_5meoVShh3flYgQWiF9g
|
24540 |
| jax/_src/pallas/mosaic_gpu/__init__.py |
sha256=mS3YSYUZFIG68T5Zo9CFGkT7wCyWRycnWV4L36CnHYU
|
581 |
| jax/_src/pallas/mosaic_gpu/core.py |
sha256=QD9jq4MRHz97JwMdzNmUoea3BIfwobIS5YrlCfm5XYI
|
17451 |
| jax/_src/pallas/mosaic_gpu/lowering.py |
sha256=ZV-9rabyEGWxd0kEs_psQAGTTTmoVFMSfCqYKUH5kds
|
62534 |
| jax/_src/pallas/mosaic_gpu/pallas_call_registration.py |
sha256=EthvrVK_W8aSVPFGWhEcYT3Y03BSwXuzzH2HY4_T0bQ
|
3518 |
| jax/_src/pallas/mosaic_gpu/pipeline.py |
sha256=QiwvxB4cuKNqiWNBJJ4Xc7He0H_BOYAzvBUUiMqHUrw
|
11190 |
| jax/_src/pallas/mosaic_gpu/primitives.py |
sha256=uAJqdDGn02nECaM1jmpEAsLbhbIODpuynI82-t5g74s
|
24221 |
| jax/_src/pallas/triton/__init__.py |
sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc
|
581 |
| jax/_src/pallas/triton/core.py |
sha256=5u6hubCM5N7aSYK5iExZpffEGrmlpqR6Op1xd3dOhDk
|
1374 |
| jax/_src/pallas/triton/lowering.py |
sha256=YONSpaVqxkV4Dh317M7-fQk_MR06G-aNkt9sb8PDp5g
|
88416 |
| jax/_src/pallas/triton/pallas_call_registration.py |
sha256=JD7nfRPCIEH4dvXhyOjb_I15ZwdlL9irmQRLvc320Ns
|
4395 |
| jax/_src/pallas/triton/primitives.py |
sha256=_UjvgpOsEaRzul3xizwzPdP1blZ1-NzChzZ4bsVmSaA
|
4228 |
| jax/_src/scipy/__init__.py |
sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg
|
581 |
| jax/_src/scipy/fft.py |
sha256=rEOuNrkRnMOofq0i8XunDVONpgMyUws_vn7HgGCjHAs
|
15266 |
| jax/_src/scipy/integrate.py |
sha256=4VtR2wYVO037Z0yiOdQbniKtpxL4boZQNIppy4J9qy0
|
2346 |
| jax/_src/scipy/linalg.py |
sha256=uNk5a101MNAn-bnvLaKnV7khyhgnur3_1Be1rLW2LLU
|
75465 |
| jax/_src/scipy/ndimage.py |
sha256=IuE4PMzY5HuxST9D0G0On1SVFRpNF367Y2cIQv8duTc
|
7092 |
| jax/_src/scipy/signal.py |
sha256=E01k5pOeAtg2has0eb8GRoaogdfBWNYFzhcCdHt7dAs
|
47453 |
| jax/_src/scipy/special.py |
sha256=oQaxHPa3lTbqdOqpYGZEb-IlhF7FgJf5bWCn2kDXePU
|
80012 |
| jax/_src/scipy/cluster/__init__.py |
sha256=S-o0CP72882Xv3l6k2PlrEx6eNvXBTk5Oaed5KF4iL8
|
581 |
| jax/_src/scipy/cluster/vq.py |
sha256=3Xfr7esTCzSawrDGZ37hu_MXJ-EqnZtff6hYmXhz0Fc
|
3043 |
| jax/_src/scipy/interpolate/__init__.py |
sha256=S-o0CP72882Xv3l6k2PlrEx6eNvXBTk5Oaed5KF4iL8
|
581 |
| jax/_src/scipy/optimize/__init__.py |
sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg
|
581 |
| jax/_src/scipy/optimize/_lbfgs.py |
sha256=kqKkdDeRYiH5prawdntOXfCs-g311NYwnY57jV8DDbw
|
7769 |
| jax/_src/scipy/optimize/bfgs.py |
sha256=JRdDBxnfNrVxK5UDyehFGF3eQTJenWfdZVmKOMheenE
|
5609 |
| jax/_src/scipy/optimize/line_search.py |
sha256=E51ROBUJYmanL1byuPpIa28HmN1vCf_BAaBC_dQZgoc
|
13235 |
| jax/_src/scipy/optimize/minimize.py |
sha256=mCojZkL0TXQhix6w0-GroM7ITU2rodYdDwmvynLsqWs
|
4979 |
| jax/_src/scipy/sparse/__init__.py |
sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg
|
581 |
| jax/_src/scipy/sparse/linalg.py |
sha256=IGX8lZ2il77S3WA50Lz7MdBiw7lqdD8mHgWRaCUHX6k
|
27245 |
| jax/_src/scipy/spatial/__init__.py |
sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc
|
581 |
| jax/_src/scipy/spatial/transform.py |
sha256=yD9HOfgQpQO3-eACTh5zW8qzV-WLcrTHY_5ZlsnBoSk
|
17525 |
| jax/_src/scipy/stats/__init__.py |
sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg
|
581 |
| jax/_src/scipy/stats/_core.py |
sha256=Gq8H86uI_wWodhyvdEy5GiNulBfW3kJZvm8Z0Qw6Vdc
|
10653 |
| jax/_src/scipy/stats/bernoulli.py |
sha256=zOvo5ZLr8y3O0byR0cp3JRPKPPHoz8oVyG3-C_5-NJo
|
4541 |
| jax/_src/scipy/stats/beta.py |
sha256=h2LF8lIveFCa-qNZ_XLD1hQIK9_dLSmlCjrqeSdz_ME
|
7988 |
| jax/_src/scipy/stats/betabinom.py |
sha256=dmhOAUk6py5Fnu-jR95KnwJ6suvvfwNF9CXNHz73CSc
|
3447 |
| jax/_src/scipy/stats/binom.py |
sha256=E7a3go8bCM5bSiC72l1pDQEuuzSpXLmgfG40iPF90wA
|
2825 |
| jax/_src/scipy/stats/cauchy.py |
sha256=26eHaNIfxwh1og3kTqnPYgivH0GTIuWVwElai4VJwv8
|
8884 |
| jax/_src/scipy/stats/chi2.py |
sha256=vPFrhpbLRPDheYenmHQ7QvZrRt8cAyRtOpU9VmemNIY
|
8008 |
| jax/_src/scipy/stats/dirichlet.py |
sha256=_wWyx07z62_EH25i3J4jS5q53oj0UH5A3-XgQzapwbY
|
3198 |
| jax/_src/scipy/stats/expon.py |
sha256=8q5nUo_N8Rl8BjMVfFKv2ydB-x7-TkKtIdZi5FVrbl4
|
2268 |
| jax/_src/scipy/stats/gamma.py |
sha256=W5B8z6LPD3Zi3cbcWHDB-RU_OuKjXopJIspf_VKZMw0
|
6987 |
| jax/_src/scipy/stats/gennorm.py |
sha256=kp22L5H-rR5n7sn9rpeWsWta-F76yZaXkO0Ovvf9zc8
|
2989 |
| jax/_src/scipy/stats/geom.py |
sha256=2hQ4LoaqNHFBahdRTGS9oyWMYdFQufJb2ByqOw9709s
|
2243 |
| jax/_src/scipy/stats/kde.py |
sha256=Mi9nMB6q3eueSbyNMfs3DokLhsjte0lYsSIdAZ9Birc
|
10141 |
| jax/_src/scipy/stats/laplace.py |
sha256=9PlK08r9wRHelqSIuKkv8KGO0oPV2AArI0Fs6rICFj0
|
3269 |
| jax/_src/scipy/stats/logistic.py |
sha256=HB3Brjwwl-yyGB_9K1lpdiQlmitsHwNLfgl0g4owFSM
|
6271 |
| jax/_src/scipy/stats/multinomial.py |
sha256=CVgpOPCVcr_wTNuF1vXcUjgsENe2mVO7gYBYTSdCOwI
|
2469 |
| jax/_src/scipy/stats/multivariate_normal.py |
sha256=Dei7nhj3j5gB1lBi9q_ZpdMHi1UY7XCr3GTzNuaWF1E
|
3433 |
| jax/_src/scipy/stats/nbinom.py |
sha256=kr6FLGYROqMMtfLfWLI1fumHBl_tSX1_Ys2icp59Hdw
|
2603 |
| jax/_src/scipy/stats/norm.py |
sha256=pUS-JhrjrFtdbfO178IeOJizZBU_WjXU6aUr8W67tQg
|
8401 |
| jax/_src/scipy/stats/pareto.py |
sha256=WcRAchdj198RF_W3egR8Bteok1dXE0PHv9ygw-pQYtM
|
2596 |
| jax/_src/scipy/stats/poisson.py |
sha256=Y43e2hY6L89go0Vc1HJCDrf61-vDZcQVIC6zVJdaYyQ
|
3348 |
| jax/_src/scipy/stats/t.py |
sha256=I32sQ9TVaes4keK5AIK6I-vX--nvl4qtZctuc0zIhG0
|
3221 |
| jax/_src/scipy/stats/truncnorm.py |
sha256=V4_Y2OEMYPepPyWVqMZKO0hRV8HwwnI6_djK9xYwdDA
|
9239 |
| jax/_src/scipy/stats/uniform.py |
sha256=pR3NYr6ZHTs6xhw-LECsRJqp5l_iBbsw689QHxr9Puc
|
4361 |
| jax/_src/scipy/stats/vonmises.py |
sha256=yxvmgBI_HWQn8U3u7wtLv9PjqOk7hRX8I4gWKfqRZKg
|
2451 |
| jax/_src/scipy/stats/wrapcauchy.py |
sha256=SpHIjzS9j3TqgAMTHXxtDz73aBfezqYUaNSsGYCky5I
|
2371 |
| jax/_src/state/__init__.py |
sha256=TU0qQETOXKzVAd1oVaIEtOJTC2OMj1PdFhsExMHFx2g
|
988 |
| jax/_src/state/discharge.py |
sha256=mI9j_FT-IRPi-j43pRhB0NEc06MGGiu5vYFrzvXniAk
|
44458 |
| jax/_src/state/indexing.py |
sha256=p8oEIfj8oIv1M3VhQBoQI2gkG4AJ19MiPayqlDZC6xw
|
9264 |
| jax/_src/state/primitives.py |
sha256=iofaTUrRSb3sWyJcJSGqpBjXkDOtBA8s43lpOd0BWXI
|
25203 |
| jax/_src/state/types.py |
sha256=0Zro4YUM4K7_cy_t4hSuHOHYBaIP2OEDkLK-goBDv70
|
12462 |
| jax/_src/state/utils.py |
sha256=7hHiQUpzt9uEyu3yh-l3PkzWV0-_65JtTcg0WuAkIyw
|
4078 |
| jax/_src/third_party/__init__.py |
sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU
|
0 |
| jax/_src/third_party/scipy/__init__.py |
sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU
|
0 |
| jax/_src/third_party/scipy/betaln.py |
sha256=TT77MW5HcPt-C6m-jHT43mGPLuAKO9ANcoEsXjl8JDE
|
2373 |
| jax/_src/third_party/scipy/interpolate.py |
sha256=5C9-SwIUrwlpBUgAKYskaSLgTbOtpwESJRPv-6cjj0Y
|
6307 |
| jax/_src/third_party/scipy/linalg.py |
sha256=SzmpCoi4kSnLTflDsRa5QdoBcijKHuwmtN9xac2TWlw
|
3786 |
| jax/_src/third_party/scipy/signal_helper.py |
sha256=Ygxut_RD0Tixc4Gt0AcW32QhNTmV5EBxLYn6RjTUipU
|
3495 |
| jax/_src/third_party/scipy/special.py |
sha256=ojE5A5mClaexr_qca69NYpBTgvBEdvVaVMRmptVh0yE
|
9155 |
| jax/example_libraries/__init__.py |
sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw
|
581 |
| jax/example_libraries/optimizers.py |
sha256=T2D3qDPWZ50kwTyDe2IXPPLbOUu04cQTdf51oHdXVvU
|
20426 |
| jax/example_libraries/stax.py |
sha256=jNCoL5RlFv9JnLssUHthGRaNcA9xY57BKPxF76Nz7u8
|
13862 |
| jax/experimental/__init__.py |
sha256=OQS_hX2SykwFBXHl4t0VHv5WKOsqpLjuVL5wXg0CZik
|
1012 |
| jax/experimental/attrs.py |
sha256=X_8-3ztzkQjs0VPW3oKh5RPlXwMLnKm2QSDkfwsvObc
|
9535 |
| jax/experimental/checkify.py |
sha256=V84hTTNrwnckaRzgvWDmpcKKA95kAakjZwwlfnRaXt0
|
1213 |
| jax/experimental/compute_on.py |
sha256=Mdg2n8JtAsrq1ap_PwUPwoqpoCNGO5rjf7T47e_iaAE
|
644 |
| jax/experimental/custom_partitioning.py |
sha256=E0IrgU4LX1P-khUlG69nf9d4SPi9wbaBmyTiyvE6PbU
|
856 |
| jax/experimental/host_callback.py |
sha256=cuZhGDILVkhFgEFjgwSRH4C__ySPwxeJjxvPVNXf-b8
|
1207 |
| jax/experimental/jet.py |
sha256=x_rZd51I4cnVKQNJKO2tbym_tRyrnhVA1Uzx1KzgwyU
|
27123 |
| jax/experimental/layout.py |
sha256=q2i-_DZq6t3TaTmtkRSwh5CiseuwbH-VMqIMBUL4ItY
|
675 |
| jax/experimental/mesh_utils.py |
sha256=X4EYn9ZS-cEOBncDPtqJZA7A4FDEOWCAw2YP7qoJMNY
|
930 |
| jax/experimental/multihost_utils.py |
sha256=-sQHcp08vioCBBlXAA-cHb5csS4lGnVDgX7Khe9bIAI
|
18376 |
| jax/experimental/ode.py |
sha256=zvQAZHbxgGHA9fAhhq7KVxGefqO9-E0kB726eXc18Pc
|
10646 |
| jax/experimental/pjit.py |
sha256=2Qi4A4tKn1p6pjxysiCY5UAsfQq_KVZ7QpMX3gRJw30
|
749 |
| jax/experimental/profiler.py |
sha256=DJ0ETyXEZ56c6LYWRY_ankgPOs5oWZejGtGXS-Ttbus
|
1424 |
| jax/experimental/rnn.py |
sha256=kxuSVk5trxvOkWqPLUJ6iZqROD3lSKxcsMA7Uq5IRok
|
18777 |
| jax/experimental/serialize_executable.py |
sha256=DYfargD1RSL8Uc65iFWOjm1MU1WArjt4Q4t5AZTTz_0
|
3006 |
| jax/experimental/shard_alike.py |
sha256=dbrhVCnWEX39TsQYRyGoClFByDmqtmDlKGtOmlT-m2M
|
646 |
| jax/experimental/shard_map.py |
sha256=1-e_2wjEzCpJ6YcVAIK4k0Eo4gtnWWKXnhwXI4ZGiN4
|
90246 |
| jax/experimental/topologies.py |
sha256=Lxxb2W6nT6AEhM2ekkNUKm6W1tgh5i-_49454NuxCkQ
|
2101 |
| jax/experimental/x64_context.py |
sha256=lUrmtrYhu0ZGwd_j8vrzFqQnwfDptgpHxZbqV94Hb30
|
1710 |
| jax/experimental/xla_metadata.py |
sha256=YoNAzPbdARYfGzGTUVoxTfHG9dDrobUvWaPI4uvHRJk
|
658 |
| jax/experimental/array_api/__init__.py |
sha256=JYhDscKw62F_3Y0uaTh2SrJADrKBOOQwwavlvXo0KKA
|
1090 |
| jax/experimental/array_serialization/__init__.py |
sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw
|
581 |
| jax/experimental/array_serialization/serialization.py |
sha256=VS2pBc2z5E7QPQ5-Xk8HmNwI5_kR7kOjvyMdguQyb8k
|
26055 |
| jax/experimental/array_serialization/serialization_test.py |
sha256=BJ-FXawRqDfl6G6SXrDpl_4XaZy0IP2B2SodDrve6n8
|
23222 |
| jax/experimental/colocated_python/__init__.py |
sha256=bpuYP80GTKaGnrDoSTwjYRb_Faoi2Kz7E0il-40CVvw
|
926 |
| jax/experimental/colocated_python/api.py |
sha256=AnaV-mxxfaxPiG9_pst54n3QCV4rFgu6bYmKNtloFLg
|
2018 |
| jax/experimental/colocated_python/func.py |
sha256=OusSq72YbwE3wpP9BDOXL3LOhA81g1JDfAcnoWbpj_0
|
17447 |
| jax/experimental/colocated_python/func_backend.py |
sha256=lXN-FRbWIWezsICyIc3QYwjpF9-tYgZV1Ut7TBN_dpE
|
1373 |
| jax/experimental/colocated_python/serialization.py |
sha256=w8pKcncrWWKZtezgWhVb7BzLuSeUAm67b-8PPqbBU0s
|
8635 |
| jax/experimental/compilation_cache/__init__.py |
sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw
|
581 |
| jax/experimental/compilation_cache/compilation_cache.py |
sha256=OIZLvS1c3Yw_EbbqlV28pC40g5dPdBB-XDrXnHcT6As
|
820 |
| jax/experimental/jax2tf/__init__.py |
sha256=ZqsPlkalGAkzdTwRIutfW1k8NmOHzJn6kfIB1A98pdw
|
944 |
| jax/experimental/jax2tf/call_tf.py |
sha256=5_UtMy3cx45NHQ2OIVXRZTJCaFKStDZoYGA1UfPcC0Y
|
28243 |
| jax/experimental/jax2tf/impl_no_xla.py |
sha256=0qzCj3AFxvRIm-iCFykCT2tFgo5ghlJd8BXcK-HDzrs
|
54474 |
| jax/experimental/jax2tf/jax2tf.py |
sha256=lrUDm2rQtJs4PpyXsQtoeFKQcCp_Jlv06bxhttSBZ9M
|
145559 |
| jax/experimental/jax2tf/examples/__init__.py |
sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw
|
581 |
| jax/experimental/jax2tf/examples/keras_reuse_main.py |
sha256=AN1R_wzL4NyCgMYEb6ab8fos2vruCL0PGj_GYAEWNEs
|
2936 |
| jax/experimental/jax2tf/examples/keras_reuse_main_test.py |
sha256=0nnk1mDlssNRhOIG4u1_WSx6KXF9C7F5Eih6f5hFWjw
|
1670 |
| jax/experimental/jax2tf/examples/mnist_lib.py |
sha256=c-Bno2RVxfY4aVgMRK9sjrCInqWVpTIBIQH3C9nPyIQ
|
11512 |
| jax/experimental/jax2tf/examples/saved_model_lib.py |
sha256=czppig2onwKslN38JDp6qx3z7iHzFqL_gAcIVmfrAN8
|
7031 |
| jax/experimental/jax2tf/examples/saved_model_main.py |
sha256=wuf8oOJ5QNr3R71ltZyAci6o7oKKMpvyn_l96ISMm6U
|
8003 |
| jax/experimental/jax2tf/examples/saved_model_main_test.py |
sha256=Ij1lZWIgrePGW8aElclPa2F9-ZW_msbAsJSTMQL799g
|
2467 |
| jax/experimental/jax2tf/examples/serving/__init__.py |
sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw
|
581 |
| jax/experimental/jax2tf/examples/serving/model_server_request.py |
sha256=De9TWLSyC5EbkQRg00S9bvVOo17HJuS_M4cztRFjOa4
|
4912 |
| jax/experimental/jax2tf/tests/__init__.py |
sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg
|
581 |
| jax/experimental/jax2tf/tests/back_compat_tf_test.py |
sha256=xOP6CFs9sD3yKiI2vkk0W4rPKdB5T5C_4IA1E5-H09g
|
5360 |
| jax/experimental/jax2tf/tests/call_tf_test.py |
sha256=Dox3xHLclyOzyWXTZm2zbY8jJPaprcI5LdEbG3nz2Dw
|
67020 |
| jax/experimental/jax2tf/tests/control_flow_ops_test.py |
sha256=sqYQJWhcY-awqj6u6IF26qRBF7xPzodkN2XnZpRMleg
|
10108 |
| jax/experimental/jax2tf/tests/converters.py |
sha256=uYSiDOPUNaxcgVwLExAgquiK6M0VFH7j55OnbruOyaM
|
1998 |
| jax/experimental/jax2tf/tests/cross_compilation_check.py |
sha256=holjJ2qso1R1VD_QSmKj-HnSoh_dkJgkOVLVqZqPIcA
|
7400 |
| jax/experimental/jax2tf/tests/jax2tf_limitations.py |
sha256=FWkCsZeRHZq93XP2No2CtiO4uLq24AcLbh6kLns1Frw
|
60812 |
| jax/experimental/jax2tf/tests/jax2tf_test.py |
sha256=MaB6CtzT4OWv6UcZUeBBxdQfxF0bc26BbBhsQBt8V14
|
67084 |
| jax/experimental/jax2tf/tests/jax_primitives_coverage_test.py |
sha256=2ouDpIkKBvLLomEsmYU8jMKIm4iy4xKjONdlEG1Fy-s
|
6645 |
| jax/experimental/jax2tf/tests/model_harness.py |
sha256=zhPlq2VnostLmvDfkPdE74A6UPec7wLRS6c_0oFwaLA
|
13934 |
| jax/experimental/jax2tf/tests/models_test_main.py |
sha256=qk_lozA90084XzxnrkmjzLd7RvEGfUIMWNM3v2FG-ac
|
8928 |
| jax/experimental/jax2tf/tests/primitives_test.py |
sha256=7hMe-LYAZRJ4gXGZrHPhdFh5mu1Wem5EHzIXJ55KXB4
|
14490 |
| jax/experimental/jax2tf/tests/savedmodel_test.py |
sha256=Ch1luggaR0F9_jRSLAovuqRSCnVw3EIUmYBYfKKJRE0
|
13067 |
| jax/experimental/jax2tf/tests/shape_poly_test.py |
sha256=WmFsKMQOzlFVsLR7O2i5o_M_duh0FIydetFq_8mEUVQ
|
126535 |
| jax/experimental/jax2tf/tests/sharding_test.py |
sha256=cGnkFXOsvqP20Q5IRaht_znOJJmVMQUQMeTZyvddkOc
|
25883 |
| jax/experimental/jax2tf/tests/tf_test_util.py |
sha256=YIli6L4hltJraWYyPbK4hOTZZ_64UafOn45MbMNpkGg
|
19962 |
| jax/experimental/key_reuse/__init__.py |
sha256=5CjqLFSLnAhm_YyrowsWA48ENwxcvcD5KQrNQJTgLpQ
|
1688 |
| jax/experimental/key_reuse/_core.py |
sha256=NVFNMCQZg0UOZhHD0RpiDqlXum6A48s2z79ETOP9rb0
|
22781 |
| jax/experimental/mosaic/__init__.py |
sha256=-MnUIUUmkSaNkO5bOpPdXrcTnKVWt9ModYX4cnlq9LY
|
883 |
| jax/experimental/mosaic/dialects.py |
sha256=eYxSlw_cu_8usVXcEA3f-cqmlxDUwcTiqLiqNAgWWfs
|
769 |
| jax/experimental/mosaic/gpu/__init__.py |
sha256=XQ6j1oFvq_t5lKLuIt_dVUZhOHe2j5AQJJ-MCf3C5ng
|
2993 |
| jax/experimental/mosaic/gpu/core.py |
sha256=ycXZ6Ogt1XkP8xMQ646Kp973JJbEb-y2t3GfX7pSlz4
|
41294 |
| jax/experimental/mosaic/gpu/dialect_lowering.py |
sha256=RcQvRlDRw_3Xd82sxAa6kVNGEK74r68BQERJZ9Saolk
|
9451 |
| jax/experimental/mosaic/gpu/fragmented_array.py |
sha256=pS2YLpLbJitsPz9jV_hyWHGTRNnd9fYI4YheEKA5-Sk
|
76814 |
| jax/experimental/mosaic/gpu/layout_inference.py |
sha256=_O6fpgzGBeQHuQmxEBhkkmTltISakDmM1To0rsmrC7Q
|
6199 |
| jax/experimental/mosaic/gpu/profiler.py |
sha256=eOKmUHrEJ1SRPAFuEadqLTwIIYryyHnXZpmPpQZlC9A
|
14514 |
| jax/experimental/mosaic/gpu/utils.py |
sha256=ZDzIEWLaCYbtChPS1sc23mBvNDjmiZVnoOHYyeJO8oY
|
35211 |
| jax/experimental/mosaic/gpu/wgmma.py |
sha256=kDsIBYlSViDyP8TsNbKP3vYoPXq4Px1S081c-OJtciU
|
16152 |
| jax/experimental/pallas/__init__.py |
sha256=RaOXR9hWOTxrIUOZHum3hmwLzNclK3XZejmENiKd7Wk
|
3295 |
| jax/experimental/pallas/gpu.py |
sha256=Tswdk79SmB-Otqbp0s-FLoyQmMGPrrVQHHUNUBDWT_I
|
865 |
| jax/experimental/pallas/mosaic_gpu.py |
sha256=FbA96Z4GmD06pcAjhDiGxiow5AFGObNFEf9eFiyNYCw
|
2823 |
| jax/experimental/pallas/tpu.py |
sha256=Dmx7XjC55WtLzn3ZFBBwAYCoK0c7IpFTdoBXwRCQC44
|
3891 |
| jax/experimental/pallas/triton.py |
sha256=8FEMurTT9d-X8tipH-706ETuSLjFkT0LGWiOHjp8zfM
|
948 |
| jax/experimental/pallas/ops/__init__.py |
sha256=vC8qlznuwgbYeFDJqP8pDUIf2gEKFrCHg3EZYdTkiwM
|
764 |
| jax/experimental/pallas/ops/gpu/__init__.py |
sha256=mS3YSYUZFIG68T5Zo9CFGkT7wCyWRycnWV4L36CnHYU
|
581 |
| jax/experimental/pallas/ops/gpu/attention.py |
sha256=u7C3fYnWEVr8Df1L7cHFK8CEFYiHoV3mdi_Q2_e6fz0
|
19616 |
| jax/experimental/pallas/ops/gpu/attention_mgpu.py |
sha256=OIaxhKCmfisEx2URA_OJGL9ocVS_0sVANjL1mNFSKDA
|
12117 |
| jax/experimental/pallas/ops/gpu/decode_attention.py |
sha256=-J86zYUnqRKU4YI6FzzqQ45xmWa5Hw3iAAiSohOLYvE
|
16321 |
| jax/experimental/pallas/ops/gpu/layer_norm.py |
sha256=OAdBY-Ijtai5IOUfXafJAtudlOUrli0g-_Qhc-t9jQU
|
11142 |
| jax/experimental/pallas/ops/gpu/rms_norm.py |
sha256=hNREA4EYeqPbF3RAijhEvtFk4jP8IiNy5LhPGcQldwU
|
10033 |
| jax/experimental/pallas/ops/gpu/softmax.py |
sha256=ydajdpaW_oAOwwdGJUx-t2TmIx9QfHhnXX_lNdtmVL4
|
2774 |
| jax/experimental/pallas/ops/tpu/__init__.py |
sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc
|
581 |
| jax/experimental/pallas/ops/tpu/all_gather.py |
sha256=oLZefmIUQ1Ag_C3kz5h114E2l4OiW9-Nn6DkpB4AC8I
|
5581 |
| jax/experimental/pallas/ops/tpu/example_kernel.py |
sha256=jzsyvU-Lk0PIorE9XCw2NDdqdp8Cn4bqsSe76DBxeEM
|
802 |
| jax/experimental/pallas/ops/tpu/flash_attention.py |
sha256=2g9jECYvE3S-HnVPmABhsQgFsiw6JU3wacYhYIGFAWs
|
50787 |
| jax/experimental/pallas/ops/tpu/matmul.py |
sha256=jfpCbL3yQ37So4yOtb9pF6K7PtA0WmV65yrzQ8yp4cs
|
2726 |
| jax/experimental/pallas/ops/tpu/megablox/__init__.py |
sha256=HRWx7FtM9_17OLitbxwIZp-A9RM61uSL6QAcBGJyw_w
|
650 |
| jax/experimental/pallas/ops/tpu/megablox/common.py |
sha256=9zg98VzHJh5l4Sf_SKly1nrXFWeWeyKoe7ZSPn6Ht38
|
1969 |
| jax/experimental/pallas/ops/tpu/megablox/gmm.py |
sha256=vNNgQIirDK7pyDViP2foec7GC8EhJuxNjR1mp3kHwjU
|
27249 |
| jax/experimental/pallas/ops/tpu/megablox/ops.py |
sha256=4Xg3O0bnP5HTWsawS0hAm8mF9HTvRHVK65RlopkA1rw
|
2903 |
| jax/experimental/pallas/ops/tpu/paged_attention/__init__.py |
sha256=Q6gFLakYVRJDRxxvqq6vmtDYup-edUHZy54j9194JfY
|
700 |
| jax/experimental/pallas/ops/tpu/paged_attention/paged_attention_kernel.py |
sha256=f6z0ukhHjObUOYoWsLc8Wxvhv8g3UVpjGxUixdnf_UE
|
21960 |
| jax/experimental/pallas/ops/tpu/paged_attention/quantization_utils.py |
sha256=5y0QHJo2yDsafF_lgeSO_izChspLHpVOvbFWZ-2d0ss
|
2556 |
| jax/experimental/pallas/ops/tpu/splash_attention/__init__.py |
sha256=8y556G1Vpmglvw8FbNzeB4x6QK6PLkKUBnMsL4vleNs
|
2738 |
| jax/experimental/pallas/ops/tpu/splash_attention/splash_attention_kernel.py |
sha256=Ra3ZkXny78WrKMPvorsP6-BdjOOZcIj6Htm1DTgWOlk
|
73759 |
| jax/experimental/pallas/ops/tpu/splash_attention/splash_attention_mask.py |
sha256=o-9hXrMgy6vq7nq8SiUFPceTGUdvvAsXNHR1bnjPa5A
|
14554 |
| jax/experimental/pallas/ops/tpu/splash_attention/splash_attention_mask_info.py |
sha256=OJLEky-Sfna6TGiBjLd1NLhzVJ1Q6AVp1XpQvsRinxM
|
32846 |
| jax/experimental/roofline/__init__.py |
sha256=ftyNH95ANQHFYSq9hsweKfJNP47sfipMm-3XFsr32yw
|
1260 |
| jax/experimental/roofline/roofline.py |
sha256=hobbkBeBhgGGzSDPWiR-X6f_7V_amtlPd0jl65k9S8k
|
9964 |
| jax/experimental/roofline/rooflines.py |
sha256=_1MGFjZaRvcZlN-E_C6nLFVfq6oEPN4slTQ1Dm-K8DQ
|
8169 |
| jax/experimental/sparse/__init__.py |
sha256=iYfDK0sdq3ErijcE7Va7cRjEHjzuJ3ykCwnHSYxNMPs
|
11186 |
| jax/experimental/sparse/_base.py |
sha256=qyiBgPa12uwS1nLpSDw8dTD1VEq0v9P2_CGTNyPCA_g
|
3239 |
| jax/experimental/sparse/_lowerings.py |
sha256=Wgpt5l3eCB6Vv8PKOrNUh6A_7i4JJ4xyhZxeI7zMvgk
|
7666 |
| jax/experimental/sparse/ad.py |
sha256=YT1jU0dsB3VoX5d0QUx9ihNUx2L_HfAh6ouJuEdhmXE
|
7784 |
| jax/experimental/sparse/api.py |
sha256=DrqEZmlIr6n6kpJjNli9J_rqMX3UO3sPPW7E8_t-u4g
|
6596 |
| jax/experimental/sparse/bcoo.py |
sha256=tisc09hy8F4P5kt3plsAYw64CWCd4QrnXRdXOxAxFIk
|
129014 |
| jax/experimental/sparse/bcsr.py |
sha256=gs4LWREQFXYokwqNSA5_zmZrYAwGT6NKfpLIWTBjROc
|
38290 |
| jax/experimental/sparse/coo.py |
sha256=2bBCeU2NXmlp7rsGsOQvn--yTrYCbya_ss0K7IjpCsg
|
24143 |
| jax/experimental/sparse/csr.py |
sha256=bBNOl1KR8WwAW0MtiBNKfebV3W2qTW58mA7lsH6Kotg
|
24396 |
| jax/experimental/sparse/linalg.py |
sha256=MOtslQMaEgK7Ky7kktDOCjltE8J5vYtbtBQAMIAUQnA
|
23442 |
| jax/experimental/sparse/nm.py |
sha256=ypXA8WaigylNbaRqJRlfXyurJC00f41VF1jxqUAiI3I
|
8568 |
| jax/experimental/sparse/random.py |
sha256=702kXl0gkbcSBthBBbyD3JAUinP_PlRDPCLMZHCdgcA
|
3882 |
| jax/experimental/sparse/test_util.py |
sha256=4JlLtN_KEl_bzF-ts9KRSbMNbzUsjEB_6IyQAhw7WQA
|
9428 |
| jax/experimental/sparse/transform.py |
sha256=VCDJJDLbkJgtcn5OEkM_mP9kv0QMD4FSSM8hpF7oxg8
|
36876 |
| jax/experimental/sparse/util.py |
sha256=TcMp_1-74yfTUU1q35lXL5ufjSBZ12xFof3-Vpnbd8E
|
4328 |
| jax/extend/__init__.py |
sha256=cL0QjoXSmMpM_j6_QBlKN6dtl30zMbvHqYA3eoHsd4g
|
1630 |
| jax/extend/backend.py |
sha256=ODjEVf5_5U37PZk5WVp8wI8thqOZO8XAdRV6Rortr8E
|
1054 |
| jax/extend/ffi.py |
sha256=Bo8zJigDbyClf75aMNeDPKKJoWQTNdJsp2fvctaVH2g
|
915 |
| jax/extend/ifrt_programs.py |
sha256=oCXLRZL4g3Syl8tQsTnjQNw8AL0zfKBlhUIUQ6Fh9gc
|
801 |
| jax/extend/linear_util.py |
sha256=Ie3ii7eVvtHFWsbtlOVu3o_T4jzP_JiXcKDnl-QKm6E
|
1080 |
| jax/extend/random.py |
sha256=w1OM6RCcGwmzIr9bdGJJTNKIcMO4ypXbzmeEORlBHjQ
|
1083 |
| jax/extend/source_info_util.py |
sha256=IxZYjLDlwOH1uE3A_vc25dy25QZ6FABMb4RqrkiA5hU
|
1215 |
| jax/extend/core/__init__.py |
sha256=o_q4S8jW_Tc0986SMqpjcifigZq-Ez2OhFZo6aBLwzM
|
1036 |
| jax/extend/core/primitives.py |
sha256=Ja7N3kkrO4Qf1HOj5_KSLy9GUfgQfmpthYNEvdaKGGA
|
6422 |
| jax/extend/mlir/__init__.py |
sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc
|
581 |
| jax/extend/mlir/ir.py |
sha256=1IirtLROpBL2nVDNSlP_3c6u09xXaKLqBkzF1JtmGz4
|
631 |
| jax/extend/mlir/passmanager.py |
sha256=1rfuCocMdFlx6It95ynBCP-8JC5xEAEG7rBdLJ7SpFI
|
640 |
| jax/extend/mlir/dialects/__init__.py |
sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc
|
581 |
| jax/extend/mlir/dialects/arith.py |
sha256=Au6mT9C4OrdpaGGwJ4_bauTFjGFQKF7wilv62mxzP1E
|
643 |
| jax/extend/mlir/dialects/builtin.py |
sha256=Y1QDRes-Arh7uBLJpySSqygBwUovalHyBq71m3uPbiQ
|
645 |
| jax/extend/mlir/dialects/chlo.py |
sha256=sfxvYEEBgGVd8fDXOlyfGtOLCCAW0xquLrSlTJFA81A
|
642 |
| jax/extend/mlir/dialects/func.py |
sha256=sCnRWz5gKtZ8Cf5NpeCIQ0zVh-nM6H95ttjzksTRrXk
|
642 |
| jax/extend/mlir/dialects/math.py |
sha256=G2gFkarEChMUEjv-vjDc4O09e-1orPhh_zHJhjbPPRQ
|
642 |
| jax/extend/mlir/dialects/memref.py |
sha256=nPR3oOrsENZ-mHjHKFSD48PpQMZHsVLfUVCwomBhv_w
|
644 |
| jax/extend/mlir/dialects/scf.py |
sha256=hmcJIF7RUqzeeP_NIOampnvCB5Jb6T8Xex4T03IeKrI
|
641 |
| jax/extend/mlir/dialects/sdy.py |
sha256=rtDlK5Z2OMDjBduAaRn08XBH-yGdZkvZogSB7RoFxBc
|
746 |
| jax/extend/mlir/dialects/sparse_tensor.py |
sha256=TaFY58y273rRITr8WxSM7LB1-sTAa03dCAjH4F8ZWaE
|
651 |
| jax/extend/mlir/dialects/stablehlo.py |
sha256=AbJQc7zghgrJh-BgRB31L7iHrNLVaEk4AELejcq7QiE
|
647 |
| jax/extend/mlir/dialects/vector.py |
sha256=xP03nScJg-t7olD8_P8SiFGVnV6Qi0Fwn9uGK0MHOXI
|
644 |
| jax/image/__init__.py |
sha256=cthOdx-HsZ6SgvW19tdNCDHmC4Proc8da1hY1hRztTI
|
1027 |
| jax/interpreters/__init__.py |
sha256=HNgKpgKpz4svFyTlHE9wMrDp5iQtg6XeeUshYBOnRko
|
690 |
| jax/interpreters/ad.py |
sha256=p0gJFj9VvrsZbZavI-xg00Y-PzEu5Rw_6d3upq5u92Y
|
2759 |
| jax/interpreters/batching.py |
sha256=QqKOP4uX4F1PSf96ZrXi6x3DOIrU8wyWqXGTlt1nC2g
|
2574 |
| jax/interpreters/mlir.py |
sha256=9Kg6EyCFqLqVIxvAo6euAo_o3UYVmZxJcfWPgcFSrBk
|
2924 |
| jax/interpreters/partial_eval.py |
sha256=Qjptpf1zUvUt6mUK-RF3JqqY3Aq9uBIB4uH2JZwBb2I
|
4163 |
| jax/interpreters/pxla.py |
sha256=VN0Mlb4IreJ9Tqez8i-QWn1EXVcrDWTLPeioTkeL-qE
|
1801 |
| jax/interpreters/xla.py |
sha256=NLL4fnTDuhZMHIiAUu5Nk1R9zlo7q3AlNd7ym4flf7U
|
1613 |
| jax/lax/__init__.py |
sha256=3uTvp2K9ET7argpl5sxaxwuOZMI29nAi4vaOMRCTAhI
|
10814 |
| jax/lax/linalg.py |
sha256=xvbvXnAOPZ_z97hjoaYzqlyM9verSaMqaL5ZrEsulK0
|
1397 |
| jax/lib/__init__.py |
sha256=wy3yxhlsb9k2th9Dfsr6eS59L2qLIxH_IoTM4cPfx3M
|
774 |
| jax/lib/xla_bridge.py |
sha256=SrvAZ1XaonXyv1MmJv6JT4_eg9q5aa8Gk8Gj2qwH24w
|
1557 |
| jax/lib/xla_client.py |
sha256=ndg7xbw7-DXqoLKcT3RaWFtdEFXpzoWNTrOqG2SlJFY
|
3886 |
| jax/lib/xla_extension.py |
sha256=xuSR42eY6codcchsSZx8rJZ2MdokxbJDNZwW9Xldz6o
|
1942 |
| jax/nn/__init__.py |
sha256=4LyqLzosWMtCv_zCd-9Gt9XmfCohbduCQrN_cRPqvVg
|
1562 |
| jax/nn/initializers.py |
sha256=finfAJmOEZdPIzY2roSIBTOmC4rKEW-rchpH9d6Dc2g
|
1469 |
| jax/numpy/__init__.py |
sha256=I2cC5hf3hBIzn8sR1l8Qm8dl-y0qRe-4-ZxvgUsZ_kc
|
12420 |
| jax/numpy/__init__.pyi |
sha256=3ohL_CEiQa-_Y_XfgkK3desgVe1vuWEtoAbjXTiEkmc
|
43446 |
| jax/numpy/fft.py |
sha256=iNatBIYRSwpXquPSgtNRMe6-dhbKkv2yoH6UJ6PFoGM
|
1084 |
| jax/numpy/linalg.py |
sha256=bnzLmLpLYSDRIICkjCEsH1WmA7wNOdSKhmO25gPvTqI
|
1419 |
| jax/ops/__init__.py |
sha256=VIAczIKYAFWawHdNmLmptUYnFQiBwHwlsa4LpgGEbIc
|
870 |
| jax/scipy/__init__.py |
sha256=DXszF0t3ZbfDeoFEP_GyBb0OOr6F-4yqsZzmen7pY58
|
1474 |
| jax/scipy/fft.py |
sha256=PVKQ4r-vl4J9Ct6LrhZWZ7jMF7frIXggiO98Bh0oVIE
|
809 |
| jax/scipy/integrate.py |
sha256=H9iq7YKgsZQ1d0SUvGcgPnEgD1TBdWJXHYjfUOGzT8Y
|
777 |
| jax/scipy/linalg.py |
sha256=jLoDaaxGZTcRu94dbVgk7qa4Nm-CHu_L8rMsvEIwzWI
|
1350 |
| jax/scipy/ndimage.py |
sha256=ei-ZfRiDXrlY4nYC1SFL7nnGHFrC_emD_NAicGhH4vU
|
788 |
| jax/scipy/signal.py |
sha256=nW4h017jShZBLL3J2NMhLwlOv46wK1MhEPeKlPTj710
|
975 |
| jax/scipy/special.py |
sha256=hT-5s-5ki7yXeQG3ZO-GMTjSrx3JHla41UNnQ0Njwsk
|
1718 |
| jax/scipy/cluster/__init__.py |
sha256=sgIbEEx62OubzDO36E3EkqjbYY_3qsErPOTQE6lsj5E
|
768 |
| jax/scipy/cluster/vq.py |
sha256=lkLSlkWFi6TLjwuzTX7hVSiC1IkCYd3OWgn8YtIAJd0
|
776 |
| jax/scipy/interpolate/__init__.py |
sha256=RI5oUXPC-8tKmYURnnrnA-e1LDbnNVOVVqkGSpagEBM
|
760 |
| jax/scipy/optimize/__init__.py |
sha256=cAw1rT1e0Mu-22vBUuJb6Q-CQF8fbIUBlKYpynB7e2s
|
822 |
| jax/scipy/sparse/__init__.py |
sha256=ugltlihEjwgorQxgfziHC-03UeYy4JzdQmTxrt1cnTo
|
757 |
| jax/scipy/sparse/linalg.py |
sha256=wpjjS1G1UNISyhZVFoJMMgIkqs-L8ASdvtQAbvEBsuk
|
810 |
| jax/scipy/spatial/__init__.py |
sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc
|
581 |
| jax/scipy/spatial/transform.py |
sha256=cm7lKi8aUepI3cOwTnjrldhFh7hBv89CR_m3pFj3r2o
|
802 |
| jax/scipy/stats/__init__.py |
sha256=ssiwAm2AdZ81MPMM11015yMNogj4DGsNfWfq507yhlo
|
1995 |
| jax/scipy/stats/bernoulli.py |
sha256=llkwAlvFH0qNU1BHdxn9CXXOBAPg0ffnbW_WQQ4LKVg
|
819 |
| jax/scipy/stats/beta.py |
sha256=5UIf6_JtqVZfAl1dDc2N3K8SkG0OV1bBkC3vWpY8-Hs
|
851 |
| jax/scipy/stats/betabinom.py |
sha256=5T6965_jHQxG4oR2YocPQRKcDo5AGxkhIyrk2iz-Kk0
|
792 |
| jax/scipy/stats/binom.py |
sha256=XGO5jDk3nRRgGwHQU6KEl9_kQ5X2qF-2TLgvm3ZUvvs
|
659 |
| jax/scipy/stats/cauchy.py |
sha256=MfAMnK9eG2wcutA5FVAyhU6iVYSSNgzI1XFT1xtNIfg
|
881 |
| jax/scipy/stats/chi2.py |
sha256=f-6i-0twBiOrak26tc7bEnWgOODA3iFwEFgi_An9udo
|
851 |
| jax/scipy/stats/dirichlet.py |
sha256=FUHSEmXKmVUneXuxgaIZM28s6GxlusbymdRYEPnDiUE
|
792 |
| jax/scipy/stats/expon.py |
sha256=D4a9KT9p2u0vRjpBS6sQgZ95vuzyZLCh-Mon8HiNq4U
|
788 |
| jax/scipy/stats/gamma.py |
sha256=TvP4BE8MROMxQL6YNDQIHSaZCfxumHkjGbCXw9O2iUo
|
852 |
| jax/scipy/stats/gennorm.py |
sha256=hqwz7ueAgdtzs1v0LHTTro2PLKRDjWxTFuu9jXwXOyI
|
804 |
| jax/scipy/stats/geom.py |
sha256=pdp2tCdXOa3CUWyKJCyrFrMR7TBX0La5tVI4ALAETI4
|
787 |
| jax/scipy/stats/laplace.py |
sha256=7Lap3E3_neHYpG5mUFlYxZnwAT1J8mTrNPnn8BSEGrA
|
804 |
| jax/scipy/stats/logistic.py |
sha256=IoHe_TMB0CH5EMAXjmqRMP8LjR4lTS_x9Zf4K7K5e2g
|
845 |
| jax/scipy/stats/multinomial.py |
sha256=hfRYQL_GeszF4sJXx5Y416JgXoXVw5vxtQIt1NmkCDU
|
794 |
| jax/scipy/stats/multivariate_normal.py |
sha256=YYqtLLZmjsLWeCIoDpxUk-rHde-g9unfo6ofg44nSBQ
|
802 |
| jax/scipy/stats/nbinom.py |
sha256=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/tools/__init__.py |
sha256=BTyMF0yh0M75AsNwF_0NSd5gHnIVpxhsLznsWkxu3XQ
|
581 |
| jax/tools/build_utils.py |
sha256=nu6hxCk0o1Z06Tydm8msAqSmoZnjMOq6kY58qpCfHwk
|
4393 |
| jax/tools/colab_tpu.py |
sha256=IPEQ1HZB4xW0MRSPGDeFAw9GoMFaphNQ0kLRIwGypmo
|
890 |
| jax/tools/jax_to_ir.py |
sha256=Z8LkCMaXfqjYFSHPSTS3L5imebmYtV3NPLy7V1WMLPU
|
8776 |
| jax/tools/pgo_nsys_converter.py |
sha256=BCxXmHnvcorM62adEbOFNYcsnTyv-Hkqnf4K7lo_wt4
|
3413 |
| jax-0.4.38.dist-info/AUTHORS |
sha256=OOBQygrI1Zc94AYJycH1r_BSCwP38Hyv3sludltJ7hY
|
313 |
| jax-0.4.38.dist-info/LICENSE |
sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA
|
11358 |
| jax-0.4.38.dist-info/METADATA |
sha256=cjGX0cC4QvzUjo1aU72yjlv6FFVDiRNGRzP5iAagy74
|
22754 |
| jax-0.4.38.dist-info/WHEEL |
sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U
|
91 |
| jax-0.4.38.dist-info/top_level.txt |
sha256=DabAh8MXQ-HM0EtjKD-Wx_D6tTIW-_de73pdG_f_KTE
|
4 |
| jax-0.4.38.dist-info/RECORD |
— |
— |