pmap
  1. jax
  2. jax
Name
..
experimental
interpreters
lax
lib
numpy
ops
scipy
BUILD
__init__.py
abstract_arrays.py
ad_util.py
api.py
api_util.py
config.py
core.py
flatten_util.py
lax_linalg.py
lax_reference.py
linear_util.py
pprint_util.py
random.py
test_util.py
tree_util.py
util.py
version.py