| Name | Last modified | Size | Description | |
|---|---|---|---|---|
| Parent Directory | - | |||
| __init__.py | 2025-02-02 11:49 | 1.0K | ||
| __pycache__/ | 2025-02-02 11:49 | - | ||
| array_api/ | 2025-02-02 11:49 | - | ||
| array_serialization/ | 2025-02-02 11:49 | - | ||
| attrs.py | 2025-02-02 11:49 | 9.2K | ||
| checkify.py | 2025-02-02 11:49 | 1.2K | ||
| compilation_cache/ | 2025-02-02 11:49 | - | ||
| compute_on.py | 2025-02-02 11:49 | 644 | ||
| custom_partitioning.py | 2025-02-02 11:49 | 856 | ||
| export/ | 2025-02-02 11:49 | - | ||
| host_callback.py | 2025-02-02 11:49 | 88K | ||
| jax2tf/ | 2025-02-02 11:49 | - | ||
| jet.py | 2025-02-02 11:49 | 26K | ||
| key_reuse/ | 2025-02-02 11:49 | - | ||
| layout.py | 2025-02-02 11:49 | 675 | ||
| mesh_utils.py | 2025-02-02 11:49 | 930 | ||
| mosaic/ | 2025-02-02 11:49 | - | ||
| multihost_utils.py | 2025-02-02 11:49 | 18K | ||
| ode.py | 2025-02-02 11:49 | 10K | ||
| pallas/ | 2025-02-02 11:49 | - | ||
| pjit.py | 2025-02-02 11:49 | 808 | ||
| profiler.py | 2025-02-02 11:49 | 1.4K | ||
| rnn.py | 2025-02-02 11:49 | 18K | ||
| serialize_executable.py | 2025-02-02 11:49 | 3.0K | ||
| shard_alike.py | 2025-02-02 11:49 | 646 | ||
| shard_map.py | 2025-02-02 11:49 | 93K | ||
| sparse/ | 2025-02-02 11:49 | - | ||
| topologies.py | 2025-02-02 11:49 | 2.1K | ||
| x64_context.py | 2025-02-02 11:49 | 1.7K | ||
| xla_metadata.py | 2025-02-02 11:49 | 658 | ||