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