jax-ml/jax
Documented errors, page 9 of 23. Back to jax-ml/jax
| Code / Message | Type | Severity | Tags |
|---|---|---|---|
| jnp.insert(): index array must be integer typed; got | exception | error | jax, insert, dtype, indexing |
| method can only be 'linear', 'lower', 'higher', 'midpoint'… | validation | error | jax, numpy, quantile, invalid-argument |
| argument type error | validation | error | jax, dot-general, dtype-mismatch, mixed-precision |
| non-bool prevent_cse together with… | error_code | error | pytree, jax, structure-mismatch, from-iterable |
| Requesting collective allocations, but no explicit thread… | exception | error | pallas, mosaic-gpu, collective, mesh, thread-axis, config |
| Scope not supported | exception | error | profiling, mosaic, gpu, immutability |
| Shape along concat dimension | validation | error | jax, pallas, concatenate, block-divisibility, valueerror |
| structure of the differentiated function | validation | error | jax, vjp, autodiff, pytree |
| The global_shards property was called on | exception | error | jax, concretization, tracer, sharding, global-shards |
| returned structured residuals from `vjp_fwd`, which… | validation | error | jax, autodiff, vjp, api-contract |
| Unknown cluster axis | exception | error | jax, pallas, mosaic-gpu, cluster, axis-name, mesh |
| vary_unreduced_cast only accepts inputs that are varying… | exception | error | jax, named-axes, manual-parallelism, aval-validation |
| Wrong number of explicit pads for convolution: expected | validation | error | jax, convolution, padding, shape-validation |
| Accumulator and RHS have incompatible shapes. Expected RHS… | validation | error | jax, pallas, tcgen05, mma, shape-mismatch |
| Cannot specify scope to a non-uniform debug_print. | validation | error | gpu, mosaic, debug-print, argument-validation |
| copy_gmem_to_smem with a barrier is only supported Hopper… | exception | error | jax, pallas, mosaic-gpu, gpu-architecture, tma, barrier, cp-async |
| Could not find location in string | validation | warning | jax, pallas, tpu, error-handling, parsing |
| CSR.tree_unflatten: invalid | exception | error | jax, sparse, csr, pytree, serialization |
| does not accept dtype . Accepted dtypes are subtypes of… | validation | error | jax, lax, dtype, cumsum |
| get_global discharge is not supported in interpret mode. | exception | error | pallas, get-global, interpret-mode, not-implemented, jax |
| Index map function for must return integer scalars. Output[… | validation | error | jax, pallas, index-map, dtype |
| indices batch dimensions not compatible for | exception | error | jax, sparse, bcoo, indices, shape-validation |
| Invalid compute type | exception | error | jax, mlir, input-validation, enum, compute-type |
| ragged_dot_general requires zero group dimensions in the… | validation | error | jax, ragged-dot, shape-validation, moe |
| Sharding has to be concrete when layout is of type | validation | error | jax, layout, sharding, validation |
| State effect not supported in vmap-of-cond. | exception | error | jax, vmap, cond, state-effects, experimental |
| The bitwidth of `x` must match the bitwidth of… | exception | error | jax, pallas, tpu, dtype, bitwidth |
| The format string expects | exception | error | pallas, format-string, debug-print, jax |
| The manually assigned | exception | error | tpu, pallas, collective-id, barrier, conflict |
| unreduced for prod is not yet supported. | exception | error | jax, pallas-sharding, notimplementederror, distributed |
| WriteEffect on input buffer | exception | error | jax, pallas, effects, input-mutation |
| All arrays must have the same rank, got | validation | error | jax, mosaic-gpu, validation, rank-mismatch |
| boolean index did not match shape of indexed array in index | validation | error | jax, boolean-mask, shape-mismatch, indexing |
| Expected the trailing dimension of the metadata to be 2, got | validation | error | jax, pallas, mosaic-gpu, sparse, tcgen05, shape, last-dim |
| function is not returning an array of the correct shape | exception | error | jax, apply-along-axis, shape-mismatch |
| Grouped convolutions are not supported on Pallas Mosaic TPU… | exception | error | jax, pallas, tpu, convolution, grouped-conv, depthwise |
| Incorrect output shape for return value # | exception | error | jax, callback, shape-mismatch, runtime-validation |
| Input must be a rank-1 array. | validation | error | jax, numpy, polynomial, roots, shape-validation |
| Last 2 dimensions of the array must be square | exception | error | jax, linalg, matrix-power, square-matrix-required |
| Only float32 and int32 results are supported, got | exception | error | jax, pallas, tpu, matmul, dtype |
| Only one of `core_axis` or `core_axis_name` can be set. | exception | error | jax, pallas, mutually-exclusive-args, validation |
| Only the type=1 case of eigh is implemented. | validation | error | jax, linalg, eigenvalue, not-implemented |
| Out-of-bounds swap of | exception | error | pallas, mosaic-gpu, interpret-mode, out-of-bounds, atomics, swap |
| Output buffering does not support lookahead. | exception | error | jax, pallas, pipeline, lookahead, output-buffer |
| Q head_dim must be the same as that of K/V . | validation | error | jax, pallas, tpu, ragged-attention, head-dim-validation |
| Scoped allocation must have the thread axis in its… | exception | error | jax, mosaic-gpu, run-scoped, collective-axes |
| The context mesh cannot be empty. Use `jax.set_mesh(mesh)`… | exception | error | jax, shard-map, mesh, context, distributed |
| array ref with memory space only works inside of a `jit`. | exception | error | jax, refs, memory-space, jit, not-implemented |
| Array shapes are not compatible for c @ Q operation: a has… | validation | error | jax, linalg, shape-mismatch, qr-decomposition |
| Eager shard_map + unreduced/reduced + partial manual is not… | exception | error | shard-map, eager, not-implemented, unreduced, jax |
| Encountered unexpected shape dimension | exception | error | jax, export, shape-polymorphism, type-error |
| Index map function for must return values to match … | validation | error | jax, pallas, block-spec, index-map |
| len() of unsized object | exception | error | jax, numpy, len, zero-d-array |
| mTLS for the JAX distributed service requires jaxlib 0.11.2… | exception | error | jax, distributed, mtls, version-mismatch |
| Not implemented: bitcast 1D | exception | error | jax, pallas, mosaic, bitcast, dtype, shape-validation |
| reduction_op not supported | exception | error | pallas, mosaic-gpu, interpret-mode, reduction, not-implemented |
| has a layout that does not match the expected layout . | validation | error | mosaic, gpu, tmem, layout-mismatch |
| start_indices arguments to dynamic_update_slice must be… | exception | error | jax, lax, dynamic-update-slice, scalar-indices |
| Symbolic dimension comparison | exception | error | jax, shape-polymorphism, comparison, inconclusive |
| The problem arose with the | exception | error | jax, tracer, concretization, jit, float-conversion |
| trunc_div supports only integer types, got | validation | error | jax, mosaic-gpu, division, dtype-check, integer-only |
| Unfolding dimensions is not supported when commuting an … | exception | error | jax, pallas, mosaic-gpu, reshape, unfolding |
| Unsupported core type | validation | error | jax, pallas, tpu, memory-space, lowering, version-skew |
| a_scale and b_scale must both be present or absent. | validation | error | jax, pallas, tcgen05, block-scaling, argument-validation |
| A ShapeDtypeStruct does not have a value and cannot be used… | exception | error | jax, shape-dtype-struct, constants |
| Acc ref must be at least 2D, got shape | validation | error | jax, pallas, tpu, mosaic, accumulator, shape |
| All arrays must have the same signedness, got | validation | error | jax, mosaic-gpu, validation, signedness |
| is not available; available versions are | exception | error | jax, array-api, version-mismatch |
| Array.__contains__: search array must be one-dimensional… | exception | error | jax, membership, shape-mismatch, contains |
| Async store only supports TMEM refs | validation | error | jax, pallas, tmem, memory-space, blackwell |
| mixes JAX mesh and Pallas mesh grid axes | exception | error | pallas, mesh, collectives, axis-name, not-implemented, jax |
| Cannot interpret value of type | exception | error | jax, type-error, abstract-value, tracing, dtype |
| concatenate expects at least one operand, got 0. | validation | error | jax, concatenate, shape-validation, empty-input |
| JVP with aliasing not supported. | exception | error | jax, pallas, jvp, autodiff, input-output-alias, notimplementederror |
| KDEs are not the same dimensionality | exception | error | jax, scipy, kde, dimensionality, shape-validation |
| mode must be one of ['full', 'same', 'valid'] | exception | error | jax, scipy, signal, convolution, argument-validation |
| was requested to map a value of non-array type along axis … | validation | error | jax, vmap, in-axes, type-error |
| Negative orders for normalization is not implemented yet. | exception | error | jax, scipy-special, legendre, not-implemented, spherical-harmonics |
| Packing must be a power of 2, got | exception | error | jax, mosaic, tmem, layout, argument-validation |
| Stack with Element indexing is not yet supported. | exception | error | jax, pallas, stack, element-indexing, not-implemented |
| static_slice: indices must be static scalars or slices. Got… | validation | error | jax, static-slice, advanced-indexing |
| Symbolic dimension cannot be raised to negative powers | exception | error | jax, shape-polymorphism, pow, negative-exponent |
| transpose with implicit broadcasting of unshaped values. Got | exception | error | jax, autodiff, vjp, dynamic-shapes, transpose, type-error |
| Unsupported block dimension type | validation | error | jax, pallas, block-spec, type-validation |
| Array.__contains__: query value must be a scalar, got | exception | error | jax, membership, scalar-required, contains |
| axis_types passed to | validation | error | jax, mesh, axis-types, type-validation |
| can only convert from an extended dtype on an array… | exception | error | jax, extended-dtype, abstract-eval, typeerror |
| Cannot specify both scratch_shapes and scratch_types. Use… | exception | error | jax, pallas, mosaic-gpu, deprecation, scratch-memory |
| head_dim of Q must be the same as that of K/V. Got | validation | error | jax, pallas, tpu, paged-attention, head-dim-validation |
| Insufficient space to capture a full trace | exception | error | profiling, mosaic, gpu, buffer-overflow |
| More than one location found in string | validation | error | jax, pallas, tpu, error-handling, parsing |
| partial_mask_blocks must be of type np.bool_ but got | validation | error | jax, pallas, tpu, splash-attention, dtype, mask |
| Please install the `etils[epath]` package to specify a… | error_code | error | jax, compilation-cache, optional-dependency, cloud-storage |
| size must be a positive integer; got | exception | error | jax, nonzero, argument-validation |
| unexpected JAX type (e.g. shape/dtype) for gradient ref… | validation | error | jax, vjp, refs, gradient-accumulation, dtype |
| Unsupported transforms for RHS | exception | error | jax, pallas, tcgen05, transform, not-implemented |
| Value ' ' with dtype is not a valid JAX array type. Only… | validation | error | jax, dtype, string-dtype, validation |
| Aliasing of scalar prefetch arguments is not currently… | validation | error | jax, pallas, tpu, aliasing, scalar-prefetch, interpret-mode |
| : arrays must have same number of dimensions, got . | exception | error | jax, sharding, named-sharding, rank-mismatch, type-error |
| Can only perform collective MMA along minormost cluster… | validation | error | jax, pallas, tcgen05, cluster, collective-mma, axis-order |