jax-ml/jax

19.0

Adjusted Score

19.0

Raw Score

100%

Time Factor

2026-08-28

Last Push

36.2K

Stars

Python

Language

688.0K

Lines of Code

1.4K

Files

13.0K

Pattern Hits

2026-08-29

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 631MEDIUM 337LOW 12078

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 13049 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 Identifiers6128 hits · 4702 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	143	def test_unrolled_fsdp_pipeline_grad_explicit_mode_program_order(self, mesh):	CODE
LOW	tests/overlap_test.py	206	def test_unrolled_fsdp_pipeline_grad_program_order_shmap(self, mesh):	CODE
LOW	tests/overlap_test.py	271	def test_unrolled_fsdp_pipeline_grad_program_order_async_decomp(self, mesh):	CODE
LOW	tests/overlap_test.py	346	def test_simple_program_order(self, mesh):	CODE
LOW	tests/overlap_test.py	365	def test_avoid_excess_precision(self):	CODE
LOW	tests/overlap_test.py	384	def test_lower_async_all_gather(self, mesh):	CODE
LOW	tests/overlap_test.py	408	def test_lower_async_psum_scatter(self, mesh):	CODE
LOW	tests/overlap_test.py	421	def test_lower_async_all_to_all(self, mesh):	CODE
LOW	tests/overlap_test.py	435	def test_lower_async_ppermute(self, mesh):	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
6068 more matches not shown…

Unused Imports3560 hits · 2516 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	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	benchmarks/pallas/mgpu_planar_snake_bench.py	23	CODE
LOW	…workflows/scripts/select_continuous_wheel_test_jobs.py	19	CODE
LOW	jax/sharding.py	18	CODE
LOW	jax/sharding.py	19	CODE
LOW	jax/sharding.py	19	CODE
3500 more matches not shown…

Docstring Block Structure570 hits · 2320 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	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	3030	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	1134	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	218	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	2401	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	2481	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	2550	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	2699	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	283	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	712	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	847	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	1047	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	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	165	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	214	Convenience wrapper for defining JVPs for each argument separately. This convenience wrapper cannot be used togethe	STRING
HIGH	jax/_src/custom_derivatives.py	600	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
510 more matches not shown…

Over-Commented Block1495 hits · 1414 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	//tests/pallas:tpu_pallas_call_print_test_tpu \	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/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/overlap_test.py	601	if op in hlo_sync:	COMMENT
LOW	tests/overlap_test.py	621	# x = jnp.ones((67, 67))	COMMENT
LOW	tests/overlap_test.py	641	# for w_shard in ws:	COMMENT
LOW	tests/overlap_test.py	661	# deps.append(maths[i - 1])	COMMENT
LOW	tests/overlap_test.py	681	# # This test shows FSDP with scan.	COMMENT
LOW	tests/overlap_test.py	701		COMMENT
LOW	tests/overlap_test.py	721	# dones = []	COMMENT
LOW	tests/overlap_test.py	741	# control_dep(maths[i], dones[i + 1])	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
1435 more matches not shown…

Decorative Section Separators157 hits · 496 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/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	2634	# ------------	COMMENT
MEDIUM⚡	tests/hijax_test.py	2636	# ------------	COMMENT
MEDIUM⚡	tests/hijax_test.py	2671	#------------	COMMENT
MEDIUM⚡	tests/hijax_test.py	2673	#------------	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
97 more matches not shown…

Deep Nesting476 hits · 374 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	2376	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/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	121	CODE
LOW	tests/fused_attention_stablehlo_test.py	804	CODE
LOW	tests/lax_numpy_test.py	2614	CODE
LOW	tests/lax_numpy_test.py	6886	CODE
LOW	tests/lax_numpy_reducers_test.py	164	CODE
LOW	tests/linalg_test.py	715	CODE
LOW	tests/linalg_test.py	1828	CODE
LOW	tests/linalg_test.py	1930	CODE
LOW	tests/linalg_test.py	1831	CODE
LOW	tests/linalg_test.py	1938	CODE
LOW	tests/profiler_test.py	54	CODE
LOW	tests/profiler_test.py	132	CODE
LOW	tests/lax_numpy_indexing_test.py	1522	CODE
LOW	tests/export_test.py	2426	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	5331	CODE
LOW	tests/api_test.py	5823	CODE
LOW	tests/api_test.py	5831	CODE
LOW	tests/documentation_coverage_test.py	142	CODE
LOW	tests/sparse_test.py	1006	CODE
LOW	tests/lax_test.py	117	CODE
LOW	tests/lax_test.py	1614	CODE
LOW	tests/lax_test.py	4624	CODE
LOW	tests/mosaic/gpu_test.py	682	CODE
LOW	tests/mosaic/gpu_test.py	1169	CODE
LOW	tests/mosaic/gpu_test.py	5691	CODE
LOW	tests/mosaic/gpu_test.py	6549	CODE
LOW	tests/mosaic/gpu_test.py	7776	CODE
LOW	tests/multiprocess/thread_guard_test.py	42	CODE
LOW	tests/multiprocess/pgle_test.py	44	CODE
LOW	tests/pallas/tpu_pallas_test.py	5835	CODE
LOW	tests/pallas/tpu_pallas_test.py	5851	CODE
LOW	tests/pallas/tpu_pallas_interpret_test.py	321	CODE
LOW	tests/pallas/tpu_pallas_interpret_test.py	417	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
416 more matches not shown…

Self-Referential Comments72 hits · 198 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	546	# This function is not defined for negative values, this makes sure they are nan	COMMENT
MEDIUM	tests/custom_api_test.py	2146	# 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	362	# Create a custom layout instead of using `arr.layout` to test the API.	COMMENT
MEDIUM	tests/layout_test.py	385	# Create a custom layout instead of using `arr.layout` to test the API.	COMMENT
MEDIUM	tests/layout_test.py	666	# Create a custom layout instead of using `arr.layout` to test the API.	COMMENT
MEDIUM	tests/layout_test.py	698	# Create a custom layout instead of using `arr.layout` to test the API.	COMMENT
MEDIUM	tests/export_test.py	2517	# 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	338	# Create a compile-only topology, but DON'T switch to CPU so the real	COMMENT
MEDIUM	tests/array_test.py	679	# Create a few arrays	COMMENT
MEDIUM	tests/hijax_test.py	66	# Define a type	COMMENT
MEDIUM⚡	tests/api_test.py	2074	# Creating an array from per-device JAX arrays calls internal	COMMENT
MEDIUM	tests/mosaic/gpu_layout_inference_test.py	1612	# 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	70	# Create an input array that shards the last dimension across	COMMENT
MEDIUM	tests/pallas/tpu_pallas_interpret_distributed_test.py	156	# Create an input array that shards the first dimension across	COMMENT
MEDIUM	tests/pallas/mosaic_gpu_test.py	8231	# Create an index-invariant output.	COMMENT
MEDIUM	tests/pallas/pallas_test.py	2533	# 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	418	# This function is similar to api_util.argnums_partial, except the error	COMMENT
MEDIUM	jax/_src/ad_checkpoint.py	926	>>> # Define a function where we explicitly name an intermediate value	STRING
MEDIUM	jax/_src/ffi.py	712	# This method is kept to support the behavior that was previously exposed	COMMENT
MEDIUM	jax/_src/interpreters/partial_eval.py	2224	# This function is conceptually the same thing as just calling eval_jaxpr,	COMMENT
MEDIUM	jax/_src/interpreters/mlir.py	1415	# 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	1400	# Create the staged eqn.	COMMENT
MEDIUM	jax/_src/pallas/mosaic/lowering.py	4330	# Create a scalar constant.	COMMENT
MEDIUM	jax/_src/pallas/mosaic/lowering.py	4333	# Create a vector constant.	COMMENT
MEDIUM	jax/_src/pallas/mosaic_gpu/lowering.py	608	# 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
MEDIUM	jax/_src/export/serialization_generated.py	244	"""This method is deprecated. Please switch to GetRootAs."""	STRING
12 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 Patterns169 hits · 137 pts

Severity	File	Line	Context
LOW	tests/overlap_test.py	126	CODE
LOW	tests/overlap_test.py	184	CODE
LOW	tests/overlap_test.py	245	CODE
LOW	tests/overlap_test.py	320	CODE
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	230	CODE
LOW	tests/mutable_array_test.py	439	CODE
LOW	tests/custom_api_test.py	3281	CODE
LOW	tests/custom_api_test.py	1194	CODE
LOW	tests/custom_api_test.py	2902	CODE
LOW	tests/custom_api_test.py	4741	CODE
LOW	tests/dtypes_test.py	795	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	502	CODE
LOW	tests/shard_map_test.py	535	CODE
LOW	tests/shard_map_test.py	572	CODE
LOW	tests/shard_map_test.py	611	CODE
LOW	tests/shard_map_test.py	1068	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	1327	CODE
LOW	tests/nn_test.py	64	CODE
LOW	tests/linalg_test.py	977	CODE
LOW	tests/state_test.py	1665	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	1440	CODE
LOW	tests/pjit_test.py	6290	CODE
LOW	tests/pjit_test.py	7913	CODE
LOW	tests/pjit_test.py	8580	CODE
LOW	tests/pjit_test.py	8803	CODE
LOW	tests/pjit_test.py	9003	CODE
LOW	tests/pjit_test.py	9109	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	1226	CODE
LOW	tests/pallas/tpu_pallas_test.py	6339	CODE
LOW	tests/pallas/tpu_pallas_test.py	6381	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	637	CODE
LOW	tests/pallas/tpu_pallas_async_test.py	759	CODE
LOW	tests/pallas/tpu_pallas_pipeline_test.py	836	CODE
LOW	tests/pallas/mgpu_collective_matmul_test.py	109	CODE
LOW	tests/pallas/tpu_ragged_paged_attention_test.py	36	CODE
LOW	tests/pallas/mosaic_gpu_test.py	9795	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	1209	CODE
LOW	jax/experimental/mosaic/gpu/examples/matmul.py	105	CODE
109 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	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	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	9200	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	373	# useful either: instead of using nondiff_argnums here, a user can just pass	COMMENT
MEDIUM	tests/lax_numpy_test.py	3460	# 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	934	# 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	458	# 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	117	# `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, 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	534	# simply return the non-PGLE profiled module from the persistent cache if it	COMMENT
LOW	jax/_src/pjit.py	1008	# checks and just return the pjit_in_shardings directly. `shard_args` will	COMMENT
LOW	jax/_src/checkify.py	372	# 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	377	# 1. ApproxTopK is internally a variadic reduce, so we can simply call	COMMENT
MEDIUM	jax/_src/lax/control_flow/loops.py	2081	# lowering. Fundamentally, we'd like to rewrite a while loop that looks like	COMMENT
MEDIUM	jax/_src/numpy/linalg.py	1488	# TODO: add custom jvp rule for more robust lstsq differentiation	COMMENT
LOW	jax/_src/pallas/pallas_call.py	669	# dimensions. For now, we just use 0.	COMMENT
LOW	jax/_src/pallas/mosaic/lowering.py	1757	# `jax.empty_ref`), but lowering expects them to exist---so we just return	COMMENT
MEDIUM	…/_src/pallas/mosaic/interpret/interpret_pallas_call.py	1652	# TODO(nrink): It would be more robust if the buffer id, i.e. `src`,	COMMENT
LOW	…/_src/pallas/mosaic/interpret/interpret_pallas_call.py	646	# callback to `get`. Should we just pass the shape to `get`?	COMMENT
LOW	jax/_src/pallas/mosaic_gpu/interpret/gpu_callbacks.py	564	# 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

Excessive Try-Catch Wrapping105 hits · 115 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	179	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	117	except Exception as e:	CODE
LOW	tests/transfer_guard_test.py	136	except Exception as e:	CODE
LOW	tests/api_test.py	5412	except Exception as e:	CODE
LOW	…as/gpu_pallas_interpret_tcgen05_race_detection_test.py	995	except Exception as e:	CODE
LOW	tests/pallas/tpu_ops_test.py	1724	except Exception: # pylint: disable=broad-except	CODE
LOW	tests/pallas/gpu_pallas_distributed_test.py	744	except Exception:	CODE
LOW	tests/pallas/gpu_pallas_distributed_test.py	798	except Exception:	CODE
LOW	tests/pallas/gpu_pallas_distributed_test.py	1425	except Exception:	CODE
LOW	tests/pallas/ops_test.py	764	except Exception as e:	CODE
LOW	tests/pallas/ops_test.py	892	except Exception as e:	CODE
LOW	tests/pallas/ops_test.py	2921	except Exception as e:	CODE
LOW	docs/array_refs.py	84	except Exception as e:	CODE
LOW⚡	docs/array_refs.py	224	except Exception as e:	CODE
LOW⚡	docs/array_refs.py	233	except Exception as e:	CODE
LOW⚡	docs/array_refs.py	242	except Exception as e:	CODE
LOW⚡	docs/array_refs.py	251	except Exception as e:	CODE
LOW	docs/array_refs.py	291	except Exception as e:	CODE
LOW	docs/array_refs.py	415	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	224	except Exception as e:	CODE
LOW⚡	docs/array_refs.md	233	except Exception as e:	CODE
LOW⚡	docs/array_refs.md	242	except Exception as e:	CODE
LOW⚡	docs/array_refs.md	251	except Exception as e:	CODE
LOW	docs/array_refs.md	290	except Exception as e:	CODE
LOW	docs/array_refs.md	407	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	437	except Exception as e:	CODE
LOW	docs/201/shard-map.md	511	except Exception as e:	CODE
LOW	docs/201/sharding.md	413	except Exception as e:	CODE
LOW	docs/201/sharding.md	455	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	2572	except Exception: # pylint: disable=broad-except	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
45 more matches not shown…

Dead Code56 hits · 106 pts

Severity	File	Line	Context
MEDIUM	tests/custom_api_test.py	381	CODE
MEDIUM	tests/custom_api_test.py	383	CODE
MEDIUM	tests/custom_api_test.py	386	CODE
MEDIUM	tests/custom_api_test.py	389	CODE
MEDIUM	tests/custom_api_test.py	392	CODE
MEDIUM	tests/custom_api_test.py	393	CODE
MEDIUM	tests/custom_api_test.py	394	CODE
MEDIUM	tests/custom_api_test.py	1860	CODE
MEDIUM	tests/custom_api_test.py	1872	CODE
MEDIUM	tests/custom_api_test.py	1875	CODE
MEDIUM	tests/custom_api_test.py	1876	CODE
MEDIUM	tests/custom_api_test.py	1877	CODE
MEDIUM	tests/custom_api_test.py	1879	CODE
MEDIUM	tests/custom_api_test.py	1880	CODE
MEDIUM	tests/custom_api_test.py	1881	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	2080	CODE
MEDIUM	tests/pmap_test.py	2082	CODE
MEDIUM	tests/pmap_test.py	2088	CODE
MEDIUM	tests/pmap_test.py	2098	CODE
MEDIUM	tests/pmap_test.py	2102	CODE
MEDIUM	tests/pmap_test.py	2117	CODE
MEDIUM	tests/pmap_test.py	2123	CODE
MEDIUM	tests/pmap_test.py	2124	CODE
MEDIUM	tests/pmap_test.py	2125	CODE
MEDIUM	tests/pmap_test.py	2126	CODE
MEDIUM	tests/pmap_test.py	2137	CODE
MEDIUM	tests/pmap_test.py	2143	CODE
MEDIUM	tests/pmap_test.py	2144	CODE
MEDIUM	tests/pmap_test.py	2145	CODE
MEDIUM	tests/pmap_test.py	2146	CODE
MEDIUM	tests/pmap_test.py	2152	CODE
MEDIUM	tests/pmap_test.py	2158	CODE
MEDIUM	tests/pmap_test.py	2308	CODE
MEDIUM	tests/pmap_test.py	2310	CODE
MEDIUM	tests/pmap_test.py	2313	CODE
MEDIUM	tests/pmap_test.py	2314	CODE
MEDIUM	tests/pmap_test.py	2316	CODE
MEDIUM	tests/pmap_test.py	2317	CODE
MEDIUM	tests/pmap_test.py	2318	CODE
MEDIUM	tests/pmap_test.py	2319	CODE
MEDIUM	tests/pmap_test.py	2331	CODE
MEDIUM	tests/lax_test.py	1776	CODE
MEDIUM	tests/lax_test.py	1808	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	3003	CODE
MEDIUM	jax/_src/pallas/core.py	1609	CODE

Modern Structural Boilerplate61 hits · 61 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	43	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	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	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	74	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	63	logger = logging.getLogger(__name__)	CODE
LOW	jax/_src/jaxpr_util.py	44	logger = logging.getLogger(__name__)	CODE
LOW	jax/_src/hardware_utils.py	96	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	1245	if not isinstance(aval, AbstractLinVal): raise TypeError(aval)	CODE
LOW	jax/_src/state/types.py	613	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	1347	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	jax/_src/export/_export.py	61	logger = logging.getLogger(__name__)	CODE
LOW	jax_plugins/cuda/__init__.py	51	logger = logging.getLogger(__name__)	CODE
LOW	jax_plugins/rocm/__init__.py	26	logger = logging.getLogger(__name__)	CODE
1 more matches not shown…

Redundant / Tautological Comments26 hits · 38 pts

Severity	File	Line	Snippet	Context
LOW	tests/array_interoperability_test.py	208	# 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	146	# 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	169	# 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	5315	# 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	927	# 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	2257	# Check if the same Ref is written to in both cond and body.	COMMENT
LOW	jax/_src/state/primitives.py	782	# Check if start is static (which it can be)	COMMENT
LOW	jax/_src/state/primitives.py	795	# 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	1244	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	1323	# 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	633	dummy_data = jax.lax.precv(	CODE
LOW⚡	tests/shard_map_test.py	641	dummy_data,	CODE
LOW⚡	tests/shard_map_test.py	645	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	4079	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	611	# 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	2954	# but we make sure to reset it to Device because the Ref owns the memory space	COMMENT
MEDIUM	jax/_src/ad_checkpoint.py	296	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	795	# 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	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

Structural Annotation Overuse10 hits · 14 pts

Severity	File	Line	Snippet	Context
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/501/data-loading.md	217	# Step 1: setup 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/linalg.py	771	# Step 1: Pivot column 0 so row 0 gets the largest magnitude element.	COMMENT
LOW	jax/_src/numpy/linalg.py	789	# Step 2: Pivot column 1 across rows 1 and 2, then eliminate row 2.	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	123	def my_function(x):	CODE
LOW	tests/aot_test.py	144	def my_function(x):	CODE
LOW	tests/source_info_test.py	279	def my_function(x):	CODE
LOW⚡	tests/api_test.py	97	def my_function():	CODE
LOW⚡	tests/api_test.py	111	def my_function():	CODE
LOW	tests/api_test.py	1629	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 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

Verbosity Indicators3 hits · 3 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
LOW	jax/_src/numpy/linalg.py	771	# Step 1: Pivot column 0 so row 0 gets the largest magnitude element.	COMMENT
LOW	jax/_src/numpy/linalg.py	789	# Step 2: Pivot column 1 across rows 1 and 2, then eliminate row 2.	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	2345	# TODO: implement full matrices case, documented here: https://people.maths.ox.ac.uk/gilesm/files/NA-08-01.pdf	COMMENT

Analysis Overview

What These Metrics Mean

Score History

Severity Breakdown

Directory Score Breakdown

Pattern Findings