jax-ml/jax

Pattern Findings

The scanner identified 13922 distinct pattern matches across 27 syntactic categories. Each entry below represents a discrete location in the source code where the engine recorded a detector match to inspect in context. Expand any category row to inspect the individual file paths, line numbers, code snippets, and the lexical context (CODE, COMMENT, or STRING) in which each match was detected.

Reading the findings table: The Severity column indicates the configured finding weight (CRITICAL / HIGH / MEDIUM / LOW). The Context column identifies whether the match occurred inside executable code, an inline comment, or a string literal — legacy comment-context matches receive a ×1.5 score weight. The ⚡ bolt icon marks clustered matches: three or more patterns within a 10-line window, each receiving an additional ×1.5 density multiplier under the legacy scoring model.

Hyper-Verbose Identifiers6707 hits · 5072 pts

Severity	File	Line	Snippet	Context
LOW	ci/parse_wheel_metadata.py	34	def _extract_wheel_package_name(wheel_name: str) -> str \| None:	CODE
LOW	ci/parse_wheel_metadata.py	52	def parse_expected_wheel_versions(wheel_dir: Path) -> dict[str, str]:	CODE
LOW⚡	tests/random_test.py	573	def test_key_construction_with_dtype(self, dtype_spec):	CODE
LOW⚡	tests/random_test.py	577	def test_key_construction_with_both_impl_and_dtype(self):	CODE
LOW⚡	tests/random_test.py	587	def test_wrap_key_data_with_dtype(self, dtype_spec):	CODE
LOW⚡	tests/random_test.py	595	def test_wrap_key_data_with_both_impl_and_dtype(self):	CODE
LOW⚡	tests/random_test.py	1219	def test_key_make_like_other_key_via_dtype(self, prng_name):	CODE
LOW⚡	tests/random_test.py	1227	def test_key_wrap_like_other_key_via_dtype(self, prng_name):	CODE
LOW⚡	tests/random_test.py	1235	def test_key_impl_from_string_error(self):	CODE
LOW⚡	tests/random_test.py	1239	def test_key_impl_from_object_error(self):	CODE
LOW⚡	tests/random_test.py	1246	def test_key_impl_builtin_is_string_name(self, name):	CODE
LOW	tests/random_test.py	199	def test_config_prngs_registered(self):	CODE
LOW	tests/random_test.py	412	def test_threefry_gpu_kernel_lowering(self):	CODE
LOW	tests/random_test.py	449	def test_threefry_split_fold_in_symmetry(self, make_key):	CODE
LOW	tests/random_test.py	462	def test_threefry_split_vmapped_fold_in_symmetry(self, make_key):	CODE
LOW	tests/random_test.py	476	def test_loggamma_nan_corner_case(self):	CODE
LOW	tests/random_test.py	549	def test_default_prng_selection(self, make_key, name, impl):	CODE
LOW	tests/random_test.py	561	def test_key_construction_with_explicit_impl_name(self, make_key, name, impl):	CODE
LOW	tests/random_test.py	613	def test_legacy_prng_key_flag(self):	CODE
LOW	tests/random_test.py	641	def test_seed_no_implicit_transfers(self, make_key):	CODE
LOW	tests/random_test.py	717	def test_construction_upgrade_flag(self):	CODE
LOW	tests/random_test.py	763	def test_key_dtype_attributes(self):	CODE
LOW	tests/random_test.py	818	def test_cpp_dispatch_aot_normal(self):	CODE
LOW	tests/random_test.py	831	def test_cpp_dispatch_aot_split(self):	CODE
LOW	tests/random_test.py	909	def test_eval_shape_keys_in_out(self):	CODE
LOW	tests/random_test.py	967	def test_dynamic_update_slice(self):	CODE
LOW	tests/random_test.py	1057	def test_device_put_replicated(self):	CODE
LOW	tests/random_test.py	1063	def test_make_array_from_callback(self):	CODE
LOW	tests/random_test.py	1075	def test_make_array_from_single_device_arrays(self):	CODE
LOW	tests/random_test.py	1085	def test_key_array_custom_jvp(self):	CODE
LOW	tests/random_test.py	1115	def test_key_array_indexing_nd(self):	CODE
LOW	tests/random_test.py	1140	def test_array_impl_attributes(self):	CODE
LOW	tests/random_test.py	1198	def test_key_make_like_other_key(self, prng_name):	CODE
LOW	tests/random_test.py	1208	def test_key_wrap_like_other_key(self, prng_name):	CODE
LOW	tests/random_test.py	1266	def test_keyarray_custom_vjp_symbolic_zeros(self):	CODE
LOW	tests/random_test.py	1281	def test_keyarray_array_conversion_fails(self):	CODE
LOW	tests/random_test.py	1305	def _double_threefry_random_bits(key, bit_width, shape):	CODE
LOW	tests/random_test.py	1569	def test_full_like_with_key_fillvalue(self):	CODE
LOW	tests/random_test.py	1590	def test_full_with_key_fillvalue(self):	CODE
LOW	tests/overlap_test.py	160	def test_unrolled_fsdp_pipeline_grad_explicit_mode_program_order(self, mesh):	CODE
LOW	tests/overlap_test.py	211	def test_unrolled_fsdp_pipeline_grad_program_order_shmap(self, mesh):	CODE
LOW	tests/overlap_test.py	268	def test_unrolled_fsdp_pipeline_grad_program_order_async_decomp(	CODE
LOW	tests/overlap_test.py	353	def test_simple_program_order(self, mesh):	CODE
LOW	tests/overlap_test.py	375	def test_avoid_excess_precision(self):	CODE
LOW	tests/overlap_test.py	391	def test_program_order_exclude_args(self, mesh):	CODE
LOW	tests/overlap_test.py	423	def test_program_order_exclude_argnames_none(self):	CODE
LOW	tests/overlap_test.py	445	def test_program_order_true_false_true_nest(self, mesh):	CODE
LOW	tests/overlap_test.py	485	def test_program_order_per_input_barrier(self, mesh):	CODE
LOW	tests/overlap_test.py	517	def test_program_order_per_input_barrier_jit(self, mesh):	CODE
LOW	tests/overlap_test.py	605	def test_program_order_true_false_true_nest_strict_in_out(self, mesh):	CODE
LOW	tests/overlap_test.py	678	def test_program_order_opt_barrier_dce(self):	CODE
LOW	tests/overlap_test.py	699	def test_program_order_unused_output_dce(self):	CODE
LOW	tests/overlap_test.py	724	def test_program_order_hiprim(self):	CODE
LOW	tests/overlap_test.py	755	def test_program_order_grad_enforce_false_in_true(self):	CODE
LOW	tests/overlap_test.py	774	def test_program_order_dce_preserves_subfunction_deduplication(self):	CODE
LOW	tests/overlap_test.py	804	def test_simple_fsdp_async_overlap_program_order(self, mesh):	CODE
LOW	tests/overlap_test.py	833	def test_async_psum_program_order(self, mesh):	CODE
LOW	tests/overlap_test.py	864	def test_async_psum_scatter_opt_barrier(self, mesh):	CODE
LOW	tests/overlap_test.py	879	def psum_scatter_async_barrier(x, a):	CODE
LOW	tests/overlap_test.py	897	def test_async_all_to_all_opt_barrier(self, mesh):	CODE
6647 more matches not shown…

Unused Imports3645 hits · 2572 pts

Severity	File	Line	Context
LOW	tests/random_test.py	15	CODE
LOW	tests/xla_transform_test.py	17	CODE
LOW	tests/cloud_tpu_cluster_test.py	15	CODE
LOW	tests/batching_test.py	15	CODE
LOW	tests/debug_info_test.py	15	CODE
LOW	tests/mutable_array_test.py	15	CODE
LOW	tests/lax_numpy_test.py	15	CODE
LOW	tests/shard_map_test.py	15	CODE
LOW	tests/export_harnesses_multi_platform_test.py	22	CODE
LOW	tests/shape_poly_test.py	15	CODE
LOW	tests/lax_vmap_test.py	15	CODE
LOW	tests/lax_numpy_indexing_test.py	15	CODE
LOW	tests/typing_test.py	21	CODE
LOW	tests/array_api_test.py	20	CODE
LOW	tests/state_test.py	15	CODE
LOW	tests/export_test.py	14	CODE
LOW	tests/export_test.py	38	CODE
LOW	tests/export_test.py	45	CODE
LOW	tests/export_test.py	51	CODE
LOW	tests/pmap_test.py	15	CODE
LOW	tests/colocated_python_test.py	36	CODE
LOW	tests/lru_cache_test.py	15	CODE
LOW	tests/compilation_cache_test.py	15	CODE
LOW	tests/hijax_test.py	15	CODE
LOW	tests/export_serialization_back_compat_test.py	53	CODE
LOW	tests/api_test.py	15	CODE
LOW	tests/roofline_test.py	14	CODE
LOW	tests/lax_test.py	14	CODE
LOW	tests/mosaic/gpu_torch_test.py	29	CODE
LOW	tests/mosaic/flash_attention_test.py	24	CODE
LOW	tests/mosaic/gpu_test.py	54	CODE
LOW	tests/multiprocess/colocated_python_test.py	24	CODE
LOW	tests/pallas/tpu_all_gather_test.py	16	CODE
LOW	tests/pallas/tpu_splash_attention_mask_test.py	15	CODE
LOW	tests/pallas/tpu_pallas_pipeline_test.py	16	CODE
LOW	tests/pallas/tpu_pallas_random_test.py	28	CODE
LOW	tests/pallas/indexing_test.py	15	CODE
LOW	tests/pallas/tpu_splash_attention_kernel_test.py	15	CODE
LOW	tests/pallas/pallas_test.py	14	CODE
LOW	docs/parallel.py	63	CODE
LOW	docs/autodidax.py	2153	CODE
LOW	jaxlib/xla_client.py	17	CODE
LOW	jaxlib/init.py	15	CODE
LOW	jaxlib/mosaic/gpu/wheel/__init__.py	17	CODE
LOW	jaxlib/mosaic/python/tpu.py	20	CODE
LOW	jaxlib/mosaic/python/tpu.py	22	CODE
LOW	jaxlib/mosaic/python/tpu.py	24	CODE
LOW	jaxlib/mosaic/python/mosaic_gpu.py	23	CODE
LOW	jaxlib/mosaic/python/mosaic_gpu.py	25	CODE
LOW	jaxlib/mosaic/python/mosaic_gpu.py	26	CODE
LOW	jaxlib/tools/build_utils.py	17	CODE
LOW	jaxlib/triton/dialect.py	19	CODE
LOW	jaxlib/triton/dialect.py	23	CODE
LOW	jaxlib/triton/dialect.py	23	CODE
LOW	jaxlib/triton/dialect.py	30	CODE
LOW	jaxlib/triton/dialect.py	31	CODE
LOW	jaxlib/thunky/python/thunky.py	19	CODE
LOW	jaxlib/thunky/python/thunky.py	20	CODE
LOW	jaxlib/thunky/python/thunky.py	21	CODE
LOW	benchmarks/pallas/mgpu_planar_snake_bench.py	23	CODE
3585 more matches not shown…

Docstring Block Structure572 hits · 2328 pts

Severity	File	Line	Snippet	Context
HIGH	jax/experimental/multihost_utils.py	549	A context manager for atomically running code on the set of live devices. THIS API IS UNDER ACTIVE DEVELOPMENT AND IS	STRING
HIGH	jax/experimental/mosaic/gpu/profiler.py	138	Measures the GPU runtime of a function using CUPTI. ``measure`` is a higher-order function that wraps a function ``f`	STRING
HIGH	…ops/tpu/splash_attention/splash_attention_mask_info.py	84	Downcast numpy array. If possible, downcast the data-type of the input array to the smallest numpy type (among np.i	STRING
HIGH	…ops/tpu/splash_attention/splash_attention_mask_info.py	326	Similar to `_process_mask` but the mask must be a dynamic array. Since the mask is dynamic, we can't know the exact n	STRING
HIGH	…ops/tpu/splash_attention/splash_attention_mask_info.py	528	Transform a dense mask into a sparse representation. The number of head and Q sequence shards are needed to create a	STRING
HIGH	…llas/ops/tpu/splash_attention/splash_attention_mask.py	97	Makes a chunked causal attention mask. Args: shape: The desired shape of the mask (q_seq_len, kv_seq_len). ch	STRING
HIGH	jax/experimental/sparse/linalg.py	43	Compute the top-k standard eigenvalues using the LOBPCG routine. LOBPCG [1] stands for Locally Optimal Block Precondi	STRING
HIGH	jax/experimental/sparse/bcoo.py	2437	Experimental batched COO matrix implemented in JAX Args: (data, indices) : data and indices in batched COO format	STRING
HIGH	jax/_src/callback.py	275	Calls a pure Python callback. Works under :func:`jit`/:func:`~vmap`/etc. For more explanation, see `External Callback	STRING
HIGH	jax/_src/tree.py	26	Call all() over the leaves of a tree. Args: tree: the pytree to evaluate is_leaf : an optionally specified fu	STRING
HIGH	jax/_src/tree.py	55	Flattens a pytree. The flattening order (i.e. the order of elements in the output list) is deterministic, correspon	STRING
HIGH	jax/_src/tree.py	91	Gets the leaves of a pytree. Args: tree: the pytree for which to get the leaves is_leaf : an optionally speci	STRING
HIGH	jax/_src/tree.py	120	Maps a multi-input function over pytree args to produce a new pytree. Args: f: function that takes ``1 + len(rest	STRING
HIGH	jax/_src/tree.py	163	Call reduce() over the leaves of a tree. Args: function: the reduction function tree: the pytree to reduce ov	STRING
HIGH	jax/_src/tree.py	203	Perform a reduction over a pytree with an associative binary operation. This function exploits the fact that the oper	STRING
HIGH	jax/_src/tree.py	245	Gets the treedef for a pytree. Args: tree: the pytree for which to get the leaves is_leaf : an optionally spe	STRING
HIGH	jax/_src/tree.py	273	Transform a tree having tree structure (outer, inner) into one having structure (inner, outer). Args: outer_treed	STRING
HIGH	jax/_src/tree.py	303	Reconstructs a pytree from the treedef and the leaves. The inverse of :func:`tree_flatten`. Args: treedef: the	STRING
HIGH	jax/_src/tree.py	335	Flattens a pytree like ``tree_flatten``, but also returns each leaf's key path. Args: tree: a pytree to flatten.	STRING
HIGH	jax/_src/tree.py	366	Gets the leaves of a pytree like ``tree_leaves`` and returns each leaf's key path. Args: tree: a pytree. If it co	STRING
HIGH	jax/_src/tree.py	395	Maps a multi-input function over pytree key path and args to produce a new pytree. This is a more powerful alternativ	STRING
HIGH	jax/_src/tree.py	433	Broadcasts a tree prefix into the full structure of a given tree. Args: prefix_tree: a pytree that is a tree	STRING
HIGH	jax/_src/xla_bridge.py	317	Creates a CPU client with the requested collectives implementation. The implementation of CPU collectives used by the	STRING
HIGH	jax/_src/config.py	352	Set up thread-local state and return a contextmanager for managing it. This function is a convenience wrapper. It def	STRING
HIGH	jax/_src/core.py	3025	Invalidate a given reference and return its final value. For more information about mutable array references, refer t	STRING
HIGH	jax/_src/dtypes.py	881	Returns the type to which a binary operation should cast its arguments. JAX implementation of :func:`numpy.promote_ty	STRING
HIGH	jax/_src/dtypes.py	1133	Check if a dtype/value is safe to cast to another dtype/value Args: input_dtype_or_value: a dtype or value (to be	STRING
HIGH	jax/_src/api.py	220	Sets up ``fun`` for just-in-time compilation with XLA. Args: fun: Function to be jitted. ``fun`` should be a pure	STRING
HIGH	jax/_src/api.py	2432	Transfer array shards to specified devices and form Array(s). Args: shards: A sequence of arrays, scalars, or (ne	STRING
HIGH	jax/_src/api.py	2512	Transfer array(s) to each specified device and form Array(s). Args: x: an array, scalar, or (nested) standard Pyt	STRING
HIGH	jax/_src/api.py	2581	Transfer ``x`` to host. If ``x`` is a pytree, then the individual buffers are copied in parallel. Args: x: An	STRING
HIGH	jax/_src/api.py	2726	A context manager that adds a user specified name to the JAX name stack. When staging out computations for just-in-ti	STRING
HIGH	jax/_src/distributed.py	277	Initializes the JAX distributed system. Calling :func:`~jax.distributed.initialize` prepares JAX for execution on m	STRING
HIGH	jax/_src/sharding_impls.py	654	Computes the global shape given the per process if possible. The returned shape will have the size of the global tens	STRING
HIGH	jax/_src/tree_util.py	127	Makes a tuple treedef from an iterable of child treedefs. Args: treedefs: iterable of PyTree structures Return	STRING
HIGH	jax/_src/tree_util.py	153	Return a list of treedefs for immediate children Args: treedef: a single PyTreeDef Returns: a list of PyTr	STRING
HIGH	jax/_src/tree_util.py	178	Return True if the treedef represents a leaf. Args: treedef: tree to check Returns: True if treedef is a l	STRING
HIGH	jax/_src/tree_util.py	206	Tests whether all elements in the given iterable are all leaves. This function is useful in advanced cases, for examp	STRING
HIGH	jax/_src/tree_util.py	354	Extends the set of types that are considered internal nodes in pytrees. This function is a thin wrapper around ``regi	STRING
HIGH	jax/_src/tree_util.py	714	Flatten the given pytree node by one level. Args: tree: A valid pytree node, either built-in or registered via	STRING
HIGH	jax/_src/tree_util.py	860	Helper to pretty-print a tuple of keys. Args: keys: A tuple of ``KeyEntry`` or any class that can be converted to	STRING
HIGH	jax/_src/tree_util.py	994	Extends the set of types that are considered internal nodes in pytrees. This function is similar to ``register_pytree	STRING
HIGH	jax/_src/tree_util.py	1048	Extends the set of types that are considered internal nodes in pytrees. This differs from ``register_pytree_with_keys	STRING
HIGH	jax/_src/tree_util.py	1246	Registers `cls` as a pytree with no leaves. Instances are treated as static by :func:`jax.jit`, :func:`jax.pmap`, etc	STRING
HIGH	jax/_src/custom_partitioning_sharding_rule.py	215	Parses the LHS or RHS of an Einsum notation like string. Converts each operand or result in the Einsum notation like	STRING
HIGH	jax/_src/array.py	760	Returns a ``jax.Array`` via data fetched from ``data_callback``. ``data_callback`` is used to fetch the data for each	STRING
HIGH	jax/_src/array.py	895	Creates distributed tensor using the data available in process. This function is a common special case of `make_array	STRING
HIGH	jax/_src/array.py	1095	Returns a ``jax.Array`` from a sequence of ``jax.Array``\s each on a single device. Every device in input ``shardi	STRING
HIGH	jax/_src/profiler.py	98	Registers a subprocess's profiler server to be profiled alongside the current process. When the current process colle	STRING
HIGH	jax/_src/mesh_utils.py	822	Creates a performant device mesh for jax.sharding.Mesh. Args: mesh_shape: shape of logical mesh, ordered by incre	STRING
HIGH	jax/_src/mesh_utils.py	908	Creates a device mesh for hybrid (e.g., ICI and DCN) parallelism. Args: mesh_shape: shape of the logical mesh for	STRING
HIGH	jax/_src/pmap.py	475	Infer axis size from the first mapped argument. shard_map already does a check on all arguments, so just look at firs	STRING
HIGH	jax/_src/pmap.py	526	Extract dynamic args and argnums after handling static args. Args: wrapped_f: The wrapped function. static_br	STRING
HIGH	jax/_src/pmap.py	564	Compute flat in_axes tuple from in_axes prefix and args structure. Args: in_axes: The original in_axes specificat	STRING
HIGH	jax/_src/pmap.py	671	Compute effective mesh devices based on context. Args: devices: The mesh devices tuple. backend: The backend	STRING
HIGH	jax/_src/custom_derivatives.py	158	Define a custom JVP rule for the function represented by this instance. Args: jvp: a Python callable represen	STRING
HIGH	jax/_src/custom_derivatives.py	207	Convenience wrapper for defining JVPs for each argument separately. This convenience wrapper cannot be used togethe	STRING
HIGH	jax/_src/custom_derivatives.py	592	Define a custom VJP rule for the function represented by this instance. Args: fwd: a Python callable represen	STRING
HIGH	jax/_src/interpreters/mlir.py	256	Translate a Python ``val`` to an IR constant. See https://docs.jax.dev/en/latest/internals/constants.html. Args:	STRING
HIGH	jax/_src/interpreters/mlir.py	288	Translate a Python ``val`` to a sequence of IR constants. See https://docs.jax.dev/en/latest/internals/constants.html	STRING
512 more matches not shown…

Over-Commented Block1538 hits · 1448 pts

Severity	File	Line	Snippet	Context
LOW	conftest.py	1	# Copyright 2021 The JAX Authors.	COMMENT
LOW	conftest.py	41	# The pytest_collection hook can be used to overwrite the collection logic, but	COMMENT
LOW	setup.py	1	# Copyright 2018 The JAX Authors.	COMMENT
LOW	build_wheel.py	1	# Copyright 2025 The JAX Authors.	COMMENT
LOW	ci/run_pytest_tpu.sh	1	#!/bin/bash	COMMENT
LOW	ci/run_pytest_rocm.sh	1	#!/bin/bash	COMMENT
LOW	ci/run_pytest_rocm.sh	21	#	COMMENT
LOW	ci/run_bazel_test_cuda_non_rbe.sh	1	#!/bin/bash	COMMENT
LOW	ci/run_pytest_oneapi.sh	1	#!/bin/bash	COMMENT
LOW	ci/run_bazel_cuda_targeted_tests.sh	1	#!/bin/bash	COMMENT
LOW	ci/run_bazel_cuda_targeted_tests.sh	21	# JAXCI_BAZEL_TARGETS: newline-separated Bazel targets.	COMMENT
LOW	ci/run_bazel_test_rocm_rbe.sh	1	#!/bin/bash	COMMENT
LOW	ci/build_artifacts.sh	1	#!/bin/bash	COMMENT
LOW	ci/build_rocm_artifacts.sh	1	#!/bin/bash	COMMENT
LOW	ci/run_bazel_test_cpu_rbe.sh	1	#!/bin/bash	COMMENT
LOW	ci/upload_rocm_logs.sh	1	#!/bin/bash	COMMENT
LOW	ci/run_bazel_test_cuda_rbe.sh	1	#!/bin/bash	COMMENT
LOW	ci/run_pytest_cuda.sh	1	#!/bin/bash	COMMENT
LOW	ci/build_oneapi_artifacts.sh	1	#!/bin/bash	COMMENT
LOW	ci/parse_wheel_metadata.py	1	# Copyright 2026 The JAX Authors.	COMMENT
LOW	ci/run_bazel_test_tpu.sh	1	#!/bin/bash	COMMENT
LOW	ci/run_bazel_test_tpu.sh	261	//jaxlib/tools:check_tpu_wheel_sources_test \	COMMENT
LOW	ci/run_pytest_cpu.sh	1	#!/bin/bash	COMMENT
LOW	ci/run_pytest_rocm_multi.sh	1	#!/bin/bash	COMMENT
LOW	ci/run_bazel_test_oneapi.sh	1	#!/bin/bash	COMMENT
LOW	ci/run_bazel_test_oneapi.sh	21	# -e: abort script if one command fails	COMMENT
LOW	ci/utilities/install_wheels_locally.sh	1	#!/bin/bash	COMMENT
LOW	ci/utilities/rocm_test_env.sh	1	#!/bin/bash	COMMENT
LOW	ci/utilities/report_resultstore_link.py	1	# Copyright 2026 The JAX Authors.	COMMENT
LOW	ci/utilities/run_docker_container.sh	1	#!/bin/bash	COMMENT
LOW	ci/utilities/run_docker_container.sh	21	# Note: While GitHub action workflows use the same Docker images, they do not	COMMENT
LOW	ci/utilities/convert_msys_paths_to_win_paths.py	1	# Copyright 2024 The JAX Authors.	COMMENT
LOW	ci/utilities/set_artifact_tag_flags.sh	1	#!/bin/bash	COMMENT
LOW	ci/utilities/collect_bazel_test_xmls.sh	1	#!/bin/bash	COMMENT
LOW	ci/utilities/collect_bazel_test_xmls.sh	21	# bazel-testlogs/tests/cpu_tests/test.xml). Pytest, by contrast, writes a	COMMENT
LOW	ci/utilities/setup_build_environment.sh	1	#!/bin/bash	COMMENT
LOW	ci/utilities/run_auditwheel.sh	1	#!/bin/bash	COMMENT
LOW	ci/utilities/generate_invocation_id.py	1	# Copyright 2026 The JAX Authors.	COMMENT
LOW	ci/utilities/setup_portserver.sh	1	#!/bin/bash	COMMENT
LOW	ci/utilities/prepare_rocm_tests.sh	1	#!/bin/bash	COMMENT
LOW	ci/postprocess/process_test_results.sh	1	#!/bin/bash	COMMENT
LOW	ci/postprocess/xml2json.py	1	# Copyright 2026 The JAX Authors.	COMMENT
LOW	tests/random_test.py	1	# Copyright 2018 The JAX Authors.	COMMENT
LOW	tests/random_test.py	641	def test_seed_no_implicit_transfers(self, make_key):	COMMENT
LOW	tests/overlap_test.py	1	# Copyright 2026 The JAX Authors.	COMMENT
LOW	tests/extend_test.py	1	# Copyright 2023 The JAX Authors.	COMMENT
LOW	tests/svd_test.py	1	# Copyright 2022 The JAX Authors.	COMMENT
LOW	tests/scipy_spatial_test.py	1	# Copyright 2023 The JAX Authors.	COMMENT
LOW	tests/scheduling_groups_test.py	1	# Copyright 2025 The JAX Authors.	COMMENT
LOW	tests/jax_to_ir_test.py	1	# Copyright 2019 The JAX Authors.	COMMENT
LOW	tests/ragged_collective_test.py	1	# Copyright 2025 The JAX Authors.	COMMENT
LOW	tests/pytorch_interoperability_test.py	1	# Copyright 2020 The JAX Authors.	COMMENT
LOW	tests/fft_test.py	1	# Copyright 2019 The JAX Authors.	COMMENT
LOW	tests/export_back_compat_test.py	1	# Copyright 2023 The JAX Authors.	COMMENT
LOW	tests/export_back_compat_test.py	1041	# stablehlo.dynamic_rbg_bit_generator is used temporarily for a	COMMENT
LOW	tests/mesh_utils_test.py	1	# Copyright 2021 The JAX Authors. All Rights Reserved.	COMMENT
LOW	tests/lax_vmap_op_test.py	1	# Copyright 2020 The JAX Authors.	COMMENT
LOW	tests/debug_nans_test.py	1	# Copyright 2019 The JAX Authors.	COMMENT
LOW	tests/lobpcg_test.py	1	# Copyright 2022 The JAX Authors.	COMMENT
LOW	tests/heap_profiler_test.py	1	# Copyright 2021 The JAX Authors.	COMMENT
1478 more matches not shown…

Decorative Section Separators180 hits · 564 pts

Severity	File	Line	Snippet	Context
MEDIUM	ci/run_pytest_tpu.sh	15	# ==============================================================================	COMMENT
MEDIUM	ci/run_pytest_rocm.sh	15	# ==============================================================================	COMMENT
MEDIUM	ci/run_pytest_rocm.sh	31	# ==============================================================================	COMMENT
MEDIUM	ci/run_pytest_rocm.sh	33	# ==============================================================================	COMMENT
MEDIUM	ci/run_bazel_test_cuda_non_rbe.sh	15	# ==============================================================================	COMMENT
MEDIUM	ci/run_pytest_oneapi.sh	15	# ==============================================================================	COMMENT
MEDIUM	ci/run_pytest_oneapi.sh	70	# ==============================================================================	COMMENT
MEDIUM	ci/run_pytest_oneapi.sh	72	# ==============================================================================	COMMENT
MEDIUM	ci/run_pytest_oneapi.sh	84	# ==============================================================================	COMMENT
MEDIUM	ci/run_pytest_oneapi.sh	87	# ==============================================================================	COMMENT
MEDIUM	ci/run_pytest_oneapi.sh	146	# ==============================================================================	COMMENT
MEDIUM	ci/run_pytest_oneapi.sh	148	# ==============================================================================	COMMENT
MEDIUM	ci/run_bazel_cuda_targeted_tests.sh	15	# ==============================================================================	COMMENT
MEDIUM	ci/run_bazel_test_rocm_rbe.sh	15	# ==============================================================================	COMMENT
MEDIUM	ci/build_artifacts.sh	15	# ==============================================================================	COMMENT
MEDIUM	ci/build_rocm_artifacts.sh	15	# ==============================================================================	COMMENT
MEDIUM	ci/run_bazel_test_cpu_rbe.sh	15	# ==============================================================================	COMMENT
MEDIUM	ci/upload_rocm_logs.sh	15	# ==============================================================================	COMMENT
MEDIUM	ci/run_bazel_test_cuda_rbe.sh	15	# ==============================================================================	COMMENT
MEDIUM	ci/run_pytest_cuda.sh	15	# ==============================================================================	COMMENT
MEDIUM⚡	ci/run_pytest_cuda.sh	45	# ==============================================================================	COMMENT
MEDIUM⚡	ci/run_pytest_cuda.sh	47	# ==============================================================================	COMMENT
MEDIUM⚡	ci/run_pytest_cuda.sh	54	# ==============================================================================	COMMENT
MEDIUM⚡	ci/run_pytest_cuda.sh	57	# ==============================================================================	COMMENT
MEDIUM	ci/run_pytest_cuda.sh	107	# ==============================================================================	COMMENT
MEDIUM	ci/run_pytest_cuda.sh	109	# ==============================================================================	COMMENT
MEDIUM	ci/build_oneapi_artifacts.sh	15	# ==============================================================================	COMMENT
MEDIUM	ci/run_bazel_test_tpu.sh	15	# ==============================================================================	COMMENT
MEDIUM	ci/run_pytest_cpu.sh	15	# ==============================================================================	COMMENT
MEDIUM	ci/run_pytest_rocm_multi.sh	15	# ==============================================================================	COMMENT
MEDIUM	ci/run_bazel_test_oneapi.sh	15	# ==============================================================================	COMMENT
MEDIUM⚡	ci/run_bazel_test_oneapi.sh	38	# ==============================================================================	COMMENT
MEDIUM⚡	ci/run_bazel_test_oneapi.sh	40	# ==============================================================================	COMMENT
MEDIUM⚡	ci/run_bazel_test_oneapi.sh	50	# ==============================================================================	COMMENT
MEDIUM⚡	ci/run_bazel_test_oneapi.sh	52	# ==============================================================================	COMMENT
MEDIUM	ci/utilities/install_wheels_locally.sh	15	# ==============================================================================	COMMENT
MEDIUM⚡	ci/utilities/rocm_test_env.sh	15	# ==============================================================================	COMMENT
MEDIUM⚡	ci/utilities/rocm_test_env.sh	22	# ==============================================================================	COMMENT
MEDIUM⚡	ci/utilities/rocm_test_env.sh	24	# ==============================================================================	COMMENT
MEDIUM⚡	ci/utilities/rocm_test_env.sh	31	# ==============================================================================	COMMENT
MEDIUM⚡	ci/utilities/rocm_test_env.sh	33	# ==============================================================================	COMMENT
MEDIUM	ci/utilities/report_resultstore_link.py	14	# ==============================================================================	COMMENT
MEDIUM⚡	ci/utilities/run_docker_container.sh	15	# ==============================================================================	COMMENT
MEDIUM	ci/utilities/convert_msys_paths_to_win_paths.py	14	# ==============================================================================	COMMENT
MEDIUM	ci/utilities/set_artifact_tag_flags.sh	15	# ==============================================================================	COMMENT
MEDIUM	ci/utilities/collect_bazel_test_xmls.sh	15	# ==============================================================================	COMMENT
MEDIUM	ci/utilities/setup_build_environment.sh	15	# ==============================================================================	COMMENT
MEDIUM	ci/utilities/run_auditwheel.sh	15	# ==============================================================================	COMMENT
MEDIUM	ci/utilities/generate_invocation_id.py	14	# ==============================================================================	COMMENT
MEDIUM	ci/utilities/setup_portserver.sh	15	# ==============================================================================	COMMENT
MEDIUM	ci/utilities/prepare_rocm_tests.sh	15	# ==============================================================================	COMMENT
MEDIUM	ci/postprocess/process_test_results.sh	15	# ==============================================================================	COMMENT
MEDIUM	tests/mesh_utils_test.py	14	# ==============================================================================	COMMENT
MEDIUM⚡	tests/hijax_test.py	3095	# ------------	STRING
MEDIUM⚡	tests/hijax_test.py	3097	# ------------	STRING
MEDIUM⚡	tests/hijax_test.py	3132	#------------	STRING
MEDIUM⚡	tests/hijax_test.py	3134	#------------	STRING
MEDIUM	tests/mosaic/gpu_constraints_test.py	14	# ==============================================================================	COMMENT
MEDIUM	tests/mosaic/gpu_torch_distributed_test.py	14	# ==============================================================================	COMMENT
MEDIUM	tests/mosaic/gpu_dialect_test.py	14	# ==============================================================================	COMMENT
120 more matches not shown…

Deep Nesting510 hits · 394 pts

Severity	File	Line	Context
LOW	conftest.py	55	CODE
LOW	ci/postprocess/xml2json.py	70	CODE
LOW	tests/random_test.py	694	CODE
LOW	tests/export_back_compat_test.py	557	CODE
LOW	tests/lax_control_flow_test.py	2379	CODE
LOW	tests/xla_transform_test.py	366	CODE
LOW	tests/errors_test.py	46	CODE
LOW	tests/ffi_test.py	45	CODE
LOW	tests/ffi_test.py	161	CODE
LOW	tests/lax_numpy_einsum_test.py	295	CODE
LOW	tests/debug_info_test.py	119	CODE
LOW	tests/api_util_test.py	31	CODE
LOW	tests/dtypes_test.py	492	CODE
LOW	tests/fused_attention_stablehlo_test.py	126	CODE
LOW	tests/fused_attention_stablehlo_test.py	851	CODE
LOW	tests/lax_numpy_test.py	2716	CODE
LOW	tests/lax_numpy_test.py	7275	CODE
LOW	tests/lax_numpy_reducers_test.py	237	CODE
LOW	tests/linalg_test.py	723	CODE
LOW	tests/linalg_test.py	1973	CODE
LOW	tests/linalg_test.py	2075	CODE
LOW	tests/linalg_test.py	1976	CODE
LOW	tests/linalg_test.py	2083	CODE
LOW	tests/profiler_test.py	54	CODE
LOW	tests/profiler_test.py	132	CODE
LOW	tests/lax_numpy_indexing_test.py	1556	CODE
LOW	tests/pgle_test.py	135	CODE
LOW	tests/pgle_test.py	221	CODE
LOW	tests/pgle_test.py	367	CODE
LOW	tests/fused_attention_stablehlo_multigpu_test.py	128	CODE
LOW	tests/sparse_bcoo_bcsr_test.py	57	CODE
LOW	tests/sparse_bcoo_bcsr_test.py	101	CODE
LOW	tests/device_test.py	24	CODE
LOW	tests/device_test.py	46	CODE
LOW	tests/compilation_cache_test.py	577	CODE
LOW	tests/array_test.py	799	CODE
LOW	tests/hijax_test.py	2084	CODE
LOW	tests/hijax_test.py	2557	CODE
LOW	tests/transfer_guard_test.py	189	CODE
LOW	tests/transfer_guard_test.py	200	CODE
LOW	tests/api_test.py	5456	CODE
LOW	tests/api_test.py	5949	CODE
LOW	tests/api_test.py	5957	CODE
LOW	tests/documentation_coverage_test.py	142	CODE
LOW	tests/sparse_test.py	990	CODE
LOW	tests/lax_test.py	118	CODE
LOW	tests/lax_test.py	1698	CODE
LOW	tests/lax_test.py	4811	CODE
LOW	tests/mosaic/gpu_test.py	683	CODE
LOW	tests/mosaic/gpu_test.py	1171	CODE
LOW	tests/mosaic/gpu_test.py	6544	CODE
LOW	tests/mosaic/gpu_test.py	7421	CODE
LOW	tests/mosaic/gpu_test.py	8671	CODE
LOW	tests/multiprocess/thread_guard_test.py	42	CODE
LOW	tests/multiprocess/pgle_test.py	44	CODE
LOW	tests/numerics/numerics_test_util.py	114	CODE
LOW	tests/numerics/numerics_test_util.py	679	CODE
LOW	tests/numerics/numerics_test_util.py	984	CODE
LOW	tests/numerics/numerics_test_util.py	1138	CODE
LOW	tests/numerics/numerics_test_util.py	1310	CODE
450 more matches not shown…

Empty Error Handlers72 hits · 342 pts

Severity	File	Line	Snippet	Context
HIGH	tests/debug_nans_test.py	277	except FloatingPointError:	CODE
HIGH	tests/errors_test.py	440	except ValueError:	CODE
HIGH	tests/lax_numpy_test.py	7244	except TypeError:	CODE
HIGH	tests/core_test.py	169	except (TypeError, ValueError):	CODE
HIGH	tests/python_callback_test.py	943	except:	CODE
HIGH	tests/python_callback_test.py	596	except RuntimeError:	CODE
HIGH	tests/linalg_test.py	77	except np.linalg.LinAlgError:	CODE
HIGH	tests/profiler_session_test.py	40	except RuntimeError:	CODE
HIGH	…as/gpu_pallas_interpret_tcgen05_race_detection_test.py	68	except:	CODE
HIGH	tests/pallas/tpu_pallas_interpret_test.py	133	except:	CODE
HIGH	…sts/pallas/gpu_pallas_interpret_race_detection_test.py	61	except:	CODE
HIGH	…ts/pallas/gpu_pallas_interpret_cluster_barrier_test.py	53	except:	CODE
HIGH	tests/pallas/tpu_pallas_interpret_distributed_test.py	73	except:	CODE
HIGH	tests/pallas/gpu_pallas_interpret_test.py	60	except:	CODE
HIGH	tests/pallas/ops_test.py	108	except (FileNotFoundError, subprocess.TimeoutExpired, OSError):	CODE
HIGH	jax/experimental/mosaic/gpu/fragmented_array.py	4451	except TxMatrixIneligible:	CODE
HIGH	jax/experimental/mosaic/gpu/fragmented_array.py	4553	except TxMatrixIneligible:	CODE
HIGH	jax/experimental/mosaic/gpu/core.py	100	except ImportError:	CODE
HIGH	jax/experimental/mosaic/gpu/core.py	1242	except FileExistsError:	CODE
HIGH	jax/experimental/mosaic/gpu/layout_inference.py	2570	except Exception:	CODE
HIGH	jax/experimental/jax2tf/jax2tf.py	709	except TypeError:	CODE
HIGH	jax/experimental/colocated_python/func.py	670	except KeyError:	CODE
HIGH	jax/experimental/colocated_python/func.py	704	except AttributeError:	CODE
HIGH	jax/tools/evaluate_sampler_bias.py	74	except ValueError:	CODE
HIGH	jax/tools/evaluate_sampler_bias.py	78	except ValueError:	CODE
HIGH	jax/_src/lru_cache.py	27	except ImportError:	CODE
HIGH	jax/_src/config.py	2176	except KeyError:	CODE
HIGH	jax/_src/util.py	515	except Exception:	CODE
HIGH	jax/_src/core.py	2139	except TypeError:	CODE
HIGH	jax/_src/core.py	1702	except IndexError:	CODE
HIGH	jax/_src/core.py	3645	except: pass	CODE
HIGH	jax/_src/tpu_info.py	641	except (ValueError, RuntimeError):	CODE
HIGH	jax/_src/dtypes.py	44	except:	CODE
HIGH	jax/_src/test_util.py	647	except RuntimeError:	CODE
HIGH	jax/_src/api.py	2868	except Exception:	CODE
HIGH	jax/_src/pretty_printer.py	57	except NameError:	CODE
HIGH	jax/_src/ad_checkpoint.py	653	except:	CODE
HIGH	jax/_src/profiler.py	202	except RuntimeError:	CODE
HIGH	jax/_src/api_util.py	430	except ValueError:	CODE
HIGH	jax/_src/api_util.py	745	except (ValueError, TypeError):	CODE
HIGH	jax/_src/api_util.py	460	except ValueError: pass # That's not the issue.	CODE
HIGH	jax/_src/api_util.py	402	except ValueError:	CODE
HIGH	jax/_src/interpreters/pxla.py	440	except AttributeError: pass # TODO(mattjj): remove float0	CODE
HIGH	jax/_src/lax/lax.py	9703	except core.InconclusiveDimensionOperation:	CODE
HIGH	jax/_src/lax/lax.py	10413	except KeyError:	CODE
HIGH	jax/_src/lax/lax.py	1835	except OverflowError:	CODE
HIGH	jax/_src/lax/slicing.py	819	except TypeError: # func is not weak referenceable	CODE
HIGH	jax/_src/lax/linalg.py	2497	except core.InconclusiveDimensionOperation:	CODE
HIGH	jax/_src/lax/utils.py	286	except TypeError:	CODE
HIGH	jax/_src/lax/utils.py	290	except AttributeError:	CODE
HIGH	jax/_src/lax/control_flow/loops.py	1700	except core.ConcretizationTypeError:	CODE
HIGH	jax/_src/numpy/lax_numpy.py	1996	except AttributeError:	CODE
HIGH	jax/_src/numpy/lax_numpy.py	7737	except TypeError:	CODE
HIGH	jax/_src/numpy/lax_numpy.py	213	except (TypeError, AssertionError): # Unsupported dtype	CODE
HIGH	jax/_src/numpy/ufuncs.py	2731	except TypeError:	CODE
HIGH	jax/_src/numpy/indexing.py	1161	except (TypeError, ValueError, IndexError):	CODE
HIGH	jax/_src/numpy/indexing.py	1172	except (TypeError, ValueError, IndexError):	CODE
HIGH	jax/_src/pallas/pallas_call.py	866	except ImportError:	CODE
HIGH	jax/_src/pallas/pallas_call.py	879	except ImportError:	CODE
HIGH	jax/_src/pallas/mosaic/lowering.py	1897	except UncacheablePrimitiveError:	CODE
12 more matches not shown…

Self-Referential Comments73 hits · 202 pts

Severity	File	Line	Snippet	Context
MEDIUM	ci/utilities/run_docker_container.sh	50	# Create a temporary file to pass any user defined JAXCI_ / JAX_ / JAXLIB_	COMMENT
MEDIUM	ci/utilities/setup_build_environment.sh	105	# Create the output directory if it doesn't exist.	COMMENT
MEDIUM	tests/ragged_collective_test.py	769	# Define a mesh with PP + EP	COMMENT
MEDIUM	tests/lax_scipy_test.py	549	# This function is not defined for negative values, this makes sure they are nan	COMMENT
MEDIUM	tests/custom_api_test.py	2235	# Create the custom function	COMMENT
MEDIUM⚡	tests/garbage_collection_guard_test.py	54	# Create a reference cycle of two jax.Arrays.	COMMENT
MEDIUM⚡	tests/garbage_collection_guard_test.py	67	# Create a reference cycle of two jax.Arrays.	COMMENT
MEDIUM	tests/layout_test.py	365	# Create a custom layout instead of using `arr.layout` to test the API.	COMMENT
MEDIUM	tests/layout_test.py	388	# Create a custom layout instead of using `arr.layout` to test the API.	COMMENT
MEDIUM	tests/layout_test.py	669	# Create a custom layout instead of using `arr.layout` to test the API.	COMMENT
MEDIUM	tests/layout_test.py	701	# Create a custom layout instead of using `arr.layout` to test the API.	COMMENT
MEDIUM	tests/export_test.py	2513	# Create a custom layout instead of using `arr.layout` to test the API.	COMMENT
MEDIUM	tests/sparse_bcoo_bcsr_test.py	1153	# Create a matrix with duplicate indices	COMMENT
MEDIUM	tests/aot_test.py	346	# Create a compile-only topology, but DON'T switch to CPU so the real	COMMENT
MEDIUM	tests/array_test.py	686	# Create a few arrays	COMMENT
MEDIUM	tests/hijax_test.py	71	# Define a type	COMMENT
MEDIUM⚡	tests/api_test.py	2095	# Creating an array from per-device JAX arrays calls internal	COMMENT
MEDIUM	tests/mosaic/gpu_layout_inference_test.py	1678	# Create a var to use in the constraint system.	COMMENT
MEDIUM	tests/multiprocess/array_test.py	613	# Create an array that is non-addressable in processes besides `pid`.	COMMENT
MEDIUM	tests/multiprocess/array_test.py	635	# Create a PRNG key array that is non-addressable in processes besides	COMMENT
MEDIUM	tests/multiprocess/array_test.py	669	# Create a sharding that is non-addressable in processes besides `pid`.	COMMENT
MEDIUM	tests/multiprocess/array_test.py	696	# Create a sharding that is non-addressable in processes besides `pid`.	COMMENT
MEDIUM	tests/multiprocess/array_test.py	723	# Create a single device sharding for a device local to process `pid`.	COMMENT
MEDIUM	tests/pallas/tpu_pallas_interpret_distributed_test.py	85	# Create an input array that shards the last dimension across	COMMENT
MEDIUM	tests/pallas/tpu_pallas_interpret_distributed_test.py	162	# Create an input array that shards the first dimension across	COMMENT
MEDIUM	tests/pallas/mosaic_gpu_test.py	9106	# Create an index-invariant output.	COMMENT
MEDIUM	tests/pallas/pallas_test.py	2546	# Create a validity mask for OOB values.	COMMENT
MEDIUM⚡	docs/sphinxext/jax_list_config_options.py	38	# Create a field list item	COMMENT
MEDIUM⚡	docs/sphinxext/jax_list_config_options.py	41	# Create the field name (label)	COMMENT
MEDIUM⚡	docs/sphinxext/jax_list_config_options.py	46	# Create the field body (content)	COMMENT
MEDIUM⚡	docs/sphinxext/jax_list_config_options.py	81	# Create a section for this option	COMMENT
MEDIUM⚡	docs/sphinxext/jax_list_config_options.py	86	# Create a title with the option name (important for TOC)	COMMENT
MEDIUM⚡	docs/sphinxext/jax_list_config_options.py	92	# Create a field list for side-by-side display	COMMENT
MEDIUM	jaxlib/tools/build_wheel.py	226	# This file is required by PEP-561. It marks jaxlib as package containing	COMMENT
MEDIUM	.github/workflows/bazel_cuda_b200_mosaic_presubmit.yml	87	# Create an array with the file patterns and	COMMENT
MEDIUM	jax/version.py	15	# This file is included as part of both jax and jaxlib. It is also	COMMENT
MEDIUM	…ops/tpu/splash_attention/splash_attention_mask_info.py	587	# Create a collection of the unique head masks in the input multi-head mask.	COMMENT
MEDIUM	…llas/ops/tpu/splash_attention/splash_attention_mask.py	374	# Define the mask function for chunk attention	COMMENT
MEDIUM	jax/experimental/pallas/ops/tpu/megablox/gmm.py	180	# Create the group ids for each grid index based on the tile counts for each	COMMENT
MEDIUM	jax/experimental/pallas/ops/tpu/megablox/gmm.py	230	# Create the m-dimension tile ids for each grid index based on the visit	COMMENT
MEDIUM	jax/experimental/pallas/ops/tpu/megablox/gmm.py	383	# Create the metadata we need for computation.	COMMENT
MEDIUM	jax/experimental/pallas/ops/tpu/megablox/gmm.py	632	# Create the metadata we need for computation.	COMMENT
MEDIUM	jax/experimental/jax2tf/call_tf.py	220	# Define the fwd and bwd custom_vjp functions	COMMENT
MEDIUM	jax/experimental/jax2tf/tests/jax2tf_test.py	349	x = tf.Variable(4.0, dtype=tf.float32) # Create a Tensorflow variable initialized to 4.0	CODE
MEDIUM	jax/experimental/jax2tf/examples/mnist_lib.py	225	# Create the model and save it	COMMENT
MEDIUM	jax/_src/test_util.py	2033	# The following function methods operate on mpmath number instances.	COMMENT
MEDIUM	jax/_src/custom_batching.py	182	# Define a class, instead of making a function closing over `rule`, so	COMMENT
MEDIUM	jax/_src/ad_checkpoint.py	400	# This function is similar to api_util.argnums_partial, except the error	COMMENT
MEDIUM	jax/_src/ad_checkpoint.py	912	>>> # Define a function where we explicitly name an intermediate value	STRING
MEDIUM	jax/_src/ffi.py	710	# This method is kept to support the behavior that was previously exposed	COMMENT
MEDIUM	jax/_src/interpreters/partial_eval.py	2307	# This function is conceptually the same thing as just calling eval_jaxpr,	COMMENT
MEDIUM	jax/_src/interpreters/mlir.py	1465	# Create a keepalives list that will be mutated during the lowering.	COMMENT
MEDIUM	jax/_src/lax/utils.py	15	# This module contains utility functions split out of jax._src.lax.lax to	COMMENT
MEDIUM	jax/_src/lax/control_flow/loops.py	1432	# Create the staged eqn.	COMMENT
MEDIUM	jax/_src/pallas/mosaic/lowering.py	4493	# Create a scalar constant.	COMMENT
MEDIUM	jax/_src/pallas/mosaic/lowering.py	4496	# Create a vector constant.	COMMENT
MEDIUM	jax/_src/pallas/mosaic_gpu/lowering.py	614	# The below code emission relies on the assumption that the first scratch	COMMENT
MEDIUM	jax/_src/pallas/mosaic_gpu/interpret/shared_memory.py	1113	# This class is a bit messy and needs to use some hard-to-reason about state	COMMENT
MEDIUM	jax/_src/lib/__init__.py	15	# This module is largely a wrapper around `jaxlib` that performs version	COMMENT
MEDIUM	jax/_src/export/serialization_generated.py	109	"""This method is deprecated. Please switch to GetRootAs."""	STRING
13 more matches not shown…

Cross-File Repetition32 hits · 160 pts

Severity	File	Snippet	Context
HIGH	docs/401/cute-dsl.md	per-thread kernel: each thread computes relu of one element.	STRING
HIGH	docs/401/cute_dsl_jax/cute_dsl_jax_kernels.py	per-thread kernel: each thread computes relu of one element.	STRING
HIGH	docs/notebooks/cute_dsl_jax.md	per-thread kernel: each thread computes relu of one element.	STRING
HIGH	docs/401/cute-dsl.md	per-thread: out[i] = max(0, x[i] + bias[i % width]).	STRING
HIGH	docs/401/cute_dsl_jax/cute_dsl_jax_kernels.py	per-thread: out[i] = max(0, x[i] + bias[i % width]).	STRING
HIGH	docs/notebooks/cute_dsl_jax.md	per-thread: out[i] = max(0, x[i] + bias[i % width]).	STRING
HIGH	docs/401/cute-dsl.md	tiled gemm: each thread accumulates output elements.	STRING
HIGH	docs/401/cute_dsl_jax/cute_dsl_jax_kernels.py	tiled gemm: each thread accumulates output elements.	STRING
HIGH	docs/notebooks/cute_dsl_jax.md	tiled gemm: each thread accumulates output elements.	STRING
HIGH	docs/401/cute-dsl.md	per-thread kernel: 2-d element-wise add using flat indexing.	STRING
HIGH	docs/401/cute_dsl_jax/cute_dsl_jax_kernels.py	per-thread kernel: 2-d element-wise add using flat indexing.	STRING
HIGH	docs/notebooks/cute_dsl_jax.md	per-thread kernel: 2-d element-wise add using flat indexing.	STRING
HIGH	jaxlib/setup.py	this class makes 'bdist_wheel' include an abi tag on the wheel.	STRING
HIGH	jax_plugins/cuda/plugin_setup.py	this class makes 'bdist_wheel' include an abi tag on the wheel.	STRING
HIGH	jax_plugins/rocm/plugin_setup.py	this class makes 'bdist_wheel' include an abi tag on the wheel.	STRING
HIGH	jax_plugins/oneapi/plugin_setup.py	this class makes 'bdist_wheel' include an abi tag on the wheel.	STRING
HIGH	jaxlib/tools/build_gpu_plugin_wheel.py	[metadata] license_files = license.txt [bdist_wheel] plat_name={tag}	STRING
HIGH	jaxlib/tools/build_mosaic_wheel.py	[metadata] license_files = license.txt [bdist_wheel] plat_name={tag}	STRING
HIGH	jaxlib/tools/build_wheel.py	[metadata] license_files = license.txt [bdist_wheel] plat_name={tag}	STRING
HIGH	jaxlib/tools/build_gpu_kernels_wheel.py	[metadata] license_files = license.txt [bdist_wheel] plat_name={tag}	STRING
HIGH	…mental/jax2tf/tests/flax_models/transformer_nlp_seq.py	global hyperparameters used to minimize obnoxious kwarg plumbing.	STRING
HIGH	…erimental/jax2tf/tests/flax_models/transformer_lm1b.py	global hyperparameters used to minimize obnoxious kwarg plumbing.	STRING
HIGH	…perimental/jax2tf/tests/flax_models/transformer_wmt.py	global hyperparameters used to minimize obnoxious kwarg plumbing.	STRING
HIGH	…mental/jax2tf/tests/flax_models/transformer_nlp_seq.py	applies addpositionembs module. by default this layer uses a fixed sinusoidal embedding table. if a learned position emb	STRING
HIGH	…erimental/jax2tf/tests/flax_models/transformer_lm1b.py	applies addpositionembs module. by default this layer uses a fixed sinusoidal embedding table. if a learned position emb	STRING
HIGH	…perimental/jax2tf/tests/flax_models/transformer_wmt.py	applies addpositionembs module. by default this layer uses a fixed sinusoidal embedding table. if a learned position emb	STRING
HIGH	…mental/jax2tf/tests/flax_models/transformer_nlp_seq.py	transformer mlp / feed-forward block. args: config: transformerconfig dataclass containing hyperparameters. out_dim: opt	STRING
HIGH	…perimental/jax2tf/tests/flax_models/transformer_wmt.py	transformer mlp / feed-forward block. args: config: transformerconfig dataclass containing hyperparameters. out_dim: opt	STRING
HIGH	…erimental/jax2tf/tests/flax_models/transformer_lm1b.py	transformer mlp / feed-forward block. args: config: transformerconfig dataclass containing hyperparameters. out_dim: opt	STRING
HIGH	jax/_src/pallas/mosaic/interpret/shared_memory.py	synchronizes the vector clocks for the cores on the given device.	STRING
HIGH	…/_src/pallas/mosaic/interpret/interpret_pallas_call.py	synchronizes the vector clocks for the cores on the given device.	STRING
HIGH	jax/_src/pallas/mosaic_gpu/interpret/shared_memory.py	synchronizes the vector clocks for the cores on the given device.	STRING

AI Structural Patterns176 hits · 141 pts

Severity	File	Line	Context
LOW	tests/overlap_test.py	143	CODE
LOW	tests/overlap_test.py	201	CODE
LOW	tests/overlap_test.py	250	CODE
LOW	tests/overlap_test.py	336	CODE
LOW	tests/lax_control_flow_test.py	2877	CODE
LOW	tests/lax_control_flow_test.py	2882	CODE
LOW	tests/xla_transform_test.py	277	CODE
LOW	tests/scaled_matmul_stablehlo_test.py	429	CODE
LOW	tests/mutable_array_test.py	526	CODE
LOW	tests/custom_api_test.py	3599	CODE
LOW	tests/custom_api_test.py	1218	CODE
LOW	tests/custom_api_test.py	3202	CODE
LOW	tests/custom_api_test.py	5121	CODE
LOW	tests/dtypes_test.py	795	CODE
LOW	tests/fused_attention_stablehlo_test.py	77	CODE
LOW	tests/fused_attention_stablehlo_test.py	89	CODE
LOW	tests/fused_attention_stablehlo_test.py	126	CODE
LOW	tests/fused_attention_stablehlo_test.py	225	CODE
LOW	tests/shard_map_test.py	509	CODE
LOW	tests/shard_map_test.py	544	CODE
LOW	tests/shard_map_test.py	583	CODE
LOW	tests/shard_map_test.py	624	CODE
LOW	tests/shard_map_test.py	1112	CODE
LOW	tests/lax_numpy_reducers_test.py	352	CODE
LOW	tests/lax_numpy_reducers_test.py	589	CODE
LOW	tests/shape_poly_test.py	1327	CODE
LOW	tests/nn_test.py	64	CODE
LOW	tests/linalg_test.py	991	CODE
LOW	tests/state_test.py	1688	CODE
LOW	tests/image_test.py	74	CODE
LOW	tests/image_test.py	103	CODE
LOW	tests/memories_test.py	1320	CODE
LOW	tests/export_test.py	779	CODE
LOW	tests/export_test.py	782	CODE
LOW	tests/pmap_test.py	1544	CODE
LOW	tests/pjit_test.py	6299	CODE
LOW	tests/pjit_test.py	7947	CODE
LOW	tests/pjit_test.py	8618	CODE
LOW	tests/pjit_test.py	8841	CODE
LOW	tests/pjit_test.py	9041	CODE
LOW	tests/pjit_test.py	9147	CODE
LOW	tests/colocated_python_test.py	583	CODE
LOW	tests/scipy_ndimage_test.py	56	CODE
LOW	tests/fused_attention_stablehlo_multigpu_test.py	79	CODE
LOW	tests/fused_attention_stablehlo_multigpu_test.py	91	CODE
LOW	tests/fused_attention_stablehlo_multigpu_test.py	128	CODE
LOW	tests/fused_attention_stablehlo_multigpu_test.py	227	CODE
LOW	tests/checkify_test.py	1329	CODE
LOW	tests/numerics/numerics_test_util.py	1138	CODE
LOW	tests/numerics/numerics_test_util.py	1450	CODE
LOW	tests/pallas/tpu_sparsecore_pallas_test.py	3838	CODE
LOW	tests/pallas/tpu_pallas_test.py	6849	CODE
LOW	tests/pallas/tpu_pallas_test.py	6891	CODE
LOW	tests/pallas/gpu_paged_attention_test.py	84	CODE
LOW	tests/pallas/tpu_pallas_async_test.py	86	CODE
LOW	tests/pallas/tpu_pallas_async_test.py	139	CODE
LOW	tests/pallas/tpu_pallas_async_test.py	195	CODE
LOW	tests/pallas/tpu_pallas_async_test.py	635	CODE
LOW	tests/pallas/tpu_pallas_async_test.py	757	CODE
LOW	tests/pallas/tpu_pallas_pipeline_test.py	858	CODE
116 more matches not shown…

Cross-Language Confusion25 hits · 138 pts

Severity	File	Line	Snippet	Context
HIGH⚡	tests/stack_test.py	34	stack = stack.push(jnp.int32(7))	CODE
HIGH⚡	tests/stack_test.py	36	stack = stack.push(jnp.int32(8))	CODE
HIGH⚡	tests/stack_test.py	41	stack = stack.push(jnp.int32(9))	CODE
HIGH⚡	tests/export_test.py	1140	stack.push(self.assertRaisesRegex(Exception, re.escape(expect_error)))	CODE
HIGH	tests/export_test.py	957	stack.push(self.assertRaisesRegex(Exception, expect_error))	CODE
HIGH	tests/export_test.py	1063	stack.push(self.assertRaisesRegex(ValueError, expect_error_outer_exp))	CODE
HIGH	tests/export_test.py	1077	stack.push(self.assertRaisesRegex(Exception, expect_error_run))	CODE
HIGH	tests/api_test.py	2679	# is undefined, even though the transpose is well-defined.	COMMENT
HIGH	jax/experimental/jax2tf/tests/shape_poly_test.py	440	stack.push(self.assertRaisesRegex(Exception, re.escape(expect_error)))	CODE
HIGH	jax/experimental/colocated_python/func.py	363	func_backend.SINGLETON_RESULT_STORE.push(uid, result_leaves)	CODE
HIGH	jax/_src/jaxpr_util.py	648	while ((match = anchorRegex.exec(line)) !== null) {{	CODE
HIGH	jax/_src/jaxpr_util.py	693	const isCurrent = currentMatchIdx !== -1 && lineAbsoluteIdx === matchingLines[currentMatchIdx];	CODE
HIGH	jax/_src/jaxpr_util.py	726	let selectedElement = null;	CODE
HIGH	jax/_src/jaxpr_util.py	734	if (lineIdx !== undefined) {{	CODE
HIGH⚡	jax/_src/jaxpr_util.py	759	while (currentIdx !== null && currentIdx !== undefined) {{	CODE
HIGH⚡	jax/_src/jaxpr_util.py	759	while (currentIdx !== null && currentIdx !== undefined) {{	CODE
HIGH⚡	jax/_src/jaxpr_util.py	764	renderedFrames.push({{file: file, func: func, line: frame.line, col: frame.col}});	CODE
HIGH	jax/_src/jaxpr_util.py	825	matchingLines.push(idx);	CODE
HIGH	jax/_src/jaxpr_util.py	829	matchingLines.push(idx);	CODE
HIGH	jax/_src/jaxpr_util.py	958	if (offsetRight > 100 && offsetRight < document.body.clientWidth - 100) {{	CODE
HIGH	jax/_src/lax/lax.py	9858	b <= a, then the result is undefined, and different implementations may	STRING
HIGH	jax/_src/numpy/linalg.py	532	discontinuous at det=0 so standard autodiff is undefined; we use sign_dot=0	STRING
HIGH	jax/_src/tpu/linalg/eigh.py	326	agenda = agenda.push(_Subproblem(offset=jnp.array(0, np.int32), size=n))	CODE
HIGH	jax/_src/tpu/linalg/eigh.py	417	agenda.push(_Subproblem(offset, rank)),	CODE
HIGH	jax/_src/tpu/linalg/eigh.py	436	agenda.push(_Subproblem(offset + rank, (b - rank))),	CODE

Excessive Try-Catch Wrapping110 hits · 122 pts

Severity	File	Line	Snippet	Context
MEDIUM	ci/utilities/convert_msys_paths_to_win_paths.py	40	print("Error: cygpath not found. Make sure it's in your PATH.")	CODE
MEDIUM	ci/utilities/convert_msys_paths_to_win_paths.py	43	print(f"Error converting path: {e}")	CODE
LOW	ci/postprocess/xml2json.py	87	except Exception as e:	CODE
LOW	tests/array_interoperability_test.py	181	except Exception as e:	CODE
LOW	tests/debug_info_test.py	111	except Exception as e:	CODE
LOW	tests/shape_poly_test.py	2089	except Exception:	CODE
LOW	tests/profiler_test.py	63	except Exception as e:	CODE
LOW	tests/transfer_guard_test.py	116	except Exception as e:	CODE
LOW	tests/transfer_guard_test.py	135	except Exception as e:	CODE
LOW⚡	tests/transfer_guard_test.py	251	except Exception as e:	CODE
LOW⚡	tests/transfer_guard_test.py	259	except Exception as e:	CODE
MEDIUM	tests/transfer_guard_test.py	246	def config_writer():	CODE
MEDIUM	tests/transfer_guard_test.py	254	def reader():	CODE
LOW	tests/api_test.py	5537	except Exception as e:	CODE
LOW	tests/pallas/tpu_ops_test.py	2048	except Exception:	CODE
LOW	tests/pallas/gpu_pallas_distributed_test.py	877	except Exception:	CODE
LOW	tests/pallas/gpu_pallas_distributed_test.py	931	except Exception:	CODE
LOW	tests/pallas/gpu_pallas_distributed_test.py	1660	except Exception:	CODE
LOW	tests/pallas/ops_test.py	777	except Exception as e:	CODE
LOW	tests/pallas/ops_test.py	905	except Exception as e:	CODE
LOW	tests/pallas/ops_test.py	2959	except Exception as e:	CODE
LOW	tests/thunky/collective_matmul_test.py	493	except Exception as e:	CODE
LOW	docs/array_refs.py	84	except Exception as e:	CODE
LOW⚡	docs/array_refs.py	225	except Exception as e:	CODE
LOW⚡	docs/array_refs.py	234	except Exception as e:	CODE
LOW⚡	docs/array_refs.py	243	except Exception as e:	CODE
LOW⚡	docs/array_refs.py	252	except Exception as e:	CODE
LOW	docs/array_refs.py	292	except Exception as e:	CODE
LOW	docs/array_refs.py	416	except Exception as e:	CODE
LOW	docs/conf.py	114	except Exception as e:	CODE
LOW	docs/parallel.md	427	except Exception as e:	CODE
LOW	docs/parallel.md	466	except Exception as e:	CODE
LOW	docs/array_refs.md	89	except Exception as e:	CODE
LOW⚡	docs/array_refs.md	225	except Exception as e:	CODE
LOW⚡	docs/array_refs.md	234	except Exception as e:	CODE
LOW⚡	docs/array_refs.md	243	except Exception as e:	CODE
LOW⚡	docs/array_refs.md	252	except Exception as e:	CODE
LOW	docs/array_refs.md	291	except Exception as e:	CODE
LOW	docs/array_refs.md	408	except Exception as e:	CODE
LOW	docs/parallel.py	381	except Exception as e:	CODE
LOW	docs/parallel.py	419	except Exception as e:	CODE
LOW	docs/301/refs.md	144	except Exception as e:	CODE
LOW	docs/sphinxext/source_include.py	72	except Exception as e:	CODE
MEDIUM	docs/sphinxext/source_include.py	40	def get_tagged_block(filepath, tag, lines_spec=None):	CODE
LOW	…atic/fault_tolerance/data_parallelism_with_recovery.py	171	except Exception as e:	CODE
LOW	docs/_static/fault_tolerance/live_devices.py	56	except Exception as e:	CODE
LOW	docs/_static/fault_tolerance/data_parallelism.py	118	except Exception as e:	CODE
MEDIUM	docs/pallas/tpu/sparse.md	148	print("Error \|result - lax.dynamic_slice\| =", diff)	CODE
LOW	docs/201/shard-map.md	439	except Exception as e:	CODE
LOW	docs/201/shard-map.md	513	except Exception as e:	CODE
LOW	docs/201/sharding.md	411	except Exception as e:	CODE
LOW	docs/201/sharding.md	453	except Exception as e:	CODE
LOW	benchmarks/pallas/mgpu_planar_snake_bench.py	75	except Exception as e:	CODE
LOW	jax/__init__.py	28	except Exception as exc:	CODE
LOW	jax/experimental/multihost_utils.py	679	except Exception as e:	CODE
LOW⚡	jax/experimental/mosaic/gpu/layout_inference.py	2570	except Exception:	CODE
LOW	jax/experimental/array_serialization/serialization.py	238	except Exception as e:	CODE
MEDIUM	jax/experimental/array_serialization/serialization.py	197	def _thread_func(self):	CODE
LOW	jax/experimental/jax2tf/call_tf.py	295	except Exception as e:	CODE
LOW	jax/experimental/jax2tf/call_tf.py	592	except Exception as e:	CODE
50 more matches not shown…

AI Slop Vocabulary47 hits · 114 pts

Severity	File	Line	Snippet	Context
MEDIUM⚡	ci/utilities/run_docker_container.sh	22	# run this script as they leverage built-in containerization features to run	COMMENT
MEDIUM	tests/export_back_compat_test.py	264	# Compute the inputs to simplify the harness	COMMENT
MEDIUM	tests/lax_numpy_operators_test.py	295	# probably add a custom test harness for unwrap that tests the period	COMMENT
LOW	tests/custom_api_test.py	383	# useful either: instead of using nondiff_argnums here, a user can just pass	COMMENT
MEDIUM	tests/lax_numpy_test.py	3562	# permissive canonicalization logic in the test harness.	COMMENT
MEDIUM	tests/export_harnesses_multi_platform_test.py	60	# If you want to run this test for only one harness, add parameter	COMMENT
MEDIUM	tests/shape_poly_test.py	3724	# Pick the dtype with the most harnesses in this group. Some harness	COMMENT
MEDIUM	tests/shape_poly_test.py	3796	# If you want to run this test for only one harness that includes "foo"	COMMENT
MEDIUM	tests/shape_poly_test.py	3853	# Update this here rather than in harness object because vmap_random_gamma is derived	COMMENT
LOW	tests/shape_poly_test.py	1910	# It is not sufficient to just use the shape of an input; it is still unused	COMMENT
MEDIUM	tests/linalg_test.py	948	# This expresses identity function, which makes us robust to, e.g., the	COMMENT
MEDIUM	tests/lax_autodiff_test.py	163	# TODO(mattjj): make some-equal checks more robust, enable second-order	COMMENT
MEDIUM	tests/sparsify_test.py	341	# Note: more comprehensive tests in sparse_test.py:test_bcoo_squeeze	COMMENT
MEDIUM	tests/sparsify_test.py	354	# Note: more comprehensive tests in sparse_test.py:test_bcoo_rev	COMMENT
MEDIUM	tests/sparsify_test.py	419	# Note: more comprehensive tests in sparse_test.py:test_bcoo_conv_general_dilated	COMMENT
LOW	tests/api_test.py	457	# We can't just use `lambda x: x` because JAX simplifies this away to an	COMMENT
LOW	tests/mosaic/gpu_layout_inference_test.py	99	# TMEM reference, so we can just return a trivial mapping.	COMMENT
LOW	docs/array_refs.py	117	# `jax.ref.swap`, but usually you'd just use NumPy-style array indexing syntax:	COMMENT
LOW	docs/hijax_custom_derivatives.py	1080	# that a primitive with a `jvp` rule can just set `lin, linearized =	COMMENT
LOW	docs/autodidax.py	1475	# call `bind`, in the primitive wrapper we can just use `make_jaxpr` to form	COMMENT
MEDIUM	…ops/tpu/splash_attention/splash_attention_mask_info.py	892	# maintain the SPMD paradigm.	COMMENT
MEDIUM	…erimental/jax2tf/tests/jax_primitives_coverage_test.py	50	# If you want to run this test for only one harness, add parameter	COMMENT
MEDIUM	…erimental/jax2tf/tests/jax_primitives_coverage_test.py	90	# f"{[u.description for u in jax_unimpl]} in harness: {harness.fullname}"))	COMMENT
MEDIUM	jax/experimental/jax2tf/tests/shape_poly_test.py	223	# Makes and tests a harness. See PolyHarness documentation.	COMMENT
LOW	jax/experimental/jax2tf/tests/shape_poly_test.py	587	# It is not sufficient to just use the shape of an input; it is still unused	COMMENT
MEDIUM	jax/experimental/jax2tf/tests/model_harness.py	77	"""Partially apply harness in order to create variables lazily.	STRING
MEDIUM	jax/experimental/jax2tf/tests/tf_test_util.py	221	# Run JAX. Should not fail, we assume that the harness has been filtered	COMMENT
MEDIUM	jax/experimental/jax2tf/tests/primitives_test.py	91	# If you want to run this test for only one harness, add parameter	COMMENT
LOW	jax/_src/compiler.py	541	# simply return the non-PGLE profiled module from the persistent cache if it	COMMENT
LOW	jax/_src/pjit.py	1014	# checks and just return the pjit_in_shardings directly. `shard_args` will	COMMENT
LOW	jax/_src/checkify.py	359	# Default non-HOP case: just call primitive and don't update error.	COMMENT
LOW	jax/_src/interpreters/batching.py	632	# if there's only agreeing batch dims and scalars, just call the primitive	COMMENT
LOW	jax/_src/lax/ann.py	384	# 1. ApproxTopK is internally a variadic reduce, so we can simply call	COMMENT
MEDIUM	jax/_src/lax/control_flow/loops.py	2178	# lowering. Fundamentally, we'd like to rewrite a while loop that looks like	COMMENT
MEDIUM	jax/_src/numpy/linalg.py	1494	# TODO: add custom jvp rule for more robust lstsq differentiation	COMMENT
LOW	jax/_src/pallas/pallas_call.py	668	# dimensions. For now, we just use 0.	COMMENT
LOW	jax/_src/pallas/mosaic/lowering.py	1786	# `jax.empty_ref`), but lowering expects them to exist---so we just return	COMMENT
MEDIUM	…/_src/pallas/mosaic/interpret/interpret_pallas_call.py	1710	# TODO(nrink): It would be more robust if the buffer id, i.e. `src`,	COMMENT
LOW	…/_src/pallas/mosaic/interpret/interpret_pallas_call.py	684	# callback to `get`. Should we just pass the shape to `get`?	COMMENT
MEDIUM	jax/_src/internal_test_util/test_harnesses.py	148	# Descriptive name of the harness, used as a testcase_name. Unique in a group.	COMMENT
MEDIUM	jax/_src/internal_test_util/test_harnesses.py	157	# partially implemented in JAX for this harness.	COMMENT
MEDIUM	jax/_src/internal_test_util/test_harnesses.py	386	# Change the testcase name to include the harness name.	COMMENT
MEDIUM	jax/_src/internal_test_util/test_harnesses.py	2005	# This first harness runs the tests for all dtypes using default values for	COMMENT
MEDIUM	jax/_src/internal_test_util/test_harnesses.py	2591	# This first harness runs the tests for all dtypes using default values for	COMMENT
MEDIUM	jax/_src/internal_test_util/test_harnesses.py	2896	# This first harness runs the tests for all dtypes and precisions using	COMMENT
MEDIUM	jax/_src/internal_test_util/test_harnesses.py	3097	# This first harness runs the tests for all dtypes and precisions using	COMMENT
MEDIUM	jax/_src/internal_test_util/test_harnesses.py	351	self.enabled = enabled # Does it apply to the current harness?	CODE

Dead Code55 hits · 104 pts

Severity	File	Line	Context
MEDIUM	tests/custom_api_test.py	391	CODE
MEDIUM	tests/custom_api_test.py	393	CODE
MEDIUM	tests/custom_api_test.py	396	CODE
MEDIUM	tests/custom_api_test.py	399	CODE
MEDIUM	tests/custom_api_test.py	402	CODE
MEDIUM	tests/custom_api_test.py	403	CODE
MEDIUM	tests/custom_api_test.py	404	CODE
MEDIUM	tests/custom_api_test.py	1949	CODE
MEDIUM	tests/custom_api_test.py	1961	CODE
MEDIUM	tests/custom_api_test.py	1964	CODE
MEDIUM	tests/custom_api_test.py	1965	CODE
MEDIUM	tests/custom_api_test.py	1966	CODE
MEDIUM	tests/custom_api_test.py	1968	CODE
MEDIUM	tests/custom_api_test.py	1969	CODE
MEDIUM	tests/custom_api_test.py	1970	CODE
MEDIUM	tests/lax_numpy_reducers_test.py	984	CODE
MEDIUM	tests/lax_numpy_reducers_test.py	985	CODE
MEDIUM	tests/lax_numpy_reducers_test.py	986	CODE
MEDIUM	tests/lax_numpy_reducers_test.py	996	CODE
MEDIUM	tests/lax_numpy_reducers_test.py	997	CODE
MEDIUM	tests/lax_numpy_reducers_test.py	998	CODE
MEDIUM	tests/pmap_test.py	2184	CODE
MEDIUM	tests/pmap_test.py	2186	CODE
MEDIUM	tests/pmap_test.py	2192	CODE
MEDIUM	tests/pmap_test.py	2202	CODE
MEDIUM	tests/pmap_test.py	2206	CODE
MEDIUM	tests/pmap_test.py	2221	CODE
MEDIUM	tests/pmap_test.py	2227	CODE
MEDIUM	tests/pmap_test.py	2228	CODE
MEDIUM	tests/pmap_test.py	2229	CODE
MEDIUM	tests/pmap_test.py	2230	CODE
MEDIUM	tests/pmap_test.py	2241	CODE
MEDIUM	tests/pmap_test.py	2247	CODE
MEDIUM	tests/pmap_test.py	2248	CODE
MEDIUM	tests/pmap_test.py	2249	CODE
MEDIUM	tests/pmap_test.py	2250	CODE
MEDIUM	tests/pmap_test.py	2256	CODE
MEDIUM	tests/pmap_test.py	2262	CODE
MEDIUM	tests/pmap_test.py	2412	CODE
MEDIUM	tests/pmap_test.py	2414	CODE
MEDIUM	tests/pmap_test.py	2417	CODE
MEDIUM	tests/pmap_test.py	2418	CODE
MEDIUM	tests/pmap_test.py	2420	CODE
MEDIUM	tests/pmap_test.py	2421	CODE
MEDIUM	tests/pmap_test.py	2422	CODE
MEDIUM	tests/pmap_test.py	2423	CODE
MEDIUM	tests/pmap_test.py	2435	CODE
MEDIUM	tests/lax_test.py	1860	CODE
MEDIUM	tests/lax_test.py	1892	CODE
MEDIUM	tests/mosaic/gpu_test.py	971	CODE
MEDIUM	tests/mosaic/gpu_test.py	973	CODE
MEDIUM	tests/mosaic/gpu_test.py	974	CODE
MEDIUM	tests/mosaic/gpu_test.py	975	CODE
MEDIUM	tests/mosaic/gpu_test.py	982	CODE
MEDIUM	jax/_src/pallas/core.py	1626	CODE

Modern Structural Boilerplate63 hits · 63 pts

Severity	File	Line	Snippet	Context
LOW	docs/sphinxext/jax_list_config_options.py	22	logger = logging.getLogger(__name__)	CODE
LOW	jaxlib/xla_client.py	61	logger = logging.getLogger(__name__)	CODE
LOW	jax/export.py	14	__all__ = ["DisabledSafetyCheck", "Exported", "export", "deserialize",	CODE
LOW	jax/debug.py	14	__all__ = ["callback", "print", "log", "DebugEffect", "OrderedDebugEffect",	CODE
LOW	jax/ref.py	15	__all__ = [	CODE
LOW	jax/interpreters/xla.py	15	__all__ = ["apply_primitive", "canonicalize_dtype_handlers", "Backend"]	CODE
LOW	jax/experimental/mosaic/gpu/utils.py	44	logger = logging.getLogger(__name__)	CODE
LOW	jax/experimental/mosaic/gpu/layout_inference.py	48	logger = logging.getLogger(__name__)	CODE
LOW	jax/experimental/roofline/roofline.py	41	logger = logging.getLogger(__name__)	CODE
LOW	…perimental/array_serialization/pytree_serialization.py	44	logger = logging.getLogger(__name__)	CODE
LOW	…perimental/array_serialization/pytree_serialization.py	61	__all__ = ["save", "load", "load_pytreedef",	CODE
LOW	…x/experimental/array_serialization/tensorstore_impl.py	53	logger = logging.getLogger(__name__)	CODE
LOW	…ntal/array_serialization/pytree_serialization_utils.py	32	logger = logging.getLogger(__name__)	CODE
LOW	…ntal/array_serialization/pytree_serialization_utils.py	44	__all__ = ["serialize_pytreedef", "deserialize_pytreedef",	CODE
LOW	jax/experimental/array_serialization/serialization.py	74	logger = logging.getLogger(__name__)	CODE
LOW	jax/experimental/jax2tf/jax2tf.py	66	logger = logging.getLogger(__name__)	CODE
LOW	jax/experimental/jax2tf/call_tf.py	55	logger = logging.getLogger(__name__)	CODE
LOW	jax/experimental/colocated_python/obj.py	50	def update_devices(self, uid: int, device_set: set[jax.Device]) -> None:	CODE
LOW	jax/_src/callback.py	47	logger = logging.getLogger(__name__)	CODE
LOW	jax/_src/lru_cache.py	33	logger = logging.getLogger(__name__)	CODE
LOW	jax/_src/xla_bridge.py	48	logger = logging.getLogger(__name__)	CODE
LOW	jax/_src/test_loader.py	41	logger = logging.getLogger(__name__)	CODE
LOW	jax/_src/config.py	37	logger = logging.getLogger(__name__)	CODE
LOW	jax/_src/logging_config.py	62	def update_logging_level_global(logging_level: str \| None) -> None:	CODE
LOW	jax/_src/compiler.py	98	logger = logging.getLogger(__name__)	CODE
LOW	jax/_src/cache_key.py	34	logger = logging.getLogger(__name__)	CODE
LOW	jax/_src/util.py	37	logger = logging.getLogger(__name__)	CODE
LOW	jax/_src/cloud_tpu_init.py	23	logger = logging.getLogger(__name__)	CODE
LOW	jax/_src/dispatch.py	75	logger = logging.getLogger(__name__)	CODE
LOW	jax/_src/debugging.py	56	logger = logging.getLogger(__name__)	CODE
LOW	jax/_src/test_util.py	75	logger = logging.getLogger(__name__)	CODE
LOW	jax/_src/compilation_cache.py	51	logger = logging.getLogger(__name__)	CODE
LOW	jax/_src/compilation_cache.py	193	def set_cache_dir(path) -> None:	CODE
LOW	jax/_src/error_check.py	129	def set_error_if(pred: Array, /, msg: str) -> None:	CODE
LOW	jax/_src/distributed.py	28	logger = logging.getLogger(__name__)	CODE
LOW	jax/_src/public_test_util.py	30	__all__ = ['check_grads', 'check_jvp', 'check_vjp']	CODE
LOW	jax/_src/ad_checkpoint.py	61	logger = logging.getLogger(__name__)	CODE
LOW	jax/_src/jaxpr_util.py	44	logger = logging.getLogger(__name__)	CODE
LOW	jax/_src/hardware_utils.py	99	logger = logging.getLogger(__name__)	CODE
LOW	jax/_src/profiler.py	48	logger = logging.getLogger(__name__)	CODE
LOW	jax/_src/profiler.py	137	def set_metadata(key: str, value: str) -> None:	CODE
LOW	jax/_src/path.py	20	__all__ = ["Path"]	CODE
LOW	jax/_src/path.py	22	logger = logging.getLogger(__name__)	CODE
LOW	jax/_src/interpreters/pxla.py	73	logger = logging.getLogger(__name__)	CODE
LOW	jax/_src/interpreters/partial_eval.py	65	logger = logging.getLogger(__name__)	CODE
LOW	jax/_src/lax/fft.py	34	__all__ = [	CODE
LOW	jax/_src/clusters/k8s_cluster.py	30	logger = logging.getLogger(__name__)	CODE
LOW	jax/_src/clusters/cloud_tpu_cluster.py	25	logger = logging.getLogger(__name__)	CODE
LOW	jax/_src/clusters/cluster.py	22	logger = logging.getLogger(__name__)	CODE
LOW	jax/_src/state/primitives.py	1275	if not isinstance(aval, AbstractLinVal): raise TypeError(aval)	CODE
LOW	jax/_src/state/types.py	641	def _setitem(self, tracer, idx, value) -> None:	CODE
LOW	jax/_src/numpy/array_constructors.py	36	logger = logging.getLogger(__name__)	CODE
LOW	jax/_src/numpy/index_tricks.py	37	__all__ = ["c_", "index_exp", "mgrid", "ogrid", "r_", "s_"]	CODE
LOW	jax/_src/numpy/array_methods.py	22	__all__ = ['register_jax_array_methods']	CODE
LOW	jax/_src/pallas/mosaic/interpret/shared_memory.py	34	logger = logging.getLogger(__name__)	CODE
LOW	…x/_src/pallas/mosaic/interpret/race_detection_state.py	25	logger = logging.getLogger(__name__)	CODE
LOW	jax/_src/pallas/triton/lowering.py	1366	def _set_attr(v: ir.Value, name: str, attr: ir.Attribute) -> None:	CODE
LOW	jax/_src/pallas/mosaic_gpu/interpret/shared_memory.py	38	logger = logging.getLogger(__name__)	CODE
LOW	…rc/pallas/mosaic_gpu/interpret/race_detection_state.py	24	logger = logging.getLogger(__name__)	CODE
LOW	jax/_src/export/_export.py	61	logger = logging.getLogger(__name__)	CODE
3 more matches not shown…

Hallucination Indicators5 hits · 60 pts

Severity	File	Line	Snippet	Context
CRITICAL	tests/mosaic/profiler_cupti_test.py	183	jax._src.lib.mosaic_gpu._mosaic_gpu_ext._cupti_init()	CODE
CRITICAL	tests/filecheck/jax_mlir_ext.filecheck.py	114	return jax._src.lib._jax.Traceback.get_traceback()	CODE
CRITICAL⚡	tests/thunky/thunky_test.py	675	asm_text = jax._src.lib.mlir.ir.StringAttr(	CODE
CRITICAL⚡	tests/thunky/thunky_test.py	678	binary = jax._src.lib.mlir.ir.StringAttr(	CODE
CRITICAL	jax/_src/pallas/mosaic_gpu/lowering.py	4109	for k, v in ctx.launch_ctx.module.operation.attributes.items():	CODE

Redundant / Tautological Comments27 hits · 40 pts

Severity	File	Line	Snippet	Context
LOW	tests/array_interoperability_test.py	210	# Check if the source device is preserved	COMMENT
LOW	tests/nn_test.py	120	# Check if float8_e8m0fnu is available	COMMENT
LOW	tests/nn_test.py	337	# Check if cudnn backend is called (only on CUDA).	COMMENT
LOW	tests/pgle_test.py	277	# Check if FDO profile file of the biggest module is not empty	COMMENT
LOW	tests/pgle_test.py	285	# Check if FDO profile file in dump directory is not empty	COMMENT
LOW	tests/multiprocess/array_test.py	297	# Check if we can specify that local input actually contains full-span	COMMENT
LOW	…allas/tpu_fusible_matmul_with_stateful_fusions_test.py	144	# Check if either input fusion reads from a Ref that is also written by the	COMMENT
LOW	tests/pallas/tpu_pallas_call_print_test.py	170	# Check if the numbers in the output match the values generated by `arange`.	COMMENT
LOW	.github/workflows/upload_metadata.yml	68	# Check if zip file is empty before unzipping	COMMENT
LOW	.github/workflows/verify-squash.yml	32	# Check if the skip label is present	COMMENT
LOW	.github/workflows/verify-squash.yml	55	# Check if squashed	COMMENT
LOW	.github/workflows/build_artifacts.yml	175	# Set shell to cmd to avoid path errors when using gcloud commands on Windows	COMMENT
LOW	jax/experimental/mosaic/gpu/fragmented_array.py	5894	# Check if input maps exactly to the end (prevents trailing dims).	COMMENT
LOW	jax/_src/test_util.py	1269	# Check if strict hermeticity is already satisfied	COMMENT
LOW	jax/_src/checkify.py	915	# Check if the first cond application will error.	COMMENT
LOW	jax/_src/pretty_printer.py	48	# Check if we're in IPython or Colab	COMMENT
LOW	jax/_src/lax/control_flow/loops.py	2354	# Check if the same Ref is written to in both cond and body.	COMMENT
LOW	jax/_src/state/primitives.py	810	# Check if start is static (which it can be)	COMMENT
LOW	jax/_src/state/primitives.py	823	# Check if we are indexing with a scalar or not. If we are indexing	COMMENT
LOW	jax/_src/numpy/setops.py	177	# Set mask to zero at locations corresponding to unique() padding.	COMMENT
LOW	jax/_src/numpy/array_constructors.py	239	# Check if object supports any of the data exchange protocols	COMMENT
LOW	jax/_src/pallas/mpmd.py	578	# Check if any batched argument reaches an unbatchable op.	COMMENT
LOW	jax/_src/pallas/core.py	1261	elif grid and isinstance(grid[0], tuple): # Check if we have a named grid	CODE
LOW	jax/_src/pallas/einshape.py	243	# Check if all RHS dims are known	COMMENT
LOW	jax/_src/scipy/signal.py	669	# Check if we can broadcast the outer axes together	COMMENT
LOW	jax/_src/scipy/signal.py	718	# Check if x and y are the same length, zero-pad if necessary	COMMENT
LOW	jax/_src/cudnn/fused_attention_stablehlo.py	1429	# Check if all required keys are present	COMMENT

Fake / Example Data21 hits · 30 pts

Severity	File	Line	Snippet	Context
LOW⚡	tests/export_back_compat_test.py	96	dummy_data = self.load_testdata(bctu.dummy_data_dict)	CODE
LOW⚡	tests/export_back_compat_test.py	98	dummy_data, platform=self.default_jax_backend())	CODE
LOW⚡	tests/export_back_compat_test.py	103	dummy_data = self.load_testdata(bctu.dummy_data_dict)	CODE
LOW⚡	tests/export_back_compat_test.py	105	dummy_data,	CODE
LOW⚡	tests/export_back_compat_test.py	113	dummy_data = self.load_testdata(bctu.dummy_data_dict)	CODE
LOW⚡	tests/export_back_compat_test.py	115	dummy_data,	CODE
LOW⚡	tests/shard_map_test.py	648	dummy_data = jax.lax.precv(	CODE
LOW⚡	tests/shard_map_test.py	656	dummy_data,	CODE
LOW⚡	tests/shard_map_test.py	660	return dummy_data	CODE
LOW⚡	docs/gpu_performance_tips.md	400	dummy_data = jnp.zeros(	CODE
LOW⚡	docs/gpu_performance_tips.md	403	dummy_data = jax.device_put(	CODE
LOW⚡	docs/gpu_performance_tips.md	404	dummy_data,	CODE
LOW⚡	docs/gpu_performance_tips.md	411	carry = weights, input_buffer, output_buffer, dummy_data, dummy_data	CODE
LOW⚡	docs/gpu_performance_tips.md	590	weights, input_buffer, dummy_data, mesh	CODE
LOW⚡	docs/gpu_performance_tips.md	599	jax.ShapeDtypeStruct.like(dummy_data),	CODE
LOW⚡	docs/gpu_performance_tips.md	609	dummy_data,	CODE
LOW⚡	docs/gpu_performance_tips.md	610	dummy_data,	CODE
LOW	docs/gpu_performance_tips.md	677	dummy_data = jax.device_put(	CODE
LOW	docs/gpu_performance_tips.md	692	)(weights, input_buffer, dummy_data)	CODE
LOW	…perimental/array_serialization/pytree_serialization.py	237	dummy_key_path = os.path.join(kvstore_path, "dummy_key")	CODE
LOW	…src/internal_test_util/export_back_compat_test_util.py	128	# The dummy_data is used for getting started for adding a new test and for	COMMENT

Slop Phrases13 hits · 26 pts

Severity	File	Line	Snippet	Context
MEDIUM	docs/array_refs.py	612	# wouldn't be for `vmap` or `shard_map`, as discussed above). Once all the	COMMENT
MEDIUM	docs/parallel.py	398	# want the compiler to choose them automatically, you can use the `@auto_axes`	COMMENT
MEDIUM	docs/ffi.py	163	# // this handler in a header, you can use the `XLA_FFI_DECLARE_HANDLER_SYMBOL`	COMMENT
MEDIUM	docs/ffi.py	211	# If you're familiar with the legacy "custom call" API, it's worth noting that you can also use {func}`~jax.ffi.register	COMMENT
MEDIUM	docs/ffi.py	211	# If you're familiar with the legacy "custom call" API, it's worth noting that you can also use {func}`~jax.ffi.register	COMMENT
MEDIUM	jax/experimental/jax2tf/examples/saved_model_lib.py	102	# names, you can use `tree.map_structure_with_path` from the `dm-tree` package	COMMENT
LOW	jax/_src/core.py	2940	# but we make sure to reset it to Device because the Ref owns the memory space	COMMENT
MEDIUM	jax/_src/ad_checkpoint.py	274	Here is a simple example:	STRING
MEDIUM	jax/_src/errors.py	201	indices, such as with :code:`.at[...].set(...)`. Here is a simple example::	STRING
MEDIUM	jax/_src/errors.py	666	Here is a simple example of code that would lead to such an error::	STRING
MEDIUM	jax/_src/lax/windowed_reductions.py	154	Here is a simple example of a windowed product over pairs in a 1-dimensional array:	STRING
MEDIUM	jax/_src/lax/slicing.py	371	For example, here is how you can extract values at particular indices using	STRING
LOW	jax/_src/pallas/core.py	812	# If you hit this, make sure you take transforms into account and use either	COMMENT

Structural Annotation Overuse22 hits · 21 pts

Severity	File	Line	Snippet	Context
LOW	tests/thunky/collective_matmul_test.py	91	# Step 0: compute local GEMM 0 (A_d @ B_d) on compute stream 0 concurrently	COMMENT
LOW	tests/thunky/collective_matmul_test.py	237	# Step 0: compute local GEMM 0 on compute stream 0	COMMENT
LOW	docs/distributed_data_loading.md	232	# Step 1: setup the Dataset for pure data parallelism (do once)	COMMENT
LOW	docs/distributed_data_loading.md	241	# Step 2: create a jax.Array of per-replica batches from the per-process batch	COMMENT
LOW	docs/distributed_data_loading.md	327	# Step 1: Set up the Dataset with a different data shard per-process (do once)	COMMENT
LOW	docs/distributed_data_loading.md	337	# Step 2: Create a jax.Array of per-replica batches from the per-process batch	COMMENT
LOW	docs/thunky.md	413	# Step 1: Launch custom Pallas Mosaic GPU TMA kernel into intermediate_ref	STRING
LOW	docs/thunky.md	415	# Step 2: Post-process intermediate_ref in declarative mode	STRING
LOW	docs/thunky.md	547	### Step 1: Unidirectional Ring Collective Matmul	STRING
LOW	docs/thunky.md	607	# Step 0: compute local GEMM 0 (A_d @ B_d) on compute stream 0 concurrently	STRING
LOW	docs/thunky.md	649	### Step 2: Bidirectional Ring with Grouped Collectives & Fused GEMMs	STRING
LOW	docs/thunky.md	760	# Step 0: compute local GEMM 0 on compute stream 0	STRING
LOW	docs/501/data-loading.md	217	# Step 1: set up the Dataset for pure data parallelism (do once)	COMMENT
LOW	docs/501/data-loading.md	226	# Step 2: create a jax.Array of per-replica batches from the per-process batch	COMMENT
LOW	docs/501/data-loading.md	312	# Step 1: Set up the Dataset with a different data shard per process (do once)	COMMENT
LOW	docs/501/data-loading.md	322	# Step 2: Create a jax.Array of per-replica batches from the per-process batch	COMMENT
LOW⚡	jax/_src/numpy/ufuncs.py	4151	# Step 1: Exact range reduction x = n + r with n in Z and r in [-0.5, 0.5].	STRING
LOW⚡	jax/_src/numpy/ufuncs.py	4158	# Step 2: Taylor coefficients c_k = (-1)^k * pi^(2k) / (2k+1)! (low to high)	STRING
LOW⚡	jax/_src/numpy/ufuncs.py	4194	# Step 3: Evaluate Q(t) via lax.polynomial:	STRING
LOW⚡	jax/_src/numpy/ufuncs.py	4202	# Step 4: Divide by (-1)^n * x (using 1 when n == 0 to avoid 0/0 in autodiff).	STRING
LOW	jax/_src/numpy/linalg.py	774	# Step 1: Pivot column 0 so row 0 gets the largest magnitude element.	COMMENT
LOW	jax/_src/numpy/linalg.py	792	# Step 2: Pivot column 1 across rows 1 and 2, then eliminate row 2.	COMMENT

Synthetic Comment Markers3 hits · 20 pts

Severity	File	Line	Snippet	Context
HIGH	docs/contributing.md	26	## Can I contribute AI generated code?	COMMENT
HIGH	docs/contributing.md	35	You are responsible for any code you contribute to JAX, regardless of whether it was written manually or generated by AI	CODE
HIGH	jaxlib/py_executable.h	286	// Python objects to keep alive as requested by user.	COMMENT

AI Response Leakage2 hits · 15 pts

Severity	File	Line	Snippet	Context
HIGH	docs/ffi.py	135	# // In this example, we treat all leading dimensions as batch dimensions, so this	COMMENT
HIGH	docs/ffi.py	178	# In this implementation, we treat all leading dimensions of the buffer as batch dimensions, and perform RMS normalizati	COMMENT

Overly Generic Function Names10 hits · 10 pts

Severity	File	Line	Snippet	Context
LOW⚡	tests/core_test.py	587	def my_function():	CODE
LOW	tests/util_test.py	80	def my_function():	CODE
LOW	tests/aot_test.py	131	def my_function(x):	CODE
LOW	tests/aot_test.py	152	def my_function(x):	CODE
LOW	tests/source_info_test.py	279	def my_function(x):	CODE
LOW⚡	tests/api_test.py	96	def my_function():	CODE
LOW⚡	tests/api_test.py	110	def my_function():	CODE
LOW	tests/api_test.py	1634	def my_function(x, flag):	CODE
LOW	jax/typing.py	47	def my_function(x: ArrayLike) -> Array:	STRING
LOW	jax/experimental/jax2tf/tests/jax2tf_test.py	306	def test_function(self):	CODE

Verbosity Indicators9 hits · 8 pts

Severity	File	Line	Snippet	Context
LOW	tests/thunky/collective_matmul_test.py	91	# Step 0: compute local GEMM 0 (A_d @ B_d) on compute stream 0 concurrently	COMMENT
LOW	tests/thunky/collective_matmul_test.py	237	# Step 0: compute local GEMM 0 on compute stream 0	COMMENT
LOW	jaxlib/python_ref_manager.h	85	// The purpose of this function is to amortize lock acquisition costs over	COMMENT
LOW⚡	jax/_src/numpy/ufuncs.py	4151	# Step 1: Exact range reduction x = n + r with n in Z and r in [-0.5, 0.5].	STRING
LOW⚡	jax/_src/numpy/ufuncs.py	4158	# Step 2: Taylor coefficients c_k = (-1)^k * pi^(2k) / (2k+1)! (low to high)	STRING
LOW⚡	jax/_src/numpy/ufuncs.py	4194	# Step 3: Evaluate Q(t) via lax.polynomial:	STRING
LOW⚡	jax/_src/numpy/ufuncs.py	4202	# Step 4: Divide by (-1)^n * x (using 1 when n == 0 to avoid 0/0 in autodiff).	STRING
LOW	jax/_src/numpy/linalg.py	774	# Step 1: Pivot column 0 so row 0 gets the largest magnitude element.	COMMENT
LOW	jax/_src/numpy/linalg.py	792	# Step 2: Pivot column 1 across rows 1 and 2, then eliminate row 2.	COMMENT

Example Usage Blocks2 hits · 4 pts

Severity	File	Line	Snippet	Context
LOW⚡	ci/utilities/run_docker_container.sh	24	# Usage:	COMMENT
LOW	jax/_src/deprecations.py	22	# Example usage:	COMMENT

Modern AI Meta-Vocabulary1 hit · 3 pts

Severity	File	Line	Snippet	Context
MEDIUM	.github/workflows/wheel_tests_continuous.yml	5	# It orchestrates the following:	COMMENT

TODO Padding2 hits · 3 pts

Severity	File	Line	Snippet	Context
LOW	jax/_src/lax/slicing.py	822	# TODO: implement this via its own primitive so we can define appropriate autodiff rules.	COMMENT
LOW	jax/_src/lax/linalg.py	2371	# TODO: implement full matrices case, documented here: https://people.maths.ox.ac.uk/gilesm/files/NA-08-01.pdf	COMMENT

Maintenance and Governance Review

Risk Review · v3

What These Metrics Mean

Pattern-Signal History

Severity Breakdown

Directory Score Breakdown

Pattern Findings