.. |
__pycache__
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
_src
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
example_libraries
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
experimental
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
extend
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
image
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
interpreters
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
lax
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
lib
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
nn
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
numpy
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
ops
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
scipy
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
tools
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
__init__.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
abstract_arrays.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
ad_checkpoint.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
api_util.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
cloud_tpu_init.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
collect_profile.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
config.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
core.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
custom_batching.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
custom_derivatives.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
custom_transpose.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
debug.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
distributed.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
dlpack.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
dtypes.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
errors.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
flatten_util.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
jaxpr_util.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
linear_util.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
monitoring.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
prng.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
profiler.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
py.typed
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
random.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
sharding.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
stages.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
test_util.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
tree_util.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
typing.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
util.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
version.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |