| Name | Size | Last Modified |
|---|---|---|
| ../ | - | |
| __init__.py | 744 B | 2026-02-14 16:43:31 |
| array.py | 10.6kB | 2026-02-14 16:43:31 |
| callback.py | 10.7kB | 2026-02-14 16:43:31 |
| checkify.py | 1.0kB | 2026-02-14 16:43:31 |
| compilation_cache/ | - | 2026-02-14 16:43:31 |
| gda_serialization/ | - | 2026-02-14 16:43:31 |
| global_device_array.py | 24.2kB | 2026-02-14 16:43:31 |
| host_callback.py | 85.5kB | 2026-02-14 16:43:31 |
| jax2tf/ | - | 2026-02-14 16:43:31 |
| jet.py | 25.0kB | 2026-02-14 16:43:31 |
| maps.py | 91.8kB | 2026-02-14 16:43:31 |
| mesh_utils.py | 14.0kB | 2026-02-14 16:43:31 |
| multihost_utils.py | 6.8kB | 2026-02-14 16:43:31 |
| ode.py | 10.4kB | 2026-02-14 16:43:31 |
| pjit.py | 76.7kB | 2026-02-14 16:43:31 |
| sharding.py | 12.6kB | 2026-02-14 16:43:31 |
| sparse/ | - | 2026-02-14 16:43:31 |
| x64_context.py | 1.7kB | 2026-02-14 16:43:31 |