https://github.com/google/jax
History
Tip revision: b4f47c4f3888fe4677aa8d4efe6883ee1a3f7bf0 authored by jax authors on 18 March 2022, 18:50:34 UTC
Merge pull request #9959 from mattjj:fix-9923
Tip revision: b4f47c4
File Mode Size
_src
example_libraries
experimental
image
interpreters
lax
lib
nn
numpy
ops
scipy
tools
__init__.py -rw-r--r-- 4.8 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-- 82.4 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-- 10.4 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-- 2.0 KB
tree_util.py -rw-r--r-- 2.5 KB
util.py -rw-r--r-- 977 bytes
version.py -rw-r--r-- 633 bytes

back to top