Name | Last modified | Size | Description | |
---|---|---|---|---|
Parent Directory | - | |||
__init__.py | 2025-01-02 23:40 | 5.6K | ||
_src/ | 2025-01-02 23:40 | - | ||
abstract_arrays.py | 2025-01-02 23:40 | 758 | ||
ad_checkpoint.py | 2025-01-02 23:40 | 711 | ||
api_util.py | 2025-01-02 23:40 | 769 | ||
cloud_tpu_init.py | 2025-01-02 23:40 | 633 | ||
collect_profile.py | 2025-01-02 23:40 | 4.6K | ||
config.py | 2025-01-02 23:40 | 682 | ||
core.py | 2025-01-02 23:40 | 99K | ||
custom_batching.py | 2025-01-02 23:40 | 657 | ||
custom_derivatives.py | 2025-01-02 23:40 | 1.0K | ||
custom_transpose.py | 2025-01-02 23:40 | 644 | ||
debug.py | 2025-01-02 23:40 | 935 | ||
distributed.py | 2025-01-02 23:40 | 638 | ||
dlpack.py | 2025-01-02 23:40 | 653 | ||
dtypes.py | 2025-01-02 23:40 | 1.0K | ||
errors.py | 2025-01-02 23:40 | 972 | ||
example_libraries/ | 2025-01-02 23:40 | - | ||
experimental/ | 2025-01-02 23:40 | - | ||
flatten_util.py | 2025-01-02 23:40 | 629 | ||
image/ | 2025-01-02 23:40 | - | ||
interpreters/ | 2025-01-02 23:40 | - | ||
jaxpr_util.py | 2025-01-02 23:40 | 6.4K | ||
lax/ | 2025-01-02 23:40 | - | ||
lib/ | 2025-01-02 23:40 | - | ||
linear_util.py | 2025-01-02 23:40 | 12K | ||
nn/ | 2025-01-02 23:40 | - | ||
numpy/ | 2025-01-02 23:40 | - | ||
ops/ | 2025-01-02 23:40 | - | ||
prng.py | 2025-01-02 23:40 | 868 | ||
profiler.py | 2025-01-02 23:40 | 1.0K | ||
py.typed | 2025-01-02 23:40 | 0 | ||
random.py | 2025-01-02 23:40 | 6.4K | ||
scipy/ | 2025-01-02 23:40 | - | ||
sharding.py | 2025-01-02 23:40 | 868 | ||
stages.py | 2025-01-02 23:40 | 1.1K | ||
test_util.py | 2025-01-02 23:40 | 1.8K | ||
tools/ | 2025-01-02 23:40 | - | ||
tree_util.py | 2025-01-02 23:40 | 2.4K | ||
util.py | 2025-01-02 23:40 | 962 | ||
version.py | 2025-01-02 23:40 | 1.0K | ||