swh:1:snp:71afc25eb6e6e055a37a962e6b91010ec35e397f
History
Tip revision: 1189c3c62f39b6ba0ec464c21201bcb5b5041238 authored by jax authors on 15 April 2022, 19:28:57 UTC
Merge pull request #10312 from hawkinsp:jaxlib
Tip revision: 1189c3c
File Mode Size
_src
example_libraries
experimental
image
interpreters
lax
lib
nn
numpy
ops
scipy
tools
__init__.py -rw-r--r-- 4.9 KB
abstract_arrays.py -rw-r--r-- 773 bytes
ad_checkpoint.py -rw-r--r-- 726 bytes
api_util.py -rw-r--r-- 784 bytes
cloud_tpu_init.py -rw-r--r-- 648 bytes
config.py -rw-r--r-- 697 bytes
core.py -rw-r--r-- 85.9 KB
custom_batching.py -rw-r--r-- 672 bytes
custom_derivatives.py -rw-r--r-- 1.0 KB
custom_transpose.py -rw-r--r-- 659 bytes
distributed.py -rw-r--r-- 641 bytes
dlpack.py -rw-r--r-- 668 bytes
dtypes.py -rw-r--r-- 1.0 KB
errors.py -rw-r--r-- 987 bytes
flatten_util.py -rw-r--r-- 644 bytes
jaxpr_util.py -rw-r--r-- 6.4 KB
linear_util.py -rw-r--r-- 11.0 KB
prng.py -rw-r--r-- 884 bytes
profiler.py -rw-r--r-- 1.1 KB
py.typed -rw-r--r-- 0 bytes
random.py -rw-r--r-- 3.7 KB
stages.py -rw-r--r-- 697 bytes
test_util.py -rw-r--r-- 1.3 KB
tree_util.py -rw-r--r-- 2.5 KB
util.py -rw-r--r-- 977 bytes
version.py -rw-r--r-- 864 bytes

back to top