https://filehashintel.com/name/jax/__init__.py https://filehashintel.com/name/jax/_src/__init__.py https://filehashintel.com/name/jax/_src/abstract_arrays.py https://filehashintel.com/name/jax/_src/ad_checkpoint.py https://filehashintel.com/name/jax/_src/ad_util.py https://filehashintel.com/name/jax/_src/api.py https://filehashintel.com/name/jax/_src/api_util.py https://filehashintel.com/name/jax/_src/array.py https://filehashintel.com/name/jax/_src/attrs.py https://filehashintel.com/name/jax/_src/basearray.py https://filehashintel.com/name/jax/_src/blocked_sampler.py https://filehashintel.com/name/jax/_src/buffer_callback.py https://filehashintel.com/name/jax/_src/cache_key.py https://filehashintel.com/name/jax/_src/callback.py https://filehashintel.com/name/jax/_src/checkify.py https://filehashintel.com/name/jax/_src/cloud_tpu_init.py https://filehashintel.com/name/jax/_src/clusters/__init__.py https://filehashintel.com/name/jax/_src/clusters/cloud_tpu_cluster.py https://filehashintel.com/name/jax/_src/clusters/cluster.py https://filehashintel.com/name/jax/_src/clusters/k8s_cluster.py https://filehashintel.com/name/jax/_src/clusters/mpi4py_cluster.py https://filehashintel.com/name/jax/_src/clusters/ompi_cluster.py https://filehashintel.com/name/jax/_src/clusters/slurm_cluster.py https://filehashintel.com/name/jax/_src/compilation_cache.py https://filehashintel.com/name/jax/_src/compilation_cache_interface.py https://filehashintel.com/name/jax/_src/compiler.py https://filehashintel.com/name/jax/_src/compute_on.py https://filehashintel.com/name/jax/_src/config.py https://filehashintel.com/name/jax/_src/core.py https://filehashintel.com/name/jax/_src/cudnn/__init__.py https://filehashintel.com/name/jax/_src/cudnn/fused_attention_stablehlo.py https://filehashintel.com/name/jax/_src/cudnn/fusion.py https://filehashintel.com/name/jax/_src/cudnn/scaled_matmul_stablehlo.py https://filehashintel.com/name/jax/_src/custom_api_util.py https://filehashintel.com/name/jax/_src/custom_batching.py https://filehashintel.com/name/jax/_src/custom_dce.py https://filehashintel.com/name/jax/_src/custom_derivatives.py https://filehashintel.com/name/jax/_src/custom_partitioning.py https://filehashintel.com/name/jax/_src/custom_partitioning_sharding_rule.py https://filehashintel.com/name/jax/_src/custom_transpose.py https://filehashintel.com/name/jax/_src/debugger/__init__.py https://filehashintel.com/name/jax/_src/debugger/cli_debugger.py https://filehashintel.com/name/jax/_src/debugger/colab_debugger.py https://filehashintel.com/name/jax/_src/debugger/colab_lib.py https://filehashintel.com/name/jax/_src/debugger/core.py https://filehashintel.com/name/jax/_src/debugger/web_debugger.py https://filehashintel.com/name/jax/_src/debugging.py https://filehashintel.com/name/jax/_src/deprecations.py https://filehashintel.com/name/jax/_src/dispatch.py https://filehashintel.com/name/jax/_src/distributed.py https://filehashintel.com/name/jax/_src/dlpack.py https://filehashintel.com/name/jax/_src/dtypes.py https://filehashintel.com/name/jax/_src/earray.py https://filehashintel.com/name/jax/_src/effects.py https://filehashintel.com/name/jax/_src/environment_info.py https://filehashintel.com/name/jax/_src/error_check.py https://filehashintel.com/name/jax/_src/errors.py https://filehashintel.com/name/jax/_src/export/__init__.py https://filehashintel.com/name/jax/_src/export/_export.py https://filehashintel.com/name/jax/_src/export/serialization.py https://filehashintel.com/name/jax/_src/export/serialization_generated.py https://filehashintel.com/name/jax/_src/export/shape_poly.py https://filehashintel.com/name/jax/_src/export/shape_poly_decision.py https://filehashintel.com/name/jax/_src/extend/__init__.py https://filehashintel.com/name/jax/_src/extend/ffi.py https://filehashintel.com/name/jax/_src/extend/random.py https://filehashintel.com/name/jax/_src/ffi.py https://filehashintel.com/name/jax/_src/flatten_util.py https://filehashintel.com/name/jax/_src/flattree.py https://filehashintel.com/name/jax/_src/frozen_dict.py https://filehashintel.com/name/jax/_src/gfile_cache.py https://filehashintel.com/name/jax/_src/hardware_utils.py https://filehashintel.com/name/jax/_src/hashable_array.py https://filehashintel.com/name/jax/_src/hijax.py https://filehashintel.com/name/jax/_src/image/__init__.py https://filehashintel.com/name/jax/_src/image/scale.py https://filehashintel.com/name/jax/_src/indexing.py https://filehashintel.com/name/jax/_src/internal_test_util/__init__.py https://filehashintel.com/name/jax/_src/internal_test_util/deprecation_module.py https://filehashintel.com/name/jax/_src/internal_test_util/export_back_compat_test_data/__init__.py https://filehashintel.com/name/jax/_src/internal_test_util/export_back_compat_test_data/annotate_data_placement.py https://filehashintel.com/name/jax/_src/internal_test_util/export_back_compat_test_data/cpu_cholesky_lapack_potrf.py https://filehashintel.com/name/jax/_src/internal_test_util/export_back_compat_test_data/cpu_eig_lapack_geev.py https://filehashintel.com/name/jax/_src/internal_test_util/export_back_compat_test_data/cpu_eigh_lapack_syev.py https://filehashintel.com/name/jax/_src/internal_test_util/export_back_compat_test_data/cpu_hessenberg_lapack_gehrd.py https://filehashintel.com/name/jax/_src/internal_test_util/export_back_compat_test_data/cpu_lu_lapack_getrf.py https://filehashintel.com/name/jax/_src/internal_test_util/export_back_compat_test_data/cpu_qr_lapack_geqrf.py https://filehashintel.com/name/jax/_src/internal_test_util/export_back_compat_test_data/cpu_schur_lapack_gees.py https://filehashintel.com/name/jax/_src/internal_test_util/export_back_compat_test_data/cpu_svd_lapack_gesdd.py https://filehashintel.com/name/jax/_src/internal_test_util/export_back_compat_test_data/cpu_triangular_solve_blas_trsm.py https://filehashintel.com/name/jax/_src/internal_test_util/export_back_compat_test_data/cpu_tridiagonal_lapack_sytrd_hetrd.py https://filehashintel.com/name/jax/_src/internal_test_util/export_back_compat_test_data/cpu_tridiagonal_solve_lapack_gtsv.py https://filehashintel.com/name/jax/_src/internal_test_util/export_back_compat_test_data/cuda_cholesky_solver_potrf.py https://filehashintel.com/name/jax/_src/internal_test_util/export_back_compat_test_data/cuda_eigh_cusolver_syev.py https://filehashintel.com/name/jax/_src/internal_test_util/export_back_compat_test_data/cuda_lu_cusolver_getrf.py https://filehashintel.com/name/jax/_src/internal_test_util/export_back_compat_test_data/cuda_lu_pivots_to_permutation.py https://filehashintel.com/name/jax/_src/internal_test_util/export_back_compat_test_data/cuda_qr_cusolver_geqrf.py https://filehashintel.com/name/jax/_src/internal_test_util/export_back_compat_test_data/cuda_svd_cusolver_gesvd.py https://filehashintel.com/name/jax/_src/internal_test_util/export_back_compat_test_data/cuda_threefry2x32.py https://filehashintel.com/name/jax/_src/internal_test_util/export_back_compat_test_data/cuda_tridiagonal_cusolver_sytrd.py https://filehashintel.com/name/jax/_src/internal_test_util/export_back_compat_test_data/cuda_tridiagonal_solve.py https://filehashintel.com/name/jax/_src/internal_test_util/export_back_compat_test_data/export_with_memory_space.py https://filehashintel.com/name/jax/_src/internal_test_util/export_back_compat_test_data/export_with_multiple_meshes.py https://filehashintel.com/name/jax/_src/internal_test_util/export_back_compat_test_data/export_with_specified_sharding.py https://filehashintel.com/name/jax/_src/internal_test_util/export_back_compat_test_data/export_with_unspecified_sharding.py https://filehashintel.com/name/jax/_src/internal_test_util/export_back_compat_test_data/gpu_eigh_solver_syev.py https://filehashintel.com/name/jax/_src/internal_test_util/export_back_compat_test_data/pallas/__init__.py https://filehashintel.com/name/jax/_src/internal_test_util/export_back_compat_test_data/pallas/mosaic_boolean_constant.py https://filehashintel.com/name/jax/_src/internal_test_util/export_back_compat_test_data/pallas/mosaic_gpu_add_one.py https://filehashintel.com/name/jax/_src/internal_test_util/export_back_compat_test_data/pallas/mosaic_matmul.py https://filehashintel.com/name/jax/_src/internal_test_util/export_back_compat_test_data/pallas/mosaic_semaphore_dma.py https://filehashintel.com/name/jax/_src/internal_test_util/export_back_compat_test_data/pallas/triton_add_one.py https://filehashintel.com/name/jax/_src/internal_test_util/export_back_compat_test_data/rocm_cholesky_solver_potrf.py https://filehashintel.com/name/jax/_src/internal_test_util/export_back_compat_test_data/rocm_eigh_hipsolver_syev.py https://filehashintel.com/name/jax/_src/internal_test_util/export_back_compat_test_data/rocm_lu_pivots_to_permutation.py https://filehashintel.com/name/jax/_src/internal_test_util/export_back_compat_test_data/rocm_lu_rocsolver_getrf.py https://filehashintel.com/name/jax/_src/internal_test_util/export_back_compat_test_data/rocm_qr_hipsolver_geqrf.py https://filehashintel.com/name/jax/_src/internal_test_util/export_back_compat_test_data/rocm_svd_hipsolver_gesvd.py https://filehashintel.com/name/jax/_src/internal_test_util/export_back_compat_test_data/rocm_threefry2x32.py https://filehashintel.com/name/jax/_src/internal_test_util/export_back_compat_test_data/rocm_tridiagonal_hipsolver_sytrd.py https://filehashintel.com/name/jax/_src/internal_test_util/export_back_compat_test_data/shardy_sharding_ops_with_different_meshes.py https://filehashintel.com/name/jax/_src/internal_test_util/export_back_compat_test_data/stablehlo_dynamic_approx_top_k.py https://filehashintel.com/name/jax/_src/internal_test_util/export_back_compat_test_data/stablehlo_dynamic_rng_bit_generator.py https://filehashintel.com/name/jax/_src/internal_test_util/export_back_compat_test_data/stablehlo_dynamic_top_k.py https://filehashintel.com/name/jax/_src/internal_test_util/export_back_compat_test_data/tpu_ApproxTopK.py https://filehashintel.com/name/jax/_src/internal_test_util/export_back_compat_test_data/tpu_Eigh.py https://filehashintel.com/name/jax/_src/internal_test_util/export_back_compat_test_data/tpu_Lu.py https://filehashintel.com/name/jax/_src/internal_test_util/export_back_compat_test_data/tpu_Qr.py https://filehashintel.com/name/jax/_src/internal_test_util/export_back_compat_test_data/tpu_Sharding.py https://filehashintel.com/name/jax/_src/internal_test_util/export_back_compat_test_data/tpu_stablehlo_dynamic_reduce_window.py https://filehashintel.com/name/jax/_src/internal_test_util/export_back_compat_test_util.py https://filehashintel.com/name/jax/_src/internal_test_util/lax_test_util.py https://filehashintel.com/name/jax/_src/internal_test_util/lazy_loader_module/__init__.py https://filehashintel.com/name/jax/_src/internal_test_util/lazy_loader_module/lazy_test_submodule.py https://filehashintel.com/name/jax/_src/internal_test_util/test_harnesses.py https://filehashintel.com/name/jax/_src/interpreters/__init__.py https://filehashintel.com/name/jax/_src/interpreters/ad.py https://filehashintel.com/name/jax/_src/interpreters/batching.py https://filehashintel.com/name/jax/_src/interpreters/mlir.py https://filehashintel.com/name/jax/_src/interpreters/partial_eval.py https://filehashintel.com/name/jax/_src/interpreters/pxla.py https://filehashintel.com/name/jax/_src/interpreters/remat.py https://filehashintel.com/name/jax/_src/interpreters/xla.py https://filehashintel.com/name/jax/_src/jaxpr_util.py https://filehashintel.com/name/jax/_src/lax/__init__.py https://filehashintel.com/name/jax/_src/lax/ann.py https://filehashintel.com/name/jax/_src/lax/control_flow/__init__.py https://filehashintel.com/name/jax/_src/lax/control_flow/common.py https://filehashintel.com/name/jax/_src/lax/control_flow/conditionals.py https://filehashintel.com/name/jax/_src/lax/control_flow/for_loop.py https://filehashintel.com/name/jax/_src/lax/control_flow/loops.py https://filehashintel.com/name/jax/_src/lax/control_flow/solves.py https://filehashintel.com/name/jax/_src/lax/convolution.py https://filehashintel.com/name/jax/_src/lax/eigh.py https://filehashintel.com/name/jax/_src/lax/eval_jaxpr.py https://filehashintel.com/name/jax/_src/lax/fft.py https://filehashintel.com/name/jax/_src/lax/lax.py https://filehashintel.com/name/jax/_src/lax/linalg.py https://filehashintel.com/name/jax/_src/lax/other.py https://filehashintel.com/name/jax/_src/lax/parallel.py https://filehashintel.com/name/jax/_src/lax/qdwh.py https://filehashintel.com/name/jax/_src/lax/scaled_dot.py https://filehashintel.com/name/jax/_src/lax/slicing.py https://filehashintel.com/name/jax/_src/lax/special.py https://filehashintel.com/name/jax/_src/lax/stack.py https://filehashintel.com/name/jax/_src/lax/svd.py https://filehashintel.com/name/jax/_src/lax/utils.py https://filehashintel.com/name/jax/_src/lax/windowed_reductions.py https://filehashintel.com/name/jax/_src/lax_reference.py https://filehashintel.com/name/jax/_src/layout.py https://filehashintel.com/name/jax/_src/lazy_loader.py https://filehashintel.com/name/jax/_src/lib/__init__.py https://filehashintel.com/name/jax/_src/lib/mlir/__init__.py https://filehashintel.com/name/jax/_src/lib/mlir/dialects/__init__.py https://filehashintel.com/name/jax/_src/lib/mosaic_gpu.py https://filehashintel.com/name/jax/_src/lib/triton.py https://filehashintel.com/name/jax/_src/linear_util.py https://filehashintel.com/name/jax/_src/literals.py https://filehashintel.com/name/jax/_src/logging_config.py https://filehashintel.com/name/jax/_src/lru_cache.py https://filehashintel.com/name/jax/_src/maps.py https://filehashintel.com/name/jax/_src/memory.py https://filehashintel.com/name/jax/_src/mesh.py https://filehashintel.com/name/jax/_src/mesh_utils.py https://filehashintel.com/name/jax/_src/monitoring.py https://filehashintel.com/name/jax/_src/named_sharding.py https://filehashintel.com/name/jax/_src/nn/__init__.py https://filehashintel.com/name/jax/_src/nn/functions.py https://filehashintel.com/name/jax/_src/nn/initializers.py https://filehashintel.com/name/jax/_src/numpy/__init__.py https://filehashintel.com/name/jax/_src/numpy/array.py https://filehashintel.com/name/jax/_src/numpy/array_api_metadata.py https://filehashintel.com/name/jax/_src/numpy/array_constructors.py https://filehashintel.com/name/jax/_src/numpy/array_creation.py https://filehashintel.com/name/jax/_src/numpy/array_methods.py https://filehashintel.com/name/jax/_src/numpy/einsum.py https://filehashintel.com/name/jax/_src/numpy/error.py https://filehashintel.com/name/jax/_src/numpy/fft.py https://filehashintel.com/name/jax/_src/numpy/hijax.py https://filehashintel.com/name/jax/_src/numpy/index_tricks.py https://filehashintel.com/name/jax/_src/numpy/indexing.py https://filehashintel.com/name/jax/_src/numpy/lax_numpy.py https://filehashintel.com/name/jax/_src/numpy/linalg.py https://filehashintel.com/name/jax/_src/numpy/polynomial.py https://filehashintel.com/name/jax/_src/numpy/reductions.py https://filehashintel.com/name/jax/_src/numpy/scalar_types.py https://filehashintel.com/name/jax/_src/numpy/setops.py https://filehashintel.com/name/jax/_src/numpy/sorting.py https://filehashintel.com/name/jax/_src/numpy/tensor_contractions.py https://filehashintel.com/name/jax/_src/numpy/ufunc_api.py https://filehashintel.com/name/jax/_src/numpy/ufuncs.py https://filehashintel.com/name/jax/_src/numpy/util.py https://filehashintel.com/name/jax/_src/numpy/vectorize.py https://filehashintel.com/name/jax/_src/numpy/window_functions.py https://filehashintel.com/name/jax/_src/op_shardings.py https://filehashintel.com/name/jax/_src/ops/__init__.py https://filehashintel.com/name/jax/_src/ops/scatter.py https://filehashintel.com/name/jax/_src/ops/special.py https://filehashintel.com/name/jax/_src/pallas/__init__.py https://filehashintel.com/name/jax/_src/pallas/core.py https://filehashintel.com/name/jax/_src/pallas/cost_estimate.py https://filehashintel.com/name/jax/_src/pallas/einshape.py https://filehashintel.com/name/jax/_src/pallas/fuser/__init__.py https://filehashintel.com/name/jax/_src/pallas/fuser/block_spec.py https://filehashintel.com/name/jax/_src/pallas/fuser/custom_evaluate.py https://filehashintel.com/name/jax/_src/pallas/fuser/custom_fusion_lib.py https://filehashintel.com/name/jax/_src/pallas/fuser/fusable.py https://filehashintel.com/name/jax/_src/pallas/fuser/fusable_dtype.py https://filehashintel.com/name/jax/_src/pallas/fuser/fuser_utils.py https://filehashintel.com/name/jax/_src/pallas/fuser/fusible.py https://filehashintel.com/name/jax/_src/pallas/fuser/fusible_dtype.py https://filehashintel.com/name/jax/_src/pallas/fuser/fusion.py https://filehashintel.com/name/jax/_src/pallas/fuser/jaxpr_fusion.py https://filehashintel.com/name/jax/_src/pallas/helpers.py https://filehashintel.com/name/jax/_src/pallas/hlo_interpreter.py https://filehashintel.com/name/jax/_src/pallas/indexing.py https://filehashintel.com/name/jax/_src/pallas/mosaic/__init__.py https://filehashintel.com/name/jax/_src/pallas/mosaic/core.py https://filehashintel.com/name/jax/_src/pallas/mosaic/error_handling.py https://filehashintel.com/name/jax/_src/pallas/mosaic/helpers.py https://filehashintel.com/name/jax/_src/pallas/mosaic/interpret.py https://filehashintel.com/name/jax/_src/pallas/mosaic/interpret/__init__.py https://filehashintel.com/name/jax/_src/pallas/mosaic/interpret/interpret_pallas_call.py https://filehashintel.com/name/jax/_src/pallas/mosaic/interpret/params.py https://filehashintel.com/name/jax/_src/pallas/mosaic/interpret/race_detection_state.py https://filehashintel.com/name/jax/_src/pallas/mosaic/interpret/shared_memory.py https://filehashintel.com/name/jax/_src/pallas/mosaic/interpret/thread_map.py https://filehashintel.com/name/jax/_src/pallas/mosaic/interpret/utils.py https://filehashintel.com/name/jax/_src/pallas/mosaic/interpret/vector_clock.py https://filehashintel.com/name/jax/_src/pallas/mosaic/kernel_regeneration_util.py https://filehashintel.com/name/jax/_src/pallas/mosaic/lowering.py https://filehashintel.com/name/jax/_src/pallas/mosaic/pallas_call_registration.py https://filehashintel.com/name/jax/_src/pallas/mosaic/pipeline.py https://filehashintel.com/name/jax/_src/pallas/mosaic/primitives.py https://filehashintel.com/name/jax/_src/pallas/mosaic/random.py https://filehashintel.com/name/jax/_src/pallas/mosaic/sc_core.py https://filehashintel.com/name/jax/_src/pallas/mosaic/sc_lowering.py https://filehashintel.com/name/jax/_src/pallas/mosaic/sc_primitives.py https://filehashintel.com/name/jax/_src/pallas/mosaic/tpu_info.py https://filehashintel.com/name/jax/_src/pallas/mosaic/verification.py https://filehashintel.com/name/jax/_src/pallas/mosaic_gpu/__init__.py https://filehashintel.com/name/jax/_src/pallas/mosaic_gpu/core.py https://filehashintel.com/name/jax/_src/pallas/mosaic_gpu/helpers.py https://filehashintel.com/name/jax/_src/pallas/mosaic_gpu/interpret/__init__.py https://filehashintel.com/name/jax/_src/pallas/mosaic_gpu/interpret/gpu_callbacks.py https://filehashintel.com/name/jax/_src/pallas/mosaic_gpu/interpret/interpret_pallas_call.py https://filehashintel.com/name/jax/_src/pallas/mosaic_gpu/interpret/jaxpr_interpret.py https://filehashintel.com/name/jax/_src/pallas/mosaic_gpu/interpret/params.py https://filehashintel.com/name/jax/_src/pallas/mosaic_gpu/interpret/shared_memory.py https://filehashintel.com/name/jax/_src/pallas/mosaic_gpu/lowering.py https://filehashintel.com/name/jax/_src/pallas/mosaic_gpu/pallas_call_registration.py https://filehashintel.com/name/jax/_src/pallas/mosaic_gpu/pipeline.py https://filehashintel.com/name/jax/_src/pallas/mosaic_gpu/primitives.py https://filehashintel.com/name/jax/_src/pallas/mosaic_gpu/torch.py https://filehashintel.com/name/jax/_src/pallas/mpmd.py https://filehashintel.com/name/jax/_src/pallas/pallas_call.py https://filehashintel.com/name/jax/_src/pallas/primitives.py https://filehashintel.com/name/jax/_src/pallas/triton/__init__.py https://filehashintel.com/name/jax/_src/pallas/triton/core.py https://filehashintel.com/name/jax/_src/pallas/triton/gpu_info.py https://filehashintel.com/name/jax/_src/pallas/triton/lowering.py https://filehashintel.com/name/jax/_src/pallas/triton/pallas_call_registration.py https://filehashintel.com/name/jax/_src/pallas/triton/primitives.py https://filehashintel.com/name/jax/_src/pallas/utils.py https://filehashintel.com/name/jax/_src/partition_spec.py https://filehashintel.com/name/jax/_src/path.py https://filehashintel.com/name/jax/_src/pickle_util.py https://filehashintel.com/name/jax/_src/pjit.py https://filehashintel.com/name/jax/_src/pmap.py https://filehashintel.com/name/jax/_src/pretty_printer.py https://filehashintel.com/name/jax/_src/prng.py https://filehashintel.com/name/jax/_src/profiler.py https://filehashintel.com/name/jax/_src/public_test_util.py https://filehashintel.com/name/jax/_src/random.py https://filehashintel.com/name/jax/_src/random/__init__.py https://filehashintel.com/name/jax/_src/random/core.py https://filehashintel.com/name/jax/_src/random/philox2x32.py https://filehashintel.com/name/jax/_src/random/philox4x32.py https://filehashintel.com/name/jax/_src/random/prng.py https://filehashintel.com/name/jax/_src/random/rbg.py https://filehashintel.com/name/jax/_src/random/stateful_rng.py https://filehashintel.com/name/jax/_src/random/threefry2x32.py https://filehashintel.com/name/jax/_src/random/threefry4x32.py https://filehashintel.com/name/jax/_src/ref.py https://filehashintel.com/name/jax/_src/scipy/__init__.py https://filehashintel.com/name/jax/_src/scipy/cluster/__init__.py https://filehashintel.com/name/jax/_src/scipy/cluster/vq.py https://filehashintel.com/name/jax/_src/scipy/fft.py https://filehashintel.com/name/jax/_src/scipy/integrate.py https://filehashintel.com/name/jax/_src/scipy/interpolate/__init__.py https://filehashintel.com/name/jax/_src/scipy/linalg.py https://filehashintel.com/name/jax/_src/scipy/ndimage.py https://filehashintel.com/name/jax/_src/scipy/optimize/__init__.py https://filehashintel.com/name/jax/_src/scipy/optimize/_lbfgs.py https://filehashintel.com/name/jax/_src/scipy/optimize/bfgs.py https://filehashintel.com/name/jax/_src/scipy/optimize/line_search.py https://filehashintel.com/name/jax/_src/scipy/optimize/minimize.py https://filehashintel.com/name/jax/_src/scipy/signal.py https://filehashintel.com/name/jax/_src/scipy/sparse/__init__.py https://filehashintel.com/name/jax/_src/scipy/sparse/linalg.py https://filehashintel.com/name/jax/_src/scipy/spatial/__init__.py https://filehashintel.com/name/jax/_src/scipy/spatial/transform.py https://filehashintel.com/name/jax/_src/scipy/special.py https://filehashintel.com/name/jax/_src/scipy/stats/__init__.py https://filehashintel.com/name/jax/_src/scipy/stats/_core.py https://filehashintel.com/name/jax/_src/scipy/stats/bernoulli.py https://filehashintel.com/name/jax/_src/scipy/stats/beta.py https://filehashintel.com/name/jax/_src/scipy/stats/betabinom.py https://filehashintel.com/name/jax/_src/scipy/stats/binom.py https://filehashintel.com/name/jax/_src/scipy/stats/cauchy.py https://filehashintel.com/name/jax/_src/scipy/stats/chi2.py https://filehashintel.com/name/jax/_src/scipy/stats/dirichlet.py https://filehashintel.com/name/jax/_src/scipy/stats/expon.py https://filehashintel.com/name/jax/_src/scipy/stats/gamma.py https://filehashintel.com/name/jax/_src/scipy/stats/gennorm.py https://filehashintel.com/name/jax/_src/scipy/stats/geom.py https://filehashintel.com/name/jax/_src/scipy/stats/gumbel_l.py https://filehashintel.com/name/jax/_src/scipy/stats/gumbel_r.py https://filehashintel.com/name/jax/_src/scipy/stats/kde.py https://filehashintel.com/name/jax/_src/scipy/stats/laplace.py https://filehashintel.com/name/jax/_src/scipy/stats/logistic.py https://filehashintel.com/name/jax/_src/scipy/stats/multinomial.py https://filehashintel.com/name/jax/_src/scipy/stats/multivariate_normal.py https://filehashintel.com/name/jax/_src/scipy/stats/nbinom.py https://filehashintel.com/name/jax/_src/scipy/stats/norm.py https://filehashintel.com/name/jax/_src/scipy/stats/pareto.py https://filehashintel.com/name/jax/_src/scipy/stats/poisson.py https://filehashintel.com/name/jax/_src/scipy/stats/t.py https://filehashintel.com/name/jax/_src/scipy/stats/truncnorm.py https://filehashintel.com/name/jax/_src/scipy/stats/uniform.py https://filehashintel.com/name/jax/_src/scipy/stats/vonmises.py https://filehashintel.com/name/jax/_src/scipy/stats/wrapcauchy.py https://filehashintel.com/name/jax/_src/shard_alike.py https://filehashintel.com/name/jax/_src/shard_map.py https://filehashintel.com/name/jax/_src/sharding.py https://filehashintel.com/name/jax/_src/sharding_impls.py https://filehashintel.com/name/jax/_src/sharding_specs.py https://filehashintel.com/name/jax/_src/source_info_util.py https://filehashintel.com/name/jax/_src/sourcemap.py https://filehashintel.com/name/jax/_src/stages.py https://filehashintel.com/name/jax/_src/state/__init__.py https://filehashintel.com/name/jax/_src/state/discharge.py https://filehashintel.com/name/jax/_src/state/indexing.py https://filehashintel.com/name/jax/_src/state/primitives.py https://filehashintel.com/name/jax/_src/state/types.py https://filehashintel.com/name/jax/_src/state/utils.py https://filehashintel.com/name/jax/_src/stateful_rng.py https://filehashintel.com/name/jax/_src/test_loader.py https://filehashintel.com/name/jax/_src/test_multiprocess.py https://filehashintel.com/name/jax/_src/test_util.py https://filehashintel.com/name/jax/_src/test_warning_util.py https://filehashintel.com/name/jax/_src/third_party/__init__.py https://filehashintel.com/name/jax/_src/third_party/numpy/__init__.py https://filehashintel.com/name/jax/_src/third_party/numpy/linalg.py https://filehashintel.com/name/jax/_src/third_party/scipy/__init__.py https://filehashintel.com/name/jax/_src/third_party/scipy/betaln.py https://filehashintel.com/name/jax/_src/third_party/scipy/interpolate.py https://filehashintel.com/name/jax/_src/third_party/scipy/linalg.py https://filehashintel.com/name/jax/_src/third_party/scipy/signal_helper.py https://filehashintel.com/name/jax/_src/third_party/scipy/special.py https://filehashintel.com/name/jax/_src/tpu/__init__.py https://filehashintel.com/name/jax/_src/tpu/linalg/__init__.py https://filehashintel.com/name/jax/_src/tpu/linalg/eigh.py https://filehashintel.com/name/jax/_src/tpu/linalg/qdwh.py https://filehashintel.com/name/jax/_src/tpu/linalg/stack.py https://filehashintel.com/name/jax/_src/tpu/linalg/svd.py https://filehashintel.com/name/jax/_src/tpu_custom_call.py https://filehashintel.com/name/jax/_src/traceback_util.py https://filehashintel.com/name/jax/_src/tree.py https://filehashintel.com/name/jax/_src/tree_util.py https://filehashintel.com/name/jax/_src/typing.py https://filehashintel.com/name/jax/_src/util.py https://filehashintel.com/name/jax/_src/xla_bridge.py https://filehashintel.com/name/jax/_src/xla_metadata.py https://filehashintel.com/name/jax/_src/xla_metadata_lib.py https://filehashintel.com/name/jax/_src/xla_transform.py https://filehashintel.com/name/jax/abstract_arrays.py https://filehashintel.com/name/jax/ad_checkpoint.py https://filehashintel.com/name/jax/ad_util.py https://filehashintel.com/name/jax/api.py https://filehashintel.com/name/jax/api_util.py https://filehashintel.com/name/jax/cloud_tpu_init.py https://filehashintel.com/name/jax/collect_profile.py https://filehashintel.com/name/jax/config.py https://filehashintel.com/name/jax/core.py https://filehashintel.com/name/jax/custom_batching.py https://filehashintel.com/name/jax/custom_derivatives.py https://filehashintel.com/name/jax/custom_transpose.py https://filehashintel.com/name/jax/debug.py https://filehashintel.com/name/jax/distributed.py https://filehashintel.com/name/jax/dlpack.py https://filehashintel.com/name/jax/dtypes.py https://filehashintel.com/name/jax/errors.py https://filehashintel.com/name/jax/example_libraries/__init__.py https://filehashintel.com/name/jax/example_libraries/optimizers.py https://filehashintel.com/name/jax/example_libraries/stax.py https://filehashintel.com/name/jax/experimental/__init__.py https://filehashintel.com/name/jax/experimental/_private_mm/__init__.py https://filehashintel.com/name/jax/experimental/_private_mm/mini_dime.py https://filehashintel.com/name/jax/experimental/_private_mm/mm.py https://filehashintel.com/name/jax/experimental/_private_mm/profile_utils.py https://filehashintel.com/name/jax/experimental/array_api/__init__.py https://filehashintel.com/name/jax/experimental/array_api/_array_methods.py https://filehashintel.com/name/jax/experimental/array_api/_constants.py https://filehashintel.com/name/jax/experimental/array_api/_creation_functions.py https://filehashintel.com/name/jax/experimental/array_api/_data_type_functions.py https://filehashintel.com/name/jax/experimental/array_api/_dtypes.py https://filehashintel.com/name/jax/experimental/array_api/_elementwise_functions.py https://filehashintel.com/name/jax/experimental/array_api/_fft_functions.py https://filehashintel.com/name/jax/experimental/array_api/_indexing_functions.py https://filehashintel.com/name/jax/experimental/array_api/_linear_algebra_functions.py https://filehashintel.com/name/jax/experimental/array_api/_manipulation_functions.py https://filehashintel.com/name/jax/experimental/array_api/_searching_functions.py https://filehashintel.com/name/jax/experimental/array_api/_set_functions.py https://filehashintel.com/name/jax/experimental/array_api/_sorting_functions.py https://filehashintel.com/name/jax/experimental/array_api/_statistical_functions.py https://filehashintel.com/name/jax/experimental/array_api/_utility_functions.py https://filehashintel.com/name/jax/experimental/array_api/_version.py https://filehashintel.com/name/jax/experimental/array_api/fft.py https://filehashintel.com/name/jax/experimental/array_api/linalg.py https://filehashintel.com/name/jax/experimental/array_serialization/__init__.py https://filehashintel.com/name/jax/experimental/array_serialization/pytree_serialization.py https://filehashintel.com/name/jax/experimental/array_serialization/pytree_serialization_utils.py https://filehashintel.com/name/jax/experimental/array_serialization/serialization.py https://filehashintel.com/name/jax/experimental/array_serialization/serialization_test.py https://filehashintel.com/name/jax/experimental/array_serialization/tensorstore_impl.py https://filehashintel.com/name/jax/experimental/attrs.py https://filehashintel.com/name/jax/experimental/buffer_callback.py https://filehashintel.com/name/jax/experimental/checkify.py https://filehashintel.com/name/jax/experimental/colocated_python/__init__.py https://filehashintel.com/name/jax/experimental/colocated_python/api.py https://filehashintel.com/name/jax/experimental/colocated_python/func.py https://filehashintel.com/name/jax/experimental/colocated_python/func_backend.py https://filehashintel.com/name/jax/experimental/colocated_python/obj.py https://filehashintel.com/name/jax/experimental/colocated_python/obj_backend.py https://filehashintel.com/name/jax/experimental/colocated_python/serialization.py https://filehashintel.com/name/jax/experimental/compilation_cache/__init__.py https://filehashintel.com/name/jax/experimental/compilation_cache/compilation_cache.py https://filehashintel.com/name/jax/experimental/compute_on.py https://filehashintel.com/name/jax/experimental/control_deps.py https://filehashintel.com/name/jax/experimental/custom_dce.py https://filehashintel.com/name/jax/experimental/custom_partitioning.py https://filehashintel.com/name/jax/experimental/export/__init__.py https://filehashintel.com/name/jax/experimental/export/_export.py https://filehashintel.com/name/jax/experimental/export/_serialization.py https://filehashintel.com/name/jax/experimental/export/_shape_poly.py https://filehashintel.com/name/jax/experimental/export/_shape_poly_decision.py https://filehashintel.com/name/jax/experimental/export/export.py https://filehashintel.com/name/jax/experimental/export/serialization.py https://filehashintel.com/name/jax/experimental/export/serialization_generated.py https://filehashintel.com/name/jax/experimental/export/shape_poly.py https://filehashintel.com/name/jax/experimental/fused.py https://filehashintel.com/name/jax/experimental/hijax.py https://filehashintel.com/name/jax/experimental/host_callback.py https://filehashintel.com/name/jax/experimental/jax2tf/__init__.py https://filehashintel.com/name/jax/experimental/jax2tf/call_tf.py https://filehashintel.com/name/jax/experimental/jax2tf/examples/__init__.py https://filehashintel.com/name/jax/experimental/jax2tf/examples/keras_reuse_main.py https://filehashintel.com/name/jax/experimental/jax2tf/examples/keras_reuse_main_test.py https://filehashintel.com/name/jax/experimental/jax2tf/examples/mnist_lib.py https://filehashintel.com/name/jax/experimental/jax2tf/examples/saved_model_lib.py https://filehashintel.com/name/jax/experimental/jax2tf/examples/saved_model_main.py https://filehashintel.com/name/jax/experimental/jax2tf/examples/saved_model_main_test.py https://filehashintel.com/name/jax/experimental/jax2tf/examples/serving/__init__.py https://filehashintel.com/name/jax/experimental/jax2tf/examples/serving/model_server_request.py https://filehashintel.com/name/jax/experimental/jax2tf/impl_no_xla.py https://filehashintel.com/name/jax/experimental/jax2tf/jax2tf.py https://filehashintel.com/name/jax/experimental/jax2tf/jax_export.py https://filehashintel.com/name/jax/experimental/jax2tf/shape_poly.py https://filehashintel.com/name/jax/experimental/jax2tf/tests/__init__.py https://filehashintel.com/name/jax/experimental/jax2tf/tests/back_compat_test.py https://filehashintel.com/name/jax/experimental/jax2tf/tests/back_compat_test_util.py https://filehashintel.com/name/jax/experimental/jax2tf/tests/back_compat_tf_test.py https://filehashintel.com/name/jax/experimental/jax2tf/tests/call_tf_test.py https://filehashintel.com/name/jax/experimental/jax2tf/tests/control_flow_ops_test.py https://filehashintel.com/name/jax/experimental/jax2tf/tests/converters.py https://filehashintel.com/name/jax/experimental/jax2tf/tests/cross_compilation_check.py https://filehashintel.com/name/jax/experimental/jax2tf/tests/jax2tf_limitations.py https://filehashintel.com/name/jax/experimental/jax2tf/tests/jax2tf_test.py https://filehashintel.com/name/jax/experimental/jax2tf/tests/jax_primitives_coverage_test.py https://filehashintel.com/name/jax/experimental/jax2tf/tests/model_harness.py https://filehashintel.com/name/jax/experimental/jax2tf/tests/models_test_main.py https://filehashintel.com/name/jax/experimental/jax2tf/tests/multi_platform_export_test.py https://filehashintel.com/name/jax/experimental/jax2tf/tests/primitive_harness.py https://filehashintel.com/name/jax/experimental/jax2tf/tests/primitives_test.py https://filehashintel.com/name/jax/experimental/jax2tf/tests/savedmodel_test.py https://filehashintel.com/name/jax/experimental/jax2tf/tests/shape_poly_test.py https://filehashintel.com/name/jax/experimental/jax2tf/tests/sharding_test.py https://filehashintel.com/name/jax/experimental/jax2tf/tests/tf_test_util.py https://filehashintel.com/name/jax/experimental/jet.py https://filehashintel.com/name/jax/experimental/key_reuse/__init__.py https://filehashintel.com/name/jax/experimental/key_reuse/_common.py https://filehashintel.com/name/jax/experimental/key_reuse/_core.py https://filehashintel.com/name/jax/experimental/key_reuse/_forwarding.py https://filehashintel.com/name/jax/experimental/key_reuse/_simple.py https://filehashintel.com/name/jax/experimental/lapax.py https://filehashintel.com/name/jax/experimental/layout.py https://filehashintel.com/name/jax/experimental/maps.py https://filehashintel.com/name/jax/experimental/mesh_utils.py https://filehashintel.com/name/jax/experimental/mosaic/__init__.py https://filehashintel.com/name/jax/experimental/mosaic/dialects.py https://filehashintel.com/name/jax/experimental/mosaic/gpu/__init__.py https://filehashintel.com/name/jax/experimental/mosaic/gpu/constraints.py https://filehashintel.com/name/jax/experimental/mosaic/gpu/core.py https://filehashintel.com/name/jax/experimental/mosaic/gpu/dialect_lowering.py https://filehashintel.com/name/jax/experimental/mosaic/gpu/dsl.py https://filehashintel.com/name/jax/experimental/mosaic/gpu/equations.py https://filehashintel.com/name/jax/experimental/mosaic/gpu/fragmented_array.py https://filehashintel.com/name/jax/experimental/mosaic/gpu/inference_utils.py https://filehashintel.com/name/jax/experimental/mosaic/gpu/launch_context.py https://filehashintel.com/name/jax/experimental/mosaic/gpu/layout_inference.py https://filehashintel.com/name/jax/experimental/mosaic/gpu/layout_inference2.py https://filehashintel.com/name/jax/experimental/mosaic/gpu/layouts.py https://filehashintel.com/name/jax/experimental/mosaic/gpu/mma.py https://filehashintel.com/name/jax/experimental/mosaic/gpu/mma_utils.py https://filehashintel.com/name/jax/experimental/mosaic/gpu/profiler.py https://filehashintel.com/name/jax/experimental/mosaic/gpu/tcgen05.py https://filehashintel.com/name/jax/experimental/mosaic/gpu/transform_inference.py https://filehashintel.com/name/jax/experimental/mosaic/gpu/utils.py https://filehashintel.com/name/jax/experimental/mosaic/gpu/wgmma.py https://filehashintel.com/name/jax/experimental/multihost_utils.py https://filehashintel.com/name/jax/experimental/ode.py https://filehashintel.com/name/jax/experimental/optimizers.py https://filehashintel.com/name/jax/experimental/pallas/__init__.py https://filehashintel.com/name/jax/experimental/pallas/fuser.py https://filehashintel.com/name/jax/experimental/pallas/gpu.py https://filehashintel.com/name/jax/experimental/pallas/mosaic_gpu.py https://filehashintel.com/name/jax/experimental/pallas/ops/__init__.py https://filehashintel.com/name/jax/experimental/pallas/ops/attention.py https://filehashintel.com/name/jax/experimental/pallas/ops/gpu/__init__.py https://filehashintel.com/name/jax/experimental/pallas/ops/gpu/all_gather_mgpu.py https://filehashintel.com/name/jax/experimental/pallas/ops/gpu/attention.py https://filehashintel.com/name/jax/experimental/pallas/ops/gpu/attention_mgpu.py https://filehashintel.com/name/jax/experimental/pallas/ops/gpu/blackwell_matmul_mgpu.py https://filehashintel.com/name/jax/experimental/pallas/ops/gpu/blackwell_ragged_dot_mgpu.py https://filehashintel.com/name/jax/experimental/pallas/ops/gpu/collective_matmul_mgpu.py https://filehashintel.com/name/jax/experimental/pallas/ops/gpu/decode_attention.py https://filehashintel.com/name/jax/experimental/pallas/ops/gpu/hopper_matmul_mgpu.py https://filehashintel.com/name/jax/experimental/pallas/ops/gpu/hopper_mixed_type_matmul_mgpu.py https://filehashintel.com/name/jax/experimental/pallas/ops/gpu/layer_norm.py https://filehashintel.com/name/jax/experimental/pallas/ops/gpu/paged_attention.py https://filehashintel.com/name/jax/experimental/pallas/ops/gpu/ragged_dot_mgpu.py https://filehashintel.com/name/jax/experimental/pallas/ops/gpu/reduce_scatter_mgpu.py https://filehashintel.com/name/jax/experimental/pallas/ops/gpu/rms_norm.py https://filehashintel.com/name/jax/experimental/pallas/ops/gpu/softmax.py https://filehashintel.com/name/jax/experimental/pallas/ops/gpu/transposed_ragged_dot_mgpu.py https://filehashintel.com/name/jax/experimental/pallas/ops/layer_norm.py https://filehashintel.com/name/jax/experimental/pallas/ops/rms_norm.py https://filehashintel.com/name/jax/experimental/pallas/ops/softmax.py https://filehashintel.com/name/jax/experimental/pallas/ops/tpu/__init__.py https://filehashintel.com/name/jax/experimental/pallas/ops/tpu/all_gather.py https://filehashintel.com/name/jax/experimental/pallas/ops/tpu/example_kernel.py https://filehashintel.com/name/jax/experimental/pallas/ops/tpu/flash_attention.py https://filehashintel.com/name/jax/experimental/pallas/ops/tpu/matmul.py https://filehashintel.com/name/jax/experimental/pallas/ops/tpu/megablox/__init__.py https://filehashintel.com/name/jax/experimental/pallas/ops/tpu/megablox/common.py https://filehashintel.com/name/jax/experimental/pallas/ops/tpu/megablox/gmm.py https://filehashintel.com/name/jax/experimental/pallas/ops/tpu/megablox/ops.py https://filehashintel.com/name/jax/experimental/pallas/ops/tpu/paged_attention/__init__.py https://filehashintel.com/name/jax/experimental/pallas/ops/tpu/paged_attention/paged_attention_kernel.py https://filehashintel.com/name/jax/experimental/pallas/ops/tpu/paged_attention/quantization_utils.py https://filehashintel.com/name/jax/experimental/pallas/ops/tpu/paged_attention/util.py https://filehashintel.com/name/jax/experimental/pallas/ops/tpu/ragged_paged_attention.py https://filehashintel.com/name/jax/experimental/pallas/ops/tpu/ragged_paged_attention/__init__.py https://filehashintel.com/name/jax/experimental/pallas/ops/tpu/ragged_paged_attention/kernel.py https://filehashintel.com/name/jax/experimental/pallas/ops/tpu/ragged_paged_attention/tuned_block_sizes.py https://filehashintel.com/name/jax/experimental/pallas/ops/tpu/random/__init__.py https://filehashintel.com/name/jax/experimental/pallas/ops/tpu/random/philox.py https://filehashintel.com/name/jax/experimental/pallas/ops/tpu/random/prng_utils.py https://filehashintel.com/name/jax/experimental/pallas/ops/tpu/random/threefry.py https://filehashintel.com/name/jax/experimental/pallas/ops/tpu/splash_attention/__init__.py https://filehashintel.com/name/jax/experimental/pallas/ops/tpu/splash_attention/splash_attention_kernel.py https://filehashintel.com/name/jax/experimental/pallas/ops/tpu/splash_attention/splash_attention_mask.py https://filehashintel.com/name/jax/experimental/pallas/ops/tpu/splash_attention/splash_attention_mask_info.py https://filehashintel.com/name/jax/experimental/pallas/tpu.py https://filehashintel.com/name/jax/experimental/pallas/tpu_sc.py https://filehashintel.com/name/jax/experimental/pallas/triton.py https://filehashintel.com/name/jax/experimental/pjit.py https://filehashintel.com/name/jax/experimental/profiler.py https://filehashintel.com/name/jax/experimental/random.py https://filehashintel.com/name/jax/experimental/rnn.py https://filehashintel.com/name/jax/experimental/roofline/__init__.py https://filehashintel.com/name/jax/experimental/roofline/roofline.py https://filehashintel.com/name/jax/experimental/roofline/rooflines.py https://filehashintel.com/name/jax/experimental/scheduling_groups.py https://filehashintel.com/name/jax/experimental/serialize_executable.py https://filehashintel.com/name/jax/experimental/shard.py https://filehashintel.com/name/jax/experimental/shard_alike.py https://filehashintel.com/name/jax/experimental/shard_map.py https://filehashintel.com/name/jax/experimental/source_mapper/__init__.py https://filehashintel.com/name/jax/experimental/source_mapper/common.py https://filehashintel.com/name/jax/experimental/source_mapper/generate_map.py https://filehashintel.com/name/jax/experimental/source_mapper/hlo.py https://filehashintel.com/name/jax/experimental/source_mapper/jaxpr.py https://filehashintel.com/name/jax/experimental/source_mapper/mlir.py https://filehashintel.com/name/jax/experimental/sparse/__init__.py https://filehashintel.com/name/jax/experimental/sparse/_base.py https://filehashintel.com/name/jax/experimental/sparse/_lowerings.py https://filehashintel.com/name/jax/experimental/sparse/ad.py https://filehashintel.com/name/jax/experimental/sparse/api.py https://filehashintel.com/name/jax/experimental/sparse/bcoo.py https://filehashintel.com/name/jax/experimental/sparse/bcsr.py https://filehashintel.com/name/jax/experimental/sparse/coo.py https://filehashintel.com/name/jax/experimental/sparse/csr.py https://filehashintel.com/name/jax/experimental/sparse/linalg.py https://filehashintel.com/name/jax/experimental/sparse/nm.py https://filehashintel.com/name/jax/experimental/sparse/random.py https://filehashintel.com/name/jax/experimental/sparse/test_util.py https://filehashintel.com/name/jax/experimental/sparse/transform.py https://filehashintel.com/name/jax/experimental/sparse/util.py https://filehashintel.com/name/jax/experimental/stax.py https://filehashintel.com/name/jax/experimental/topologies.py https://filehashintel.com/name/jax/experimental/transfer.py https://filehashintel.com/name/jax/experimental/x64_context.py https://filehashintel.com/name/jax/experimental/xla_metadata.py https://filehashintel.com/name/jax/export.py https://filehashintel.com/name/jax/extend/__init__.py https://filehashintel.com/name/jax/extend/backend.py https://filehashintel.com/name/jax/extend/core.py https://filehashintel.com/name/jax/extend/core/__init__.py https://filehashintel.com/name/jax/extend/core/primitives.py https://filehashintel.com/name/jax/extend/ffi.py https://filehashintel.com/name/jax/extend/ifrt_programs.py https://filehashintel.com/name/jax/extend/linear_util.py https://filehashintel.com/name/jax/extend/lowering.py https://filehashintel.com/name/jax/extend/mlir/__init__.py https://filehashintel.com/name/jax/extend/mlir/dialects/__init__.py https://filehashintel.com/name/jax/extend/mlir/dialects/arith.py https://filehashintel.com/name/jax/extend/mlir/dialects/builtin.py https://filehashintel.com/name/jax/extend/mlir/dialects/chlo.py https://filehashintel.com/name/jax/extend/mlir/dialects/func.py https://filehashintel.com/name/jax/extend/mlir/dialects/math.py https://filehashintel.com/name/jax/extend/mlir/dialects/memref.py https://filehashintel.com/name/jax/extend/mlir/dialects/mhlo.py https://filehashintel.com/name/jax/extend/mlir/dialects/mpmd.py https://filehashintel.com/name/jax/extend/mlir/dialects/scf.py https://filehashintel.com/name/jax/extend/mlir/dialects/sdy.py https://filehashintel.com/name/jax/extend/mlir/dialects/sparse_tensor.py https://filehashintel.com/name/jax/extend/mlir/dialects/stablehlo.py https://filehashintel.com/name/jax/extend/mlir/dialects/vector.py https://filehashintel.com/name/jax/extend/mlir/ir.py https://filehashintel.com/name/jax/extend/mlir/passmanager.py https://filehashintel.com/name/jax/extend/pallas.py https://filehashintel.com/name/jax/extend/random.py https://filehashintel.com/name/jax/extend/sharding.py https://filehashintel.com/name/jax/extend/source_info_util.py https://filehashintel.com/name/jax/extend/xla.py https://filehashintel.com/name/jax/ffi.py https://filehashintel.com/name/jax/flatten_util.py https://filehashintel.com/name/jax/image/__init__.py https://filehashintel.com/name/jax/interpreters/__init__.py https://filehashintel.com/name/jax/interpreters/ad.py https://filehashintel.com/name/jax/interpreters/algsimp.py https://filehashintel.com/name/jax/interpreters/batching.py https://filehashintel.com/name/jax/interpreters/cse2.py https://filehashintel.com/name/jax/interpreters/mlir.py https://filehashintel.com/name/jax/interpreters/parallel.py https://filehashintel.com/name/jax/interpreters/partial_eval.py https://filehashintel.com/name/jax/interpreters/pxla.py https://filehashintel.com/name/jax/interpreters/xla.py https://filehashintel.com/name/jax/lax.py https://filehashintel.com/name/jax/lax/__init__.py https://filehashintel.com/name/jax/lax/linalg.py https://filehashintel.com/name/jax/lax_linalg.py https://filehashintel.com/name/jax/lax_reference.py https://filehashintel.com/name/jax/lib/__init__.py https://filehashintel.com/name/jax/lib/pywrap_xla.py https://filehashintel.com/name/jax/lib/xla_bridge.py https://filehashintel.com/name/jax/lib/xla_client.py https://filehashintel.com/name/jax/lib/xla_data_pb2.py https://filehashintel.com/name/jax/lib/xla_extension.py https://filehashintel.com/name/jax/linear_util.py https://filehashintel.com/name/jax/memory.py https://filehashintel.com/name/jax/monitoring.py https://filehashintel.com/name/jax/nn/__init__.py https://filehashintel.com/name/jax/nn/initializers.py https://filehashintel.com/name/jax/numpy/__init__.py https://filehashintel.com/name/jax/numpy/fft.py https://filehashintel.com/name/jax/numpy/lax_numpy.py https://filehashintel.com/name/jax/numpy/linalg.py https://filehashintel.com/name/jax/ops/__init__.py https://filehashintel.com/name/jax/ops/scatter.py https://filehashintel.com/name/jax/pprint_util.py https://filehashintel.com/name/jax/prng.py https://filehashintel.com/name/jax/profiler.py https://filehashintel.com/name/jax/random.py https://filehashintel.com/name/jax/ref.py https://filehashintel.com/name/jax/scipy/__init__.py https://filehashintel.com/name/jax/scipy/cluster/__init__.py https://filehashintel.com/name/jax/scipy/cluster/vq.py https://filehashintel.com/name/jax/scipy/fft.py https://filehashintel.com/name/jax/scipy/integrate.py https://filehashintel.com/name/jax/scipy/interpolate/__init__.py https://filehashintel.com/name/jax/scipy/linalg.py https://filehashintel.com/name/jax/scipy/misc.py https://filehashintel.com/name/jax/scipy/ndimage.py https://filehashintel.com/name/jax/scipy/optimize/__init__.py https://filehashintel.com/name/jax/scipy/signal.py https://filehashintel.com/name/jax/scipy/sparse/__init__.py https://filehashintel.com/name/jax/scipy/sparse/linalg.py https://filehashintel.com/name/jax/scipy/spatial/__init__.py https://filehashintel.com/name/jax/scipy/spatial/transform.py https://filehashintel.com/name/jax/scipy/special.py https://filehashintel.com/name/jax/scipy/stats/__init__.py https://filehashintel.com/name/jax/scipy/stats/bernoulli.py https://filehashintel.com/name/jax/scipy/stats/beta.py https://filehashintel.com/name/jax/scipy/stats/betabinom.py https://filehashintel.com/name/jax/scipy/stats/binom.py https://filehashintel.com/name/jax/scipy/stats/cauchy.py https://filehashintel.com/name/jax/scipy/stats/chi2.py https://filehashintel.com/name/jax/scipy/stats/dirichlet.py https://filehashintel.com/name/jax/scipy/stats/expon.py https://filehashintel.com/name/jax/scipy/stats/gamma.py https://filehashintel.com/name/jax/scipy/stats/gennorm.py https://filehashintel.com/name/jax/scipy/stats/geom.py https://filehashintel.com/name/jax/scipy/stats/gumbel_l.py https://filehashintel.com/name/jax/scipy/stats/gumbel_r.py https://filehashintel.com/name/jax/scipy/stats/laplace.py https://filehashintel.com/name/jax/scipy/stats/logistic.py https://filehashintel.com/name/jax/scipy/stats/multinomial.py https://filehashintel.com/name/jax/scipy/stats/multivariate_normal.py https://filehashintel.com/name/jax/scipy/stats/nbinom.py https://filehashintel.com/name/jax/scipy/stats/norm.py https://filehashintel.com/name/jax/scipy/stats/pareto.py https://filehashintel.com/name/jax/scipy/stats/poisson.py https://filehashintel.com/name/jax/scipy/stats/t.py https://filehashintel.com/name/jax/scipy/stats/truncnorm.py https://filehashintel.com/name/jax/scipy/stats/uniform.py https://filehashintel.com/name/jax/scipy/stats/vonmises.py https://filehashintel.com/name/jax/scipy/stats/wrapcauchy.py https://filehashintel.com/name/jax/sharding.py https://filehashintel.com/name/jax/stages.py https://filehashintel.com/name/jax/test_util.py https://filehashintel.com/name/jax/tools/__init__.py https://filehashintel.com/name/jax/tools/build_utils.py https://filehashintel.com/name/jax/tools/colab_tpu.py https://filehashintel.com/name/jax/tools/jax_to_ir.py https://filehashintel.com/name/jax/tools/pgo_nsys_converter.py https://filehashintel.com/name/jax/tree.py https://filehashintel.com/name/jax/tree_util.py https://filehashintel.com/name/jax/typing.py https://filehashintel.com/name/jax/util.py https://filehashintel.com/name/jax/version.py