jax-ml/jax
Documented errors, page 5 of 23. Back to jax-ml/jax
| Code / Message | Type | Severity | Tags |
|---|---|---|---|
| expected deg >= 0 | validation | error | jax, numpy, polynomial, polyfit, argument-validation |
| Failed to infer the output layout of multimem_load_reduce… | exception | error | jax, pallas, layout, multimem, tpu |
| Invalid equality constraint | exception | error | jax, shape-polymorphism, constraints, normalization |
| Out-of-bounds block index | validation | error | jax, pallas, tpu, blockspec, out-of-bounds |
| remat optimization for custom_vjp does not support forward… | exception | error | jax, custom-vjp, remat, side-effects |
| `thread_id` must be zero when allocating a buffer for all… | exception | error | jax, pallas, mosaic-gpu, interpret-mode, allocation, warp-specialization |
| Tuple arity mismatch | validation | error | jax, pytree, arity-mismatch, tuple |
| All block dimensions must be Elements or none of them can… | validation | error | jax, pallas, tpu, block-shape, element-window |
| Can only store scalars in warp-level lowering. | exception | error | jax, pallas, mosaic-gpu, warp-semantics, store |
| dynamic grid bounds not supported in the Mosaic GPU backend | exception | error | pallas, mosaic-gpu, grid, not-implemented |
| Expected to be ( ,) where `max_num_seqs` is… | validation | error | jax, pallas, tpu, ragged-attention, shape-validation |
| get not supported yet for block shape | exception | error | jax, pallas, element-indexing, get, not-implemented |
| new string config value must be of type str, got | validation | error | jax, config, typeerror, string |
| preduced is a Invariant->Reduced collective. This means… | exception | error | jax, sharding, collectives, transpose, mesh |
| has no `sharding`. | exception | error | jax, aval, attribute-error, tokens |
| The is_fully_addressable property was called on | exception | error | jax, concretization, tracer, sharding, distributed |
| The 'out' argument to jnp. | validation | error | jax, numpy, cumsum, cumprod, out-argument |
| Total q tokens must be less or equal to . | validation | error | jax, pallas, tpu, ragged-attention, cumulative-length-validation |
| unsupported pointer type | exception | error | jax, triton, pallas, load, pointer, validation |
| Unsupported TMEM ref | exception | error | jax, pallas, tmem, tcgen05, aliasing, not-implemented |
| But the tree structures differ: | validation | error | jax, vjp, pytree, refs |
| Cannot map in_axis= for BCSR array with n_batch= . in_axes… | exception | error | jax, sparse, bcsr, vmap, batching |
| Cannot slice TMEM with multiple tiles along rows. | exception | error | jax, mosaic, tmem, slicing, layout, unsupported-operation |
| __dlpack__ only supported for unsharded arrays. | exception | error | jax, dlpack, sharding, buffer-protocol, interop |
| Expected assume_a to be one of | validation | error | jax, linalg, solve, argument-validation, version-migration |
| must be a pytree prefix with bool leaves or a tuple-tree of… | validation | error | jax, pytree, validation, bool-prefix |
| Length of sharding.spec | exception | error | jax, sharding, partition-spec, shape-mismatch |
| Ref unions can't be assigned to. | exception | error | jax, pallas, mosaic-gpu, ref-union, immutable, read-only |
| Sharding on sequence dim is not allowed. | validation | error | jax, sharding, sequence-parallel, spmd, cudnn |
| TMEM layout is not supported | exception | error | jax, mosaic, tmem, load, layout, unsupported-operation |
| Dimensions with parallel semantics must form a prefix of… | validation | error | jax, pallas, tpu, grid, dimension-semantics |
| Explicitly requested dtype | validation | error | jax, x64, dtype, configuration |
| Get only supports loading scalars from SMEM. | exception | error | jax, pallas, tpu, sparsecore, scalar-load |
| MemoryRef type must be a ShapedArray, got | validation | error | jax, pallas, aval, type-validation |
| Must provide valid mode for gumbel got | validation | error | jax, random, gumbel, mode, input-validation |
| Attempting to convert array of shape | exception | error | jax, bitcast, dtype, shape-validation |
| coordinator_bind_address should be defined. | validation | error | jax, distributed, network, bind-address |
| ddof and correction can't be provided simultaneously. | validation | error | jax, var, ddof, parameter-conflict |
| is not a valid dtype for JAX type promotion. | validation | error | jax, type-promotion, dtype, lattice |
| matmul between two sparse objects. | exception | error | jax, sparse, coo, matmul, not-implemented |
| Not implemented: accuracy | exception | error | jax, pallas, mosaic-gpu, rsqrt, accuracy, not-implemented |
| points and xi must have same trailing dim | exception | error | jax, scipy, kde, shape-validation |
| collective attribute is only defined for TMEM refs | exception | error | jax, pallas, mosaic-gpu, tmem, collective-ops |
| Could not find memory addressable by device | validation | error | jax, sharding, memory-kind, tpu |
| Expected kind to be on of | exception | error | jax, scipy, linalg, argument-validation |
| () output shapes must match , got and | validation | error | jax, custom-linear-solve, shape-mismatch, linear-algebra |
| PRNG keys must be loaded from SMEM. Did you set the memory… | validation | error | jax, pallas, tpu, prng, random, smem |
| reduction operation does not have an identity, so to use a… | validation | error | jax, ufunc, reduce, where-mask |
| shard_map _specs argument must be a pytree of… | exception | error | shard-map, partition-spec, typeerror, jax |
| Unexpected unhandled transforms | exception | error | jax, pallas, mosaic-gpu, internal-invariant, transforms, async-store |
| You can't use two different ScalarSubcoreMeshes. | exception | error | jax, pallas, tpu, sparsecore, mesh-compat |
| Array dtype mismatch: expected | validation | error | jax, pallas, inline-mgpu, dtype-mismatch |
| Axis does not refer to a GPU mesh axis (available axes: )… | exception | error | jax, pallas, mesh, collective, axis-name |
| cannot select an axis to squeeze out which has size not… | validation | error | jax, squeeze, size-one-required, shape-validation |
| conv_general_dilated lhs feature dimension size divided by… | validation | error | jax, lax, convolution, feature-group-count, kernel-shape |
| devices already specified | validation | error | jax, colocated-python, devices, api-misuse |
| k_pages and v_pages must have the same shape. Got | validation | error | jax, pallas, tpu, paged-attention, shape-validation |
| Must semaphores of the following types: . Got . | exception | error | pallas, semaphore, dtype-validation, jax |
| Pure callbacks do not support transpose. Please use… | exception | error | jax, autodiff, vjp, transpose, pure-callback, custom-vjp |
| serializing DisabledSafetyCheck | exception | error | jax, export, serialization, flatbuffers |
| tensordot requires both axes lists to be either ints… | validation | error | jax, tensordot, type-error, axes |
| The names should be exclusive and should not intersect in… | validation | error | pytree, namedtuple, jax, pickle, tree-flatten-with-path |
| unreduced for max is not yet supported. | exception | error | jax, sharding, notimplementederror, distributed |
| Unsupported reduction for f32. Only min, max, absmin, and… | exception | error | mosaic, gpu, tcgen05, reduction, tensor-core, jax |
| can only convert from extended dtype to its representation… | exception | error | jax, extended-dtype, dtype-conversion, representation |
| cannot splat a tensor | validation | error | jax, pallas, triton, splat, scalar-shape, reshape, typeerror |
| Cluster barriers are not arrays | validation | error | jax, pallas, barrier, cluster, api-misuse |
| Expected dict, got . | validation | error | jax, pytree, type-mismatch, dict |
| Only byte-aligned bitcasts are supported. | exception | error | jax, pallas, mosaic-gpu, smem, bitcast, alignment, sub-byte |
| Shardy is used, but sharding propagation callbacks instead… | exception | error | jax, shardy, custom-partitioning, migration, tpu |
| Sparse meta layout stores unsupported. | exception | error | mosaic, gpu, tcgen05, sparse, not-implemented, tensor-memory, jax |
| the number of data points must exceed order to scale the… | validation | error | jax, polyfit, covariance, insufficient-data |
| Array inputs to associative_scan must have the same first… | validation | error | jax, associative-scan, shape-mismatch |
| Attempting to get contents of allocation with key | exception | error | jax, pallas, mosaic, shared-memory, key-collision, interpret-mode |
| betainc does not support complex-valued inputs. | validation | error | jax, scipy-special, complex-dtype, dtype-validation |
| Cannot bitcast a ()-shaped array to a dtype with a… | exception | error | jax, pallas, sparsecore, bitcast, scalar |
| cuDNN doesn't support right window | validation | error | jax, nn, attention, cudnn, sliding-window, gpu |
| dtype argument to `t` must be a float dtype, got | validation | error | jax, random, t-distribution, dtype, input-validation |
| expi does not support complex-valued inputs. | exception | error | jax, expi, complex-numbers, unsupported-dtype |
| in op must not repeat; got: . | exception | error | jax, lax, gather, scatter, duplicate-dims |
| Non-zero offset is not supported for ptr_as_memref | validation | error | gpu, mosaic, memref, offset, llvm |
| prevent_cse must be a bool or tuple of bools, got | validation | error | pytree, unflatten, jax, leaf-count |
| Commuting a `UntilingTransform` with a `ReshapeTransform`… | exception | error | jax, pallas, mosaic-gpu, reshape, transforms |
| multi_dot: input arrays must all be two-dimensional, except… | exception | error | jax, numpy, linalg, matmul, shape-validation |
| pad operand and padding_value must be same dtype: got | validation | error | jax, pad, dtype-mismatch |
| threefry_2x32 requires uint32 arguments, got | exception | error | jax, prng, threefry, uint32, dtype |
| Unsupported conv precision | exception | error | jax, pallas, tpu, convolution, precision |
| Unsupported register bitwidth | exception | error | jax, mosaic, gpu, ptx, register-width, not-implemented |
| Argument ' ' of type ' ' is not a valid JAX type | exception | error | jax, type-error, jit, argument-validation |
| Can't transpose the swizzled dimension. | exception | error | jax, pallas, mosaic-gpu, transpose, swizzle |
| Every block dimension must be either a multiple or factor… | exception | error | jax, pallas, tile, divisibility, shape-mismatch |
| invalid entry in choice array | validation | error | jax, choose, index-out-of-range, valueerror |
| Malformed pytree proto (invalid node type) | validation | error | jax, pytree, serialization, protobuf |
| Number of tensordot axes | validation | error | jax, tensordot, shape-mismatch, numpy |
| 'select must have a value in | exception | error | jax, argument-validation, linalg, eigenvalues |
| subset_by_index must be a tuple of size 2. | exception | error | jax, tpu, svd, linalg, argument-validation |
| The first argument to householder_product must have at… | validation | error | jax, linalg, householder, shape-validation, qr |
| Expected base_key to be a typed PRNG key; got | exception | error | jax, prng, stateful-rng, typed-key |
| Expected value and mask to have the same shape, but got… | validation | error | jax, pallas, tpu, shape-mismatch, masked-store |
| Factor is not used in the rule, but size is provided | exception | error | jax, sharding, factor-sizes, validation |