| Name | Last modified | Size | Description | |
|---|---|---|---|---|
| Parent Directory | - | |||
| x64_context.py | 2025-08-11 08:34 | 1.7K | ||
| topologies.py | 2025-08-11 08:34 | 2.1K | ||
| sparse/ | 2025-08-11 08:34 | - | ||
| shard_map.py | 2025-08-11 08:34 | 85K | ||
| shard_alike.py | 2025-08-11 08:34 | 646 | ||
| serialize_executable.py | 2025-08-11 08:34 | 3.0K | ||
| rnn.py | 2025-08-11 08:34 | 18K | ||
| profiler.py | 2025-08-11 08:34 | 1.4K | ||
| pjit.py | 2025-08-11 08:34 | 808 | ||
| pallas/ | 2025-08-11 08:34 | - | ||
| ode.py | 2025-08-11 08:34 | 10K | ||
| multihost_utils.py | 2025-08-11 08:34 | 18K | ||
| mosaic/ | 2025-08-11 08:34 | - | ||
| mesh_utils.py | 2025-08-11 08:34 | 28K | ||
| maps.py | 2025-08-11 08:34 | 1.6K | ||
| layout.py | 2025-08-11 08:34 | 675 | ||
| key_reuse/ | 2025-08-11 08:34 | - | ||
| jet.py | 2025-08-11 08:34 | 26K | ||
| jax2tf/ | 2025-08-11 08:34 | - | ||
| host_callback.py | 2025-08-11 08:34 | 83K | ||
| export/ | 2025-08-11 08:34 | - | ||
| custom_partitioning.py | 2025-08-11 08:34 | 22K | ||
| compute_on.py | 2025-08-11 08:34 | 644 | ||
| compilation_cache/ | 2025-08-11 08:34 | - | ||
| checkify.py | 2025-08-11 08:34 | 1.2K | ||
| attrs.py | 2025-08-11 08:34 | 9.2K | ||
| array_serialization/ | 2025-08-11 08:34 | - | ||
| array_api/ | 2025-08-11 08:34 | - | ||
| __pycache__/ | 2025-08-11 08:34 | - | ||
| __init__.py | 2025-08-11 08:34 | 930 | ||