jax-ml/jax
Documented errors, page 18 of 23. Back to jax-ml/jax
| Code / Message | Type | Severity | Tags |
|---|---|---|---|
| Can only store scalars to SMEM | exception | error | jax, pallas, tpu, smem, shape-mismatch |
| Cannot enter TracebackScope recursively. | validation | error | jax, traceback, context-manager |
| Cannot select from Refs of different types | validation | error | jax, multiref, type-mismatch |
| cotangent type does not match function output, expected | validation | error | jax, linear-transpose, dtype, shape-mismatch |
| dtype must be a complex floating-point type; got | exception | error | jax, scipy, linalg, dtype, dft |
| Expected a string or dtype-like object; got | validation | error | jax, dlpack, dtype, none-guard |
| float32 top_k is not supported on TPUv3 or older | exception | error | jax, pallas, tpu, top-k, hardware-version |
| Incorrect lengths for f and s. The length of s along the… | exception | error | jax, leslie-matrix, off-by-one, argument-length-validation |
| N must be a multiple of 8 and <= 256, got | validation | error | tcgen05, matmul, mosaic, gpu, shape-validation |
| Named reductions not implemented for jnp. | validation | error | jax, named-axes, pmap, reductions |
| new enum value must be None or in | validation | error | jax, config, valueerror, enum |
| Non-decrementing wait is not supported. | validation | error | jax, pallas, tpu, semaphore, interpret-mode, not-implemented |
| process_id must be a nonnegative int. Got process_id= | validation | error | jax, distributed, type-error, rank |
| reduce_precision: mantissa_bits must be non-negative; got | validation | error | jax, reduce-precision, quantization, validation |
| scaled_matmul requires all inputs to be 3-dimensional arrays | validation | error | jax, nn, matmul, float8, mxfp8, shape-validation |
| State effect not supported in cond partial-eval. | exception | error | jax, cond, partial-eval, state-effects, autodiff |
| string directive must be placed at the beginning | validation | error | jax, r-c-index-tricks, directive, numpy-compat |
| The array passed to from_dlpack must have __dlpack__ and… | validation | error | jax, dlpack, type-check, buffer-protocol |
| Unknown algorithm ' '. Expected 'fast' or 'stable'. | validation | error | jax, nn, standardize, enum-argument |
| axis_index_groups must all be the same size for TPU lowering | exception | error | jax, tpu, collectives, lowering |
| cannot handle multidimensional fweights | exception | error | jax, covariance, weights, shape-validation |
| Cannot signal on a non-()-shaped semaphore | exception | error | jax, pallas, mosaic, dma, semaphore, shape-validation |
| coordinator_address should be defined. | validation | error | jax, distributed, coordinator, config |
| Custom VJP bwd rule was registered with defvjp_with_logs… | exception | error | jax, custom-vjp, api-contract, logging |
| duplicate value in 'axis | validation | error | jax, reductions, axis, duplicate |
| Expected both or neither of scales to be specified. | validation | error | mosaic, gpu, tcgen05, mma, scales, argument-validation |
| Expected fill values to have dtype | exception | error | jax, nonzero, fill-value, dtype |
| Factor represents a whole dimension; do not specify its size | exception | error | jax, sharding, factor-sizes, redundant-config |
| Illegal nan_policy value | exception | error | jax, scipy, stats, mode, nan-policy, argument-validation |
| In order to best JIT compile `mode`, we cannot know whether… | exception | error | jax, scipy, stats, mode, jit, not-implemented, nan-policy |
| in1 and in2 must have the same number of dimensions | exception | error | jax, scipy, signal, convolution, shape-mismatch |
| Invalid StrongLRUCache pickle version, got | validation | error | jax, lru-cache, pickle, versioning |
| Length of major_to_minor and the rank of the value should… | validation | error | jax, layout, rank-mismatch, validation |
| Mesh is not supported by the Mosaic GPU backend | exception | error | pallas, mosaic-gpu, mesh, not-implemented |
| mulhi requires integer inputs, got | validation | error | jax, dtype, integer, mulhi, type-validation |
| Multi-ref TransformedRef requires a single transform | exception | error | jax, multiref, api-restriction |
| new string config value must be None or of type str, got | validation | error | jax, config, valueerror, string |
| One Hot indexing is only supported for up to 50 leading… | exception | error | jax, take-along-axis, rank-limit, einsum |
| Only equal-sized splits are supported. | exception | error | jax, triton, pallas, split, shape, not-implemented |
| Scatter only supports storing to VMEM, got | exception | error | jax, pallas, sparsecore, scatter, memory-space |
| Sizes passed to split must be nonnegative, got | validation | error | jax, split, negative-size, validation |
| Symbolic dimension cannot be raised to non-integer powers | exception | error | jax, shape-polymorphism, pow, validation |
| tensordot axes argument must be an int, a pair of ints, or… | validation | error | jax, tensordot, invalid-argument, numpy |
| The stored value has dtype | exception | error | jax, pallas, mosaic-gpu, dtype-mismatch, async-store |
| Tiling does not divide grid . | validation | error | jax, pallas, mosaic-gpu, nd-loop, tiling, divisibility |
| TMEM transpose not allowed. | validation | error | jax, pallas, tcgen05, tmem, transpose |
| top_level_all_gather works when all mesh axes of context… | validation | error | jax, all-gather, explicit-mesh, context-mesh |
| `unreduced` and `reduced` argument to PartitionSpec cannot… | validation | error | jax, sharding, partition-spec, mesh |
| Unsupported memory space when lowering memref.cast | exception | error | jax, mosaic-gpu, memref, cast, memory-space |
| Addition between sparse matrices of different shapes. | exception | error | jax, sparse, sparsify, addition, shape-mismatch |
| Argument to register_custom_call_partitioner was not a… | validation | error | jax, xla, custom-call, capsule, validation |
| bcoo_extract with unequal batch dimensions. | exception | error | jax, sparse, bcoo, vmap, batching |
| collective_id has to be specified when using a custom… | exception | error | tpu, pallas, collective-id, barrier, missing-argument |
| complete_tx is only supported on Hopper+ hardware | validation | error | jax, mosaic-gpu, barrier, tma, gpu-architecture, hopper |
| conv_general_dilated batch_group_count must be a positive… | validation | error | jax, lax, convolution, batch-group-count |
| cuDNN not found. | exception | error | jax, cuda, cudnn, gpu, shared-library |
| Get does not support loading from | exception | error | jax, pallas, tpu, sparsecore, memory-space |
| incompatible numbers of samples and fweights | exception | error | jax, covariance, weights, length-mismatch |
| masks should be of type Mask | validation | error | jax, splash-attention, mask, type-validation |
| nperseg must be a positive integer | exception | error | jax, scipy, stft, nperseg, jit-tracer, argument-validation |
| Number of samples, , must be non-negative. | exception | error | jax, linspace, argument-validation |
| relayout_p is not supported with Lane semantics. | exception | error | pallas, mosaic-gpu, layout, not-implemented |
| scaled_matmul requires inputs a and b to have matching… | validation | error | jax, nn, matmul, float8, shape-validation |
| `seq_lengths` can only be int32. | exception | error | jax, lstm, rnn, dtype, int32 |
| The input array has rank | validation | error | jax, numpy, cumulative-sum, axis-argument |
| with_memory_space_constraint only supports HBM, VMEM, SMEM… | exception | error | jax, pallas, tpu, memory-space, unsupported-operation |
| Cannot specify both out_shape and out_type. Use out_type. | exception | error | jax, pallas, mosaic-gpu, deprecation, api-migration |
| convolution dimension_numbers must be tuple/list or None… | validation | error | jax, convolution, dimension-numbers, type-error |
| Expected a single dimension when passing a single index | validation | error | jax, mosaic-gpu, cluster, api-misuse |
| Expected mode of type `LayoutMode`. Got type | validation | error | jax, layout, type-validation, enum |
| For 'valid' mode, One input must be at least as large as… | exception | error | jax, scipy, signal, convolution, shape-mismatch |
| does not have enough range to generate sparse indices of… | validation | error | jax, sparse, bcoo, random, index-overflow |
| must have at least one non-None value in in_axes or… | validation | error | jax, vmap, in-axes, axis-size |
| One of src or dst must be a BufferedRef. | exception | error | jax, pallas, pipeline, copy |
| Only up to 32 barriers per group supported | validation | error | jax, mosaic-gpu, barrier, hardware-limit, cuda |
| out_specs passed to shard_map should be equal to the… | exception | error | shard-map, unreduced-kind, out-specs, sharding-mismatch, jax |
| ragged_dot_general expects exactly one lhs ragged dimension. | validation | error | jax, ragged-dot-general, dimension-numbers |
| Semaphore occurs as both fixed-id and internal. | exception | error | jax, pallas, mosaic, semaphore, id-collision, interpret-mode |
| Stored array has shape | exception | error | mosaic, gpu, tcgen05, shape-mismatch, tensor-memory, jax |
| transpose permutation must be a tuple/list/ndarray, got | exception | error | jax, sparse, bcoo, transpose, type-check |
| unstack requires arrays with rank > 0, however a scalar… | validation | error | jax, unstack, scalar, rank-error |
| Unsupported A register array shape | validation | error | jax, mosaic-gpu, wgmma, shape, swizzle |
| Unsupported op type | exception | error | mosaic, gpu, scf, control-flow, not-implemented |
| varying and unreduced cannot have common mesh axes. Got… | validation | error | jax, sharding, manual-axis-type, validation |
| z must be a 1D array. | validation | error | jax, scipy-special, legendre, not-implemented, argument-validation |
| A address calculation for multiple M tiles | validation | error | gpu, mosaic, tcgen05, tmem, not-implemented, mma |
| Array shapes are not compatible for Q @ c operation: a has… | validation | error | jax, linalg, shape-mismatch, qr-decomposition |
| convolution dimension_numbers elements must be strings, got | validation | error | jax, convolution, dimension-numbers, type-error |
| convolve2d() only supports 2-dimensional inputs. | exception | error | jax, scipy, convolution, rank-mismatch, shape-validation |
| Default value must be of type str or None, got | exception | error | jax, config, typeerror, enum |
| dtype argument to `triangular` must be a float dtype, got | validation | error | jax, random, dtype-validation |
| First argument to bcoo_extract should be a BCOO array. Got | exception | error | jax, sparse, type-check, bcoo |
| for linearize support, subclass | validation | error | jax, autodiff, linearize, not-implemented |
| fweights must be integer. | exception | error | jax, covariance, weights, dtype-validation |
| MMA lhs tiling does not fit swizzle. | validation | error | jax, pallas, tcgen05, swizzle, tiling, layout |
| cannot accept args which are reduced. Got and axes= | validation | error | jax, sharding, collectives, reduced, spmd |
| out_dtype should be an integer type; got | exception | error | jax, dtype, searchsorted, argument-validation |
| Range is (at least partially) out of bounds for allocation… | exception | error | jax, pallas, mosaic, index-error, out-of-bounds, write |
| Requires libtpu >= 0.1.0 | exception | error | jax, pallas, tpu, libtpu, version, convolution |
| Scalars are not supported in async_store_smem | exception | error | jax, pallas, mosaic-gpu, scalar, async-store, not-implemented |