| .. |
|
__pycache__
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
clusters
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
cudnn
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
debugger
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
export
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
extend
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
image
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
internal_test_util
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
interpreters
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
lax
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
lib
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
nn
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
numpy
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
ops
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
pallas
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
scipy
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
state
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
third_party
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
tpu
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
__init__.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
abstract_arrays.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
ad_checkpoint.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
ad_util.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
api_util.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
api.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
array.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
basearray.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
basearray.pyi
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
blocked_sampler.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
buffer_callback.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
cache_key.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
callback.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
checkify.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
cloud_tpu_init.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
compilation_cache_interface.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
compilation_cache.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
compiler.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
compute_on.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
config.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
core.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
custom_api_util.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
custom_batching.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_derivatives.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
custom_partitioning_sharding_rule.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
custom_partitioning.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
custom_transpose.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
debugging.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
deprecations.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
dispatch.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
distributed.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
dlpack.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
dtypes.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
earray.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
effects.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
environment_info.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
error_check.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
errors.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
ffi.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
flatten_util.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
frozen_dict.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
hardware_utils.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
hashable_array.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
hijax.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
jaxpr_util.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
lax_reference.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
layout.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
lazy_loader.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
linear_util.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
logging_config.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
lru_cache.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
memory.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
mesh_utils.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
mesh.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
monitoring.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
named_sharding.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
op_shardings.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
partition_spec.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
path.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
pickle_util.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
pjit.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
pretty_printer.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
prng.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
profiler.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
public_test_util.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
random.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
ref.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 |
|
sharding_impls.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
sharding_specs.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
sharding.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
source_info_util.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
sourcemap.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
stages.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
test_loader.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
test_multiprocess.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
test_util.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
test_warning_util.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
tpu_custom_call.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
traceback_util.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
tree_util.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
tree.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
typing.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
util.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
xla_bridge.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
xla_metadata_lib.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
|
xla_metadata.py
|
POC_DEMO
|
2025-11-28 09:08:33 +05:30 |
