File | Mode | Size |
---|---|---|
experimental | ||
image | ||
interpreters | ||
lax | ||
lib | ||
nn | ||
numpy | ||
ops | ||
scipy | ||
third_party | ||
tools | ||
BUILD | -rw-r--r-- | 2.4 KB |
__init__.py | -rw-r--r-- | 2.3 KB |
abstract_arrays.py | -rw-r--r-- | 2.4 KB |
ad_util.py | -rw-r--r-- | 2.4 KB |
api.py | -rw-r--r-- | 104.5 KB |
api_util.py | -rw-r--r-- | 5.7 KB |
config.py | -rw-r--r-- | 5.0 KB |
core.py | -rw-r--r-- | 51.5 KB |
custom_derivatives.py | -rw-r--r-- | 24.6 KB |
dlpack.py | -rw-r--r-- | 2.0 KB |
dtypes.py | -rw-r--r-- | 7.9 KB |
flatten_util.py | -rw-r--r-- | 1.4 KB |
jaxpr_util.py | -rw-r--r-- | 3.6 KB |
lax_linalg.py | -rw-r--r-- | 37.0 KB |
lax_reference.py | -rw-r--r-- | 16.3 KB |
lazy.py | -rw-r--r-- | 9.1 KB |
linear_util.py | -rw-r--r-- | 8.8 KB |
pprint_util.py | -rw-r--r-- | 1.7 KB |
profiler.py | -rw-r--r-- | 5.4 KB |
random.py | -rw-r--r-- | 55.4 KB |
source_info_util.py | -rw-r--r-- | 1.8 KB |
test_util.py | -rw-r--r-- | 31.0 KB |
traceback_util.py | -rw-r--r-- | 5.8 KB |
tree_util.py | -rw-r--r-- | 9.8 KB |
util.py | -rw-r--r-- | 5.9 KB |
version.py | -rw-r--r-- | 598 bytes |