choice
  1. jax
  2. jax
Name
..
experimental
image
interpreters
lax
lib
nn
numpy
ops
scipy
third_party
tools
BUILD
__init__.py
abstract_arrays.py
ad_util.py
api.py
api_util.py
config.py
core.py
custom_derivatives.py
dlpack.py
dtypes.py
flatten_util.py
jaxpr_util.py
lax_linalg.py
lax_reference.py
lazy.py
linear_util.py
pprint_util.py
profiler.py
random.py
source_info_util.py
test_util.py
traceback_util.py
tree_util.py
util.py
version.py