swh:1:snp:71afc25eb6e6e055a37a962e6b91010ec35e397f
History
Tip revision: e27af48510b57e5d9f1ae17f256bd67fa9db7941 authored by jax authors on 06 October 2023, 20:21:49 UTC
Merge pull request #17989 from skye:version
Tip revision: e27af48
File Mode Size
_src
example_libraries
experimental
extend
image
interpreters
lax
lib
nn
numpy
ops
scipy
tools
BUILD -rw-r--r-- 19.7 KB
__init__.py -rw-r--r-- 9.2 KB
ad_checkpoint.py -rw-r--r-- 802 bytes
api_util.py -rw-r--r-- 769 bytes
cloud_tpu_init.py -rw-r--r-- 633 bytes
collect_profile.py -rw-r--r-- 4.6 KB
config.py -rw-r--r-- 696 bytes
core.py -rw-r--r-- 6.3 KB
custom_batching.py -rw-r--r-- 657 bytes
custom_derivatives.py -rw-r--r-- 1.3 KB
custom_transpose.py -rw-r--r-- 644 bytes
debug.py -rw-r--r-- 1.1 KB
distributed.py -rw-r--r-- 638 bytes
dlpack.py -rw-r--r-- 653 bytes
dtypes.py -rw-r--r-- 1.1 KB
errors.py -rw-r--r-- 1.2 KB
flatten_util.py -rw-r--r-- 645 bytes
linear_util.py -rw-r--r-- 2.4 KB
monitoring.py -rw-r--r-- 1.2 KB
prng.py -rw-r--r-- 2.3 KB
profiler.py -rw-r--r-- 1.1 KB
py.typed -rw-r--r-- 0 bytes
random.py -rw-r--r-- 8.2 KB
sharding.py -rw-r--r-- 1.2 KB
stages.py -rw-r--r-- 1.2 KB
test_util.py -rw-r--r-- 835 bytes
tree_util.py -rw-r--r-- 4.0 KB
typing.py -rw-r--r-- 2.9 KB
util.py -rw-r--r-- 1.1 KB
version.py -rw-r--r-- 4.9 KB

back to top