| .. |
|
__pycache__
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
array_serialization
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
colocated_python
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
compilation_cache
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
jax2tf
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
key_reuse
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
mosaic
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
pallas
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
roofline
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
source_mapper
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
sparse
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
__init__.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
buffer_callback.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
checkify.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
compute_on.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
custom_dce.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
custom_partitioning.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
host_callback.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
jet.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
layout.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
mesh_utils.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
multihost_utils.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
ode.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
pjit.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
profiler.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
rnn.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
scheduling_groups.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
serialize_executable.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
shard_alike.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
shard_map.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
topologies.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
transfer.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
x64_context.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
xla_metadata.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
