jax-ml/jax
Documented errors, page 8 of 23. Back to jax-ml/jax
| Code / Message | Type | Severity | Tags |
|---|---|---|---|
| np.delete(arr, obj): for boolean indices, obj must be… | exception | error | jax, numpy-compat, boolean-mask, shape-mismatch |
| Only one of src or dst can be a BufferedRef. | exception | error | jax, pallas, pipeline, copy |
| Only tiled and WG strided layouts are supported by… | validation | error | jax, pallas, layout, multimem, unsupported-layout |
| stack expects at least one operand, got 0. | validation | error | jax, stack, empty-input, shape-validation |
| strides must be a sequence of positive integers; got | exception | error | jax, sparse, bcoo, strides, positive-value-required |
| Swizzle requires the trailing dimension to be of size , but… | exception | error | jax, pallas, mosaic-gpu, swizzle, layout, shape-mismatch |
| TPU version must be 4 or higher. | exception | error | jax, pallas, tpu, hardware-version, not-implemented |
| Addition between sparse matrices with different batch/dense… | exception | error | jax, sparse, sparsify, addition, batch-dims |
| All arrays must have matching shapes along non-concatenated… | validation | error | jax, mosaic-gpu, validation, shape-mismatch |
| all keys need to be the same shape | validation | error | jax, lexsort, shape-mismatch, validation, sorting |
| Inferred layout not found for operand | exception | error | jax, mosaic, layout-inference, internal |
| input must be 1d or non-empty square 2d array. | validation | error | jax, numpy-poly, input-shape |
| jax.distributed.initialize() must be called before any JAX… | exception | critical | jax, distributed, xla-backend, ordering |
| Mismatched list depths in jax.numpy.block | validation | error | jax, block, shape-mismatch, valueerror |
| has an unsupported layout | exception | error | jax, mosaic, gpu, layout, tiled-layout |
| out_sharding passed to | validation | error | jax, sharding, named-sharding, reduction, distributed |
| must not be 0.0. | validation | error | jax, pallas, tpu, softcap, attention, config-validation |
| Splat layout does not support multimem | exception | error | jax, mosaic-gpu, splat-layout, multimem, tma |
| The kernel function in mpmd_map | exception | error | jax, pallas, mpmd, kernel-api, inplace-semantics |
| Unknown keyword arguments | exception | error | jax, custom-partitioning, shardy, sharding-rule |
| Unrecognized expected one of , or a compatible input for… | validation | warning | jax, dtype, array-api, validation |
| Accumulator aval mismatch: expected | validation | error | jax, vjp, accumulator, internal-api, dtype |
| dma_start not implemented in LoJAX yet. | exception | error | jax, pallas, mosaic, lojax, dma, not-implemented, tpu |
| Index map function for must not capture constants | validation | error | jax, pallas, index-map, closure |
| Invalid axis for operand shape | validation | error | jax, argmax, argmin, axis, validation |
| iteration over a 0-d array | exception | error | jax, numpy-interop, iteration, scalar, typeerror |
| lax.bitcast_convert_type does not support bool or complex… | exception | error | jax, bitcast, complex, bool, typeerror |
| len(strides) = ; expected | exception | error | jax, sparse, bcoo, strides, validation |
| LHS must be a TMEM/SMEM Ref. | validation | error | jax, pallas, tcgen05, smem, tmem, memory-space |
| No layout inference rule defined for | exception | error | jax, mosaic, not-implemented, unsupported-op |
| Only 2D gathers/scatters for async load/store are supported. | validation | error | jax, mosaic-gpu, tma, gather, async-load, rank, layout-inference |
| is not a memref. | validation | error | mosaic, gpu, tmem, mlir, type-validation |
| subset_by_index not supported on CPU and GPU | validation | error | jax, eigh, subset-by-index, cpu, gpu, not-implemented |
| The 'out' argument to jnp.compress is not supported. | exception | error | jax, unsupported-argument, numpy-compat, out-parameter |
| Triggering __jax_array__() during abstractification is no… | exception | error | jax, pytree, jax-array, custom-class, tracing |
| Unsupported swizzle, expected 16, 32, 64 or 128, but got | validation | error | gpu, mosaic, tcgen05, swizzle, validation |
| address_space not supported | exception | error | gpu, mosaic, address-space, nvptx, lowering |
| can only specify one unknown axis size with a `-1` value… | exception | error | jax, reshape, invalid-shape |
| Cannot interpret ' ' as a data type.\n\nDid you… | exception | error | jax, shape-validation, dtype, typo-detection |
| collective argument must be provided when TMEM layout is… | exception | error | jax, mosaic, tmem, missing-argument |
| convolution dimension_numbers | validation | error | jax, convolution, dimension-numbers, shape-validation |
| Dtype is not a valid JAX array type. Only arrays of numeric… | validation | error | jax, dtype, validation, numeric-types |
| Expected a FragmentedArray, but got | validation | error | jax, pallas, fragmented-array, lowering |
| jnp.argpartition for complex dtype is not implemented. | validation | error | jax, argpartition, complex-dtype, not-implemented, sorting |
| mode= not understood. Must be 'raise', 'wrap', or 'clip | validation | error | jax, choose, invalid-argument, valueerror |
| reduce_sum on operand | validation | error | jax, float16, bfloat16, reduction, mixed-precision, config-flag |
| repeated axis in lax.expand_dims | validation | error | jax, lax, expand-dims, axis-canonicalization, duplicate-axis |
| tmem_addr_ref must contain a single element, got | exception | error | jax, mosaic, tmem, memref, shape-validation |
| 0th dimension of all xs should be replicated. Got | validation | error | jax, scan, sharding, multi-device, jaxmesh |
| Barriers are not arrays | validation | error | jax, pallas, barrier, api-misuse |
| convolve2d() only supports boundary='fill', fillvalue=0 | exception | error | jax, scipy, convolution, not-implemented, boundary |
| duplicate value in 'axes' of reduction | validation | error | jax, lax, reduction, axes, duplicate |
| Expected a memref type but got | validation | error | mosaic-gpu, memref, type-validation, shared-memory |
| Mesh of an aval must be an AbstractMesh. Got | exception | error | jax, sharding, mesh, abstract-mesh, type-mismatch |
| mode='slice' is not valid for polymorphic shapes. | validation | error | jax, polymorphic-shapes, export, slicing |
| Not implemented: the 2nd minor dim can not be perfectly… | exception | error | jax, pallas, mosaic, bitcast, dtype, packing, shape-validation |
| only integers, slices (`:`), ellipsis (`...`), newaxis… | validation | error | jax, indexing, type-error |
| Sharding the kv sequence dimension is not supported | validation | error | jax, tpu, splash-attention, sharding, partition-spec |
| The iteration bounds and dimension semantics attributes… | exception | error | tpu, pallas, mlir, subkernel, validation |
| Unknown character | validation | error | jax, sharding, reserved-character, rule-syntax |
| Unsupported reduction for i32. Only min and max are… | exception | error | jax, mosaic, tmem, reduction, dtype, unsupported-operation |
| Cannot pull iota along dimension | exception | error | jax, pallas, iota, block-spec, validation |
| Cannot retrieve the architecture without an insertion point | validation | error | jax, mosaic-gpu, mlir, insertion-point, architecture |
| CollapseShapeOp with empty reassociation is not supported. | validation | error | jax, mosaic-gpu, memref, collapse-shape, layout-inference, not-implemented |
| collective_axes is not supported in pallas_call. Use… | validation | error | jax, pallas, mosaic-gpu, collective-axes, unsupported-feature |
| must form a tree prefix of the corresponding values (up to… | validation | error | jax, pytree, tree-prefix |
| Higher-order AD not supported | exception | error | jax, pallas, tpu, splash-attention, autodiff |
| JAX tried to execute function | exception | error | jax, config, no-execution, environment, diagnostics |
| Only arrays with tiled layouts can be sliced | exception | error | mosaic, gpu, slicing, layout |
| Only workgroup memory is supported but got | exception | error | jax, pallas, mosaic-gpu, smem, memory-space, bitcast |
| philox2x32_random_bits got invalid prng key. | exception | error | jax, prng, philox, key-validation |
| Sharding on head dim is not allowed. | validation | error | jax, sharding, tensor-parallel, spmd, cudnn |
| The inputs to the closure produced by closure_convert must… | exception | error | jax, closure-convert, pytree, typeerror |
| Unsupported type for JSON serialization | exception | error | jax, json-serialization, async-collectives, config |
| Accumulator shape does not match value shape | validation | error | jax, pallas, mosaic-gpu, wgmma, shape-mismatch |
| All arrays must have the same layout, got | validation | error | jax, mosaic-gpu, validation, layout-mismatch |
| axis_index_groups not supported in vmap collectives. Please… | exception | error | jax, vmap, collectives, not-implemented |
| Can not tile strides when tiled dimensions have been… | exception | error | jax, mosaic, gpu, tiling, strides, transpose |
| Can't use ``defjvps`` with ``nondiff_argnums``. | exception | error | jax, custom-jvp, defjvps, api-misuse |
| Custom JVP rule must produce primal and tangent outputs… | exception | error | jax, custom-jvp, tangent-mismatch, multi-output |
| Integer indexing not implemented for tiled dimensions (only… | exception | error | mosaic, gpu, slicing, tiled-layout |
| `lengths` and `q` must have the same batch size | validation | error | jax, pallas, tpu, paged-attention, shape-validation |
| out_specs passed to shard_map should be equal to the… | exception | error | shard-map, unreduced, out-specs, sharding-mismatch, jax |
| packed, collective and layout arguments are only supported… | validation | error | jax, pallas, mosaic-gpu, smem, gmem, tmem, api-misuse |
| pmap in_axes must be an int, None, or (nested) container… | validation | error | jax, pmap, in-axes, type-validation |
| run_scoped_p with collective axes is not supported | validation | error | jax, pallas, tpu, interpret-mode, not-implemented, collectives |
| shardings spec for batch dim should be same, but got lhs | validation | error | jax, fp8, sharding, batch-dim, matmul |
| The delete() method was called on | exception | error | jax, concretization, tracer, memory-management, delete |
| The sharding must divide the mask blocks evenly between… | validation | error | jax, tpu, splash-attention, sharding, dynamic-mask |
| Unsupported cast: -> | exception | error | jax, pallas, tpu, dtype-cast, unsigned-integer |
| Unsupported | exception | error | jax, pallas, tpu, dot-general, preferred-element-type, mixed-precision |
| Bitcast 1D ref with bitwidth change is not supported. | exception | error | jax, bitcast, shape-restriction |
| Can only transfer integer bytes | exception | error | jax, pallas, mosaic-gpu, sub-byte-dtype, alignment, async-store |
| Can't make a peer reference into a multicast reference. | exception | error | jax, pallas, mosaic-gpu, multicast, peer-memory |
| Cannot concatenate arrays with different numbers of… | validation | error | jax, concatenate, rank-mismatch, shape-validation |
| Computation failed on | exception | error | jax, pallas, mosaic, interpret-mode, threading, error-propagation |
| Concatenation with Element indexing is not yet supported. | exception | error | jax, pallas, concatenate, element-indexing, not-implemented |
| convolution dimension_numbers elements must each have the… | validation | error | jax, convolution, dimension-numbers, validation |
| -dimensional array given. Array must be at least… | exception | error | jax, linalg, matrix-power, ndim-validation |
| expected 1D or 2D array for y | validation | error | jax, numpy, polynomial, polyfit, shape-validation |