![[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-07-22 06:26 | 930 | |
| __pycache__/ | 2025-07-22 06:26 | - | |
| array_api/ | 2025-07-22 06:26 | - | |
| array_serialization/ | 2025-07-22 06:26 | - | |
| attrs.py | 2025-07-22 06:26 | 9.2K | |
| checkify.py | 2025-07-22 06:26 | 1.2K | |
| compilation_cache/ | 2025-07-22 06:26 | - | |
| compute_on.py | 2025-07-22 06:26 | 644 | |
| custom_partitioning.py | 2025-07-22 06:26 | 22K | |
| export/ | 2025-07-22 06:26 | - | |
| host_callback.py | 2025-07-22 06:26 | 83K | |
| jax2tf/ | 2025-07-22 06:26 | - | |
| jet.py | 2025-07-22 06:26 | 26K | |
| key_reuse/ | 2025-07-22 06:26 | - | |
| layout.py | 2025-07-22 06:26 | 675 | |
| maps.py | 2025-07-22 06:26 | 1.6K | |
| mesh_utils.py | 2025-07-22 06:26 | 28K | |
| mosaic/ | 2025-07-22 06:26 | - | |
| multihost_utils.py | 2025-07-22 06:26 | 18K | |
| ode.py | 2025-07-22 06:26 | 10K | |
| pallas/ | 2025-07-22 06:26 | - | |
| pjit.py | 2025-07-22 06:26 | 808 | |
| profiler.py | 2025-07-22 06:26 | 1.4K | |
| rnn.py | 2025-07-22 06:26 | 18K | |
| serialize_executable.py | 2025-07-22 06:26 | 3.0K | |
| shard_alike.py | 2025-07-22 06:26 | 646 | |
| shard_map.py | 2025-07-22 06:26 | 85K | |
| sparse/ | 2025-07-22 06:26 | - | |
| topologies.py | 2025-07-22 06:26 | 2.1K | |
| x64_context.py | 2025-07-22 06:26 | 1.7K | |