![[ICO]](/icons/blank.gif){kind=icon}
![[PARENTDIR]](/icons/back.gif){kind=icon}
![[TXT]](/icons/text.gif){kind=icon}
![[DIR]](/icons/folder.gif){kind=icon}
| Name | Last modified | Size | Description |
|---|---|---|---|---|
| Parent Directory | - | ||
| __init__.py | 2025-10-11 15:11 | 1.0K | |
| __pycache__/ | 2025-10-11 15:11 | - | |
| array_api/ | 2025-10-11 15:11 | - | |
| array_serialization/ | 2025-10-11 15:11 | - | |
| attrs.py | 2025-10-11 15:11 | 9.3K | |
| checkify.py | 2025-10-11 15:11 | 1.2K | |
| colocated_python/ | 2025-10-11 15:11 | - | |
| compilation_cache/ | 2025-10-11 15:11 | - | |
| compute_on.py | 2025-10-11 15:11 | 644 | |
| custom_partitioning.py | 2025-10-11 15:11 | 856 | |
| host_callback.py | 2025-10-11 15:11 | 1.2K | |
| jax2tf/ | 2025-10-11 15:11 | - | |
| jet.py | 2025-10-11 15:11 | 26K | |
| key_reuse/ | 2025-10-11 15:11 | - | |
| layout.py | 2025-10-11 15:11 | 675 | |
| mesh_utils.py | 2025-10-11 15:11 | 930 | |
| mosaic/ | 2025-10-11 15:11 | - | |
| multihost_utils.py | 2025-10-11 15:11 | 18K | |
| ode.py | 2025-10-11 15:11 | 10K | |
| pallas/ | 2025-10-11 15:11 | - | |
| pjit.py | 2025-10-11 15:11 | 749 | |
| profiler.py | 2025-10-11 15:11 | 1.4K | |
| rnn.py | 2025-10-11 15:11 | 18K | |
| roofline/ | 2025-10-11 15:11 | - | |
| serialize_executable.py | 2025-10-11 15:11 | 2.9K | |
| shard_alike.py | 2025-10-11 15:11 | 646 | |
| shard_map.py | 2025-10-11 15:11 | 88K | |
| sparse/ | 2025-10-11 15:11 | - | |
| topologies.py | 2025-10-11 15:11 | 2.1K | |
| x64_context.py | 2025-10-11 15:11 | 1.7K | |
| xla_metadata.py | 2025-10-11 15:11 | 658 | |