python-jax
Version 0.7.0-1 from cachyos (any)
- Description
- Composable transformations of Python+NumPy programs
- Homepage
- https://github.com/jax-ml/jax/
- License(s)
- Apache-2.0
- Build Date
- Packager
- CachyOS <admin@cachyos.org>
- Package Size
- 5.2 MB
- Installed Size
- 23.79 MB
- SHA256 Sum
- c86e532eae28245959f834ccc051a48ea6417ebede6f76f109d2111b1c9c41d1
- Provides
- N/A
- Conflicts With
- N/A
- Replaces
- N/A
- Package Files
- usr/usr/lib/usr/lib/python3.13/usr/lib/python3.13/site-packages/usr/lib/python3.13/site-packages/jax-0.7.0.dist-info/usr/lib/python3.13/site-packages/jax-0.7.0.dist-info/METADATAusr/lib/python3.13/site-packages/jax-0.7.0.dist-info/RECORDusr/lib/python3.13/site-packages/jax-0.7.0.dist-info/WHEELusr/lib/python3.13/site-packages/jax-0.7.0.dist-info/licenses/usr/lib/python3.13/site-packages/jax-0.7.0.dist-info/licenses/AUTHORSusr/lib/python3.13/site-packages/jax-0.7.0.dist-info/licenses/LICENSEusr/lib/python3.13/site-packages/jax-0.7.0.dist-info/top_level.txtusr/lib/python3.13/site-packages/jax/usr/lib/python3.13/site-packages/jax/__init__.pyusr/lib/python3.13/site-packages/jax/__pycache__/usr/lib/python3.13/site-packages/jax/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/__pycache__/ad_checkpoint.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/__pycache__/api_util.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/__pycache__/cloud_tpu_init.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/__pycache__/collect_profile.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/__pycache__/core.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/__pycache__/custom_batching.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/__pycache__/custom_derivatives.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/__pycache__/custom_transpose.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/__pycache__/debug.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/__pycache__/distributed.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/__pycache__/dlpack.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/__pycache__/dtypes.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/__pycache__/errors.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/__pycache__/export.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/__pycache__/ffi.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/__pycache__/flatten_util.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/__pycache__/monitoring.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/__pycache__/profiler.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/__pycache__/random.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/__pycache__/sharding.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/__pycache__/stages.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/__pycache__/test_util.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/__pycache__/tree.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/__pycache__/tree_util.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/__pycache__/typing.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/__pycache__/util.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/__pycache__/version.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/usr/lib/python3.13/site-packages/jax/_src/__init__.pyusr/lib/python3.13/site-packages/jax/_src/__pycache__/usr/lib/python3.13/site-packages/jax/_src/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/abstract_arrays.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/ad_checkpoint.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/ad_util.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/api.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/api_util.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/array.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/basearray.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/blocked_sampler.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/buffer_callback.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/cache_key.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/callback.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/checkify.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/cloud_tpu_init.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/compilation_cache.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/compilation_cache_interface.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/compiler.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/compute_on.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/config.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/core.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/custom_api_util.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/custom_batching.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/custom_dce.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/custom_derivatives.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/custom_partitioning.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/custom_partitioning_sharding_rule.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/custom_transpose.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/debugging.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/deprecations.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/dispatch.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/distributed.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/dlpack.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/dtypes.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/earray.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/effects.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/environment_info.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/error_check.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/errors.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/ffi.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/flatten_util.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/frozen_dict.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/hardware_utils.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/hashable_array.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/jaxpr_util.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/lax_reference.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/layout.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/lazy_loader.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/linear_util.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/logging_config.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/lru_cache.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/mesh.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/mesh_utils.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/monitoring.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/named_sharding.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/op_shardings.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/partition_spec.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/path.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/pickle_util.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/pjit.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/pretty_printer.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/prng.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/profiler.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/public_test_util.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/random.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/shard_alike.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/shard_map.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/sharding.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/sharding_impls.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/sharding_specs.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/source_info_util.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/sourcemap.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/stages.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/test_loader.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/test_multiprocess.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/test_util.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/test_warning_util.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/tpu_custom_call.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/traceback_util.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/tree.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/tree_util.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/typing.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/util.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/xla_bridge.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/xla_metadata.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/__pycache__/xla_metadata_lib.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/abstract_arrays.pyusr/lib/python3.13/site-packages/jax/_src/ad_checkpoint.pyusr/lib/python3.13/site-packages/jax/_src/ad_util.pyusr/lib/python3.13/site-packages/jax/_src/api.pyusr/lib/python3.13/site-packages/jax/_src/api_util.pyusr/lib/python3.13/site-packages/jax/_src/array.pyusr/lib/python3.13/site-packages/jax/_src/basearray.pyusr/lib/python3.13/site-packages/jax/_src/basearray.pyiusr/lib/python3.13/site-packages/jax/_src/blocked_sampler.pyusr/lib/python3.13/site-packages/jax/_src/buffer_callback.pyusr/lib/python3.13/site-packages/jax/_src/cache_key.pyusr/lib/python3.13/site-packages/jax/_src/callback.pyusr/lib/python3.13/site-packages/jax/_src/checkify.pyusr/lib/python3.13/site-packages/jax/_src/cloud_tpu_init.pyusr/lib/python3.13/site-packages/jax/_src/clusters/usr/lib/python3.13/site-packages/jax/_src/clusters/__init__.pyusr/lib/python3.13/site-packages/jax/_src/clusters/__pycache__/usr/lib/python3.13/site-packages/jax/_src/clusters/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/clusters/__pycache__/cloud_tpu_cluster.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/clusters/__pycache__/cluster.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/clusters/__pycache__/k8s_cluster.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/clusters/__pycache__/mpi4py_cluster.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/clusters/__pycache__/ompi_cluster.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/clusters/__pycache__/slurm_cluster.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/clusters/cloud_tpu_cluster.pyusr/lib/python3.13/site-packages/jax/_src/clusters/cluster.pyusr/lib/python3.13/site-packages/jax/_src/clusters/k8s_cluster.pyusr/lib/python3.13/site-packages/jax/_src/clusters/mpi4py_cluster.pyusr/lib/python3.13/site-packages/jax/_src/clusters/ompi_cluster.pyusr/lib/python3.13/site-packages/jax/_src/clusters/slurm_cluster.pyusr/lib/python3.13/site-packages/jax/_src/compilation_cache.pyusr/lib/python3.13/site-packages/jax/_src/compilation_cache_interface.pyusr/lib/python3.13/site-packages/jax/_src/compiler.pyusr/lib/python3.13/site-packages/jax/_src/compute_on.pyusr/lib/python3.13/site-packages/jax/_src/config.pyusr/lib/python3.13/site-packages/jax/_src/core.pyusr/lib/python3.13/site-packages/jax/_src/cudnn/usr/lib/python3.13/site-packages/jax/_src/cudnn/__init__.pyusr/lib/python3.13/site-packages/jax/_src/cudnn/__pycache__/usr/lib/python3.13/site-packages/jax/_src/cudnn/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/cudnn/__pycache__/fused_attention_stablehlo.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/cudnn/__pycache__/fusion.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/cudnn/__pycache__/scaled_matmul_stablehlo.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/cudnn/fused_attention_stablehlo.pyusr/lib/python3.13/site-packages/jax/_src/cudnn/fusion.pyusr/lib/python3.13/site-packages/jax/_src/cudnn/scaled_matmul_stablehlo.pyusr/lib/python3.13/site-packages/jax/_src/custom_api_util.pyusr/lib/python3.13/site-packages/jax/_src/custom_batching.pyusr/lib/python3.13/site-packages/jax/_src/custom_dce.pyusr/lib/python3.13/site-packages/jax/_src/custom_derivatives.pyusr/lib/python3.13/site-packages/jax/_src/custom_partitioning.pyusr/lib/python3.13/site-packages/jax/_src/custom_partitioning_sharding_rule.pyusr/lib/python3.13/site-packages/jax/_src/custom_transpose.pyusr/lib/python3.13/site-packages/jax/_src/debugger/usr/lib/python3.13/site-packages/jax/_src/debugger/__init__.pyusr/lib/python3.13/site-packages/jax/_src/debugger/__pycache__/usr/lib/python3.13/site-packages/jax/_src/debugger/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/debugger/__pycache__/cli_debugger.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/debugger/__pycache__/colab_debugger.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/debugger/__pycache__/colab_lib.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/debugger/__pycache__/core.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/debugger/__pycache__/web_debugger.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/debugger/cli_debugger.pyusr/lib/python3.13/site-packages/jax/_src/debugger/colab_debugger.pyusr/lib/python3.13/site-packages/jax/_src/debugger/colab_lib.pyusr/lib/python3.13/site-packages/jax/_src/debugger/core.pyusr/lib/python3.13/site-packages/jax/_src/debugger/web_debugger.pyusr/lib/python3.13/site-packages/jax/_src/debugging.pyusr/lib/python3.13/site-packages/jax/_src/deprecations.pyusr/lib/python3.13/site-packages/jax/_src/dispatch.pyusr/lib/python3.13/site-packages/jax/_src/distributed.pyusr/lib/python3.13/site-packages/jax/_src/dlpack.pyusr/lib/python3.13/site-packages/jax/_src/dtypes.pyusr/lib/python3.13/site-packages/jax/_src/earray.pyusr/lib/python3.13/site-packages/jax/_src/effects.pyusr/lib/python3.13/site-packages/jax/_src/environment_info.pyusr/lib/python3.13/site-packages/jax/_src/error_check.pyusr/lib/python3.13/site-packages/jax/_src/errors.pyusr/lib/python3.13/site-packages/jax/_src/export/usr/lib/python3.13/site-packages/jax/_src/export/__init__.pyusr/lib/python3.13/site-packages/jax/_src/export/__pycache__/usr/lib/python3.13/site-packages/jax/_src/export/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/export/__pycache__/_export.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/export/__pycache__/serialization.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/export/__pycache__/serialization_generated.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/export/__pycache__/shape_poly.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/export/__pycache__/shape_poly_decision.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/export/_export.pyusr/lib/python3.13/site-packages/jax/_src/export/serialization.pyusr/lib/python3.13/site-packages/jax/_src/export/serialization_generated.pyusr/lib/python3.13/site-packages/jax/_src/export/shape_poly.pyusr/lib/python3.13/site-packages/jax/_src/export/shape_poly_decision.pyusr/lib/python3.13/site-packages/jax/_src/extend/usr/lib/python3.13/site-packages/jax/_src/extend/__init__.pyusr/lib/python3.13/site-packages/jax/_src/extend/__pycache__/usr/lib/python3.13/site-packages/jax/_src/extend/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/extend/__pycache__/random.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/extend/random.pyusr/lib/python3.13/site-packages/jax/_src/ffi.pyusr/lib/python3.13/site-packages/jax/_src/flatten_util.pyusr/lib/python3.13/site-packages/jax/_src/frozen_dict.pyusr/lib/python3.13/site-packages/jax/_src/hardware_utils.pyusr/lib/python3.13/site-packages/jax/_src/hashable_array.pyusr/lib/python3.13/site-packages/jax/_src/image/usr/lib/python3.13/site-packages/jax/_src/image/__init__.pyusr/lib/python3.13/site-packages/jax/_src/image/__pycache__/usr/lib/python3.13/site-packages/jax/_src/image/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/image/__pycache__/scale.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/image/scale.pyusr/lib/python3.13/site-packages/jax/_src/internal_test_util/usr/lib/python3.13/site-packages/jax/_src/internal_test_util/__init__.pyusr/lib/python3.13/site-packages/jax/_src/internal_test_util/__pycache__/usr/lib/python3.13/site-packages/jax/_src/internal_test_util/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/internal_test_util/__pycache__/deprecation_module.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/internal_test_util/__pycache__/export_back_compat_test_util.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/internal_test_util/__pycache__/lax_test_util.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/internal_test_util/__pycache__/test_harnesses.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/internal_test_util/deprecation_module.pyusr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_data/usr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_data/__init__.pyusr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/usr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/annotate_data_placement.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/cpu_cholesky_lapack_potrf.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/cpu_eig_lapack_geev.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/cpu_eigh_lapack_syev.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/cpu_hessenberg_lapack_gehrd.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/cpu_lu_lapack_getrf.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/cpu_qr_lapack_geqrf.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/cpu_schur_lapack_gees.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/cpu_svd_lapack_gesdd.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/cpu_triangular_solve_blas_trsm.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/cpu_tridiagonal_lapack_sytrd_hetrd.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/cpu_tridiagonal_solve_lapack_gtsv.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/cuda_eigh_cusolver_syev.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/cuda_lu_cusolver_getrf.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/cuda_lu_pivots_to_permutation.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/cuda_qr_cusolver_geqrf.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/cuda_svd_cusolver_gesvd.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/cuda_threefry2x32.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/cuda_tridiagonal_cusolver_sytrd.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/cuda_tridiagonal_solve.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/rocm_eigh_hipsolver_syev.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/shardy_sharding_ops_with_different_meshes.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/stablehlo_dynamic_approx_top_k.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/stablehlo_dynamic_rng_bit_generator.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/stablehlo_dynamic_top_k.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/tpu_ApproxTopK.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/tpu_Eigh.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/tpu_Lu.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/tpu_Qr.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/tpu_Sharding.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/tpu_stablehlo_dynamic_reduce_window.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_data/annotate_data_placement.pyusr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_data/cpu_cholesky_lapack_potrf.pyusr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_data/cpu_eig_lapack_geev.pyusr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_data/cpu_eigh_lapack_syev.pyusr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_data/cpu_hessenberg_lapack_gehrd.pyusr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_data/cpu_lu_lapack_getrf.pyusr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_data/cpu_qr_lapack_geqrf.pyusr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_data/cpu_schur_lapack_gees.pyusr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_data/cpu_svd_lapack_gesdd.pyusr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_data/cpu_triangular_solve_blas_trsm.pyusr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_data/cpu_tridiagonal_lapack_sytrd_hetrd.pyusr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_data/cpu_tridiagonal_solve_lapack_gtsv.pyusr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_data/cuda_eigh_cusolver_syev.pyusr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_data/cuda_lu_cusolver_getrf.pyusr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_data/cuda_lu_pivots_to_permutation.pyusr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_data/cuda_qr_cusolver_geqrf.pyusr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_data/cuda_svd_cusolver_gesvd.pyusr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_data/cuda_threefry2x32.pyusr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_data/cuda_tridiagonal_cusolver_sytrd.pyusr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_data/cuda_tridiagonal_solve.pyusr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_data/pallas/usr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_data/pallas/__init__.pyusr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_data/pallas/__pycache__/usr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_data/pallas/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_data/pallas/__pycache__/mosaic_gpu_add_one.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_data/pallas/__pycache__/mosaic_matmul.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_data/pallas/__pycache__/mosaic_semaphore_dma.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_data/pallas/__pycache__/triton_add_one.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_data/pallas/mosaic_gpu_add_one.pyusr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_data/pallas/mosaic_matmul.pyusr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_data/pallas/mosaic_semaphore_dma.pyusr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_data/pallas/triton_add_one.pyusr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_data/rocm_eigh_hipsolver_syev.pyusr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_data/shardy_sharding_ops_with_different_meshes.pyusr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_data/stablehlo_dynamic_approx_top_k.pyusr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_data/stablehlo_dynamic_rng_bit_generator.pyusr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_data/stablehlo_dynamic_top_k.pyusr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_data/tpu_ApproxTopK.pyusr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_data/tpu_Eigh.pyusr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_data/tpu_Lu.pyusr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_data/tpu_Qr.pyusr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_data/tpu_Sharding.pyusr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_data/tpu_stablehlo_dynamic_reduce_window.pyusr/lib/python3.13/site-packages/jax/_src/internal_test_util/export_back_compat_test_util.pyusr/lib/python3.13/site-packages/jax/_src/internal_test_util/lax_test_util.pyusr/lib/python3.13/site-packages/jax/_src/internal_test_util/lazy_loader_module/usr/lib/python3.13/site-packages/jax/_src/internal_test_util/lazy_loader_module/__init__.pyusr/lib/python3.13/site-packages/jax/_src/internal_test_util/lazy_loader_module/__pycache__/usr/lib/python3.13/site-packages/jax/_src/internal_test_util/lazy_loader_module/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/internal_test_util/lazy_loader_module/__pycache__/lazy_test_submodule.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/internal_test_util/lazy_loader_module/lazy_test_submodule.pyusr/lib/python3.13/site-packages/jax/_src/internal_test_util/test_harnesses.pyusr/lib/python3.13/site-packages/jax/_src/interpreters/usr/lib/python3.13/site-packages/jax/_src/interpreters/__init__.pyusr/lib/python3.13/site-packages/jax/_src/interpreters/__pycache__/usr/lib/python3.13/site-packages/jax/_src/interpreters/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/interpreters/__pycache__/ad.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/interpreters/__pycache__/batching.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/interpreters/__pycache__/mlir.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/interpreters/__pycache__/partial_eval.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/interpreters/__pycache__/pxla.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/interpreters/__pycache__/xla.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/interpreters/ad.pyusr/lib/python3.13/site-packages/jax/_src/interpreters/batching.pyusr/lib/python3.13/site-packages/jax/_src/interpreters/mlir.pyusr/lib/python3.13/site-packages/jax/_src/interpreters/partial_eval.pyusr/lib/python3.13/site-packages/jax/_src/interpreters/pxla.pyusr/lib/python3.13/site-packages/jax/_src/interpreters/xla.pyusr/lib/python3.13/site-packages/jax/_src/jaxpr_util.pyusr/lib/python3.13/site-packages/jax/_src/lax/usr/lib/python3.13/site-packages/jax/_src/lax/__init__.pyusr/lib/python3.13/site-packages/jax/_src/lax/__pycache__/usr/lib/python3.13/site-packages/jax/_src/lax/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/lax/__pycache__/ann.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/lax/__pycache__/convolution.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/lax/__pycache__/fft.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/lax/__pycache__/lax.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/lax/__pycache__/linalg.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/lax/__pycache__/other.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/lax/__pycache__/parallel.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/lax/__pycache__/slicing.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/lax/__pycache__/special.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/lax/__pycache__/utils.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/lax/__pycache__/windowed_reductions.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/lax/ann.pyusr/lib/python3.13/site-packages/jax/_src/lax/control_flow/usr/lib/python3.13/site-packages/jax/_src/lax/control_flow/__init__.pyusr/lib/python3.13/site-packages/jax/_src/lax/control_flow/__pycache__/usr/lib/python3.13/site-packages/jax/_src/lax/control_flow/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/lax/control_flow/__pycache__/common.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/lax/control_flow/__pycache__/conditionals.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/lax/control_flow/__pycache__/for_loop.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/lax/control_flow/__pycache__/loops.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/lax/control_flow/__pycache__/solves.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/lax/control_flow/common.pyusr/lib/python3.13/site-packages/jax/_src/lax/control_flow/conditionals.pyusr/lib/python3.13/site-packages/jax/_src/lax/control_flow/for_loop.pyusr/lib/python3.13/site-packages/jax/_src/lax/control_flow/loops.pyusr/lib/python3.13/site-packages/jax/_src/lax/control_flow/solves.pyusr/lib/python3.13/site-packages/jax/_src/lax/convolution.pyusr/lib/python3.13/site-packages/jax/_src/lax/fft.pyusr/lib/python3.13/site-packages/jax/_src/lax/lax.pyusr/lib/python3.13/site-packages/jax/_src/lax/linalg.pyusr/lib/python3.13/site-packages/jax/_src/lax/other.pyusr/lib/python3.13/site-packages/jax/_src/lax/parallel.pyusr/lib/python3.13/site-packages/jax/_src/lax/slicing.pyusr/lib/python3.13/site-packages/jax/_src/lax/special.pyusr/lib/python3.13/site-packages/jax/_src/lax/utils.pyusr/lib/python3.13/site-packages/jax/_src/lax/windowed_reductions.pyusr/lib/python3.13/site-packages/jax/_src/lax_reference.pyusr/lib/python3.13/site-packages/jax/_src/layout.pyusr/lib/python3.13/site-packages/jax/_src/lazy_loader.pyusr/lib/python3.13/site-packages/jax/_src/lib/usr/lib/python3.13/site-packages/jax/_src/lib/__init__.pyusr/lib/python3.13/site-packages/jax/_src/lib/__pycache__/usr/lib/python3.13/site-packages/jax/_src/lib/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/lib/__pycache__/mosaic_gpu.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/lib/__pycache__/triton.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/lib/mlir/usr/lib/python3.13/site-packages/jax/_src/lib/mlir/__init__.pyusr/lib/python3.13/site-packages/jax/_src/lib/mlir/__pycache__/usr/lib/python3.13/site-packages/jax/_src/lib/mlir/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/lib/mlir/dialects/usr/lib/python3.13/site-packages/jax/_src/lib/mlir/dialects/__init__.pyusr/lib/python3.13/site-packages/jax/_src/lib/mlir/dialects/__pycache__/usr/lib/python3.13/site-packages/jax/_src/lib/mlir/dialects/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/lib/mosaic_gpu.pyusr/lib/python3.13/site-packages/jax/_src/lib/triton.pyusr/lib/python3.13/site-packages/jax/_src/linear_util.pyusr/lib/python3.13/site-packages/jax/_src/logging_config.pyusr/lib/python3.13/site-packages/jax/_src/lru_cache.pyusr/lib/python3.13/site-packages/jax/_src/mesh.pyusr/lib/python3.13/site-packages/jax/_src/mesh_utils.pyusr/lib/python3.13/site-packages/jax/_src/monitoring.pyusr/lib/python3.13/site-packages/jax/_src/named_sharding.pyusr/lib/python3.13/site-packages/jax/_src/nn/usr/lib/python3.13/site-packages/jax/_src/nn/__init__.pyusr/lib/python3.13/site-packages/jax/_src/nn/__pycache__/usr/lib/python3.13/site-packages/jax/_src/nn/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/nn/__pycache__/functions.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/nn/__pycache__/initializers.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/nn/functions.pyusr/lib/python3.13/site-packages/jax/_src/nn/initializers.pyusr/lib/python3.13/site-packages/jax/_src/numpy/usr/lib/python3.13/site-packages/jax/_src/numpy/__init__.pyusr/lib/python3.13/site-packages/jax/_src/numpy/__pycache__/usr/lib/python3.13/site-packages/jax/_src/numpy/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/numpy/__pycache__/array_api_metadata.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/numpy/__pycache__/array_constructors.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/numpy/__pycache__/array_creation.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/numpy/__pycache__/array_methods.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/numpy/__pycache__/einsum.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/numpy/__pycache__/error.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/numpy/__pycache__/fft.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/numpy/__pycache__/index_tricks.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/numpy/__pycache__/indexing.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/numpy/__pycache__/lax_numpy.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/numpy/__pycache__/linalg.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/numpy/__pycache__/polynomial.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/numpy/__pycache__/reductions.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/numpy/__pycache__/scalar_types.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/numpy/__pycache__/setops.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/numpy/__pycache__/sorting.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/numpy/__pycache__/tensor_contractions.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/numpy/__pycache__/ufunc_api.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/numpy/__pycache__/ufuncs.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/numpy/__pycache__/util.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/numpy/__pycache__/vectorize.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/numpy/__pycache__/window_functions.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/numpy/array_api_metadata.pyusr/lib/python3.13/site-packages/jax/_src/numpy/array_constructors.pyusr/lib/python3.13/site-packages/jax/_src/numpy/array_creation.pyusr/lib/python3.13/site-packages/jax/_src/numpy/array_methods.pyusr/lib/python3.13/site-packages/jax/_src/numpy/einsum.pyusr/lib/python3.13/site-packages/jax/_src/numpy/error.pyusr/lib/python3.13/site-packages/jax/_src/numpy/fft.pyusr/lib/python3.13/site-packages/jax/_src/numpy/index_tricks.pyusr/lib/python3.13/site-packages/jax/_src/numpy/indexing.pyusr/lib/python3.13/site-packages/jax/_src/numpy/lax_numpy.pyusr/lib/python3.13/site-packages/jax/_src/numpy/linalg.pyusr/lib/python3.13/site-packages/jax/_src/numpy/polynomial.pyusr/lib/python3.13/site-packages/jax/_src/numpy/reductions.pyusr/lib/python3.13/site-packages/jax/_src/numpy/scalar_types.pyusr/lib/python3.13/site-packages/jax/_src/numpy/setops.pyusr/lib/python3.13/site-packages/jax/_src/numpy/sorting.pyusr/lib/python3.13/site-packages/jax/_src/numpy/tensor_contractions.pyusr/lib/python3.13/site-packages/jax/_src/numpy/ufunc_api.pyusr/lib/python3.13/site-packages/jax/_src/numpy/ufuncs.pyusr/lib/python3.13/site-packages/jax/_src/numpy/util.pyusr/lib/python3.13/site-packages/jax/_src/numpy/vectorize.pyusr/lib/python3.13/site-packages/jax/_src/numpy/window_functions.pyusr/lib/python3.13/site-packages/jax/_src/op_shardings.pyusr/lib/python3.13/site-packages/jax/_src/ops/usr/lib/python3.13/site-packages/jax/_src/ops/__init__.pyusr/lib/python3.13/site-packages/jax/_src/ops/__pycache__/usr/lib/python3.13/site-packages/jax/_src/ops/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/ops/__pycache__/scatter.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/ops/__pycache__/special.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/ops/scatter.pyusr/lib/python3.13/site-packages/jax/_src/ops/special.pyusr/lib/python3.13/site-packages/jax/_src/pallas/usr/lib/python3.13/site-packages/jax/_src/pallas/__init__.pyusr/lib/python3.13/site-packages/jax/_src/pallas/__pycache__/usr/lib/python3.13/site-packages/jax/_src/pallas/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/pallas/__pycache__/core.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/pallas/__pycache__/cost_estimate.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/pallas/__pycache__/helpers.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/pallas/__pycache__/hlo_interpreter.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/pallas/__pycache__/pallas_call.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/pallas/__pycache__/primitives.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/pallas/__pycache__/utils.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/pallas/core.pyusr/lib/python3.13/site-packages/jax/_src/pallas/cost_estimate.pyusr/lib/python3.13/site-packages/jax/_src/pallas/fuser/usr/lib/python3.13/site-packages/jax/_src/pallas/fuser/__init__.pyusr/lib/python3.13/site-packages/jax/_src/pallas/fuser/__pycache__/usr/lib/python3.13/site-packages/jax/_src/pallas/fuser/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/pallas/fuser/__pycache__/block_spec.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/pallas/fuser/__pycache__/custom_evaluate.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/pallas/fuser/__pycache__/fuser_utils.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/pallas/fuser/__pycache__/fusible.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/pallas/fuser/__pycache__/fusible_dtype.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/pallas/fuser/__pycache__/fusion.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/pallas/fuser/__pycache__/jaxpr_fusion.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/pallas/fuser/block_spec.pyusr/lib/python3.13/site-packages/jax/_src/pallas/fuser/custom_evaluate.pyusr/lib/python3.13/site-packages/jax/_src/pallas/fuser/fuser_utils.pyusr/lib/python3.13/site-packages/jax/_src/pallas/fuser/fusible.pyusr/lib/python3.13/site-packages/jax/_src/pallas/fuser/fusible_dtype.pyusr/lib/python3.13/site-packages/jax/_src/pallas/fuser/fusion.pyusr/lib/python3.13/site-packages/jax/_src/pallas/fuser/jaxpr_fusion.pyusr/lib/python3.13/site-packages/jax/_src/pallas/helpers.pyusr/lib/python3.13/site-packages/jax/_src/pallas/hlo_interpreter.pyusr/lib/python3.13/site-packages/jax/_src/pallas/mosaic/usr/lib/python3.13/site-packages/jax/_src/pallas/mosaic/__init__.pyusr/lib/python3.13/site-packages/jax/_src/pallas/mosaic/__pycache__/usr/lib/python3.13/site-packages/jax/_src/pallas/mosaic/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/pallas/mosaic/__pycache__/core.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/pallas/mosaic/__pycache__/error_handling.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/pallas/mosaic/__pycache__/helpers.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/pallas/mosaic/__pycache__/interpret.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/pallas/mosaic/__pycache__/lowering.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/pallas/mosaic/__pycache__/pallas_call_registration.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/pallas/mosaic/__pycache__/pipeline.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/pallas/mosaic/__pycache__/primitives.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/pallas/mosaic/__pycache__/random.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/pallas/mosaic/__pycache__/verification.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/pallas/mosaic/core.pyusr/lib/python3.13/site-packages/jax/_src/pallas/mosaic/error_handling.pyusr/lib/python3.13/site-packages/jax/_src/pallas/mosaic/helpers.pyusr/lib/python3.13/site-packages/jax/_src/pallas/mosaic/interpret.pyusr/lib/python3.13/site-packages/jax/_src/pallas/mosaic/lowering.pyusr/lib/python3.13/site-packages/jax/_src/pallas/mosaic/pallas_call_registration.pyusr/lib/python3.13/site-packages/jax/_src/pallas/mosaic/pipeline.pyusr/lib/python3.13/site-packages/jax/_src/pallas/mosaic/primitives.pyusr/lib/python3.13/site-packages/jax/_src/pallas/mosaic/random.pyusr/lib/python3.13/site-packages/jax/_src/pallas/mosaic/verification.pyusr/lib/python3.13/site-packages/jax/_src/pallas/mosaic_gpu/usr/lib/python3.13/site-packages/jax/_src/pallas/mosaic_gpu/__init__.pyusr/lib/python3.13/site-packages/jax/_src/pallas/mosaic_gpu/__pycache__/usr/lib/python3.13/site-packages/jax/_src/pallas/mosaic_gpu/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/pallas/mosaic_gpu/__pycache__/core.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/pallas/mosaic_gpu/__pycache__/helpers.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/pallas/mosaic_gpu/__pycache__/lowering.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/pallas/mosaic_gpu/__pycache__/pallas_call_registration.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/pallas/mosaic_gpu/__pycache__/pipeline.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/pallas/mosaic_gpu/__pycache__/primitives.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/pallas/mosaic_gpu/core.pyusr/lib/python3.13/site-packages/jax/_src/pallas/mosaic_gpu/helpers.pyusr/lib/python3.13/site-packages/jax/_src/pallas/mosaic_gpu/lowering.pyusr/lib/python3.13/site-packages/jax/_src/pallas/mosaic_gpu/pallas_call_registration.pyusr/lib/python3.13/site-packages/jax/_src/pallas/mosaic_gpu/pipeline.pyusr/lib/python3.13/site-packages/jax/_src/pallas/mosaic_gpu/primitives.pyusr/lib/python3.13/site-packages/jax/_src/pallas/pallas_call.pyusr/lib/python3.13/site-packages/jax/_src/pallas/primitives.pyusr/lib/python3.13/site-packages/jax/_src/pallas/triton/usr/lib/python3.13/site-packages/jax/_src/pallas/triton/__init__.pyusr/lib/python3.13/site-packages/jax/_src/pallas/triton/__pycache__/usr/lib/python3.13/site-packages/jax/_src/pallas/triton/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/pallas/triton/__pycache__/core.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/pallas/triton/__pycache__/lowering.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/pallas/triton/__pycache__/pallas_call_registration.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/pallas/triton/__pycache__/primitives.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/pallas/triton/core.pyusr/lib/python3.13/site-packages/jax/_src/pallas/triton/lowering.pyusr/lib/python3.13/site-packages/jax/_src/pallas/triton/pallas_call_registration.pyusr/lib/python3.13/site-packages/jax/_src/pallas/triton/primitives.pyusr/lib/python3.13/site-packages/jax/_src/pallas/utils.pyusr/lib/python3.13/site-packages/jax/_src/partition_spec.pyusr/lib/python3.13/site-packages/jax/_src/path.pyusr/lib/python3.13/site-packages/jax/_src/pickle_util.pyusr/lib/python3.13/site-packages/jax/_src/pjit.pyusr/lib/python3.13/site-packages/jax/_src/pretty_printer.pyusr/lib/python3.13/site-packages/jax/_src/prng.pyusr/lib/python3.13/site-packages/jax/_src/profiler.pyusr/lib/python3.13/site-packages/jax/_src/public_test_util.pyusr/lib/python3.13/site-packages/jax/_src/random.pyusr/lib/python3.13/site-packages/jax/_src/scipy/usr/lib/python3.13/site-packages/jax/_src/scipy/__init__.pyusr/lib/python3.13/site-packages/jax/_src/scipy/__pycache__/usr/lib/python3.13/site-packages/jax/_src/scipy/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/scipy/__pycache__/fft.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/scipy/__pycache__/integrate.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/scipy/__pycache__/linalg.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/scipy/__pycache__/ndimage.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/scipy/__pycache__/signal.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/scipy/__pycache__/special.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/scipy/cluster/usr/lib/python3.13/site-packages/jax/_src/scipy/cluster/__init__.pyusr/lib/python3.13/site-packages/jax/_src/scipy/cluster/__pycache__/usr/lib/python3.13/site-packages/jax/_src/scipy/cluster/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/scipy/cluster/__pycache__/vq.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/scipy/cluster/vq.pyusr/lib/python3.13/site-packages/jax/_src/scipy/fft.pyusr/lib/python3.13/site-packages/jax/_src/scipy/integrate.pyusr/lib/python3.13/site-packages/jax/_src/scipy/linalg.pyusr/lib/python3.13/site-packages/jax/_src/scipy/ndimage.pyusr/lib/python3.13/site-packages/jax/_src/scipy/optimize/usr/lib/python3.13/site-packages/jax/_src/scipy/optimize/__init__.pyusr/lib/python3.13/site-packages/jax/_src/scipy/optimize/__pycache__/usr/lib/python3.13/site-packages/jax/_src/scipy/optimize/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/scipy/optimize/__pycache__/_lbfgs.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/scipy/optimize/__pycache__/bfgs.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/scipy/optimize/__pycache__/line_search.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/scipy/optimize/__pycache__/minimize.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/scipy/optimize/_lbfgs.pyusr/lib/python3.13/site-packages/jax/_src/scipy/optimize/bfgs.pyusr/lib/python3.13/site-packages/jax/_src/scipy/optimize/line_search.pyusr/lib/python3.13/site-packages/jax/_src/scipy/optimize/minimize.pyusr/lib/python3.13/site-packages/jax/_src/scipy/signal.pyusr/lib/python3.13/site-packages/jax/_src/scipy/sparse/usr/lib/python3.13/site-packages/jax/_src/scipy/sparse/__init__.pyusr/lib/python3.13/site-packages/jax/_src/scipy/sparse/__pycache__/usr/lib/python3.13/site-packages/jax/_src/scipy/sparse/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/scipy/sparse/__pycache__/linalg.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/scipy/sparse/linalg.pyusr/lib/python3.13/site-packages/jax/_src/scipy/spatial/usr/lib/python3.13/site-packages/jax/_src/scipy/spatial/__init__.pyusr/lib/python3.13/site-packages/jax/_src/scipy/spatial/__pycache__/usr/lib/python3.13/site-packages/jax/_src/scipy/spatial/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/scipy/spatial/__pycache__/transform.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/scipy/spatial/transform.pyusr/lib/python3.13/site-packages/jax/_src/scipy/special.pyusr/lib/python3.13/site-packages/jax/_src/scipy/stats/usr/lib/python3.13/site-packages/jax/_src/scipy/stats/__init__.pyusr/lib/python3.13/site-packages/jax/_src/scipy/stats/__pycache__/usr/lib/python3.13/site-packages/jax/_src/scipy/stats/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/scipy/stats/__pycache__/_core.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/scipy/stats/__pycache__/bernoulli.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/scipy/stats/__pycache__/beta.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/scipy/stats/__pycache__/betabinom.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/scipy/stats/__pycache__/binom.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/scipy/stats/__pycache__/cauchy.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/scipy/stats/__pycache__/chi2.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/scipy/stats/__pycache__/dirichlet.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/scipy/stats/__pycache__/expon.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/scipy/stats/__pycache__/gamma.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/scipy/stats/__pycache__/gennorm.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/scipy/stats/__pycache__/geom.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/scipy/stats/__pycache__/gumbel_l.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/scipy/stats/__pycache__/gumbel_r.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/scipy/stats/__pycache__/kde.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/scipy/stats/__pycache__/laplace.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/scipy/stats/__pycache__/logistic.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/scipy/stats/__pycache__/multinomial.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/scipy/stats/__pycache__/multivariate_normal.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/scipy/stats/__pycache__/nbinom.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/scipy/stats/__pycache__/norm.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/scipy/stats/__pycache__/pareto.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/scipy/stats/__pycache__/poisson.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/scipy/stats/__pycache__/t.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/scipy/stats/__pycache__/truncnorm.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/scipy/stats/__pycache__/uniform.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/scipy/stats/__pycache__/vonmises.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/scipy/stats/__pycache__/wrapcauchy.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/scipy/stats/_core.pyusr/lib/python3.13/site-packages/jax/_src/scipy/stats/bernoulli.pyusr/lib/python3.13/site-packages/jax/_src/scipy/stats/beta.pyusr/lib/python3.13/site-packages/jax/_src/scipy/stats/betabinom.pyusr/lib/python3.13/site-packages/jax/_src/scipy/stats/binom.pyusr/lib/python3.13/site-packages/jax/_src/scipy/stats/cauchy.pyusr/lib/python3.13/site-packages/jax/_src/scipy/stats/chi2.pyusr/lib/python3.13/site-packages/jax/_src/scipy/stats/dirichlet.pyusr/lib/python3.13/site-packages/jax/_src/scipy/stats/expon.pyusr/lib/python3.13/site-packages/jax/_src/scipy/stats/gamma.pyusr/lib/python3.13/site-packages/jax/_src/scipy/stats/gennorm.pyusr/lib/python3.13/site-packages/jax/_src/scipy/stats/geom.pyusr/lib/python3.13/site-packages/jax/_src/scipy/stats/gumbel_l.pyusr/lib/python3.13/site-packages/jax/_src/scipy/stats/gumbel_r.pyusr/lib/python3.13/site-packages/jax/_src/scipy/stats/kde.pyusr/lib/python3.13/site-packages/jax/_src/scipy/stats/laplace.pyusr/lib/python3.13/site-packages/jax/_src/scipy/stats/logistic.pyusr/lib/python3.13/site-packages/jax/_src/scipy/stats/multinomial.pyusr/lib/python3.13/site-packages/jax/_src/scipy/stats/multivariate_normal.pyusr/lib/python3.13/site-packages/jax/_src/scipy/stats/nbinom.pyusr/lib/python3.13/site-packages/jax/_src/scipy/stats/norm.pyusr/lib/python3.13/site-packages/jax/_src/scipy/stats/pareto.pyusr/lib/python3.13/site-packages/jax/_src/scipy/stats/poisson.pyusr/lib/python3.13/site-packages/jax/_src/scipy/stats/t.pyusr/lib/python3.13/site-packages/jax/_src/scipy/stats/truncnorm.pyusr/lib/python3.13/site-packages/jax/_src/scipy/stats/uniform.pyusr/lib/python3.13/site-packages/jax/_src/scipy/stats/vonmises.pyusr/lib/python3.13/site-packages/jax/_src/scipy/stats/wrapcauchy.pyusr/lib/python3.13/site-packages/jax/_src/shard_alike.pyusr/lib/python3.13/site-packages/jax/_src/shard_map.pyusr/lib/python3.13/site-packages/jax/_src/sharding.pyusr/lib/python3.13/site-packages/jax/_src/sharding_impls.pyusr/lib/python3.13/site-packages/jax/_src/sharding_specs.pyusr/lib/python3.13/site-packages/jax/_src/source_info_util.pyusr/lib/python3.13/site-packages/jax/_src/sourcemap.pyusr/lib/python3.13/site-packages/jax/_src/stages.pyusr/lib/python3.13/site-packages/jax/_src/state/usr/lib/python3.13/site-packages/jax/_src/state/__init__.pyusr/lib/python3.13/site-packages/jax/_src/state/__pycache__/usr/lib/python3.13/site-packages/jax/_src/state/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/state/__pycache__/discharge.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/state/__pycache__/indexing.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/state/__pycache__/primitives.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/state/__pycache__/types.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/state/__pycache__/utils.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/state/discharge.pyusr/lib/python3.13/site-packages/jax/_src/state/indexing.pyusr/lib/python3.13/site-packages/jax/_src/state/primitives.pyusr/lib/python3.13/site-packages/jax/_src/state/types.pyusr/lib/python3.13/site-packages/jax/_src/state/utils.pyusr/lib/python3.13/site-packages/jax/_src/test_loader.pyusr/lib/python3.13/site-packages/jax/_src/test_multiprocess.pyusr/lib/python3.13/site-packages/jax/_src/test_util.pyusr/lib/python3.13/site-packages/jax/_src/test_warning_util.pyusr/lib/python3.13/site-packages/jax/_src/third_party/usr/lib/python3.13/site-packages/jax/_src/third_party/__init__.pyusr/lib/python3.13/site-packages/jax/_src/third_party/__pycache__/usr/lib/python3.13/site-packages/jax/_src/third_party/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/third_party/scipy/usr/lib/python3.13/site-packages/jax/_src/third_party/scipy/__init__.pyusr/lib/python3.13/site-packages/jax/_src/third_party/scipy/__pycache__/usr/lib/python3.13/site-packages/jax/_src/third_party/scipy/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/third_party/scipy/__pycache__/betaln.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/third_party/scipy/__pycache__/interpolate.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/third_party/scipy/__pycache__/linalg.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/third_party/scipy/__pycache__/signal_helper.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/third_party/scipy/__pycache__/special.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/third_party/scipy/betaln.pyusr/lib/python3.13/site-packages/jax/_src/third_party/scipy/interpolate.pyusr/lib/python3.13/site-packages/jax/_src/third_party/scipy/linalg.pyusr/lib/python3.13/site-packages/jax/_src/third_party/scipy/signal_helper.pyusr/lib/python3.13/site-packages/jax/_src/third_party/scipy/special.pyusr/lib/python3.13/site-packages/jax/_src/tpu/usr/lib/python3.13/site-packages/jax/_src/tpu/__init__.pyusr/lib/python3.13/site-packages/jax/_src/tpu/__pycache__/usr/lib/python3.13/site-packages/jax/_src/tpu/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/tpu/linalg/usr/lib/python3.13/site-packages/jax/_src/tpu/linalg/__init__.pyusr/lib/python3.13/site-packages/jax/_src/tpu/linalg/__pycache__/usr/lib/python3.13/site-packages/jax/_src/tpu/linalg/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/tpu/linalg/__pycache__/eigh.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/tpu/linalg/__pycache__/qdwh.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/tpu/linalg/__pycache__/stack.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/tpu/linalg/__pycache__/svd.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/_src/tpu/linalg/eigh.pyusr/lib/python3.13/site-packages/jax/_src/tpu/linalg/qdwh.pyusr/lib/python3.13/site-packages/jax/_src/tpu/linalg/stack.pyusr/lib/python3.13/site-packages/jax/_src/tpu/linalg/svd.pyusr/lib/python3.13/site-packages/jax/_src/tpu_custom_call.pyusr/lib/python3.13/site-packages/jax/_src/traceback_util.pyusr/lib/python3.13/site-packages/jax/_src/tree.pyusr/lib/python3.13/site-packages/jax/_src/tree_util.pyusr/lib/python3.13/site-packages/jax/_src/typing.pyusr/lib/python3.13/site-packages/jax/_src/util.pyusr/lib/python3.13/site-packages/jax/_src/xla_bridge.pyusr/lib/python3.13/site-packages/jax/_src/xla_metadata.pyusr/lib/python3.13/site-packages/jax/_src/xla_metadata_lib.pyusr/lib/python3.13/site-packages/jax/ad_checkpoint.pyusr/lib/python3.13/site-packages/jax/api_util.pyusr/lib/python3.13/site-packages/jax/cloud_tpu_init.pyusr/lib/python3.13/site-packages/jax/collect_profile.pyusr/lib/python3.13/site-packages/jax/core.pyusr/lib/python3.13/site-packages/jax/custom_batching.pyusr/lib/python3.13/site-packages/jax/custom_derivatives.pyusr/lib/python3.13/site-packages/jax/custom_transpose.pyusr/lib/python3.13/site-packages/jax/debug.pyusr/lib/python3.13/site-packages/jax/distributed.pyusr/lib/python3.13/site-packages/jax/dlpack.pyusr/lib/python3.13/site-packages/jax/dtypes.pyusr/lib/python3.13/site-packages/jax/errors.pyusr/lib/python3.13/site-packages/jax/example_libraries/usr/lib/python3.13/site-packages/jax/example_libraries/__init__.pyusr/lib/python3.13/site-packages/jax/example_libraries/__pycache__/usr/lib/python3.13/site-packages/jax/example_libraries/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/example_libraries/__pycache__/optimizers.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/example_libraries/__pycache__/stax.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/example_libraries/optimizers.pyusr/lib/python3.13/site-packages/jax/example_libraries/stax.pyusr/lib/python3.13/site-packages/jax/experimental/usr/lib/python3.13/site-packages/jax/experimental/__init__.pyusr/lib/python3.13/site-packages/jax/experimental/__pycache__/usr/lib/python3.13/site-packages/jax/experimental/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/__pycache__/buffer_callback.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/__pycache__/checkify.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/__pycache__/compute_on.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/__pycache__/custom_dce.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/__pycache__/custom_partitioning.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/__pycache__/host_callback.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/__pycache__/jet.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/__pycache__/layout.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/__pycache__/mesh_utils.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/__pycache__/multihost_utils.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/__pycache__/ode.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/__pycache__/pjit.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/__pycache__/profiler.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/__pycache__/rnn.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/__pycache__/serialize_executable.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/__pycache__/shard_alike.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/__pycache__/shard_map.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/__pycache__/topologies.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/__pycache__/transfer.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/__pycache__/x64_context.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/__pycache__/xla_metadata.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/_private_mm/usr/lib/python3.13/site-packages/jax/experimental/_private_mm/__init__.pyusr/lib/python3.13/site-packages/jax/experimental/_private_mm/__pycache__/usr/lib/python3.13/site-packages/jax/experimental/_private_mm/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/_private_mm/__pycache__/mini_dime.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/_private_mm/__pycache__/mm.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/_private_mm/__pycache__/profile_utils.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/_private_mm/mini_dime.pyusr/lib/python3.13/site-packages/jax/experimental/_private_mm/mm.pyusr/lib/python3.13/site-packages/jax/experimental/_private_mm/profile_utils.pyusr/lib/python3.13/site-packages/jax/experimental/array_serialization/usr/lib/python3.13/site-packages/jax/experimental/array_serialization/__init__.pyusr/lib/python3.13/site-packages/jax/experimental/array_serialization/__pycache__/usr/lib/python3.13/site-packages/jax/experimental/array_serialization/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/array_serialization/__pycache__/pytree_serialization.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/array_serialization/__pycache__/pytree_serialization_utils.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/array_serialization/__pycache__/serialization.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/array_serialization/__pycache__/serialization_test.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/array_serialization/__pycache__/tensorstore_impl.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/array_serialization/pytree_serialization.pyusr/lib/python3.13/site-packages/jax/experimental/array_serialization/pytree_serialization_utils.pyusr/lib/python3.13/site-packages/jax/experimental/array_serialization/serialization.pyusr/lib/python3.13/site-packages/jax/experimental/array_serialization/serialization_test.pyusr/lib/python3.13/site-packages/jax/experimental/array_serialization/tensorstore_impl.pyusr/lib/python3.13/site-packages/jax/experimental/buffer_callback.pyusr/lib/python3.13/site-packages/jax/experimental/checkify.pyusr/lib/python3.13/site-packages/jax/experimental/colocated_python/usr/lib/python3.13/site-packages/jax/experimental/colocated_python/__init__.pyusr/lib/python3.13/site-packages/jax/experimental/colocated_python/__pycache__/usr/lib/python3.13/site-packages/jax/experimental/colocated_python/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/colocated_python/__pycache__/api.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/colocated_python/__pycache__/func.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/colocated_python/__pycache__/func_backend.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/colocated_python/__pycache__/obj.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/colocated_python/__pycache__/obj_backend.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/colocated_python/__pycache__/serialization.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/colocated_python/api.pyusr/lib/python3.13/site-packages/jax/experimental/colocated_python/func.pyusr/lib/python3.13/site-packages/jax/experimental/colocated_python/func_backend.pyusr/lib/python3.13/site-packages/jax/experimental/colocated_python/obj.pyusr/lib/python3.13/site-packages/jax/experimental/colocated_python/obj_backend.pyusr/lib/python3.13/site-packages/jax/experimental/colocated_python/serialization.pyusr/lib/python3.13/site-packages/jax/experimental/compilation_cache/usr/lib/python3.13/site-packages/jax/experimental/compilation_cache/__init__.pyusr/lib/python3.13/site-packages/jax/experimental/compilation_cache/__pycache__/usr/lib/python3.13/site-packages/jax/experimental/compilation_cache/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/compilation_cache/__pycache__/compilation_cache.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/compilation_cache/compilation_cache.pyusr/lib/python3.13/site-packages/jax/experimental/compute_on.pyusr/lib/python3.13/site-packages/jax/experimental/custom_dce.pyusr/lib/python3.13/site-packages/jax/experimental/custom_partitioning.pyusr/lib/python3.13/site-packages/jax/experimental/host_callback.pyusr/lib/python3.13/site-packages/jax/experimental/jax2tf/usr/lib/python3.13/site-packages/jax/experimental/jax2tf/__init__.pyusr/lib/python3.13/site-packages/jax/experimental/jax2tf/__pycache__/usr/lib/python3.13/site-packages/jax/experimental/jax2tf/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/jax2tf/__pycache__/call_tf.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/jax2tf/__pycache__/impl_no_xla.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/jax2tf/__pycache__/jax2tf.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/jax2tf/call_tf.pyusr/lib/python3.13/site-packages/jax/experimental/jax2tf/examples/usr/lib/python3.13/site-packages/jax/experimental/jax2tf/examples/__init__.pyusr/lib/python3.13/site-packages/jax/experimental/jax2tf/examples/__pycache__/usr/lib/python3.13/site-packages/jax/experimental/jax2tf/examples/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/jax2tf/examples/__pycache__/keras_reuse_main.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/jax2tf/examples/__pycache__/keras_reuse_main_test.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/jax2tf/examples/__pycache__/mnist_lib.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/jax2tf/examples/__pycache__/saved_model_lib.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/jax2tf/examples/__pycache__/saved_model_main.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/jax2tf/examples/__pycache__/saved_model_main_test.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/jax2tf/examples/keras_reuse_main.pyusr/lib/python3.13/site-packages/jax/experimental/jax2tf/examples/keras_reuse_main_test.pyusr/lib/python3.13/site-packages/jax/experimental/jax2tf/examples/mnist_lib.pyusr/lib/python3.13/site-packages/jax/experimental/jax2tf/examples/saved_model_lib.pyusr/lib/python3.13/site-packages/jax/experimental/jax2tf/examples/saved_model_main.pyusr/lib/python3.13/site-packages/jax/experimental/jax2tf/examples/saved_model_main_test.pyusr/lib/python3.13/site-packages/jax/experimental/jax2tf/examples/serving/usr/lib/python3.13/site-packages/jax/experimental/jax2tf/examples/serving/__init__.pyusr/lib/python3.13/site-packages/jax/experimental/jax2tf/examples/serving/__pycache__/usr/lib/python3.13/site-packages/jax/experimental/jax2tf/examples/serving/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/jax2tf/examples/serving/__pycache__/model_server_request.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/jax2tf/examples/serving/model_server_request.pyusr/lib/python3.13/site-packages/jax/experimental/jax2tf/impl_no_xla.pyusr/lib/python3.13/site-packages/jax/experimental/jax2tf/jax2tf.pyusr/lib/python3.13/site-packages/jax/experimental/jax2tf/tests/usr/lib/python3.13/site-packages/jax/experimental/jax2tf/tests/__init__.pyusr/lib/python3.13/site-packages/jax/experimental/jax2tf/tests/__pycache__/usr/lib/python3.13/site-packages/jax/experimental/jax2tf/tests/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/jax2tf/tests/__pycache__/back_compat_tf_test.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/jax2tf/tests/__pycache__/call_tf_test.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/jax2tf/tests/__pycache__/control_flow_ops_test.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/jax2tf/tests/__pycache__/converters.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/jax2tf/tests/__pycache__/cross_compilation_check.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/jax2tf/tests/__pycache__/jax2tf_limitations.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/jax2tf/tests/__pycache__/jax2tf_test.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/jax2tf/tests/__pycache__/jax_primitives_coverage_test.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/jax2tf/tests/__pycache__/model_harness.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/jax2tf/tests/__pycache__/models_test_main.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/jax2tf/tests/__pycache__/primitives_test.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/jax2tf/tests/__pycache__/savedmodel_test.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/jax2tf/tests/__pycache__/shape_poly_test.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/jax2tf/tests/__pycache__/sharding_test.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/jax2tf/tests/__pycache__/tf_test_util.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/jax2tf/tests/back_compat_tf_test.pyusr/lib/python3.13/site-packages/jax/experimental/jax2tf/tests/call_tf_test.pyusr/lib/python3.13/site-packages/jax/experimental/jax2tf/tests/control_flow_ops_test.pyusr/lib/python3.13/site-packages/jax/experimental/jax2tf/tests/converters.pyusr/lib/python3.13/site-packages/jax/experimental/jax2tf/tests/cross_compilation_check.pyusr/lib/python3.13/site-packages/jax/experimental/jax2tf/tests/jax2tf_limitations.pyusr/lib/python3.13/site-packages/jax/experimental/jax2tf/tests/jax2tf_test.pyusr/lib/python3.13/site-packages/jax/experimental/jax2tf/tests/jax_primitives_coverage_test.pyusr/lib/python3.13/site-packages/jax/experimental/jax2tf/tests/model_harness.pyusr/lib/python3.13/site-packages/jax/experimental/jax2tf/tests/models_test_main.pyusr/lib/python3.13/site-packages/jax/experimental/jax2tf/tests/primitives_test.pyusr/lib/python3.13/site-packages/jax/experimental/jax2tf/tests/savedmodel_test.pyusr/lib/python3.13/site-packages/jax/experimental/jax2tf/tests/shape_poly_test.pyusr/lib/python3.13/site-packages/jax/experimental/jax2tf/tests/sharding_test.pyusr/lib/python3.13/site-packages/jax/experimental/jax2tf/tests/tf_test_util.pyusr/lib/python3.13/site-packages/jax/experimental/jet.pyusr/lib/python3.13/site-packages/jax/experimental/key_reuse/usr/lib/python3.13/site-packages/jax/experimental/key_reuse/__init__.pyusr/lib/python3.13/site-packages/jax/experimental/key_reuse/__pycache__/usr/lib/python3.13/site-packages/jax/experimental/key_reuse/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/key_reuse/__pycache__/_core.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/key_reuse/_core.pyusr/lib/python3.13/site-packages/jax/experimental/layout.pyusr/lib/python3.13/site-packages/jax/experimental/mesh_utils.pyusr/lib/python3.13/site-packages/jax/experimental/mosaic/usr/lib/python3.13/site-packages/jax/experimental/mosaic/__init__.pyusr/lib/python3.13/site-packages/jax/experimental/mosaic/__pycache__/usr/lib/python3.13/site-packages/jax/experimental/mosaic/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/mosaic/__pycache__/dialects.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/mosaic/dialects.pyusr/lib/python3.13/site-packages/jax/experimental/mosaic/gpu/usr/lib/python3.13/site-packages/jax/experimental/mosaic/gpu/__init__.pyusr/lib/python3.13/site-packages/jax/experimental/mosaic/gpu/__pycache__/usr/lib/python3.13/site-packages/jax/experimental/mosaic/gpu/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/mosaic/gpu/__pycache__/core.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/mosaic/gpu/__pycache__/dialect_lowering.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/mosaic/gpu/__pycache__/equations.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/mosaic/gpu/__pycache__/fragmented_array.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/mosaic/gpu/__pycache__/inference_utils.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/mosaic/gpu/__pycache__/launch_context.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/mosaic/gpu/__pycache__/layout_inference.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/mosaic/gpu/__pycache__/layout_inference2.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/mosaic/gpu/__pycache__/layouts.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/mosaic/gpu/__pycache__/mma_utils.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/mosaic/gpu/__pycache__/profiler.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/mosaic/gpu/__pycache__/tcgen05.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/mosaic/gpu/__pycache__/transform_inference.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/mosaic/gpu/__pycache__/utils.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/mosaic/gpu/__pycache__/wgmma.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/mosaic/gpu/core.pyusr/lib/python3.13/site-packages/jax/experimental/mosaic/gpu/dialect_lowering.pyusr/lib/python3.13/site-packages/jax/experimental/mosaic/gpu/equations.pyusr/lib/python3.13/site-packages/jax/experimental/mosaic/gpu/examples/usr/lib/python3.13/site-packages/jax/experimental/mosaic/gpu/examples/__init__.pyusr/lib/python3.13/site-packages/jax/experimental/mosaic/gpu/examples/__pycache__/usr/lib/python3.13/site-packages/jax/experimental/mosaic/gpu/examples/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/mosaic/gpu/examples/__pycache__/flash_attention.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/mosaic/gpu/examples/__pycache__/matmul.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/mosaic/gpu/examples/__pycache__/matmul_blackwell.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/mosaic/gpu/examples/flash_attention.pyusr/lib/python3.13/site-packages/jax/experimental/mosaic/gpu/examples/matmul.pyusr/lib/python3.13/site-packages/jax/experimental/mosaic/gpu/examples/matmul_blackwell.pyusr/lib/python3.13/site-packages/jax/experimental/mosaic/gpu/fragmented_array.pyusr/lib/python3.13/site-packages/jax/experimental/mosaic/gpu/inference_utils.pyusr/lib/python3.13/site-packages/jax/experimental/mosaic/gpu/launch_context.pyusr/lib/python3.13/site-packages/jax/experimental/mosaic/gpu/layout_inference.pyusr/lib/python3.13/site-packages/jax/experimental/mosaic/gpu/layout_inference2.pyusr/lib/python3.13/site-packages/jax/experimental/mosaic/gpu/layouts.pyusr/lib/python3.13/site-packages/jax/experimental/mosaic/gpu/mma_utils.pyusr/lib/python3.13/site-packages/jax/experimental/mosaic/gpu/profiler.pyusr/lib/python3.13/site-packages/jax/experimental/mosaic/gpu/tcgen05.pyusr/lib/python3.13/site-packages/jax/experimental/mosaic/gpu/transform_inference.pyusr/lib/python3.13/site-packages/jax/experimental/mosaic/gpu/utils.pyusr/lib/python3.13/site-packages/jax/experimental/mosaic/gpu/wgmma.pyusr/lib/python3.13/site-packages/jax/experimental/multihost_utils.pyusr/lib/python3.13/site-packages/jax/experimental/ode.pyusr/lib/python3.13/site-packages/jax/experimental/pallas/usr/lib/python3.13/site-packages/jax/experimental/pallas/__init__.pyusr/lib/python3.13/site-packages/jax/experimental/pallas/__pycache__/usr/lib/python3.13/site-packages/jax/experimental/pallas/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/pallas/__pycache__/fuser.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/pallas/__pycache__/mosaic_gpu.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/pallas/__pycache__/tpu.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/pallas/__pycache__/triton.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/pallas/fuser.pyusr/lib/python3.13/site-packages/jax/experimental/pallas/mosaic_gpu.pyusr/lib/python3.13/site-packages/jax/experimental/pallas/ops/usr/lib/python3.13/site-packages/jax/experimental/pallas/ops/__init__.pyusr/lib/python3.13/site-packages/jax/experimental/pallas/ops/__pycache__/usr/lib/python3.13/site-packages/jax/experimental/pallas/ops/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/pallas/ops/gpu/usr/lib/python3.13/site-packages/jax/experimental/pallas/ops/gpu/__init__.pyusr/lib/python3.13/site-packages/jax/experimental/pallas/ops/gpu/__pycache__/usr/lib/python3.13/site-packages/jax/experimental/pallas/ops/gpu/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/pallas/ops/gpu/__pycache__/attention.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/pallas/ops/gpu/__pycache__/attention_mgpu.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/pallas/ops/gpu/__pycache__/blackwell_matmul_mgpu.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/pallas/ops/gpu/__pycache__/collective_matmul_mgpu.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/pallas/ops/gpu/__pycache__/decode_attention.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/pallas/ops/gpu/__pycache__/layer_norm.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/pallas/ops/gpu/__pycache__/paged_attention.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/pallas/ops/gpu/__pycache__/ragged_dot_mgpu.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/pallas/ops/gpu/__pycache__/rms_norm.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/pallas/ops/gpu/__pycache__/softmax.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/pallas/ops/gpu/attention.pyusr/lib/python3.13/site-packages/jax/experimental/pallas/ops/gpu/attention_mgpu.pyusr/lib/python3.13/site-packages/jax/experimental/pallas/ops/gpu/blackwell_matmul_mgpu.pyusr/lib/python3.13/site-packages/jax/experimental/pallas/ops/gpu/collective_matmul_mgpu.pyusr/lib/python3.13/site-packages/jax/experimental/pallas/ops/gpu/decode_attention.pyusr/lib/python3.13/site-packages/jax/experimental/pallas/ops/gpu/layer_norm.pyusr/lib/python3.13/site-packages/jax/experimental/pallas/ops/gpu/paged_attention.pyusr/lib/python3.13/site-packages/jax/experimental/pallas/ops/gpu/ragged_dot_mgpu.pyusr/lib/python3.13/site-packages/jax/experimental/pallas/ops/gpu/rms_norm.pyusr/lib/python3.13/site-packages/jax/experimental/pallas/ops/gpu/softmax.pyusr/lib/python3.13/site-packages/jax/experimental/pallas/ops/tpu/usr/lib/python3.13/site-packages/jax/experimental/pallas/ops/tpu/__init__.pyusr/lib/python3.13/site-packages/jax/experimental/pallas/ops/tpu/__pycache__/usr/lib/python3.13/site-packages/jax/experimental/pallas/ops/tpu/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/pallas/ops/tpu/__pycache__/all_gather.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/pallas/ops/tpu/__pycache__/example_kernel.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/pallas/ops/tpu/__pycache__/flash_attention.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/pallas/ops/tpu/__pycache__/matmul.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/pallas/ops/tpu/all_gather.pyusr/lib/python3.13/site-packages/jax/experimental/pallas/ops/tpu/example_kernel.pyusr/lib/python3.13/site-packages/jax/experimental/pallas/ops/tpu/flash_attention.pyusr/lib/python3.13/site-packages/jax/experimental/pallas/ops/tpu/matmul.pyusr/lib/python3.13/site-packages/jax/experimental/pallas/ops/tpu/megablox/usr/lib/python3.13/site-packages/jax/experimental/pallas/ops/tpu/megablox/__init__.pyusr/lib/python3.13/site-packages/jax/experimental/pallas/ops/tpu/megablox/__pycache__/usr/lib/python3.13/site-packages/jax/experimental/pallas/ops/tpu/megablox/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/pallas/ops/tpu/megablox/__pycache__/common.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/pallas/ops/tpu/megablox/__pycache__/gmm.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/pallas/ops/tpu/megablox/__pycache__/ops.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/pallas/ops/tpu/megablox/common.pyusr/lib/python3.13/site-packages/jax/experimental/pallas/ops/tpu/megablox/gmm.pyusr/lib/python3.13/site-packages/jax/experimental/pallas/ops/tpu/megablox/ops.pyusr/lib/python3.13/site-packages/jax/experimental/pallas/ops/tpu/paged_attention/usr/lib/python3.13/site-packages/jax/experimental/pallas/ops/tpu/paged_attention/__init__.pyusr/lib/python3.13/site-packages/jax/experimental/pallas/ops/tpu/paged_attention/__pycache__/usr/lib/python3.13/site-packages/jax/experimental/pallas/ops/tpu/paged_attention/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/pallas/ops/tpu/paged_attention/__pycache__/paged_attention_kernel.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/pallas/ops/tpu/paged_attention/__pycache__/quantization_utils.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/pallas/ops/tpu/paged_attention/__pycache__/util.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/pallas/ops/tpu/paged_attention/paged_attention_kernel.pyusr/lib/python3.13/site-packages/jax/experimental/pallas/ops/tpu/paged_attention/quantization_utils.pyusr/lib/python3.13/site-packages/jax/experimental/pallas/ops/tpu/paged_attention/util.pyusr/lib/python3.13/site-packages/jax/experimental/pallas/ops/tpu/ragged_paged_attention/usr/lib/python3.13/site-packages/jax/experimental/pallas/ops/tpu/ragged_paged_attention/__init__.pyusr/lib/python3.13/site-packages/jax/experimental/pallas/ops/tpu/ragged_paged_attention/__pycache__/usr/lib/python3.13/site-packages/jax/experimental/pallas/ops/tpu/ragged_paged_attention/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/pallas/ops/tpu/ragged_paged_attention/__pycache__/kernel.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/pallas/ops/tpu/ragged_paged_attention/__pycache__/tuned_block_sizes.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/pallas/ops/tpu/ragged_paged_attention/kernel.pyusr/lib/python3.13/site-packages/jax/experimental/pallas/ops/tpu/ragged_paged_attention/tuned_block_sizes.pyusr/lib/python3.13/site-packages/jax/experimental/pallas/ops/tpu/random/usr/lib/python3.13/site-packages/jax/experimental/pallas/ops/tpu/random/__init__.pyusr/lib/python3.13/site-packages/jax/experimental/pallas/ops/tpu/random/__pycache__/usr/lib/python3.13/site-packages/jax/experimental/pallas/ops/tpu/random/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/pallas/ops/tpu/random/__pycache__/philox.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/pallas/ops/tpu/random/__pycache__/prng_utils.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/pallas/ops/tpu/random/__pycache__/threefry.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/pallas/ops/tpu/random/philox.pyusr/lib/python3.13/site-packages/jax/experimental/pallas/ops/tpu/random/prng_utils.pyusr/lib/python3.13/site-packages/jax/experimental/pallas/ops/tpu/random/threefry.pyusr/lib/python3.13/site-packages/jax/experimental/pallas/ops/tpu/splash_attention/usr/lib/python3.13/site-packages/jax/experimental/pallas/ops/tpu/splash_attention/__init__.pyusr/lib/python3.13/site-packages/jax/experimental/pallas/ops/tpu/splash_attention/__pycache__/usr/lib/python3.13/site-packages/jax/experimental/pallas/ops/tpu/splash_attention/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/pallas/ops/tpu/splash_attention/__pycache__/splash_attention_kernel.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/pallas/ops/tpu/splash_attention/__pycache__/splash_attention_mask.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/pallas/ops/tpu/splash_attention/__pycache__/splash_attention_mask_info.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/pallas/ops/tpu/splash_attention/splash_attention_kernel.pyusr/lib/python3.13/site-packages/jax/experimental/pallas/ops/tpu/splash_attention/splash_attention_mask.pyusr/lib/python3.13/site-packages/jax/experimental/pallas/ops/tpu/splash_attention/splash_attention_mask_info.pyusr/lib/python3.13/site-packages/jax/experimental/pallas/tpu.pyusr/lib/python3.13/site-packages/jax/experimental/pallas/triton.pyusr/lib/python3.13/site-packages/jax/experimental/pjit.pyusr/lib/python3.13/site-packages/jax/experimental/profiler.pyusr/lib/python3.13/site-packages/jax/experimental/rnn.pyusr/lib/python3.13/site-packages/jax/experimental/roofline/usr/lib/python3.13/site-packages/jax/experimental/roofline/__init__.pyusr/lib/python3.13/site-packages/jax/experimental/roofline/__pycache__/usr/lib/python3.13/site-packages/jax/experimental/roofline/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/roofline/__pycache__/roofline.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/roofline/__pycache__/rooflines.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/roofline/roofline.pyusr/lib/python3.13/site-packages/jax/experimental/roofline/rooflines.pyusr/lib/python3.13/site-packages/jax/experimental/serialize_executable.pyusr/lib/python3.13/site-packages/jax/experimental/shard_alike.pyusr/lib/python3.13/site-packages/jax/experimental/shard_map.pyusr/lib/python3.13/site-packages/jax/experimental/source_mapper/usr/lib/python3.13/site-packages/jax/experimental/source_mapper/__init__.pyusr/lib/python3.13/site-packages/jax/experimental/source_mapper/__pycache__/usr/lib/python3.13/site-packages/jax/experimental/source_mapper/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/source_mapper/__pycache__/common.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/source_mapper/__pycache__/generate_map.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/source_mapper/__pycache__/hlo.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/source_mapper/__pycache__/jaxpr.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/source_mapper/__pycache__/mlir.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/source_mapper/common.pyusr/lib/python3.13/site-packages/jax/experimental/source_mapper/generate_map.pyusr/lib/python3.13/site-packages/jax/experimental/source_mapper/hlo.pyusr/lib/python3.13/site-packages/jax/experimental/source_mapper/jaxpr.pyusr/lib/python3.13/site-packages/jax/experimental/source_mapper/mlir.pyusr/lib/python3.13/site-packages/jax/experimental/sparse/usr/lib/python3.13/site-packages/jax/experimental/sparse/__init__.pyusr/lib/python3.13/site-packages/jax/experimental/sparse/__pycache__/usr/lib/python3.13/site-packages/jax/experimental/sparse/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/sparse/__pycache__/_base.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/sparse/__pycache__/_lowerings.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/sparse/__pycache__/ad.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/sparse/__pycache__/api.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/sparse/__pycache__/bcoo.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/sparse/__pycache__/bcsr.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/sparse/__pycache__/coo.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/sparse/__pycache__/csr.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/sparse/__pycache__/linalg.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/sparse/__pycache__/nm.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/sparse/__pycache__/random.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/sparse/__pycache__/test_util.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/sparse/__pycache__/transform.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/sparse/__pycache__/util.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/experimental/sparse/_base.pyusr/lib/python3.13/site-packages/jax/experimental/sparse/_lowerings.pyusr/lib/python3.13/site-packages/jax/experimental/sparse/ad.pyusr/lib/python3.13/site-packages/jax/experimental/sparse/api.pyusr/lib/python3.13/site-packages/jax/experimental/sparse/bcoo.pyusr/lib/python3.13/site-packages/jax/experimental/sparse/bcsr.pyusr/lib/python3.13/site-packages/jax/experimental/sparse/coo.pyusr/lib/python3.13/site-packages/jax/experimental/sparse/csr.pyusr/lib/python3.13/site-packages/jax/experimental/sparse/linalg.pyusr/lib/python3.13/site-packages/jax/experimental/sparse/nm.pyusr/lib/python3.13/site-packages/jax/experimental/sparse/random.pyusr/lib/python3.13/site-packages/jax/experimental/sparse/test_util.pyusr/lib/python3.13/site-packages/jax/experimental/sparse/transform.pyusr/lib/python3.13/site-packages/jax/experimental/sparse/util.pyusr/lib/python3.13/site-packages/jax/experimental/topologies.pyusr/lib/python3.13/site-packages/jax/experimental/transfer.pyusr/lib/python3.13/site-packages/jax/experimental/x64_context.pyusr/lib/python3.13/site-packages/jax/experimental/xla_metadata.pyusr/lib/python3.13/site-packages/jax/export.pyusr/lib/python3.13/site-packages/jax/extend/usr/lib/python3.13/site-packages/jax/extend/__init__.pyusr/lib/python3.13/site-packages/jax/extend/__pycache__/usr/lib/python3.13/site-packages/jax/extend/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/extend/__pycache__/backend.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/extend/__pycache__/ffi.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/extend/__pycache__/ifrt_programs.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/extend/__pycache__/linear_util.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/extend/__pycache__/random.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/extend/__pycache__/sharding.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/extend/__pycache__/source_info_util.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/extend/backend.pyusr/lib/python3.13/site-packages/jax/extend/core/usr/lib/python3.13/site-packages/jax/extend/core/__init__.pyusr/lib/python3.13/site-packages/jax/extend/core/__pycache__/usr/lib/python3.13/site-packages/jax/extend/core/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/extend/core/__pycache__/primitives.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/extend/core/primitives.pyusr/lib/python3.13/site-packages/jax/extend/ffi.pyusr/lib/python3.13/site-packages/jax/extend/ifrt_programs.pyusr/lib/python3.13/site-packages/jax/extend/linear_util.pyusr/lib/python3.13/site-packages/jax/extend/mlir/usr/lib/python3.13/site-packages/jax/extend/mlir/__init__.pyusr/lib/python3.13/site-packages/jax/extend/mlir/__pycache__/usr/lib/python3.13/site-packages/jax/extend/mlir/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/extend/mlir/__pycache__/ir.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/extend/mlir/__pycache__/passmanager.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/extend/mlir/dialects/usr/lib/python3.13/site-packages/jax/extend/mlir/dialects/__init__.pyusr/lib/python3.13/site-packages/jax/extend/mlir/dialects/__pycache__/usr/lib/python3.13/site-packages/jax/extend/mlir/dialects/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/extend/mlir/dialects/__pycache__/arith.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/extend/mlir/dialects/__pycache__/builtin.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/extend/mlir/dialects/__pycache__/chlo.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/extend/mlir/dialects/__pycache__/func.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/extend/mlir/dialects/__pycache__/math.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/extend/mlir/dialects/__pycache__/memref.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/extend/mlir/dialects/__pycache__/scf.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/extend/mlir/dialects/__pycache__/sdy.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/extend/mlir/dialects/__pycache__/sparse_tensor.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/extend/mlir/dialects/__pycache__/stablehlo.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/extend/mlir/dialects/__pycache__/vector.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/extend/mlir/dialects/arith.pyusr/lib/python3.13/site-packages/jax/extend/mlir/dialects/builtin.pyusr/lib/python3.13/site-packages/jax/extend/mlir/dialects/chlo.pyusr/lib/python3.13/site-packages/jax/extend/mlir/dialects/func.pyusr/lib/python3.13/site-packages/jax/extend/mlir/dialects/math.pyusr/lib/python3.13/site-packages/jax/extend/mlir/dialects/memref.pyusr/lib/python3.13/site-packages/jax/extend/mlir/dialects/scf.pyusr/lib/python3.13/site-packages/jax/extend/mlir/dialects/sdy.pyusr/lib/python3.13/site-packages/jax/extend/mlir/dialects/sparse_tensor.pyusr/lib/python3.13/site-packages/jax/extend/mlir/dialects/stablehlo.pyusr/lib/python3.13/site-packages/jax/extend/mlir/dialects/vector.pyusr/lib/python3.13/site-packages/jax/extend/mlir/ir.pyusr/lib/python3.13/site-packages/jax/extend/mlir/passmanager.pyusr/lib/python3.13/site-packages/jax/extend/random.pyusr/lib/python3.13/site-packages/jax/extend/sharding.pyusr/lib/python3.13/site-packages/jax/extend/source_info_util.pyusr/lib/python3.13/site-packages/jax/ffi.pyusr/lib/python3.13/site-packages/jax/flatten_util.pyusr/lib/python3.13/site-packages/jax/image/usr/lib/python3.13/site-packages/jax/image/__init__.pyusr/lib/python3.13/site-packages/jax/image/__pycache__/usr/lib/python3.13/site-packages/jax/image/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/interpreters/usr/lib/python3.13/site-packages/jax/interpreters/__init__.pyusr/lib/python3.13/site-packages/jax/interpreters/__pycache__/usr/lib/python3.13/site-packages/jax/interpreters/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/interpreters/__pycache__/ad.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/interpreters/__pycache__/batching.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/interpreters/__pycache__/mlir.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/interpreters/__pycache__/partial_eval.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/interpreters/__pycache__/pxla.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/interpreters/__pycache__/xla.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/interpreters/ad.pyusr/lib/python3.13/site-packages/jax/interpreters/batching.pyusr/lib/python3.13/site-packages/jax/interpreters/mlir.pyusr/lib/python3.13/site-packages/jax/interpreters/partial_eval.pyusr/lib/python3.13/site-packages/jax/interpreters/pxla.pyusr/lib/python3.13/site-packages/jax/interpreters/xla.pyusr/lib/python3.13/site-packages/jax/lax/usr/lib/python3.13/site-packages/jax/lax/__init__.pyusr/lib/python3.13/site-packages/jax/lax/__pycache__/usr/lib/python3.13/site-packages/jax/lax/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/lax/__pycache__/linalg.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/lax/linalg.pyusr/lib/python3.13/site-packages/jax/lib/usr/lib/python3.13/site-packages/jax/lib/__init__.pyusr/lib/python3.13/site-packages/jax/lib/__pycache__/usr/lib/python3.13/site-packages/jax/lib/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/lib/__pycache__/xla_bridge.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/lib/__pycache__/xla_client.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/lib/__pycache__/xla_extension.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/lib/xla_bridge.pyusr/lib/python3.13/site-packages/jax/lib/xla_client.pyusr/lib/python3.13/site-packages/jax/lib/xla_extension.pyusr/lib/python3.13/site-packages/jax/monitoring.pyusr/lib/python3.13/site-packages/jax/nn/usr/lib/python3.13/site-packages/jax/nn/__init__.pyusr/lib/python3.13/site-packages/jax/nn/__init__.pyiusr/lib/python3.13/site-packages/jax/nn/__pycache__/usr/lib/python3.13/site-packages/jax/nn/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/nn/__pycache__/initializers.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/nn/initializers.pyusr/lib/python3.13/site-packages/jax/numpy/usr/lib/python3.13/site-packages/jax/numpy/__init__.pyusr/lib/python3.13/site-packages/jax/numpy/__init__.pyiusr/lib/python3.13/site-packages/jax/numpy/__pycache__/usr/lib/python3.13/site-packages/jax/numpy/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/numpy/__pycache__/fft.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/numpy/__pycache__/linalg.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/numpy/fft.pyusr/lib/python3.13/site-packages/jax/numpy/linalg.pyusr/lib/python3.13/site-packages/jax/ops/usr/lib/python3.13/site-packages/jax/ops/__init__.pyusr/lib/python3.13/site-packages/jax/ops/__pycache__/usr/lib/python3.13/site-packages/jax/ops/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/profiler.pyusr/lib/python3.13/site-packages/jax/py.typedusr/lib/python3.13/site-packages/jax/random.pyusr/lib/python3.13/site-packages/jax/scipy/usr/lib/python3.13/site-packages/jax/scipy/__init__.pyusr/lib/python3.13/site-packages/jax/scipy/__pycache__/usr/lib/python3.13/site-packages/jax/scipy/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/scipy/__pycache__/fft.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/scipy/__pycache__/integrate.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/scipy/__pycache__/linalg.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/scipy/__pycache__/ndimage.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/scipy/__pycache__/signal.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/scipy/__pycache__/special.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/scipy/cluster/usr/lib/python3.13/site-packages/jax/scipy/cluster/__init__.pyusr/lib/python3.13/site-packages/jax/scipy/cluster/__pycache__/usr/lib/python3.13/site-packages/jax/scipy/cluster/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/scipy/cluster/__pycache__/vq.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/scipy/cluster/vq.pyusr/lib/python3.13/site-packages/jax/scipy/fft.pyusr/lib/python3.13/site-packages/jax/scipy/integrate.pyusr/lib/python3.13/site-packages/jax/scipy/interpolate/usr/lib/python3.13/site-packages/jax/scipy/interpolate/__init__.pyusr/lib/python3.13/site-packages/jax/scipy/interpolate/__pycache__/usr/lib/python3.13/site-packages/jax/scipy/interpolate/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/scipy/linalg.pyusr/lib/python3.13/site-packages/jax/scipy/ndimage.pyusr/lib/python3.13/site-packages/jax/scipy/optimize/usr/lib/python3.13/site-packages/jax/scipy/optimize/__init__.pyusr/lib/python3.13/site-packages/jax/scipy/optimize/__pycache__/usr/lib/python3.13/site-packages/jax/scipy/optimize/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/scipy/signal.pyusr/lib/python3.13/site-packages/jax/scipy/sparse/usr/lib/python3.13/site-packages/jax/scipy/sparse/__init__.pyusr/lib/python3.13/site-packages/jax/scipy/sparse/__pycache__/usr/lib/python3.13/site-packages/jax/scipy/sparse/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/scipy/sparse/__pycache__/linalg.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/scipy/sparse/linalg.pyusr/lib/python3.13/site-packages/jax/scipy/spatial/usr/lib/python3.13/site-packages/jax/scipy/spatial/__init__.pyusr/lib/python3.13/site-packages/jax/scipy/spatial/__pycache__/usr/lib/python3.13/site-packages/jax/scipy/spatial/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/scipy/spatial/__pycache__/transform.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/scipy/spatial/transform.pyusr/lib/python3.13/site-packages/jax/scipy/special.pyusr/lib/python3.13/site-packages/jax/scipy/stats/usr/lib/python3.13/site-packages/jax/scipy/stats/__init__.pyusr/lib/python3.13/site-packages/jax/scipy/stats/__pycache__/usr/lib/python3.13/site-packages/jax/scipy/stats/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/scipy/stats/__pycache__/bernoulli.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/scipy/stats/__pycache__/beta.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/scipy/stats/__pycache__/betabinom.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/scipy/stats/__pycache__/binom.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/scipy/stats/__pycache__/cauchy.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/scipy/stats/__pycache__/chi2.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/scipy/stats/__pycache__/dirichlet.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/scipy/stats/__pycache__/expon.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/scipy/stats/__pycache__/gamma.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/scipy/stats/__pycache__/gennorm.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/scipy/stats/__pycache__/geom.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/scipy/stats/__pycache__/gumbel_l.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/scipy/stats/__pycache__/gumbel_r.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/scipy/stats/__pycache__/laplace.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/scipy/stats/__pycache__/logistic.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/scipy/stats/__pycache__/multinomial.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/scipy/stats/__pycache__/multivariate_normal.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/scipy/stats/__pycache__/nbinom.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/scipy/stats/__pycache__/norm.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/scipy/stats/__pycache__/pareto.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/scipy/stats/__pycache__/poisson.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/scipy/stats/__pycache__/t.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/scipy/stats/__pycache__/truncnorm.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/scipy/stats/__pycache__/uniform.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/scipy/stats/__pycache__/vonmises.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/scipy/stats/__pycache__/wrapcauchy.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/scipy/stats/bernoulli.pyusr/lib/python3.13/site-packages/jax/scipy/stats/beta.pyusr/lib/python3.13/site-packages/jax/scipy/stats/betabinom.pyusr/lib/python3.13/site-packages/jax/scipy/stats/binom.pyusr/lib/python3.13/site-packages/jax/scipy/stats/cauchy.pyusr/lib/python3.13/site-packages/jax/scipy/stats/chi2.pyusr/lib/python3.13/site-packages/jax/scipy/stats/dirichlet.pyusr/lib/python3.13/site-packages/jax/scipy/stats/expon.pyusr/lib/python3.13/site-packages/jax/scipy/stats/gamma.pyusr/lib/python3.13/site-packages/jax/scipy/stats/gennorm.pyusr/lib/python3.13/site-packages/jax/scipy/stats/geom.pyusr/lib/python3.13/site-packages/jax/scipy/stats/gumbel_l.pyusr/lib/python3.13/site-packages/jax/scipy/stats/gumbel_r.pyusr/lib/python3.13/site-packages/jax/scipy/stats/laplace.pyusr/lib/python3.13/site-packages/jax/scipy/stats/logistic.pyusr/lib/python3.13/site-packages/jax/scipy/stats/multinomial.pyusr/lib/python3.13/site-packages/jax/scipy/stats/multivariate_normal.pyusr/lib/python3.13/site-packages/jax/scipy/stats/nbinom.pyusr/lib/python3.13/site-packages/jax/scipy/stats/norm.pyusr/lib/python3.13/site-packages/jax/scipy/stats/pareto.pyusr/lib/python3.13/site-packages/jax/scipy/stats/poisson.pyusr/lib/python3.13/site-packages/jax/scipy/stats/t.pyusr/lib/python3.13/site-packages/jax/scipy/stats/truncnorm.pyusr/lib/python3.13/site-packages/jax/scipy/stats/uniform.pyusr/lib/python3.13/site-packages/jax/scipy/stats/vonmises.pyusr/lib/python3.13/site-packages/jax/scipy/stats/wrapcauchy.pyusr/lib/python3.13/site-packages/jax/sharding.pyusr/lib/python3.13/site-packages/jax/stages.pyusr/lib/python3.13/site-packages/jax/test_util.pyusr/lib/python3.13/site-packages/jax/tools/usr/lib/python3.13/site-packages/jax/tools/__init__.pyusr/lib/python3.13/site-packages/jax/tools/__pycache__/usr/lib/python3.13/site-packages/jax/tools/__pycache__/__init__.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/tools/__pycache__/colab_tpu.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/tools/__pycache__/jax_to_ir.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/tools/__pycache__/pgo_nsys_converter.cpython-313.opt-1.pycusr/lib/python3.13/site-packages/jax/tools/colab_tpu.pyusr/lib/python3.13/site-packages/jax/tools/jax_to_ir.pyusr/lib/python3.13/site-packages/jax/tools/pgo_nsys_converter.pyusr/lib/python3.13/site-packages/jax/tree.pyusr/lib/python3.13/site-packages/jax/tree_util.pyusr/lib/python3.13/site-packages/jax/typing.pyusr/lib/python3.13/site-packages/jax/util.pyusr/lib/python3.13/site-packages/jax/version.pyusr/share/usr/share/licenses/usr/share/licenses/python-jax/usr/share/licenses/python-jax/LICENSE