| .. |
|
__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 |
