Name | Last modified | Size | Description | |
---|---|---|---|---|
Parent Directory | - | |||
__init__.py | 2025-01-02 22:55 | 5.2K | ||
_src/ | 2025-01-02 22:55 | - | ||
abstract_arrays.py | 2025-01-02 22:55 | 752 | ||
ad_checkpoint.py | 2025-01-02 22:55 | 705 | ||
api_util.py | 2025-01-02 22:55 | 763 | ||
cloud_tpu_init.py | 2025-01-02 22:55 | 627 | ||
collect_profile.py | 2025-01-02 22:55 | 4.5K | ||
config.py | 2025-01-02 22:55 | 676 | ||
core.py | 2025-01-02 22:55 | 98K | ||
custom_batching.py | 2025-01-02 22:55 | 651 | ||
custom_derivatives.py | 2025-01-02 22:55 | 1.0K | ||
custom_transpose.py | 2025-01-02 22:55 | 638 | ||
debug.py | 2025-01-02 22:55 | 769 | ||
distributed.py | 2025-01-02 22:55 | 632 | ||
dlpack.py | 2025-01-02 22:55 | 647 | ||
dtypes.py | 2025-01-02 22:55 | 1.0K | ||
errors.py | 2025-01-02 22:55 | 966 | ||
example_libraries/ | 2025-01-02 22:55 | - | ||
experimental/ | 2025-01-02 22:55 | - | ||
flatten_util.py | 2025-01-02 22:55 | 623 | ||
image/ | 2025-01-02 22:55 | - | ||
interpreters/ | 2025-01-02 22:55 | - | ||
jaxpr_util.py | 2025-01-02 22:55 | 6.4K | ||
lax/ | 2025-01-02 22:55 | - | ||
lib/ | 2025-01-02 22:55 | - | ||
linear_util.py | 2025-01-02 22:55 | 12K | ||
nn/ | 2025-01-02 22:55 | - | ||
numpy/ | 2025-01-02 22:55 | - | ||
ops/ | 2025-01-02 22:55 | - | ||
prng.py | 2025-01-02 22:55 | 862 | ||
profiler.py | 2025-01-02 22:55 | 1.0K | ||
py.typed | 2025-01-02 22:55 | 0 | ||
random.py | 2025-01-02 22:55 | 5.8K | ||
scipy/ | 2025-01-02 22:55 | - | ||
stages.py | 2025-01-02 22:55 | 676 | ||
test_util.py | 2025-01-02 22:55 | 1.8K | ||
tools/ | 2025-01-02 22:55 | - | ||
tree_util.py | 2025-01-02 22:55 | 2.4K | ||
util.py | 2025-01-02 22:55 | 956 | ||
version.py | 2025-01-02 22:55 | 1.0K | ||