| Name | Last modified | Size | Description | |
|---|---|---|---|---|
| Parent Directory | - | |||
| api.py | 2025-08-11 08:34 | 126K | ||
| core.py | 2025-08-11 08:34 | 120K | ||
| pjit.py | 2025-08-11 08:34 | 113K | ||
| random.py | 2025-08-11 08:34 | 97K | ||
| maps.py | 2025-08-11 08:34 | 85K | ||
| test_util.py | 2025-08-11 08:34 | 67K | ||
| config.py | 2025-08-11 08:34 | 64K | ||
| sharding_impls.py | 2025-08-11 08:34 | 59K | ||
| custom_derivatives.py | 2025-08-11 08:34 | 59K | ||
| checkify.py | 2025-08-11 08:34 | 52K | ||
| array.py | 2025-08-11 08:34 | 45K | ||
| tree_util.py | 2025-08-11 08:34 | 45K | ||
| prng.py | 2025-08-11 08:34 | 44K | ||
| xla_bridge.py | 2025-08-11 08:34 | 43K | ||
| ad_checkpoint.py | 2025-08-11 08:34 | 36K | ||
| dtypes.py | 2025-08-11 08:34 | 29K | ||
| stages.py | 2025-08-11 08:34 | 28K | ||
| api_util.py | 2025-08-11 08:34 | 27K | ||
| compiler.py | 2025-08-11 08:34 | 25K | ||
| debugging.py | 2025-08-11 08:34 | 24K | ||
| errors.py | 2025-08-11 08:34 | 24K | ||
| dispatch.py | 2025-08-11 08:34 | 21K | ||
| util.py | 2025-08-11 08:34 | 19K | ||
| callback.py | 2025-08-11 08:34 | 18K | ||
| lax_reference.py | 2025-08-11 08:34 | 18K | ||
| tpu_custom_call.py | 2025-08-11 08:34 | 17K | ||
| profiler.py | 2025-08-11 08:34 | 16K | ||
| pretty_printer.py | 2025-08-11 08:34 | 15K | ||
| linear_util.py | 2025-08-11 08:34 | 13K | ||
| mesh.py | 2025-08-11 08:34 | 11K | ||
| dlpack.py | 2025-08-11 08:34 | 11K | ||
| basearray.pyi | 2025-08-11 08:34 | 10K | ||
| cache_key.py | 2025-08-11 08:34 | 10K | ||
| custom_batching.py | 2025-08-11 08:34 | 10K | ||
| public_test_util.py | 2025-08-11 08:34 | 10K | ||
| compilation_cache.py | 2025-08-11 08:34 | 9.8K | ||
| source_info_util.py | 2025-08-11 08:34 | 9.7K | ||
| tree.py | 2025-08-11 08:34 | 9.1K | ||
| traceback_util.py | 2025-08-11 08:34 | 8.9K | ||
| distributed.py | 2025-08-11 08:34 | 8.7K | ||
| sharding_specs.py | 2025-08-11 08:34 | 8.4K | ||
| custom_transpose.py | 2025-08-11 08:34 | 8.1K | ||
| sharding.py | 2025-08-11 08:34 | 7.6K | ||
| sourcemap.py | 2025-08-11 08:34 | 6.5K | ||
| jaxpr_util.py | 2025-08-11 08:34 | 6.3K | ||
| lru_cache.py | 2025-08-11 08:34 | 5.8K | ||
| effects.py | 2025-08-11 08:34 | 4.6K | ||
| basearray.py | 2025-08-11 08:34 | 4.5K | ||
| earray.py | 2025-08-11 08:34 | 4.0K | ||
| monitoring.py | 2025-08-11 08:34 | 3.8K | ||
| op_shardings.py | 2025-08-11 08:34 | 3.8K | ||
| deprecations.py | 2025-08-11 08:34 | 3.8K | ||
| shard_alike.py | 2025-08-11 08:34 | 3.7K | ||
| ad_util.py | 2025-08-11 08:34 | 3.7K | ||
| flatten_util.py | 2025-08-11 08:34 | 3.5K | ||
| layout.py | 2025-08-11 08:34 | 3.3K | ||
| cloud_tpu_init.py | 2025-08-11 08:34 | 3.2K | ||
| typing.py | 2025-08-11 08:34 | 3.2K | ||
| abstract_arrays.py | 2025-08-11 08:34 | 2.6K | ||
| pickle_util.py | 2025-08-11 08:34 | 2.3K | ||
| gfile_cache.py | 2025-08-11 08:34 | 2.0K | ||
| logging_config.py | 2025-08-11 08:34 | 2.0K | ||
| environment_info.py | 2025-08-11 08:34 | 2.0K | ||
| compute_on.py | 2025-08-11 08:34 | 1.8K | ||
| hardware_utils.py | 2025-08-11 08:34 | 1.7K | ||
| partition_spec.py | 2025-08-11 08:34 | 1.6K | ||
| lazy_loader.py | 2025-08-11 08:34 | 1.3K | ||
| path.py | 2025-08-11 08:34 | 1.1K | ||
| custom_api_util.py | 2025-08-11 08:34 | 876 | ||
| compilation_cache_in..> | 2025-08-11 08:34 | 863 | ||
| __init__.py | 2025-08-11 08:34 | 581 | ||
| third_party/ | 2025-08-11 08:34 | - | ||
| state/ | 2025-08-11 08:34 | - | ||
| scipy/ | 2025-08-11 08:34 | - | ||
| pallas/ | 2025-08-11 08:34 | - | ||
| ops/ | 2025-08-11 08:34 | - | ||
| numpy/ | 2025-08-11 08:34 | - | ||
| nn/ | 2025-08-11 08:34 | - | ||
| lib/ | 2025-08-11 08:34 | - | ||
| lax/ | 2025-08-11 08:34 | - | ||
| interpreters/ | 2025-08-11 08:34 | - | ||
| internal_test_util/ | 2025-08-11 08:34 | - | ||
| image/ | 2025-08-11 08:34 | - | ||
| extend/ | 2025-08-11 08:34 | - | ||
| export/ | 2025-08-11 08:34 | - | ||
| debugger/ | 2025-08-11 08:34 | - | ||
| cudnn/ | 2025-08-11 08:34 | - | ||
| clusters/ | 2025-08-11 08:34 | - | ||
| __pycache__/ | 2025-08-11 08:34 | - | ||