jax-ml/jax

18.9

Adjusted Score

18.9

Raw Score

100%

Time Factor

2026-07-14

Last Push

36.0K

Stars

Python

Language

669.7K

Lines of Code

1.4K

Files

12.6K

Pattern Hits

2026-07-14

Scan Date

0.45

HC Hit Rate

What These Metrics Mean

Adjusted Score: Primary synthetic code indicator. Raw score normalised per 1,000 lines of code and multiplied by the temporal discount factor. This is the definitive comparative metric — use it to rank repositories by AI authorship density.
Raw Score: The unmodified sum of all severity-weighted, context-multiplied pattern match scores before temporal discounting. Reflects the absolute signal strength independent of when the repository was last active.
Time Factor: The temporal discount multiplier (0–100%) applied to the raw score. Repositories last updated before ChatGPT's launch (Nov 2022) receive a 5% factor. Full signal is only assigned to repositories active in the post-adoption era (Jan 2024+).
Pattern Hits: Total count of individual pattern matches across all files and categories. A high hit count with a low score may indicate a very large codebase with isolated AI snippets; a low count with a high score indicates dense, concentrated AI signatures.
HC Hit Rate: High+Critical pattern hits per file, averaged across the repository. This orthogonal signal catches repositories where a few files are densely packed with high-severity AI tells — a strong indicator even when the normalised score appears moderate due to codebase size.
Lines of Code / Files: Total lines and files analysed. The scanner examines 94 file extensions. These denominators are used to normalise the score, enabling fair comparison between repositories of vastly different sizes.

Score History

This chart maps the temporal evolution of the adjusted synthetic code score across successive scan runs. An upward trajectory indicates ongoing incorporation of AI-generated code or expanding LLM-assisted scaffolding; a stable or declining trajectory may reflect active human refactoring, code removal, or the adoption of stricter authorship policies. The dashed secondary line (right axis) independently tracks total raw pattern hit count, which can diverge from the normalised score when codebase size changes significantly between scans.

Severity Breakdown

Classifies detected patterns by their diagnostic confidence and structural impact. CRITICAL patterns (coefficient 10) represent definitive synthetic signatures — hallucinated imports, explicit LLM attribution metadata — virtually never produced by human authors. HIGH (5) indicates strong structural tells such as cross-file repetition or cross-linguistic idioms. MEDIUM (2) covers recognisable conversational padding and AI-specific vocabulary. LOW (1) captures subtle indicators like tautological comments and generic boilerplate that require density to carry independent signal.

CRITICAL 3HIGH 628MEDIUM 309LOW 11655

Directory Score Breakdown

This horizontal bar chart decomposes the repository's raw synthetic code score by top-level directory, allowing you to pinpoint precisely which modules or components carry the highest AI authorship density. Directories with disproportionately high scores relative to their size warrant targeted manual review: concentrated AI signatures often trace back to mass-generated configuration layers, auto-ported test suites, LLM-scaffolded boilerplate classes, or entire subsystems authored under heavy copilot assistance. Use this view to prioritise your human code-review effort.

Pattern Findings

The scanner identified 12595 distinct pattern matches across 26 syntactic categories. Each entry below represents a discrete location in the source code where the engine recorded a statistically significant AI authorship indicator. 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 diagnostic confidence level (CRITICAL / HIGH / MEDIUM / LOW). The Context column identifies whether the match occurred inside executable code, an inline comment, or a string literal — comment-context matches receive a ×1.5 weight because LLMs systematically over-annotate. The ⚡ bolt icon marks clustered matches: three or more patterns within a 10-line window, each receiving an additional ×1.5 density multiplier as dense clusters constitute far stronger evidence of synthetic authorship than isolated hits.

Hyper-Verbose Identifiers5799 hits · 4489 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	569	def test_key_construction_with_dtype(self, dtype_spec):	CODE
LOW⚡	tests/random_test.py	573	def test_key_construction_with_both_impl_and_dtype(self):	CODE
LOW⚡	tests/random_test.py	583	def test_wrap_key_data_with_dtype(self, dtype_spec):	CODE
LOW⚡	tests/random_test.py	591	def test_wrap_key_data_with_both_impl_and_dtype(self):	CODE
LOW⚡	tests/random_test.py	1215	def test_key_make_like_other_key_via_dtype(self, prng_name):	CODE
LOW⚡	tests/random_test.py	1223	def test_key_wrap_like_other_key_via_dtype(self, prng_name):	CODE
LOW⚡	tests/random_test.py	1231	def test_key_impl_from_string_error(self):	CODE
LOW⚡	tests/random_test.py	1235	def test_key_impl_from_object_error(self):	CODE
LOW⚡	tests/random_test.py	1242	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	445	def test_threefry_split_fold_in_symmetry(self, make_key):	CODE
LOW	tests/random_test.py	458	def test_threefry_split_vmapped_fold_in_symmetry(self, make_key):	CODE
LOW	tests/random_test.py	472	def test_loggamma_nan_corner_case(self):	CODE
LOW	tests/random_test.py	545	def test_default_prng_selection(self, make_key, name, impl):	CODE
LOW	tests/random_test.py	557	def test_key_construction_with_explicit_impl_name(self, make_key, name, impl):	CODE
LOW	tests/random_test.py	609	def test_legacy_prng_key_flag(self):	CODE
LOW	tests/random_test.py	637	def test_seed_no_implicit_transfers(self, make_key):	CODE
LOW	tests/random_test.py	713	def test_construction_upgrade_flag(self):	CODE
LOW	tests/random_test.py	759	def test_key_dtype_attributes(self):	CODE
LOW	tests/random_test.py	814	def test_cpp_dispatch_aot_normal(self):	CODE
LOW	tests/random_test.py	827	def test_cpp_dispatch_aot_split(self):	CODE
LOW	tests/random_test.py	905	def test_eval_shape_keys_in_out(self):	CODE
LOW	tests/random_test.py	963	def test_dynamic_update_slice(self):	CODE
LOW	tests/random_test.py	1053	def test_device_put_replicated(self):	CODE
LOW	tests/random_test.py	1059	def test_make_array_from_callback(self):	CODE
LOW	tests/random_test.py	1071	def test_make_array_from_single_device_arrays(self):	CODE
LOW	tests/random_test.py	1081	def test_key_array_custom_jvp(self):	CODE
LOW	tests/random_test.py	1111	def test_key_array_indexing_nd(self):	CODE
LOW	tests/random_test.py	1136	def test_array_impl_attributes(self):	CODE
LOW	tests/random_test.py	1194	def test_key_make_like_other_key(self, prng_name):	CODE
LOW	tests/random_test.py	1204	def test_key_wrap_like_other_key(self, prng_name):	CODE
LOW	tests/random_test.py	1262	def test_keyarray_custom_vjp_symbolic_zeros(self):	CODE
LOW	tests/random_test.py	1277	def test_keyarray_array_conversion_fails(self):	CODE
LOW	tests/random_test.py	1301	def _double_threefry_random_bits(key, bit_width, shape):	CODE
LOW	tests/random_test.py	1565	def test_full_like_with_key_fillvalue(self):	CODE
LOW	tests/random_test.py	1586	def test_full_with_key_fillvalue(self):	CODE
LOW⚡	tests/extend_test.py	84	def test_key_make_with_custom_impl(self):	CODE
LOW⚡	tests/extend_test.py	92	def test_key_wrap_with_custom_impl(self):	CODE
LOW⚡	tests/extend_test.py	102	def test_key_make_with_custom_impl_via_dtype(self):	CODE
LOW⚡	tests/extend_test.py	109	def test_key_wrap_with_custom_impl_via_dtype(self):	CODE
LOW⚡	tests/extend_test.py	118	def test_key_dtype_and_spec_with_custom_impl(self):	CODE
LOW	tests/extend_test.py	130	def test_unknown_platform_error(self):	CODE
LOW	tests/extend_test.py	139	def test_hlo_sharding_roundtrip(self):	CODE
LOW	tests/scheduling_groups_test.py	67	def test_xla_metadata_call_inlineable(self):	CODE
LOW	tests/scheduling_groups_test.py	88	def test_xla_metadata_call_inlineable_remat_in_scan(self):	CODE
LOW	tests/scheduling_groups_test.py	105	def test_xla_metadata_call_deduplication(self):	CODE
LOW	tests/scheduling_groups_test.py	133	def test_xla_metadata_call_deduplication_remat(self):	CODE
LOW	tests/scheduling_groups_test.py	162	def test_xla_metadata_call_deduplication_kwargs(self):	CODE
LOW	tests/jax_to_ir_test.py	65	def test_jax_to_hlo_with_constants(self):	CODE
LOW	tests/jax_to_ir_test.py	85	def test_parse_shape_str_invalid(self):	CODE
LOW	tests/ragged_collective_test.py	134	def test_ragged_all_to_all_grad(self, axis_name, mesh_axes):	CODE
LOW	tests/ragged_collective_test.py	210	def test_ragged_all_to_all_axis_index_groups(self, axis_name, mesh_axes):	CODE
LOW	tests/ragged_collective_test.py	304	def test_ragged_all_to_all_degenerate_groups(self, axis_name, mesh_axes):	CODE
LOW	tests/ragged_collective_test.py	387	def test_ragged_all_to_all_vmap_multi_dim_operand(self):	CODE
LOW	tests/ragged_collective_test.py	483	def test_ragged_all_to_all_vmap(	CODE
LOW	tests/ragged_collective_test.py	584	def test_ragged_all_to_all_vmap_unsupported_axis_index_groups(self):	CODE
LOW	tests/ragged_collective_test.py	642	def test_ragged_all_to_all_errors(self):	CODE
5739 more matches not shown…

Unused Imports3518 hits · 2479 pts

Severity	File	Line	Context
LOW	tests/random_test.py	15	CODE
LOW	tests/xla_transform_test.py	17	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	53	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	26	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	61	CODE
LOW	docs/autodidax.py	2156	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	jax/sharding.py	18	CODE
LOW	jax/sharding.py	19	CODE
LOW	jax/sharding.py	19	CODE
LOW	jax/sharding.py	19	CODE
LOW	jax/sharding.py	19	CODE
3458 more matches not shown…

Docstring Block Structure567 hits · 2308 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	137	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	269	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	376	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	2967	Invalidate a given reference and return its final value. For more information about mutable array references, refer t	STRING
HIGH	jax/_src/dtypes.py	882	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	1130	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	215	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	2304	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	2384	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	2453	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	2602	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	233	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	338	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	690	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	836	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	953	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	1007	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	1193	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	711	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	846	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	1046	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	94	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	789	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	870	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	187	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	236	Convenience wrapper for defining JVPs for each argument separately. This convenience wrapper cannot be used togethe	STRING
HIGH	jax/_src/custom_derivatives.py	621	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	259	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	291	Translate a Python ``val`` to a sequence of IR constants. See https://docs.jax.dev/en/latest/internals/constants.html	STRING
507 more matches not shown…

Over-Commented Block1484 hits · 1410 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_bazel_test_cuda_non_rbe.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	221	//tests/pallas:tpu_pallas_call_print_test_tpu \	COMMENT
LOW	ci/run_pytest_cpu.sh	1	#!/bin/bash	COMMENT
LOW	ci/utilities/install_wheels_locally.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/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/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		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
LOW	tests/custom_root_test.py	1	# Copyright 2018 The JAX Authors.	COMMENT
LOW	tests/mosaic_test.py	1	# Copyright 2023 The JAX Authors.	COMMENT
LOW	tests/lax_control_flow_test.py	1	# Copyright 2018 The JAX Authors.	COMMENT
LOW	tests/lax_control_flow_test.py	2221		COMMENT
LOW	tests/array_interoperability_test.py	1	# Copyright 2020 The JAX Authors.	COMMENT
LOW	tests/xla_transform_test.py	1	# Copyright 2026 The JAX Authors.	COMMENT
LOW	tests/lax_scipy_test.py	1	# Copyright 2018 The JAX Authors.	COMMENT
LOW	tests/errors_test.py	1	# Copyright 2020 The JAX Authors.	COMMENT
LOW	tests/distributed_initialize_test.py	1	# Copyright 2025 The JAX Authors.	COMMENT
1424 more matches not shown…

Decorative Section Separators129 hits · 387 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	52	# ==============================================================================	COMMENT
MEDIUM⚡	ci/run_pytest_rocm.sh	54	# ==============================================================================	COMMENT
MEDIUM⚡	ci/run_pytest_rocm.sh	61	# ==============================================================================	COMMENT
MEDIUM⚡	ci/run_pytest_rocm.sh	64	# ==============================================================================	COMMENT
MEDIUM	ci/run_pytest_rocm.sh	117	# ==============================================================================	COMMENT
MEDIUM	ci/run_pytest_rocm.sh	119	# ==============================================================================	COMMENT
MEDIUM	ci/run_bazel_test_cuda_non_rbe.sh	15	# ==============================================================================	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/utilities/install_wheels_locally.sh	15	# ==============================================================================	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/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/postprocess/process_test_results.sh	15	# ==============================================================================	COMMENT
MEDIUM	tests/mesh_utils_test.py	14	# ==============================================================================	COMMENT
MEDIUM⚡	tests/hijax_test.py	2530	# ------------	COMMENT
MEDIUM⚡	tests/hijax_test.py	2532	# ------------	COMMENT
MEDIUM⚡	tests/hijax_test.py	2603	#------------	COMMENT
MEDIUM⚡	tests/hijax_test.py	2605	#------------	COMMENT
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
MEDIUM	tests/mosaic/gpu_torch_test.py	14	# ==============================================================================	COMMENT
MEDIUM	tests/mosaic/profiler_cupti_test.py	14	# ==============================================================================	COMMENT
MEDIUM	tests/mosaic/flash_attention_test.py	14	# ==============================================================================	COMMENT
MEDIUM	tests/mosaic/gpu_multidevice_test.py	14	# ==============================================================================	COMMENT
MEDIUM	tests/mosaic/gpu_layout_inference_test.py	14	# ==============================================================================	COMMENT
MEDIUM	tests/mosaic/matmul_test.py	14	# ==============================================================================	COMMENT
MEDIUM	tests/mosaic/gpu_test.py	14	# ==============================================================================	COMMENT
MEDIUM	tests/mosaic/gpu_distributed_test.py	14	# ==============================================================================	COMMENT
MEDIUM	tests/pallas/mgpu_ragged_dot_test.py	14	# ==============================================================================	COMMENT
MEDIUM	tests/pallas/mgpu_torch_test.py	14	# ==============================================================================	COMMENT
MEDIUM	tests/pallas/mgpu_examples_test.py	14	# ==============================================================================	COMMENT
MEDIUM	tests/pallas/mgpu_attention_test.py	14	# ==============================================================================	COMMENT
MEDIUM	tests/pallas/mgpu_collective_matmul_test.py	14	# ==============================================================================	COMMENT
MEDIUM	tests/pallas/mgpu_matmul_test.py	14	# ==============================================================================	COMMENT
MEDIUM	docs/notebooks/cute_dsl_jax/cute_dsl_jax_kernels.py	284	# ── Vector Add ────────────────────────────────────────────────────	COMMENT
MEDIUM	docs/notebooks/cute_dsl_jax/cute_dsl_jax_kernels.py	297	# ── SAXPY ─────────────────────────────────────────────────────────	COMMENT
69 more matches not shown…

Deep Nesting452 hits · 355 pts

Severity	File	Line	Context
LOW	conftest.py	55	CODE
LOW	ci/postprocess/xml2json.py	70	CODE
LOW	tests/random_test.py	690	CODE
LOW	tests/export_back_compat_test.py	557	CODE
LOW	tests/lax_control_flow_test.py	2376	CODE
LOW	tests/xla_transform_test.py	326	CODE
LOW	tests/errors_test.py	44	CODE
LOW	tests/ffi_test.py	45	CODE
LOW	tests/lax_numpy_einsum_test.py	295	CODE
LOW	tests/debug_info_test.py	119	CODE
LOW	tests/api_util_test.py	28	CODE
LOW	tests/dtypes_test.py	494	CODE
LOW	tests/fused_attention_stablehlo_test.py	121	CODE
LOW	tests/lax_numpy_test.py	2588	CODE
LOW	tests/lax_numpy_test.py	6768	CODE
LOW	tests/lax_numpy_reducers_test.py	164	CODE
LOW	tests/linalg_test.py	708	CODE
LOW	tests/linalg_test.py	1786	CODE
LOW	tests/linalg_test.py	1888	CODE
LOW	tests/linalg_test.py	1789	CODE
LOW	tests/linalg_test.py	1896	CODE
LOW	tests/profiler_test.py	54	CODE
LOW	tests/profiler_test.py	133	CODE
LOW	tests/lax_numpy_indexing_test.py	1491	CODE
LOW	tests/export_test.py	2395	CODE
LOW	tests/control_deps_test.py	50	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	122	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/transfer_guard_test.py	190	CODE
LOW	tests/transfer_guard_test.py	201	CODE
LOW	tests/api_test.py	5001	CODE
LOW	tests/api_test.py	5322	CODE
LOW	tests/api_test.py	5330	CODE
LOW	tests/documentation_coverage_test.py	141	CODE
LOW	tests/sparse_test.py	1006	CODE
LOW	tests/lax_test.py	118	CODE
LOW	tests/lax_test.py	1613	CODE
LOW	tests/lax_test.py	4588	CODE
LOW	tests/mosaic/gpu_test.py	682	CODE
LOW	tests/mosaic/gpu_test.py	1169	CODE
LOW	tests/mosaic/gpu_test.py	5464	CODE
LOW	tests/mosaic/gpu_test.py	6168	CODE
LOW	tests/mosaic/gpu_test.py	7390	CODE
LOW	tests/multiprocess/thread_guard_test.py	42	CODE
LOW	tests/multiprocess/pgle_test.py	44	CODE
LOW	tests/pallas/tpu_pallas_interpret_test.py	313	CODE
LOW	tests/pallas/tpu_pallas_interpret_test.py	409	CODE
LOW	tests/pallas/triton_pallas_test.py	150	CODE
LOW	…allas/tpu_fusible_matmul_with_stateful_fusions_test.py	84	CODE
LOW	…ts/pallas/gpu_pallas_interpret_cluster_barrier_test.py	375	CODE
LOW	…ts/pallas/gpu_pallas_interpret_cluster_barrier_test.py	380	CODE
LOW	tests/pallas/einshape_test.py	60	CODE
LOW	tests/pallas/gpu_pallas_interpret_test.py	639	CODE
392 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	772	# Define a mesh with PP + EP	COMMENT
MEDIUM	tests/lax_scipy_test.py	546	# This function is not defined for negative values, this makes sure they are nan	COMMENT
MEDIUM	tests/custom_api_test.py	2129	# 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	361	# Create a custom layout instead of using `arr.layout` to test the API.	COMMENT
MEDIUM	tests/layout_test.py	384	# Create a custom layout instead of using `arr.layout` to test the API.	COMMENT
MEDIUM	tests/layout_test.py	665	# Create a custom layout instead of using `arr.layout` to test the API.	COMMENT
MEDIUM	tests/layout_test.py	697	# Create a custom layout instead of using `arr.layout` to test the API.	COMMENT
MEDIUM	tests/export_test.py	2486	# 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	337	# Create a compile-only topology, but DON'T switch to CPU so the real	COMMENT
MEDIUM	tests/array_test.py	667	# Create a few arrays	COMMENT
MEDIUM	tests/hijax_test.py	64	# Define a type	COMMENT
MEDIUM⚡	tests/api_test.py	2043	# Creating an array from per-device JAX arrays calls internal	COMMENT
MEDIUM	tests/mosaic/gpu_layout_inference_test.py	1527	# 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	63	# Create an input array that shards the last dimension across	COMMENT
MEDIUM	tests/pallas/tpu_pallas_interpret_distributed_test.py	149	# Create an input array that shards the first dimension across	COMMENT
MEDIUM	tests/pallas/mosaic_gpu_test.py	7525	# Create an index-invariant output.	COMMENT
MEDIUM	tests/pallas/pallas_test.py	2539	# 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	2071	# The following function methods operate on mpmath number instances.	COMMENT
MEDIUM	jax/_src/custom_batching.py	178	# Define a class, instead of making a function closing over `rule`, so	COMMENT
MEDIUM	jax/_src/ad_checkpoint.py	411	# This function is similar to api_util.argnums_partial, except the error	COMMENT
MEDIUM	jax/_src/ad_checkpoint.py	953	>>> # 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	253	# Create the input tracers for the staged-out (unknown-value) call.	COMMENT
MEDIUM	jax/_src/interpreters/partial_eval.py	2433	# This function is conceptually the same thing as just calling eval_jaxpr,	COMMENT
MEDIUM	jax/_src/interpreters/mlir.py	1391	# 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	1359	# Create the staged eqn.	COMMENT
MEDIUM	jax/_src/pallas/mosaic/lowering.py	3887	# Create a scalar constant.	COMMENT
MEDIUM	jax/_src/pallas/mosaic/lowering.py	3890	# Create a vector constant.	COMMENT
MEDIUM	jax/_src/pallas/mosaic_gpu/lowering.py	619	# The below code emission relies on the assumption that the first scratch	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/new_docs/401/cute-dsl.md	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/notebooks/cute_dsl_jax/cute_dsl_jax_kernels.py	per-thread kernel: each thread computes relu of one element.	STRING
HIGH	docs/new_docs/401/cute-dsl.md	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/notebooks/cute_dsl_jax/cute_dsl_jax_kernels.py	per-thread: out[i] = max(0, x[i] + bias[i % width]).	STRING
HIGH	docs/new_docs/401/cute-dsl.md	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/notebooks/cute_dsl_jax/cute_dsl_jax_kernels.py	tiled gemm: each thread accumulates output elements.	STRING
HIGH	docs/new_docs/401/cute-dsl.md	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	docs/notebooks/cute_dsl_jax/cute_dsl_jax_kernels.py	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 Patterns161 hits · 132 pts

Severity	File	Line	Context
LOW	tests/lax_control_flow_test.py	2874	CODE
LOW	tests/lax_control_flow_test.py	2879	CODE
LOW	tests/xla_transform_test.py	277	CODE
LOW	tests/scaled_matmul_stablehlo_test.py	233	CODE
LOW	tests/mutable_array_test.py	429	CODE
LOW	tests/custom_api_test.py	3220	CODE
LOW	tests/custom_api_test.py	1187	CODE
LOW	tests/custom_api_test.py	2843	CODE
LOW	tests/dtypes_test.py	797	CODE
LOW	tests/fused_attention_stablehlo_test.py	76	CODE
LOW	tests/fused_attention_stablehlo_test.py	88	CODE
LOW	tests/shard_map_test.py	503	CODE
LOW	tests/shard_map_test.py	536	CODE
LOW	tests/shard_map_test.py	573	CODE
LOW	tests/shard_map_test.py	612	CODE
LOW	tests/shard_map_test.py	1069	CODE
LOW	tests/lax_numpy_reducers_test.py	278	CODE
LOW	tests/lax_numpy_reducers_test.py	511	CODE
LOW	tests/shape_poly_test.py	1278	CODE
LOW	tests/nn_test.py	64	CODE
LOW	tests/linalg_test.py	973	CODE
LOW	tests/state_test.py	1606	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	760	CODE
LOW	tests/export_test.py	763	CODE
LOW	tests/pmap_test.py	1439	CODE
LOW	tests/pjit_test.py	6288	CODE
LOW	tests/pjit_test.py	7898	CODE
LOW	tests/pjit_test.py	8565	CODE
LOW	tests/pjit_test.py	8788	CODE
LOW	tests/pjit_test.py	8977	CODE
LOW	tests/pjit_test.py	9083	CODE
LOW	tests/pjit_test.py	11390	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	77	CODE
LOW	tests/fused_attention_stablehlo_multigpu_test.py	89	CODE
LOW	tests/checkify_test.py	1205	CODE
LOW	tests/pallas/tpu_pallas_test.py	5689	CODE
LOW	tests/pallas/tpu_pallas_test.py	5731	CODE
LOW	tests/pallas/gpu_paged_attention_test.py	77	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	636	CODE
LOW	tests/pallas/tpu_pallas_async_test.py	758	CODE
LOW	tests/pallas/tpu_pallas_pipeline_test.py	655	CODE
LOW	tests/pallas/mgpu_collective_matmul_test.py	108	CODE
LOW	tests/pallas/tpu_ragged_paged_attention_test.py	36	CODE
LOW	tests/pallas/mosaic_gpu_test.py	9088	CODE
LOW	…sts/pallas/tpu_splash_attention_kernel_sharded_test.py	263	CODE
LOW	jaxlib/xla_client.py	66	CODE
LOW	jax/experimental/mosaic/gpu/launch_context.py	1191	CODE
LOW	jax/experimental/mosaic/gpu/examples/matmul.py	105	CODE
LOW	jax/experimental/mosaic/gpu/examples/matmul.py	321	CODE
LOW	jax/experimental/pallas/ops/gpu/attention.py	220	CODE
LOW	jax/experimental/pallas/ops/gpu/decode_attention.py	247	CODE
LOW	jax/experimental/pallas/ops/gpu/decode_attention.py	304	CODE
101 more matches not shown…

Cross-Language Confusion24 hits · 130 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	1121	stack.push(self.assertRaisesRegex(Exception, re.escape(expect_error)))	CODE
HIGH	tests/export_test.py	938	stack.push(self.assertRaisesRegex(Exception, expect_error))	CODE
HIGH	tests/export_test.py	1044	stack.push(self.assertRaisesRegex(ValueError, expect_error_outer_exp))	CODE
HIGH	tests/export_test.py	1058	stack.push(self.assertRaisesRegex(Exception, expect_error_run))	CODE
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	9050	b <= a, then the result is undefined, and different implementations may	STRING
HIGH	jax/_src/numpy/linalg.py	529	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

AI Slop Vocabulary48 hits · 116 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	294	# probably add a custom test harness for unwrap that tests the period	COMMENT
LOW	tests/custom_api_test.py	366	# useful either: instead of using nondiff_argnums here, a user can just pass	COMMENT
MEDIUM	tests/lax_numpy_test.py	3387	# 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	3670	# Pick the dtype with the most harnesses in this group. Some harness	COMMENT
MEDIUM	tests/shape_poly_test.py	3742	# If you want to run this test for only one harness that includes "foo"	COMMENT
MEDIUM	tests/shape_poly_test.py	3799	# Update this here rather than in harness object because vmap_random_gamma is derived	COMMENT
LOW	tests/shape_poly_test.py	1861	# It is not sufficient to just use the shape of an input; it is still unused	COMMENT
MEDIUM	tests/linalg_test.py	927	# This expresses identity function, which makes us robust to, e.g., the	COMMENT
MEDIUM	tests/lax_autodiff_test.py	159	# 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	454	# We can't just use `lambda x: x` because JAX simplifies this away to an	COMMENT
LOW	tests/mosaic/gpu_layout_inference_test.py	91	# TMEM reference, so we can just return a trivial mapping.	COMMENT
LOW	docs/array_refs.py	115	# `jax.ref.swap`, but usually you'd just use NumPy-style array indexing syntax:	COMMENT
LOW	docs/hijax_custom_derivatives.py	1058	# that a primitive with a `jvp` rule can just set `lin = linearize_from_jvp`	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	489	# simply return the non-PGLE profiled module from the persistent cache if it	COMMENT
LOW	jax/_src/pjit.py	1049	# checks and just return the pjit_in_shardings directly. `shard_args` will	COMMENT
LOW	jax/_src/checkify.py	350	# Default non-HOP case: just call primitive and don't update error.	COMMENT
LOW	jax/_src/interpreters/batching.py	635	# if there's only agreeing batch dims and scalars, just call the primitive	COMMENT
LOW	jax/_src/lax/ann.py	367	# 1. ApproxTopK is internally a variadic reduce, so we can simply call	COMMENT
MEDIUM	jax/_src/lax/control_flow/loops.py	2074	# lowering. Fundamentally, we'd like to rewrite a while loop that looks like	COMMENT
MEDIUM	jax/_src/numpy/linalg.py	1445	# TODO: add custom jvp rule for more robust lstsq differentiation	COMMENT
LOW	jax/_src/pallas/pallas_call.py	674	# dimensions. For now, we just use 0.	COMMENT
LOW	jax/_src/pallas/mosaic/lowering.py	1714	# `jax.empty_ref`), but lowering expects them to exist---so we just return	COMMENT
MEDIUM	…/_src/pallas/mosaic/interpret/interpret_pallas_call.py	1595	# TODO(nrink): It would be more robust if the buffer id, i.e. `src`,	COMMENT
LOW	…/_src/pallas/mosaic/interpret/interpret_pallas_call.py	645	# callback to `get`. Should we just pass the shape to `get`?	COMMENT
LOW	jax/_src/pallas/mosaic_gpu/interpret/gpu_callbacks.py	583	# 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	2003	# This first harness runs the tests for all dtypes using default values for	COMMENT
MEDIUM	jax/_src/internal_test_util/test_harnesses.py	2589	# This first harness runs the tests for all dtypes using default values for	COMMENT
MEDIUM	jax/_src/internal_test_util/test_harnesses.py	2894	# This first harness runs the tests for all dtypes and precisions using	COMMENT
MEDIUM	jax/_src/internal_test_util/test_harnesses.py	3095	# 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 Code56 hits · 106 pts

Severity	File	Line	Context
MEDIUM	tests/custom_api_test.py	374	CODE
MEDIUM	tests/custom_api_test.py	376	CODE
MEDIUM	tests/custom_api_test.py	379	CODE
MEDIUM	tests/custom_api_test.py	382	CODE
MEDIUM	tests/custom_api_test.py	385	CODE
MEDIUM	tests/custom_api_test.py	386	CODE
MEDIUM	tests/custom_api_test.py	387	CODE
MEDIUM	tests/custom_api_test.py	1843	CODE
MEDIUM	tests/custom_api_test.py	1855	CODE
MEDIUM	tests/custom_api_test.py	1858	CODE
MEDIUM	tests/custom_api_test.py	1859	CODE
MEDIUM	tests/custom_api_test.py	1860	CODE
MEDIUM	tests/custom_api_test.py	1862	CODE
MEDIUM	tests/custom_api_test.py	1863	CODE
MEDIUM	tests/custom_api_test.py	1864	CODE
MEDIUM	tests/lax_numpy_reducers_test.py	907	CODE
MEDIUM	tests/lax_numpy_reducers_test.py	908	CODE
MEDIUM	tests/lax_numpy_reducers_test.py	909	CODE
MEDIUM	tests/lax_numpy_reducers_test.py	919	CODE
MEDIUM	tests/lax_numpy_reducers_test.py	920	CODE
MEDIUM	tests/lax_numpy_reducers_test.py	921	CODE
MEDIUM	tests/pmap_test.py	2077	CODE
MEDIUM	tests/pmap_test.py	2079	CODE
MEDIUM	tests/pmap_test.py	2085	CODE
MEDIUM	tests/pmap_test.py	2095	CODE
MEDIUM	tests/pmap_test.py	2099	CODE
MEDIUM	tests/pmap_test.py	2114	CODE
MEDIUM	tests/pmap_test.py	2120	CODE
MEDIUM	tests/pmap_test.py	2121	CODE
MEDIUM	tests/pmap_test.py	2122	CODE
MEDIUM	tests/pmap_test.py	2123	CODE
MEDIUM	tests/pmap_test.py	2134	CODE
MEDIUM	tests/pmap_test.py	2140	CODE
MEDIUM	tests/pmap_test.py	2141	CODE
MEDIUM	tests/pmap_test.py	2142	CODE
MEDIUM	tests/pmap_test.py	2143	CODE
MEDIUM	tests/pmap_test.py	2149	CODE
MEDIUM	tests/pmap_test.py	2155	CODE
MEDIUM	tests/pmap_test.py	2305	CODE
MEDIUM	tests/pmap_test.py	2307	CODE
MEDIUM	tests/pmap_test.py	2310	CODE
MEDIUM	tests/pmap_test.py	2311	CODE
MEDIUM	tests/pmap_test.py	2313	CODE
MEDIUM	tests/pmap_test.py	2314	CODE
MEDIUM	tests/pmap_test.py	2315	CODE
MEDIUM	tests/pmap_test.py	2316	CODE
MEDIUM	tests/pmap_test.py	2328	CODE
MEDIUM	tests/lax_test.py	1775	CODE
MEDIUM	tests/lax_test.py	1807	CODE
MEDIUM	tests/mosaic/gpu_test.py	969	CODE
MEDIUM	tests/mosaic/gpu_test.py	971	CODE
MEDIUM	tests/mosaic/gpu_test.py	972	CODE
MEDIUM	tests/mosaic/gpu_test.py	973	CODE
MEDIUM	tests/mosaic/gpu_test.py	980	CODE
MEDIUM	jax/_src/lax/control_flow/loops.py	2995	CODE
MEDIUM	jax/_src/pallas/core.py	1705	CODE

Excessive Try-Catch Wrapping96 hits · 106 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	180	except Exception as e:	CODE
LOW	tests/debug_info_test.py	111	except Exception as e:	CODE
LOW	tests/shape_poly_test.py	2040	except Exception:	CODE
LOW	tests/profiler_test.py	63	except Exception as e:	CODE
LOW	tests/transfer_guard_test.py	117	except Exception as e:	CODE
LOW	tests/transfer_guard_test.py	136	except Exception as e:	CODE
LOW	tests/api_test.py	5083	except Exception as e:	CODE
LOW	tests/pallas/gpu_pallas_distributed_test.py	650	except Exception:	CODE
LOW	tests/pallas/gpu_pallas_distributed_test.py	701	except Exception:	CODE
LOW	tests/pallas/gpu_pallas_distributed_test.py	1318	except Exception:	CODE
LOW	tests/pallas/ops_test.py	726	except Exception as e:	CODE
LOW	tests/pallas/ops_test.py	854	except Exception as e:	CODE
LOW	tests/pallas/ops_test.py	2716	except Exception as e:	CODE
LOW	docs/array_refs.py	82	except Exception as e:	CODE
LOW⚡	docs/array_refs.py	222	except Exception as e:	CODE
LOW⚡	docs/array_refs.py	231	except Exception as e:	CODE
LOW⚡	docs/array_refs.py	240	except Exception as e:	CODE
LOW⚡	docs/array_refs.py	249	except Exception as e:	CODE
LOW	docs/array_refs.py	289	except Exception as e:	CODE
LOW	docs/array_refs.py	413	except Exception as e:	CODE
LOW	docs/parallel.md	425	except Exception as e:	CODE
LOW	docs/parallel.md	464	except Exception as e:	CODE
LOW	docs/array_refs.md	87	except Exception as e:	CODE
LOW⚡	docs/array_refs.md	222	except Exception as e:	CODE
LOW⚡	docs/array_refs.md	231	except Exception as e:	CODE
LOW⚡	docs/array_refs.md	240	except Exception as e:	CODE
LOW⚡	docs/array_refs.md	249	except Exception as e:	CODE
LOW	docs/array_refs.md	288	except Exception as e:	CODE
LOW	docs/array_refs.md	405	except Exception as e:	CODE
LOW	docs/parallel.py	379	except Exception as e:	CODE
LOW	docs/parallel.py	417	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
LOW	docs/new_docs/301/refs.md	144	except Exception as e:	CODE
LOW	docs/new_docs/201/shard-map.md	439	except Exception as e:	CODE
LOW	docs/new_docs/201/shard-map.md	513	except Exception as e:	CODE
LOW	docs/new_docs/201/sharding.md	403	except Exception as e:	CODE
LOW	docs/new_docs/201/sharding.md	442	except Exception as e:	CODE
MEDIUM	docs/pallas/tpu/sparse.md	148	print("Error \|result - lax.dynamic_slice\| =", diff)	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/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
LOW	…erimental/jax2tf/tests/jax_primitives_coverage_test.py	72	except Exception as e:	CODE
LOW	jax/experimental/jax2tf/tests/models_test_main.py	206	except Exception as e:	CODE
LOW	jax/experimental/jax2tf/tests/tf_test_util.py	246	except Exception as e:	CODE
LOW	jax/experimental/jax2tf/tests/primitives_test.py	135	except Exception as e:	CODE
LOW	jax/experimental/jax2tf/examples/mnist_lib.py	76	except Exception as e:	CODE
LOW	jax/experimental/colocated_python/obj_backend.py	115	except Exception as exc:	CODE
LOW	jax/experimental/sparse/bcsr.py	892	except Exception:	CODE
LOW	jax/_src/environment_info.py	30	except Exception:	CODE
MEDIUM	jax/_src/environment_info.py	27	def try_nvidia_smi() -> str \| None:	CODE
36 more matches not shown…

Modern Structural Boilerplate63 hits · 64 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	42	logger = logging.getLogger(__name__)	CODE
LOW	jax/experimental/mosaic/gpu/layout_inference.py	46	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	33	logger = logging.getLogger(__name__)	CODE
LOW	…ntal/array_serialization/pytree_serialization_utils.py	45	__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	64	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	46	logger = logging.getLogger(__name__)	CODE
LOW	jax/_src/lru_cache.py	32	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	69	logger = logging.getLogger(__name__)	CODE
LOW	jax/_src/cache_key.py	34	logger = logging.getLogger(__name__)	CODE
LOW	jax/_src/hijax.py	183	def update_from_loval(self, state: QDD, val: HiVal, /, *lo_vals: LoVal) -> None:	CODE
LOW	jax/_src/hijax.py	186	def update_from_loval2(self, state, val, lo_vals_ft, /) -> None:	CODE
LOW	jax/_src/hijax.py	269	def update_from_loval(self, box_state: BoxTypeState, box, *lo_vals) -> None: # pyrefly: ignore[bad-override]	CODE
LOW	jax/_src/hijax.py	1207	def update_from_loval2(self, qdd, log: Log, lo_ft) -> None:	CODE
LOW	jax/_src/util.py	38	logger = logging.getLogger(__name__)	CODE
LOW	jax/_src/dispatch.py	74	logger = logging.getLogger(__name__)	CODE
LOW	jax/_src/debugging.py	56	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	63	logger = logging.getLogger(__name__)	CODE
LOW	jax/_src/jaxpr_util.py	44	logger = logging.getLogger(__name__)	CODE
LOW	jax/_src/hardware_utils.py	90	logger = logging.getLogger(__name__)	CODE
LOW	jax/_src/profiler.py	46	logger = logging.getLogger(__name__)	CODE
LOW	jax/_src/profiler.py	139	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	63	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	1221	if not isinstance(aval, AbstractLinVal): raise TypeError(aval)	CODE
LOW	jax/_src/state/types.py	598	def _setitem(self, tracer, idx, value) -> None:	CODE
LOW	jax/_src/numpy/array_constructors.py	37	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	jax/_src/pallas/triton/lowering.py	1349	def _set_attr(v: ir.Value, name: str, attr: ir.Attribute) -> None:	CODE
LOW	jax/_src/pallas/mosaic_gpu/interpret/shared_memory.py	33	logger = logging.getLogger(__name__)	CODE
LOW	jax/_src/export/_export.py	61	logger = logging.getLogger(__name__)	CODE
3 more matches not shown…

Redundant / Tautological Comments26 hits · 38 pts

Severity	File	Line	Snippet	Context
LOW	tests/array_interoperability_test.py	209	# 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	328	# 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	145	# 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	202	# Check if the numbers in the output match the values generated by `arange`.	COMMENT
LOW	.github/workflows/upload_metadata.yml	67	# 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	169	# Set shell to cmd to avoid path errors when using gcloud commands on Windows	COMMENT
LOW	jax/experimental/mosaic/gpu/fragmented_array.py	5033	# Check if input maps exactly to the end (prevents trailing dims).	COMMENT
LOW	jax/_src/test_util.py	1307	# Check if strict hermeticity is already satisfied	COMMENT
LOW	jax/_src/checkify.py	867	# 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	2251	# Check if the same Ref is written to in both cond and body.	COMMENT
LOW	jax/_src/state/primitives.py	764	# Check if start is static (which it can be)	COMMENT
LOW	jax/_src/state/primitives.py	777	# 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	252	# Check if object supports any of the data exchange protocols	COMMENT
LOW	jax/_src/pallas/core.py	1250	elif grid and isinstance(grid[0], tuple): # Check if we have a named grid	CODE
LOW	jax/_src/pallas/einshape.py	232	# Check if all RHS dims are known	COMMENT
LOW	jax/_src/scipy/signal.py	681	# Check if we can broadcast the outer axes together	COMMENT
LOW	jax/_src/scipy/signal.py	730	# Check if x and y are the same length, zero-pad if necessary	COMMENT
LOW	jax/_src/cudnn/fused_attention_stablehlo.py	1265	# 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	634	dummy_data = jax.lax.precv(	CODE
LOW⚡	tests/shard_map_test.py	642	dummy_data,	CODE
LOW⚡	tests/shard_map_test.py	646	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

Hallucination Indicators3 hits · 30 pts

Severity	File	Line	Snippet	Context
CRITICAL	tests/mosaic/profiler_cupti_test.py	96	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	jax/_src/pallas/mosaic_gpu/lowering.py	4034	for k, v in ctx.launch_ctx.module.operation.attributes.items():	CODE

Slop Phrases13 hits · 26 pts

Severity	File	Line	Snippet	Context
MEDIUM	docs/array_refs.py	609	# wouldn't be for `vmap` or `shard_map`, as discussed above). Once all the	COMMENT
MEDIUM	docs/parallel.py	396	# want the compiler to choose them automatically, you can use the `@auto_axes`	COMMENT
MEDIUM	docs/ffi.py	161	# // this handler in a header, you can use the `XLA_FFI_DECLARE_HANDLER_SYMBOL`	COMMENT
MEDIUM	docs/ffi.py	209	# 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	209	# 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	2899	# but we make sure to reset it to Device because the Ref owns the memory space	COMMENT
MEDIUM	jax/_src/ad_checkpoint.py	289	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	801	# If you hit this, make sure you take transforms into account and use either	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	287	// 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	133	# // In this example, we treat all leading dimensions as batch dimensions, so this	COMMENT
HIGH	docs/ffi.py	176	# In this implementation, we treat all leading dimensions of the buffer as batch dimensions, and perform RMS normalizati	COMMENT

Structural Annotation Overuse8 hits · 12 pts

Severity	File	Line	Snippet	Context
LOW	docs/distributed_data_loading.md	230	# Step 1: setup the Dataset for pure data parallelism (do once)	COMMENT
LOW	docs/distributed_data_loading.md	239	# Step 2: create a jax.Array of per-replica batches from the per-process batch	COMMENT
LOW	docs/distributed_data_loading.md	325	# Step 1: Set up the Dataset with a different data shard per-process (do once)	COMMENT
LOW	docs/distributed_data_loading.md	335	# Step 2: Create a jax.Array of per-replica batches from the per-process batch	COMMENT
LOW	docs/new_docs/501/data-loading.md	221	# Step 1: setup the Dataset for pure data parallelism (do once)	COMMENT
LOW	docs/new_docs/501/data-loading.md	230	# Step 2: create a jax.Array of per-replica batches from the per-process batch	COMMENT
LOW	docs/new_docs/501/data-loading.md	316	# Step 1: Set up the Dataset with a different data shard per-process (do once)	COMMENT
LOW	docs/new_docs/501/data-loading.md	326	# Step 2: Create a jax.Array of per-replica batches from the per-process batch	COMMENT

Overly Generic Function Names10 hits · 10 pts

Severity	File	Line	Snippet	Context
LOW⚡	tests/core_test.py	584	def my_function():	CODE
LOW	tests/util_test.py	80	def my_function():	CODE
LOW	tests/aot_test.py	122	def my_function(x):	CODE
LOW	tests/aot_test.py	143	def my_function(x):	CODE
LOW	tests/source_info_test.py	279	def my_function(x):	CODE
LOW⚡	tests/api_test.py	93	def my_function():	CODE
LOW⚡	tests/api_test.py	107	def my_function():	CODE
LOW	tests/api_test.py	1622	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

Example Usage Blocks3 hits · 5 pts

Severity	File	Line	Snippet	Context
LOW⚡	ci/utilities/run_docker_container.sh	24	# Usage:	COMMENT
LOW	jaxlib/mosaic/dialect/tpu/util.h	110	// Example 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	2332	# TODO: implement full matrices case, documented here: https://people.maths.ox.ac.uk/gilesm/files/NA-08-01.pdf	COMMENT

Verbosity Indicators1 hit · 2 pts

Severity	File	Line	Snippet	Context
LOW	jaxlib/python_ref_manager.h	85	// The purpose of this function is to amortize lock acquisition costs over	COMMENT

Analysis Overview

What These Metrics Mean

Score History

Severity Breakdown

Directory Score Breakdown

Pattern Findings