![[ICO]](/icons/blank.gif){kind=icon}
![[PARENTDIR]](/icons/back.gif){kind=icon}
![[TXT]](/icons/text.gif){kind=icon}
![[DIR]](/icons/folder.gif){kind=icon}
![[ ]](/icons/unknown.gif){kind=icon}
| Name | Last modified | Size | Description |
|---|---|---|---|---|
| Parent Directory | - | ||
| xla_metadata.py | 2025-02-28 16:52 | 1.9K | |
| xla_bridge.py | 2025-02-28 16:52 | 45K | |
| util.py | 2025-02-28 16:52 | 20K | |
| typing.py | 2025-02-28 16:52 | 3.2K | |
| tree_util.py | 2025-02-28 16:52 | 47K | |
| tree.py | 2025-02-28 16:52 | 12K | |
| traceback_util.py | 2025-02-28 16:52 | 8.9K | |
| tpu_custom_call.py | 2025-02-28 16:52 | 27K | |
| third_party/ | 2025-02-28 16:52 | - | |
| test_warning_util.py | 2025-02-28 16:52 | 4.0K | |
| test_util.py | 2025-02-28 16:52 | 84K | |
| state/ | 2025-02-28 16:52 | - | |
| stages.py | 2025-02-28 16:52 | 29K | |
| sourcemap.py | 2025-02-28 16:52 | 6.5K | |
| source_info_util.py | 2025-02-28 16:52 | 9.6K | |
| sharding_specs.py | 2025-02-28 16:52 | 8.4K | |
| sharding_impls.py | 2025-02-28 16:52 | 51K | |
| sharding.py | 2025-02-28 16:52 | 8.0K | |
| shard_alike.py | 2025-02-28 16:52 | 3.9K | |
| scipy/ | 2025-02-28 16:52 | - | |
| random.py | 2025-02-28 16:52 | 98K | |
| public_test_util.py | 2025-02-28 16:52 | 13K | |
| profiler.py | 2025-02-28 16:52 | 16K | |
| prng.py | 2025-02-28 16:52 | 44K | |
| pretty_printer.py | 2025-02-28 16:52 | 15K | |
| pjit.py | 2025-02-28 16:52 | 128K | |
| pickle_util.py | 2025-02-28 16:52 | 2.3K | |
| path.py | 2025-02-28 16:52 | 1.4K | |
| partition_spec.py | 2025-02-28 16:52 | 2.3K | |
| pallas/ | 2025-02-28 16:52 | - | |
| ops/ | 2025-02-28 16:52 | - | |
| op_shardings.py | 2025-02-28 16:52 | 3.8K | |
| numpy/ | 2025-02-28 16:52 | - | |
| nn/ | 2025-02-28 16:52 | - | |
| named_sharding.py | 2025-02-28 16:52 | 21K | |
| monitoring.py | 2025-02-28 16:52 | 5.1K | |
| mesh_utils.py | 2025-02-28 16:52 | 32K | |
| mesh.py | 2025-02-28 16:52 | 19K | |
| lru_cache.py | 2025-02-28 16:52 | 6.6K | |
| logging_config.py | 2025-02-28 16:52 | 4.3K | |
| linear_util.py | 2025-02-28 16:52 | 17K | |
| lib/ | 2025-02-28 16:52 | - | |
| lazy_loader.py | 2025-02-28 16:52 | 1.8K | |
| layout.py | 2025-02-28 16:52 | 5.2K | |
| lax_reference.py | 2025-02-28 16:52 | 19K | |
| lax/ | 2025-02-28 16:52 | - | |
| jaxpr_util.py | 2025-02-28 16:52 | 6.3K | |
| interpreters/ | 2025-02-28 16:52 | - | |
| internal_test_util/ | 2025-02-28 16:52 | - | |
| image/ | 2025-02-28 16:52 | - | |
| hardware_utils.py | 2025-02-28 16:52 | 1.7K | |
| flatten_util.py | 2025-02-28 16:52 | 3.5K | |
| ffi.py | 2025-02-28 16:52 | 30K | |
| extend/ | 2025-02-28 16:52 | - | |
| export/ | 2025-02-28 16:52 | - | |
| errors.py | 2025-02-28 16:52 | 24K | |
| error_check.py | 2025-02-28 16:52 | 2.8K | |
| environment_info.py | 2025-02-28 16:52 | 1.8K | |
| effects.py | 2025-02-28 16:52 | 4.6K | |
| earray.py | 2025-02-28 16:52 | 4.3K | |
| dtypes.py | 2025-02-28 16:52 | 36K | |
| dlpack.py | 2025-02-28 16:52 | 11K | |
| distributed.py | 2025-02-28 16:52 | 12K | |
| dispatch.py | 2025-02-28 16:52 | 24K | |
| deprecations.py | 2025-02-28 16:52 | 4.7K | |
| debugging.py | 2025-02-28 16:52 | 27K | |
| debugger/ | 2025-02-28 16:52 | - | |
| custom_transpose.py | 2025-02-28 16:52 | 8.5K | |
| custom_partitioning_sharding_rule.py | 2025-02-28 16:52 | 18K | |
| custom_partitioning.py | 2025-02-28 16:52 | 25K | |
| custom_derivatives.py | 2025-02-28 16:52 | 76K | |
| custom_dce.py | 2025-02-28 16:52 | 17K | |
| custom_batching.py | 2025-02-28 16:52 | 15K | |
| custom_api_util.py | 2025-02-28 16:52 | 876 | |
| cudnn/ | 2025-02-28 16:52 | - | |
| core.py | 2025-02-28 16:52 | 112K | |
| config.py | 2025-02-28 16:52 | 67K | |
| compute_on.py | 2025-02-28 16:52 | 1.9K | |
| compiler.py | 2025-02-28 16:52 | 29K | |
| compilation_cache_interface.py | 2025-02-28 16:52 | 887 | |
| compilation_cache.py | 2025-02-28 16:52 | 12K | |
| clusters/ | 2025-02-28 16:52 | - | |
| cloud_tpu_init.py | 2025-02-28 16:52 | 4.2K | |
| checkify.py | 2025-02-28 16:52 | 56K | |
| callback.py | 2025-02-28 16:52 | 33K | |
| cache_key.py | 2025-02-28 16:52 | 13K | |
| blocked_sampler.py | 2025-02-28 16:52 | 5.5K | |
| basearray.pyi | 2025-02-28 16:52 | 13K | |
| basearray.py | 2025-02-28 16:52 | 6.4K | |
| array.py | 2025-02-28 16:52 | 49K | |
| api_util.py | 2025-02-28 16:52 | 28K | |
| api.py | 2025-02-28 16:52 | 121K | |
| ad_util.py | 2025-02-28 16:52 | 4.1K | |
| ad_checkpoint.py | 2025-02-28 16:52 | 39K | |
| abstract_arrays.py | 2025-02-28 16:52 | 2.8K | |
| __pycache__/ | 2025-02-28 16:52 | - | |
| __init__.py | 2025-02-28 16:52 | 581 | |