lib/python3.9/site-packages/jax-0.4.23.dist-info/AUTHORS
lib/python3.9/site-packages/jax-0.4.23.dist-info/INSTALLER
lib/python3.9/site-packages/jax-0.4.23.dist-info/LICENSE
lib/python3.9/site-packages/jax-0.4.23.dist-info/METADATA
lib/python3.9/site-packages/jax-0.4.23.dist-info/RECORD
lib/python3.9/site-packages/jax-0.4.23.dist-info/REQUESTED
lib/python3.9/site-packages/jax-0.4.23.dist-info/WHEEL
lib/python3.9/site-packages/jax-0.4.23.dist-info/direct_url.json
lib/python3.9/site-packages/jax-0.4.23.dist-info/top_level.txt
lib/python3.9/site-packages/jax/__init__.py
lib/python3.9/site-packages/jax/__pycache__/__init__.cpython-39.pyc
lib/python3.9/site-packages/jax/__pycache__/ad_checkpoint.cpython-39.pyc
lib/python3.9/site-packages/jax/__pycache__/api_util.cpython-39.pyc
lib/python3.9/site-packages/jax/__pycache__/cloud_tpu_init.cpython-39.pyc
lib/python3.9/site-packages/jax/__pycache__/collect_profile.cpython-39.pyc
lib/python3.9/site-packages/jax/__pycache__/config.cpython-39.pyc
lib/python3.9/site-packages/jax/__pycache__/core.cpython-39.pyc
lib/python3.9/site-packages/jax/__pycache__/custom_batching.cpython-39.pyc
lib/python3.9/site-packages/jax/__pycache__/custom_derivatives.cpython-39.pyc
lib/python3.9/site-packages/jax/__pycache__/custom_transpose.cpython-39.pyc
lib/python3.9/site-packages/jax/__pycache__/debug.cpython-39.pyc
lib/python3.9/site-packages/jax/__pycache__/distributed.cpython-39.pyc
lib/python3.9/site-packages/jax/__pycache__/dlpack.cpython-39.pyc
lib/python3.9/site-packages/jax/__pycache__/dtypes.cpython-39.pyc
lib/python3.9/site-packages/jax/__pycache__/errors.cpython-39.pyc
lib/python3.9/site-packages/jax/__pycache__/flatten_util.cpython-39.pyc
lib/python3.9/site-packages/jax/__pycache__/linear_util.cpython-39.pyc
lib/python3.9/site-packages/jax/__pycache__/monitoring.cpython-39.pyc
lib/python3.9/site-packages/jax/__pycache__/prng.cpython-39.pyc
lib/python3.9/site-packages/jax/__pycache__/profiler.cpython-39.pyc
lib/python3.9/site-packages/jax/__pycache__/random.cpython-39.pyc
lib/python3.9/site-packages/jax/__pycache__/sharding.cpython-39.pyc
lib/python3.9/site-packages/jax/__pycache__/stages.cpython-39.pyc
lib/python3.9/site-packages/jax/__pycache__/test_util.cpython-39.pyc
lib/python3.9/site-packages/jax/__pycache__/tree_util.cpython-39.pyc
lib/python3.9/site-packages/jax/__pycache__/typing.cpython-39.pyc
lib/python3.9/site-packages/jax/__pycache__/util.cpython-39.pyc
lib/python3.9/site-packages/jax/__pycache__/version.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/__init__.py
lib/python3.9/site-packages/jax/_src/__pycache__/__init__.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/__pycache__/abstract_arrays.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/__pycache__/ad_checkpoint.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/__pycache__/ad_util.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/__pycache__/api.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/__pycache__/api_util.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/__pycache__/array.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/__pycache__/basearray.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/__pycache__/cache_key.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/__pycache__/callback.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/__pycache__/checkify.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/__pycache__/cloud_tpu_init.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/__pycache__/compilation_cache.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/__pycache__/compilation_cache_interface.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/__pycache__/compiler.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/__pycache__/config.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/__pycache__/core.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/__pycache__/custom_api_util.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/__pycache__/custom_batching.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/__pycache__/custom_derivatives.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/__pycache__/custom_transpose.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/__pycache__/debugging.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/__pycache__/deprecations.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/__pycache__/dispatch.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/__pycache__/distributed.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/__pycache__/dlpack.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/__pycache__/dtypes.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/__pycache__/effects.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/__pycache__/environment_info.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/__pycache__/errors.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/__pycache__/flatten_util.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/__pycache__/gfile_cache.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/__pycache__/jaxpr_util.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/__pycache__/lax_reference.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/__pycache__/layout.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/__pycache__/lazy_loader.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/__pycache__/linear_util.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/__pycache__/logging_config.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/__pycache__/maps.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/__pycache__/mesh.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/__pycache__/monitoring.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/__pycache__/op_shardings.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/__pycache__/partition_spec.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/__pycache__/path.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/__pycache__/pickle_util.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/__pycache__/pjit.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/__pycache__/pretty_printer.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/__pycache__/prng.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/__pycache__/profiler.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/__pycache__/public_test_util.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/__pycache__/random.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/__pycache__/sharding.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/__pycache__/sharding_impls.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/__pycache__/sharding_specs.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/__pycache__/source_info_util.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/__pycache__/stages.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/__pycache__/test_util.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/__pycache__/tpu_custom_call.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/__pycache__/traceback_util.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/__pycache__/tree_util.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/__pycache__/typing.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/__pycache__/util.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/__pycache__/xla_bridge.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/abstract_arrays.py
lib/python3.9/site-packages/jax/_src/ad_checkpoint.py
lib/python3.9/site-packages/jax/_src/ad_util.py
lib/python3.9/site-packages/jax/_src/api.py
lib/python3.9/site-packages/jax/_src/api_util.py
lib/python3.9/site-packages/jax/_src/array.py
lib/python3.9/site-packages/jax/_src/basearray.py
lib/python3.9/site-packages/jax/_src/basearray.pyi
lib/python3.9/site-packages/jax/_src/cache_key.py
lib/python3.9/site-packages/jax/_src/callback.py
lib/python3.9/site-packages/jax/_src/checkify.py
lib/python3.9/site-packages/jax/_src/cloud_tpu_init.py
lib/python3.9/site-packages/jax/_src/clusters/__init__.py
lib/python3.9/site-packages/jax/_src/clusters/__pycache__/__init__.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/clusters/__pycache__/cloud_tpu_cluster.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/clusters/__pycache__/cluster.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/clusters/__pycache__/ompi_cluster.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/clusters/__pycache__/slurm_cluster.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/clusters/cloud_tpu_cluster.py
lib/python3.9/site-packages/jax/_src/clusters/cluster.py
lib/python3.9/site-packages/jax/_src/clusters/ompi_cluster.py
lib/python3.9/site-packages/jax/_src/clusters/slurm_cluster.py
lib/python3.9/site-packages/jax/_src/compilation_cache.py
lib/python3.9/site-packages/jax/_src/compilation_cache_interface.py
lib/python3.9/site-packages/jax/_src/compiler.py
lib/python3.9/site-packages/jax/_src/config.py
lib/python3.9/site-packages/jax/_src/core.py
lib/python3.9/site-packages/jax/_src/custom_api_util.py
lib/python3.9/site-packages/jax/_src/custom_batching.py
lib/python3.9/site-packages/jax/_src/custom_derivatives.py
lib/python3.9/site-packages/jax/_src/custom_transpose.py
lib/python3.9/site-packages/jax/_src/debugger/__init__.py
lib/python3.9/site-packages/jax/_src/debugger/__pycache__/__init__.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/debugger/__pycache__/cli_debugger.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/debugger/__pycache__/colab_debugger.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/debugger/__pycache__/colab_lib.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/debugger/__pycache__/core.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/debugger/__pycache__/web_debugger.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/debugger/cli_debugger.py
lib/python3.9/site-packages/jax/_src/debugger/colab_debugger.py
lib/python3.9/site-packages/jax/_src/debugger/colab_lib.py
lib/python3.9/site-packages/jax/_src/debugger/core.py
lib/python3.9/site-packages/jax/_src/debugger/web_debugger.py
lib/python3.9/site-packages/jax/_src/debugging.py
lib/python3.9/site-packages/jax/_src/deprecations.py
lib/python3.9/site-packages/jax/_src/dispatch.py
lib/python3.9/site-packages/jax/_src/distributed.py
lib/python3.9/site-packages/jax/_src/dlpack.py
lib/python3.9/site-packages/jax/_src/dtypes.py
lib/python3.9/site-packages/jax/_src/effects.py
lib/python3.9/site-packages/jax/_src/environment_info.py
lib/python3.9/site-packages/jax/_src/errors.py
lib/python3.9/site-packages/jax/_src/extend/__init__.py
lib/python3.9/site-packages/jax/_src/extend/__pycache__/__init__.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/extend/__pycache__/random.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/extend/random.py
lib/python3.9/site-packages/jax/_src/flatten_util.py
lib/python3.9/site-packages/jax/_src/gfile_cache.py
lib/python3.9/site-packages/jax/_src/image/__init__.py
lib/python3.9/site-packages/jax/_src/image/__pycache__/__init__.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/image/__pycache__/scale.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/image/scale.py
lib/python3.9/site-packages/jax/_src/internal_test_util/__init__.py
lib/python3.9/site-packages/jax/_src/internal_test_util/__pycache__/__init__.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/internal_test_util/__pycache__/deprecation_module.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/internal_test_util/__pycache__/export_back_compat_test_util.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/internal_test_util/__pycache__/lax_test_util.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/internal_test_util/__pycache__/test_harnesses.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/internal_test_util/deprecation_module.py
lib/python3.9/site-packages/jax/_src/internal_test_util/export_back_compat_test_util.py
lib/python3.9/site-packages/jax/_src/internal_test_util/lax_test_util.py
lib/python3.9/site-packages/jax/_src/internal_test_util/lazy_loader_module/__init__.py
lib/python3.9/site-packages/jax/_src/internal_test_util/lazy_loader_module/__pycache__/__init__.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/internal_test_util/lazy_loader_module/__pycache__/lazy_test_submodule.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/internal_test_util/lazy_loader_module/lazy_test_submodule.py
lib/python3.9/site-packages/jax/_src/internal_test_util/test_harnesses.py
lib/python3.9/site-packages/jax/_src/interpreters/__init__.py
lib/python3.9/site-packages/jax/_src/interpreters/__pycache__/__init__.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/interpreters/__pycache__/ad.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/interpreters/__pycache__/batching.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/interpreters/__pycache__/mlir.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/interpreters/__pycache__/partial_eval.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/interpreters/__pycache__/pxla.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/interpreters/__pycache__/xla.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/interpreters/ad.py
lib/python3.9/site-packages/jax/_src/interpreters/batching.py
lib/python3.9/site-packages/jax/_src/interpreters/mlir.py
lib/python3.9/site-packages/jax/_src/interpreters/partial_eval.py
lib/python3.9/site-packages/jax/_src/interpreters/pxla.py
lib/python3.9/site-packages/jax/_src/interpreters/xla.py
lib/python3.9/site-packages/jax/_src/jaxpr_util.py
lib/python3.9/site-packages/jax/_src/lax/__init__.py
lib/python3.9/site-packages/jax/_src/lax/__pycache__/__init__.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/lax/__pycache__/ann.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/lax/__pycache__/convolution.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/lax/__pycache__/eigh.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/lax/__pycache__/fft.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/lax/__pycache__/lax.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/lax/__pycache__/linalg.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/lax/__pycache__/other.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/lax/__pycache__/parallel.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/lax/__pycache__/qdwh.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/lax/__pycache__/slicing.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/lax/__pycache__/special.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/lax/__pycache__/stack.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/lax/__pycache__/svd.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/lax/__pycache__/utils.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/lax/__pycache__/windowed_reductions.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/lax/ann.py
lib/python3.9/site-packages/jax/_src/lax/control_flow/__init__.py
lib/python3.9/site-packages/jax/_src/lax/control_flow/__pycache__/__init__.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/lax/control_flow/__pycache__/common.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/lax/control_flow/__pycache__/conditionals.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/lax/control_flow/__pycache__/for_loop.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/lax/control_flow/__pycache__/loops.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/lax/control_flow/__pycache__/solves.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/lax/control_flow/common.py
lib/python3.9/site-packages/jax/_src/lax/control_flow/conditionals.py
lib/python3.9/site-packages/jax/_src/lax/control_flow/for_loop.py
lib/python3.9/site-packages/jax/_src/lax/control_flow/loops.py
lib/python3.9/site-packages/jax/_src/lax/control_flow/solves.py
lib/python3.9/site-packages/jax/_src/lax/convolution.py
lib/python3.9/site-packages/jax/_src/lax/eigh.py
lib/python3.9/site-packages/jax/_src/lax/fft.py
lib/python3.9/site-packages/jax/_src/lax/lax.py
lib/python3.9/site-packages/jax/_src/lax/linalg.py
lib/python3.9/site-packages/jax/_src/lax/other.py
lib/python3.9/site-packages/jax/_src/lax/parallel.py
lib/python3.9/site-packages/jax/_src/lax/qdwh.py
lib/python3.9/site-packages/jax/_src/lax/slicing.py
lib/python3.9/site-packages/jax/_src/lax/special.py
lib/python3.9/site-packages/jax/_src/lax/stack.py
lib/python3.9/site-packages/jax/_src/lax/svd.py
lib/python3.9/site-packages/jax/_src/lax/utils.py
lib/python3.9/site-packages/jax/_src/lax/windowed_reductions.py
lib/python3.9/site-packages/jax/_src/lax_reference.py
lib/python3.9/site-packages/jax/_src/layout.py
lib/python3.9/site-packages/jax/_src/lazy_loader.py
lib/python3.9/site-packages/jax/_src/lib/__init__.py
lib/python3.9/site-packages/jax/_src/lib/__pycache__/__init__.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/lib/mlir/__init__.py
lib/python3.9/site-packages/jax/_src/lib/mlir/__pycache__/__init__.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/lib/mlir/dialects/__init__.py
lib/python3.9/site-packages/jax/_src/lib/mlir/dialects/__pycache__/__init__.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/linear_util.py
lib/python3.9/site-packages/jax/_src/logging_config.py
lib/python3.9/site-packages/jax/_src/maps.py
lib/python3.9/site-packages/jax/_src/mesh.py
lib/python3.9/site-packages/jax/_src/monitoring.py
lib/python3.9/site-packages/jax/_src/nn/__init__.py
lib/python3.9/site-packages/jax/_src/nn/__pycache__/__init__.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/nn/__pycache__/functions.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/nn/__pycache__/initializers.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/nn/functions.py
lib/python3.9/site-packages/jax/_src/nn/initializers.py
lib/python3.9/site-packages/jax/_src/numpy/__init__.py
lib/python3.9/site-packages/jax/_src/numpy/__pycache__/__init__.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/numpy/__pycache__/array_methods.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/numpy/__pycache__/fft.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/numpy/__pycache__/index_tricks.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/numpy/__pycache__/lax_numpy.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/numpy/__pycache__/linalg.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/numpy/__pycache__/polynomial.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/numpy/__pycache__/reductions.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/numpy/__pycache__/setops.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/numpy/__pycache__/ufunc_api.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/numpy/__pycache__/ufuncs.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/numpy/__pycache__/util.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/numpy/__pycache__/vectorize.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/numpy/array_methods.py
lib/python3.9/site-packages/jax/_src/numpy/fft.py
lib/python3.9/site-packages/jax/_src/numpy/index_tricks.py
lib/python3.9/site-packages/jax/_src/numpy/lax_numpy.py
lib/python3.9/site-packages/jax/_src/numpy/linalg.py
lib/python3.9/site-packages/jax/_src/numpy/polynomial.py
lib/python3.9/site-packages/jax/_src/numpy/reductions.py
lib/python3.9/site-packages/jax/_src/numpy/setops.py
lib/python3.9/site-packages/jax/_src/numpy/ufunc_api.py
lib/python3.9/site-packages/jax/_src/numpy/ufuncs.py
lib/python3.9/site-packages/jax/_src/numpy/util.py
lib/python3.9/site-packages/jax/_src/numpy/vectorize.py
lib/python3.9/site-packages/jax/_src/op_shardings.py
lib/python3.9/site-packages/jax/_src/ops/__init__.py
lib/python3.9/site-packages/jax/_src/ops/__pycache__/__init__.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/ops/__pycache__/scatter.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/ops/__pycache__/special.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/ops/scatter.py
lib/python3.9/site-packages/jax/_src/ops/special.py
lib/python3.9/site-packages/jax/_src/pallas/__init__.py
lib/python3.9/site-packages/jax/_src/pallas/__pycache__/__init__.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/pallas/__pycache__/core.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/pallas/__pycache__/indexing.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/pallas/__pycache__/pallas_call.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/pallas/__pycache__/primitives.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/pallas/__pycache__/utils.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/pallas/core.py
lib/python3.9/site-packages/jax/_src/pallas/indexing.py
lib/python3.9/site-packages/jax/_src/pallas/mosaic/__init__.py
lib/python3.9/site-packages/jax/_src/pallas/mosaic/__pycache__/__init__.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/pallas/mosaic/__pycache__/core.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/pallas/mosaic/__pycache__/kernel_regeneration_util.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/pallas/mosaic/__pycache__/lowering.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/pallas/mosaic/__pycache__/pallas_call_registration.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/pallas/mosaic/__pycache__/primitives.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/pallas/mosaic/core.py
lib/python3.9/site-packages/jax/_src/pallas/mosaic/kernel_regeneration_util.py
lib/python3.9/site-packages/jax/_src/pallas/mosaic/lowering.py
lib/python3.9/site-packages/jax/_src/pallas/mosaic/pallas_call_registration.py
lib/python3.9/site-packages/jax/_src/pallas/mosaic/primitives.py
lib/python3.9/site-packages/jax/_src/pallas/pallas_call.py
lib/python3.9/site-packages/jax/_src/pallas/primitives.py
lib/python3.9/site-packages/jax/_src/pallas/triton/__init__.py
lib/python3.9/site-packages/jax/_src/pallas/triton/__pycache__/__init__.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/pallas/triton/__pycache__/lowering.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/pallas/triton/lowering.py
lib/python3.9/site-packages/jax/_src/pallas/utils.py
lib/python3.9/site-packages/jax/_src/partition_spec.py
lib/python3.9/site-packages/jax/_src/path.py
lib/python3.9/site-packages/jax/_src/pickle_util.py
lib/python3.9/site-packages/jax/_src/pjit.py
lib/python3.9/site-packages/jax/_src/pretty_printer.py
lib/python3.9/site-packages/jax/_src/prng.py
lib/python3.9/site-packages/jax/_src/profiler.py
lib/python3.9/site-packages/jax/_src/public_test_util.py
lib/python3.9/site-packages/jax/_src/random.py
lib/python3.9/site-packages/jax/_src/scipy/__init__.py
lib/python3.9/site-packages/jax/_src/scipy/__pycache__/__init__.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/scipy/__pycache__/fft.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/scipy/__pycache__/integrate.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/scipy/__pycache__/linalg.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/scipy/__pycache__/ndimage.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/scipy/__pycache__/signal.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/scipy/__pycache__/special.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/scipy/cluster/__init__.py
lib/python3.9/site-packages/jax/_src/scipy/cluster/__pycache__/__init__.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/scipy/cluster/__pycache__/vq.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/scipy/cluster/vq.py
lib/python3.9/site-packages/jax/_src/scipy/fft.py
lib/python3.9/site-packages/jax/_src/scipy/integrate.py
lib/python3.9/site-packages/jax/_src/scipy/interpolate/__init__.py
lib/python3.9/site-packages/jax/_src/scipy/interpolate/__pycache__/__init__.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/scipy/linalg.py
lib/python3.9/site-packages/jax/_src/scipy/ndimage.py
lib/python3.9/site-packages/jax/_src/scipy/optimize/__init__.py
lib/python3.9/site-packages/jax/_src/scipy/optimize/__pycache__/__init__.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/scipy/optimize/__pycache__/_lbfgs.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/scipy/optimize/__pycache__/bfgs.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/scipy/optimize/__pycache__/line_search.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/scipy/optimize/__pycache__/minimize.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/scipy/optimize/_lbfgs.py
lib/python3.9/site-packages/jax/_src/scipy/optimize/bfgs.py
lib/python3.9/site-packages/jax/_src/scipy/optimize/line_search.py
lib/python3.9/site-packages/jax/_src/scipy/optimize/minimize.py
lib/python3.9/site-packages/jax/_src/scipy/signal.py
lib/python3.9/site-packages/jax/_src/scipy/sparse/__init__.py
lib/python3.9/site-packages/jax/_src/scipy/sparse/__pycache__/__init__.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/scipy/sparse/__pycache__/linalg.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/scipy/sparse/linalg.py
lib/python3.9/site-packages/jax/_src/scipy/spatial/__init__.py
lib/python3.9/site-packages/jax/_src/scipy/spatial/__pycache__/__init__.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/scipy/spatial/__pycache__/transform.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/scipy/spatial/transform.py
lib/python3.9/site-packages/jax/_src/scipy/special.py
lib/python3.9/site-packages/jax/_src/scipy/stats/__init__.py
lib/python3.9/site-packages/jax/_src/scipy/stats/__pycache__/__init__.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/scipy/stats/__pycache__/_core.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/scipy/stats/__pycache__/bernoulli.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/scipy/stats/__pycache__/beta.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/scipy/stats/__pycache__/betabinom.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/scipy/stats/__pycache__/binom.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/scipy/stats/__pycache__/cauchy.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/scipy/stats/__pycache__/chi2.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/scipy/stats/__pycache__/dirichlet.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/scipy/stats/__pycache__/expon.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/scipy/stats/__pycache__/gamma.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/scipy/stats/__pycache__/gennorm.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/scipy/stats/__pycache__/geom.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/scipy/stats/__pycache__/kde.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/scipy/stats/__pycache__/laplace.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/scipy/stats/__pycache__/logistic.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/scipy/stats/__pycache__/multinomial.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/scipy/stats/__pycache__/multivariate_normal.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/scipy/stats/__pycache__/nbinom.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/scipy/stats/__pycache__/norm.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/scipy/stats/__pycache__/pareto.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/scipy/stats/__pycache__/poisson.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/scipy/stats/__pycache__/t.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/scipy/stats/__pycache__/truncnorm.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/scipy/stats/__pycache__/uniform.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/scipy/stats/__pycache__/vonmises.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/scipy/stats/__pycache__/wrapcauchy.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/scipy/stats/_core.py
lib/python3.9/site-packages/jax/_src/scipy/stats/bernoulli.py
lib/python3.9/site-packages/jax/_src/scipy/stats/beta.py
lib/python3.9/site-packages/jax/_src/scipy/stats/betabinom.py
lib/python3.9/site-packages/jax/_src/scipy/stats/binom.py
lib/python3.9/site-packages/jax/_src/scipy/stats/cauchy.py
lib/python3.9/site-packages/jax/_src/scipy/stats/chi2.py
lib/python3.9/site-packages/jax/_src/scipy/stats/dirichlet.py
lib/python3.9/site-packages/jax/_src/scipy/stats/expon.py
lib/python3.9/site-packages/jax/_src/scipy/stats/gamma.py
lib/python3.9/site-packages/jax/_src/scipy/stats/gennorm.py
lib/python3.9/site-packages/jax/_src/scipy/stats/geom.py
lib/python3.9/site-packages/jax/_src/scipy/stats/kde.py
lib/python3.9/site-packages/jax/_src/scipy/stats/laplace.py
lib/python3.9/site-packages/jax/_src/scipy/stats/logistic.py
lib/python3.9/site-packages/jax/_src/scipy/stats/multinomial.py
lib/python3.9/site-packages/jax/_src/scipy/stats/multivariate_normal.py
lib/python3.9/site-packages/jax/_src/scipy/stats/nbinom.py
lib/python3.9/site-packages/jax/_src/scipy/stats/norm.py
lib/python3.9/site-packages/jax/_src/scipy/stats/pareto.py
lib/python3.9/site-packages/jax/_src/scipy/stats/poisson.py
lib/python3.9/site-packages/jax/_src/scipy/stats/t.py
lib/python3.9/site-packages/jax/_src/scipy/stats/truncnorm.py
lib/python3.9/site-packages/jax/_src/scipy/stats/uniform.py
lib/python3.9/site-packages/jax/_src/scipy/stats/vonmises.py
lib/python3.9/site-packages/jax/_src/scipy/stats/wrapcauchy.py
lib/python3.9/site-packages/jax/_src/sharding.py
lib/python3.9/site-packages/jax/_src/sharding_impls.py
lib/python3.9/site-packages/jax/_src/sharding_specs.py
lib/python3.9/site-packages/jax/_src/source_info_util.py
lib/python3.9/site-packages/jax/_src/stages.py
lib/python3.9/site-packages/jax/_src/state/__init__.py
lib/python3.9/site-packages/jax/_src/state/__pycache__/__init__.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/state/__pycache__/discharge.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/state/__pycache__/primitives.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/state/__pycache__/types.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/state/__pycache__/utils.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/state/discharge.py
lib/python3.9/site-packages/jax/_src/state/primitives.py
lib/python3.9/site-packages/jax/_src/state/types.py
lib/python3.9/site-packages/jax/_src/state/utils.py
lib/python3.9/site-packages/jax/_src/test_util.py
lib/python3.9/site-packages/jax/_src/third_party/__init__.py
lib/python3.9/site-packages/jax/_src/third_party/__pycache__/__init__.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/third_party/numpy/__init__.py
lib/python3.9/site-packages/jax/_src/third_party/numpy/__pycache__/__init__.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/third_party/numpy/__pycache__/linalg.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/third_party/numpy/linalg.py
lib/python3.9/site-packages/jax/_src/third_party/scipy/__init__.py
lib/python3.9/site-packages/jax/_src/third_party/scipy/__pycache__/__init__.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/third_party/scipy/__pycache__/betaln.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/third_party/scipy/__pycache__/interpolate.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/third_party/scipy/__pycache__/linalg.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/third_party/scipy/__pycache__/signal_helper.cpython-39.pyc
lib/python3.9/site-packages/jax/_src/third_party/scipy/betaln.py
lib/python3.9/site-packages/jax/_src/third_party/scipy/interpolate.py
lib/python3.9/site-packages/jax/_src/third_party/scipy/linalg.py
lib/python3.9/site-packages/jax/_src/third_party/scipy/signal_helper.py
lib/python3.9/site-packages/jax/_src/tpu_custom_call.py
lib/python3.9/site-packages/jax/_src/traceback_util.py
lib/python3.9/site-packages/jax/_src/tree_util.py
lib/python3.9/site-packages/jax/_src/typing.py
lib/python3.9/site-packages/jax/_src/util.py
lib/python3.9/site-packages/jax/_src/xla_bridge.py
lib/python3.9/site-packages/jax/ad_checkpoint.py
lib/python3.9/site-packages/jax/api_util.py
lib/python3.9/site-packages/jax/cloud_tpu_init.py
lib/python3.9/site-packages/jax/collect_profile.py
lib/python3.9/site-packages/jax/config.py
lib/python3.9/site-packages/jax/core.py
lib/python3.9/site-packages/jax/custom_batching.py
lib/python3.9/site-packages/jax/custom_derivatives.py
lib/python3.9/site-packages/jax/custom_transpose.py
lib/python3.9/site-packages/jax/debug.py
lib/python3.9/site-packages/jax/distributed.py
lib/python3.9/site-packages/jax/dlpack.py
lib/python3.9/site-packages/jax/dtypes.py
lib/python3.9/site-packages/jax/errors.py
lib/python3.9/site-packages/jax/example_libraries/__init__.py
lib/python3.9/site-packages/jax/example_libraries/__pycache__/__init__.cpython-39.pyc
lib/python3.9/site-packages/jax/example_libraries/__pycache__/optimizers.cpython-39.pyc
lib/python3.9/site-packages/jax/example_libraries/__pycache__/stax.cpython-39.pyc
lib/python3.9/site-packages/jax/example_libraries/optimizers.py
lib/python3.9/site-packages/jax/example_libraries/stax.py
lib/python3.9/site-packages/jax/experimental/__init__.py
lib/python3.9/site-packages/jax/experimental/__pycache__/__init__.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/__pycache__/checkify.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/__pycache__/custom_partitioning.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/__pycache__/host_callback.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/__pycache__/jet.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/__pycache__/layout.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/__pycache__/maps.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/__pycache__/mesh_utils.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/__pycache__/multihost_utils.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/__pycache__/ode.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/__pycache__/pjit.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/__pycache__/profiler.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/__pycache__/rnn.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/__pycache__/serialize_executable.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/__pycache__/shard_map.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/__pycache__/topologies.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/__pycache__/x64_context.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/array_api/__init__.py
lib/python3.9/site-packages/jax/experimental/array_api/__pycache__/__init__.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/array_api/__pycache__/_array_methods.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/array_api/__pycache__/_constants.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/array_api/__pycache__/_creation_functions.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/array_api/__pycache__/_data_type_functions.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/array_api/__pycache__/_dtypes.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/array_api/__pycache__/_elementwise_functions.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/array_api/__pycache__/_fft_functions.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/array_api/__pycache__/_indexing_functions.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/array_api/__pycache__/_linear_algebra_functions.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/array_api/__pycache__/_manipulation_functions.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/array_api/__pycache__/_searching_functions.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/array_api/__pycache__/_set_functions.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/array_api/__pycache__/_sorting_functions.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/array_api/__pycache__/_statistical_functions.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/array_api/__pycache__/_utility_functions.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/array_api/__pycache__/_version.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/array_api/__pycache__/fft.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/array_api/__pycache__/linalg.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/array_api/_array_methods.py
lib/python3.9/site-packages/jax/experimental/array_api/_constants.py
lib/python3.9/site-packages/jax/experimental/array_api/_creation_functions.py
lib/python3.9/site-packages/jax/experimental/array_api/_data_type_functions.py
lib/python3.9/site-packages/jax/experimental/array_api/_dtypes.py
lib/python3.9/site-packages/jax/experimental/array_api/_elementwise_functions.py
lib/python3.9/site-packages/jax/experimental/array_api/_fft_functions.py
lib/python3.9/site-packages/jax/experimental/array_api/_indexing_functions.py
lib/python3.9/site-packages/jax/experimental/array_api/_linear_algebra_functions.py
lib/python3.9/site-packages/jax/experimental/array_api/_manipulation_functions.py
lib/python3.9/site-packages/jax/experimental/array_api/_searching_functions.py
lib/python3.9/site-packages/jax/experimental/array_api/_set_functions.py
lib/python3.9/site-packages/jax/experimental/array_api/_sorting_functions.py
lib/python3.9/site-packages/jax/experimental/array_api/_statistical_functions.py
lib/python3.9/site-packages/jax/experimental/array_api/_utility_functions.py
lib/python3.9/site-packages/jax/experimental/array_api/_version.py
lib/python3.9/site-packages/jax/experimental/array_api/fft.py
lib/python3.9/site-packages/jax/experimental/array_api/linalg.py
lib/python3.9/site-packages/jax/experimental/array_serialization/__init__.py
lib/python3.9/site-packages/jax/experimental/array_serialization/__pycache__/__init__.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/array_serialization/__pycache__/serialization.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/array_serialization/__pycache__/serialization_test.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/array_serialization/serialization.py
lib/python3.9/site-packages/jax/experimental/array_serialization/serialization_test.py
lib/python3.9/site-packages/jax/experimental/checkify.py
lib/python3.9/site-packages/jax/experimental/compilation_cache/__init__.py
lib/python3.9/site-packages/jax/experimental/compilation_cache/__pycache__/__init__.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/compilation_cache/__pycache__/compilation_cache.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/compilation_cache/compilation_cache.py
lib/python3.9/site-packages/jax/experimental/custom_partitioning.py
lib/python3.9/site-packages/jax/experimental/export/__init__.py
lib/python3.9/site-packages/jax/experimental/export/__pycache__/__init__.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/export/__pycache__/export.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/export/__pycache__/serialization.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/export/__pycache__/serialization_generated.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/export/__pycache__/shape_poly.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/export/export.py
lib/python3.9/site-packages/jax/experimental/export/serialization.py
lib/python3.9/site-packages/jax/experimental/export/serialization_generated.py
lib/python3.9/site-packages/jax/experimental/export/shape_poly.py
lib/python3.9/site-packages/jax/experimental/host_callback.py
lib/python3.9/site-packages/jax/experimental/jax2tf/__init__.py
lib/python3.9/site-packages/jax/experimental/jax2tf/__pycache__/__init__.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/jax2tf/__pycache__/call_tf.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/jax2tf/__pycache__/impl_no_xla.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/jax2tf/__pycache__/jax2tf.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/jax2tf/call_tf.py
lib/python3.9/site-packages/jax/experimental/jax2tf/examples/__init__.py
lib/python3.9/site-packages/jax/experimental/jax2tf/examples/__pycache__/__init__.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/jax2tf/examples/__pycache__/keras_reuse_main.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/jax2tf/examples/__pycache__/keras_reuse_main_test.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/jax2tf/examples/__pycache__/mnist_lib.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/jax2tf/examples/__pycache__/saved_model_lib.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/jax2tf/examples/__pycache__/saved_model_main.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/jax2tf/examples/__pycache__/saved_model_main_test.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/jax2tf/examples/keras_reuse_main.py
lib/python3.9/site-packages/jax/experimental/jax2tf/examples/keras_reuse_main_test.py
lib/python3.9/site-packages/jax/experimental/jax2tf/examples/mnist_lib.py
lib/python3.9/site-packages/jax/experimental/jax2tf/examples/saved_model_lib.py
lib/python3.9/site-packages/jax/experimental/jax2tf/examples/saved_model_main.py
lib/python3.9/site-packages/jax/experimental/jax2tf/examples/saved_model_main_test.py
lib/python3.9/site-packages/jax/experimental/jax2tf/examples/serving/__init__.py
lib/python3.9/site-packages/jax/experimental/jax2tf/examples/serving/__pycache__/__init__.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/jax2tf/examples/serving/__pycache__/model_server_request.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/jax2tf/examples/serving/model_server_request.py
lib/python3.9/site-packages/jax/experimental/jax2tf/impl_no_xla.py
lib/python3.9/site-packages/jax/experimental/jax2tf/jax2tf.py
lib/python3.9/site-packages/jax/experimental/jax2tf/tests/__init__.py
lib/python3.9/site-packages/jax/experimental/jax2tf/tests/__pycache__/__init__.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/jax2tf/tests/__pycache__/back_compat_test.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/jax2tf/tests/__pycache__/back_compat_tf_test.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/jax2tf/tests/__pycache__/call_tf_test.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/jax2tf/tests/__pycache__/control_flow_ops_test.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/jax2tf/tests/__pycache__/converters.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/jax2tf/tests/__pycache__/cross_compilation_check.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/jax2tf/tests/__pycache__/jax2tf_limitations.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/jax2tf/tests/__pycache__/jax2tf_test.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/jax2tf/tests/__pycache__/jax_primitives_coverage_test.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/jax2tf/tests/__pycache__/model_harness.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/jax2tf/tests/__pycache__/models_test_main.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/jax2tf/tests/__pycache__/primitives_test.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/jax2tf/tests/__pycache__/savedmodel_test.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/jax2tf/tests/__pycache__/shape_poly_test.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/jax2tf/tests/__pycache__/sharding_test.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/jax2tf/tests/__pycache__/tf_test_util.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/jax2tf/tests/back_compat_test.py
lib/python3.9/site-packages/jax/experimental/jax2tf/tests/back_compat_tf_test.py
lib/python3.9/site-packages/jax/experimental/jax2tf/tests/call_tf_test.py
lib/python3.9/site-packages/jax/experimental/jax2tf/tests/control_flow_ops_test.py
lib/python3.9/site-packages/jax/experimental/jax2tf/tests/converters.py
lib/python3.9/site-packages/jax/experimental/jax2tf/tests/cross_compilation_check.py
lib/python3.9/site-packages/jax/experimental/jax2tf/tests/jax2tf_limitations.py
lib/python3.9/site-packages/jax/experimental/jax2tf/tests/jax2tf_test.py
lib/python3.9/site-packages/jax/experimental/jax2tf/tests/jax_primitives_coverage_test.py
lib/python3.9/site-packages/jax/experimental/jax2tf/tests/model_harness.py
lib/python3.9/site-packages/jax/experimental/jax2tf/tests/models_test_main.py
lib/python3.9/site-packages/jax/experimental/jax2tf/tests/primitives_test.py
lib/python3.9/site-packages/jax/experimental/jax2tf/tests/savedmodel_test.py
lib/python3.9/site-packages/jax/experimental/jax2tf/tests/shape_poly_test.py
lib/python3.9/site-packages/jax/experimental/jax2tf/tests/sharding_test.py
lib/python3.9/site-packages/jax/experimental/jax2tf/tests/tf_test_util.py
lib/python3.9/site-packages/jax/experimental/jet.py
lib/python3.9/site-packages/jax/experimental/key_reuse/__init__.py
lib/python3.9/site-packages/jax/experimental/key_reuse/__pycache__/__init__.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/key_reuse/__pycache__/_common.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/key_reuse/__pycache__/_core.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/key_reuse/__pycache__/_forwarding.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/key_reuse/__pycache__/_simple.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/key_reuse/_common.py
lib/python3.9/site-packages/jax/experimental/key_reuse/_core.py
lib/python3.9/site-packages/jax/experimental/key_reuse/_forwarding.py
lib/python3.9/site-packages/jax/experimental/key_reuse/_simple.py
lib/python3.9/site-packages/jax/experimental/layout.py
lib/python3.9/site-packages/jax/experimental/maps.py
lib/python3.9/site-packages/jax/experimental/mesh_utils.py
lib/python3.9/site-packages/jax/experimental/mosaic/__init__.py
lib/python3.9/site-packages/jax/experimental/mosaic/__pycache__/__init__.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/mosaic/__pycache__/dialects.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/mosaic/dialects.py
lib/python3.9/site-packages/jax/experimental/multihost_utils.py
lib/python3.9/site-packages/jax/experimental/ode.py
lib/python3.9/site-packages/jax/experimental/pallas/__init__.py
lib/python3.9/site-packages/jax/experimental/pallas/__pycache__/__init__.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/pallas/__pycache__/gpu.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/pallas/__pycache__/tpu.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/pallas/gpu.py
lib/python3.9/site-packages/jax/experimental/pallas/ops/__init__.py
lib/python3.9/site-packages/jax/experimental/pallas/ops/__pycache__/__init__.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/pallas/ops/__pycache__/attention.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/pallas/ops/__pycache__/layer_norm.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/pallas/ops/__pycache__/rms_norm.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/pallas/ops/__pycache__/softmax.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/pallas/ops/attention.py
lib/python3.9/site-packages/jax/experimental/pallas/ops/layer_norm.py
lib/python3.9/site-packages/jax/experimental/pallas/ops/rms_norm.py
lib/python3.9/site-packages/jax/experimental/pallas/ops/softmax.py
lib/python3.9/site-packages/jax/experimental/pallas/ops/tpu/__init__.py
lib/python3.9/site-packages/jax/experimental/pallas/ops/tpu/__pycache__/__init__.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/pallas/ops/tpu/__pycache__/all_gather.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/pallas/ops/tpu/__pycache__/flash_attention.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/pallas/ops/tpu/all_gather.py
lib/python3.9/site-packages/jax/experimental/pallas/ops/tpu/flash_attention.py
lib/python3.9/site-packages/jax/experimental/pallas/tpu.py
lib/python3.9/site-packages/jax/experimental/pjit.py
lib/python3.9/site-packages/jax/experimental/profiler.py
lib/python3.9/site-packages/jax/experimental/rnn.py
lib/python3.9/site-packages/jax/experimental/serialize_executable.py
lib/python3.9/site-packages/jax/experimental/shard_map.py
lib/python3.9/site-packages/jax/experimental/sparse/__init__.py
lib/python3.9/site-packages/jax/experimental/sparse/__pycache__/__init__.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/sparse/__pycache__/_base.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/sparse/__pycache__/_lowerings.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/sparse/__pycache__/ad.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/sparse/__pycache__/api.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/sparse/__pycache__/bcoo.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/sparse/__pycache__/bcsr.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/sparse/__pycache__/coo.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/sparse/__pycache__/csr.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/sparse/__pycache__/linalg.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/sparse/__pycache__/random.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/sparse/__pycache__/test_util.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/sparse/__pycache__/transform.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/sparse/__pycache__/util.cpython-39.pyc
lib/python3.9/site-packages/jax/experimental/sparse/_base.py
lib/python3.9/site-packages/jax/experimental/sparse/_lowerings.py
lib/python3.9/site-packages/jax/experimental/sparse/ad.py
lib/python3.9/site-packages/jax/experimental/sparse/api.py
lib/python3.9/site-packages/jax/experimental/sparse/bcoo.py
lib/python3.9/site-packages/jax/experimental/sparse/bcsr.py
lib/python3.9/site-packages/jax/experimental/sparse/coo.py
lib/python3.9/site-packages/jax/experimental/sparse/csr.py
lib/python3.9/site-packages/jax/experimental/sparse/linalg.py
lib/python3.9/site-packages/jax/experimental/sparse/random.py
lib/python3.9/site-packages/jax/experimental/sparse/test_util.py
lib/python3.9/site-packages/jax/experimental/sparse/transform.py
lib/python3.9/site-packages/jax/experimental/sparse/util.py
lib/python3.9/site-packages/jax/experimental/topologies.py
lib/python3.9/site-packages/jax/experimental/x64_context.py
lib/python3.9/site-packages/jax/extend/__init__.py
lib/python3.9/site-packages/jax/extend/__pycache__/__init__.cpython-39.pyc
lib/python3.9/site-packages/jax/extend/__pycache__/core.cpython-39.pyc
lib/python3.9/site-packages/jax/extend/__pycache__/linear_util.cpython-39.pyc
lib/python3.9/site-packages/jax/extend/__pycache__/random.cpython-39.pyc
lib/python3.9/site-packages/jax/extend/__pycache__/source_info_util.cpython-39.pyc
lib/python3.9/site-packages/jax/extend/core.py
lib/python3.9/site-packages/jax/extend/linear_util.py
lib/python3.9/site-packages/jax/extend/mlir/__init__.py
lib/python3.9/site-packages/jax/extend/mlir/__pycache__/__init__.cpython-39.pyc
lib/python3.9/site-packages/jax/extend/mlir/__pycache__/ir.cpython-39.pyc
lib/python3.9/site-packages/jax/extend/mlir/__pycache__/passmanager.cpython-39.pyc
lib/python3.9/site-packages/jax/extend/mlir/dialects/__init__.py
lib/python3.9/site-packages/jax/extend/mlir/dialects/__pycache__/__init__.cpython-39.pyc
lib/python3.9/site-packages/jax/extend/mlir/dialects/__pycache__/arith.cpython-39.pyc
lib/python3.9/site-packages/jax/extend/mlir/dialects/__pycache__/builtin.cpython-39.pyc
lib/python3.9/site-packages/jax/extend/mlir/dialects/__pycache__/chlo.cpython-39.pyc
lib/python3.9/site-packages/jax/extend/mlir/dialects/__pycache__/func.cpython-39.pyc
lib/python3.9/site-packages/jax/extend/mlir/dialects/__pycache__/math.cpython-39.pyc
lib/python3.9/site-packages/jax/extend/mlir/dialects/__pycache__/memref.cpython-39.pyc
lib/python3.9/site-packages/jax/extend/mlir/dialects/__pycache__/mhlo.cpython-39.pyc
lib/python3.9/site-packages/jax/extend/mlir/dialects/__pycache__/scf.cpython-39.pyc
lib/python3.9/site-packages/jax/extend/mlir/dialects/__pycache__/sparse_tensor.cpython-39.pyc
lib/python3.9/site-packages/jax/extend/mlir/dialects/__pycache__/stablehlo.cpython-39.pyc
lib/python3.9/site-packages/jax/extend/mlir/dialects/__pycache__/vector.cpython-39.pyc
lib/python3.9/site-packages/jax/extend/mlir/dialects/arith.py
lib/python3.9/site-packages/jax/extend/mlir/dialects/builtin.py
lib/python3.9/site-packages/jax/extend/mlir/dialects/chlo.py
lib/python3.9/site-packages/jax/extend/mlir/dialects/func.py
lib/python3.9/site-packages/jax/extend/mlir/dialects/math.py
lib/python3.9/site-packages/jax/extend/mlir/dialects/memref.py
lib/python3.9/site-packages/jax/extend/mlir/dialects/mhlo.py
lib/python3.9/site-packages/jax/extend/mlir/dialects/scf.py
lib/python3.9/site-packages/jax/extend/mlir/dialects/sparse_tensor.py
lib/python3.9/site-packages/jax/extend/mlir/dialects/stablehlo.py
lib/python3.9/site-packages/jax/extend/mlir/dialects/vector.py
lib/python3.9/site-packages/jax/extend/mlir/ir.py
lib/python3.9/site-packages/jax/extend/mlir/passmanager.py
lib/python3.9/site-packages/jax/extend/random.py
lib/python3.9/site-packages/jax/extend/source_info_util.py
lib/python3.9/site-packages/jax/flatten_util.py
lib/python3.9/site-packages/jax/image/__init__.py
lib/python3.9/site-packages/jax/image/__pycache__/__init__.cpython-39.pyc
lib/python3.9/site-packages/jax/interpreters/__init__.py
lib/python3.9/site-packages/jax/interpreters/__pycache__/__init__.cpython-39.pyc
lib/python3.9/site-packages/jax/interpreters/__pycache__/ad.cpython-39.pyc
lib/python3.9/site-packages/jax/interpreters/__pycache__/batching.cpython-39.pyc
lib/python3.9/site-packages/jax/interpreters/__pycache__/mlir.cpython-39.pyc
lib/python3.9/site-packages/jax/interpreters/__pycache__/partial_eval.cpython-39.pyc
lib/python3.9/site-packages/jax/interpreters/__pycache__/pxla.cpython-39.pyc
lib/python3.9/site-packages/jax/interpreters/__pycache__/xla.cpython-39.pyc
lib/python3.9/site-packages/jax/interpreters/ad.py
lib/python3.9/site-packages/jax/interpreters/batching.py
lib/python3.9/site-packages/jax/interpreters/mlir.py
lib/python3.9/site-packages/jax/interpreters/partial_eval.py
lib/python3.9/site-packages/jax/interpreters/pxla.py
lib/python3.9/site-packages/jax/interpreters/xla.py
lib/python3.9/site-packages/jax/lax/__init__.py
lib/python3.9/site-packages/jax/lax/__pycache__/__init__.cpython-39.pyc
lib/python3.9/site-packages/jax/lax/__pycache__/linalg.cpython-39.pyc
lib/python3.9/site-packages/jax/lax/linalg.py
lib/python3.9/site-packages/jax/lib/__init__.py
lib/python3.9/site-packages/jax/lib/__pycache__/__init__.cpython-39.pyc
lib/python3.9/site-packages/jax/lib/__pycache__/xla_bridge.cpython-39.pyc
lib/python3.9/site-packages/jax/lib/xla_bridge.py
lib/python3.9/site-packages/jax/linear_util.py
lib/python3.9/site-packages/jax/monitoring.py
lib/python3.9/site-packages/jax/nn/__init__.py
lib/python3.9/site-packages/jax/nn/__pycache__/__init__.cpython-39.pyc
lib/python3.9/site-packages/jax/nn/__pycache__/initializers.cpython-39.pyc
lib/python3.9/site-packages/jax/nn/initializers.py
lib/python3.9/site-packages/jax/numpy/__init__.py
lib/python3.9/site-packages/jax/numpy/__init__.pyi
lib/python3.9/site-packages/jax/numpy/__pycache__/__init__.cpython-39.pyc
lib/python3.9/site-packages/jax/numpy/__pycache__/fft.cpython-39.pyc
lib/python3.9/site-packages/jax/numpy/__pycache__/linalg.cpython-39.pyc
lib/python3.9/site-packages/jax/numpy/fft.py
lib/python3.9/site-packages/jax/numpy/linalg.py
lib/python3.9/site-packages/jax/ops/__init__.py
lib/python3.9/site-packages/jax/ops/__pycache__/__init__.cpython-39.pyc
lib/python3.9/site-packages/jax/prng.py
lib/python3.9/site-packages/jax/profiler.py
lib/python3.9/site-packages/jax/py.typed
lib/python3.9/site-packages/jax/random.py
lib/python3.9/site-packages/jax/scipy/__init__.py
lib/python3.9/site-packages/jax/scipy/__pycache__/__init__.cpython-39.pyc
lib/python3.9/site-packages/jax/scipy/__pycache__/fft.cpython-39.pyc
lib/python3.9/site-packages/jax/scipy/__pycache__/integrate.cpython-39.pyc
lib/python3.9/site-packages/jax/scipy/__pycache__/linalg.cpython-39.pyc
lib/python3.9/site-packages/jax/scipy/__pycache__/ndimage.cpython-39.pyc
lib/python3.9/site-packages/jax/scipy/__pycache__/signal.cpython-39.pyc
lib/python3.9/site-packages/jax/scipy/__pycache__/special.cpython-39.pyc
lib/python3.9/site-packages/jax/scipy/cluster/__init__.py
lib/python3.9/site-packages/jax/scipy/cluster/__pycache__/__init__.cpython-39.pyc
lib/python3.9/site-packages/jax/scipy/cluster/__pycache__/vq.cpython-39.pyc
lib/python3.9/site-packages/jax/scipy/cluster/vq.py
lib/python3.9/site-packages/jax/scipy/fft.py
lib/python3.9/site-packages/jax/scipy/integrate.py
lib/python3.9/site-packages/jax/scipy/interpolate/__init__.py
lib/python3.9/site-packages/jax/scipy/interpolate/__pycache__/__init__.cpython-39.pyc
lib/python3.9/site-packages/jax/scipy/linalg.py
lib/python3.9/site-packages/jax/scipy/ndimage.py
lib/python3.9/site-packages/jax/scipy/optimize/__init__.py
lib/python3.9/site-packages/jax/scipy/optimize/__pycache__/__init__.cpython-39.pyc
lib/python3.9/site-packages/jax/scipy/signal.py
lib/python3.9/site-packages/jax/scipy/sparse/__init__.py
lib/python3.9/site-packages/jax/scipy/sparse/__pycache__/__init__.cpython-39.pyc
lib/python3.9/site-packages/jax/scipy/sparse/__pycache__/linalg.cpython-39.pyc
lib/python3.9/site-packages/jax/scipy/sparse/linalg.py
lib/python3.9/site-packages/jax/scipy/spatial/__init__.py
lib/python3.9/site-packages/jax/scipy/spatial/__pycache__/__init__.cpython-39.pyc
lib/python3.9/site-packages/jax/scipy/spatial/__pycache__/transform.cpython-39.pyc
lib/python3.9/site-packages/jax/scipy/spatial/transform.py
lib/python3.9/site-packages/jax/scipy/special.py
lib/python3.9/site-packages/jax/scipy/stats/__init__.py
lib/python3.9/site-packages/jax/scipy/stats/__pycache__/__init__.cpython-39.pyc
lib/python3.9/site-packages/jax/scipy/stats/__pycache__/bernoulli.cpython-39.pyc
lib/python3.9/site-packages/jax/scipy/stats/__pycache__/beta.cpython-39.pyc
lib/python3.9/site-packages/jax/scipy/stats/__pycache__/betabinom.cpython-39.pyc
lib/python3.9/site-packages/jax/scipy/stats/__pycache__/binom.cpython-39.pyc
lib/python3.9/site-packages/jax/scipy/stats/__pycache__/cauchy.cpython-39.pyc
lib/python3.9/site-packages/jax/scipy/stats/__pycache__/chi2.cpython-39.pyc
lib/python3.9/site-packages/jax/scipy/stats/__pycache__/dirichlet.cpython-39.pyc
lib/python3.9/site-packages/jax/scipy/stats/__pycache__/expon.cpython-39.pyc
lib/python3.9/site-packages/jax/scipy/stats/__pycache__/gamma.cpython-39.pyc
lib/python3.9/site-packages/jax/scipy/stats/__pycache__/gennorm.cpython-39.pyc
lib/python3.9/site-packages/jax/scipy/stats/__pycache__/geom.cpython-39.pyc
lib/python3.9/site-packages/jax/scipy/stats/__pycache__/laplace.cpython-39.pyc
lib/python3.9/site-packages/jax/scipy/stats/__pycache__/logistic.cpython-39.pyc
lib/python3.9/site-packages/jax/scipy/stats/__pycache__/multinomial.cpython-39.pyc
lib/python3.9/site-packages/jax/scipy/stats/__pycache__/multivariate_normal.cpython-39.pyc
lib/python3.9/site-packages/jax/scipy/stats/__pycache__/nbinom.cpython-39.pyc
lib/python3.9/site-packages/jax/scipy/stats/__pycache__/norm.cpython-39.pyc
lib/python3.9/site-packages/jax/scipy/stats/__pycache__/pareto.cpython-39.pyc
lib/python3.9/site-packages/jax/scipy/stats/__pycache__/poisson.cpython-39.pyc
lib/python3.9/site-packages/jax/scipy/stats/__pycache__/t.cpython-39.pyc
lib/python3.9/site-packages/jax/scipy/stats/__pycache__/truncnorm.cpython-39.pyc
lib/python3.9/site-packages/jax/scipy/stats/__pycache__/uniform.cpython-39.pyc
lib/python3.9/site-packages/jax/scipy/stats/__pycache__/vonmises.cpython-39.pyc
lib/python3.9/site-packages/jax/scipy/stats/__pycache__/wrapcauchy.cpython-39.pyc
lib/python3.9/site-packages/jax/scipy/stats/bernoulli.py
lib/python3.9/site-packages/jax/scipy/stats/beta.py
lib/python3.9/site-packages/jax/scipy/stats/betabinom.py
lib/python3.9/site-packages/jax/scipy/stats/binom.py
lib/python3.9/site-packages/jax/scipy/stats/cauchy.py
lib/python3.9/site-packages/jax/scipy/stats/chi2.py
lib/python3.9/site-packages/jax/scipy/stats/dirichlet.py
lib/python3.9/site-packages/jax/scipy/stats/expon.py
lib/python3.9/site-packages/jax/scipy/stats/gamma.py
lib/python3.9/site-packages/jax/scipy/stats/gennorm.py
lib/python3.9/site-packages/jax/scipy/stats/geom.py
lib/python3.9/site-packages/jax/scipy/stats/laplace.py
lib/python3.9/site-packages/jax/scipy/stats/logistic.py
lib/python3.9/site-packages/jax/scipy/stats/multinomial.py
lib/python3.9/site-packages/jax/scipy/stats/multivariate_normal.py
lib/python3.9/site-packages/jax/scipy/stats/nbinom.py
lib/python3.9/site-packages/jax/scipy/stats/norm.py
lib/python3.9/site-packages/jax/scipy/stats/pareto.py
lib/python3.9/site-packages/jax/scipy/stats/poisson.py
lib/python3.9/site-packages/jax/scipy/stats/t.py
lib/python3.9/site-packages/jax/scipy/stats/truncnorm.py
lib/python3.9/site-packages/jax/scipy/stats/uniform.py
lib/python3.9/site-packages/jax/scipy/stats/vonmises.py
lib/python3.9/site-packages/jax/scipy/stats/wrapcauchy.py
lib/python3.9/site-packages/jax/sharding.py
lib/python3.9/site-packages/jax/stages.py
lib/python3.9/site-packages/jax/test_util.py
lib/python3.9/site-packages/jax/tools/__init__.py
lib/python3.9/site-packages/jax/tools/__pycache__/__init__.cpython-39.pyc
lib/python3.9/site-packages/jax/tools/__pycache__/build_utils.cpython-39.pyc
lib/python3.9/site-packages/jax/tools/__pycache__/colab_tpu.cpython-39.pyc
lib/python3.9/site-packages/jax/tools/__pycache__/jax_to_ir.cpython-39.pyc
lib/python3.9/site-packages/jax/tools/build_utils.py
lib/python3.9/site-packages/jax/tools/colab_tpu.py
lib/python3.9/site-packages/jax/tools/jax_to_ir.py
lib/python3.9/site-packages/jax/tree_util.py
lib/python3.9/site-packages/jax/typing.py
lib/python3.9/site-packages/jax/util.py
lib/python3.9/site-packages/jax/version.py
