jax-ml/jax
Documented errors, page 13 of 23. Back to jax-ml/jax
| Code / Message | Type | Severity | Tags |
|---|---|---|---|
| Out-of-bounds masked swap of | exception | error | pallas, mosaic-gpu, interpret-mode, out-of-bounds, atomics, mask |
| psend currently only supports manual sharding | exception | error | jax, psend, shard-map, manual-sharding |
| Second dimension must be divisible by batch_size | validation | error | jax, pallas, shape-validation, divisibility |
| Symbolic dimension not supported for . | exception | error | jax, shape-polymorphism, type-error, coercion |
| TMEM can only be sliced, not indexed | exception | error | jax, mosaic, tmem, slicing, unsupported-operation |
| Unmapped values passed to vmap cannot be sharded along the… | validation | error | jax, vmap, sharding, mesh |
| Unsupported combination of input dtype | exception | error | jax, pallas, tpu, argmax, dtype |
| Unsupported constant type | exception | error | jax, pallas, mlir, constant-folding, internal |
| Unsupported pipeline mode | validation | error | jax, pallas, tpu, pipeline-mode, api-misuse |
| with_layout_constraint in eager mode can only be applied to… | validation | error | jax, layout, eager-mode |
| Breakpoints must be non-negative and less than length of… | exception | error | jax, scipy, detrend, breakpoints, index-out-of-range |
| Can only await on CTA-local barriers | validation | error | jax, mosaic-gpu, barrier, scope, ptx |
| Can't convert to ref | exception | error | jax, pallas, run-scoped, ref, type-error |
| cannot cast to | validation | error | jax, triton, pallas, dtype, cast, gpu |
| Expected mesh of type `jax.sharding.AbstractMesh`. Got type | validation | error | jax, mesh, type-validation, abstract-mesh |
| Indexing into a ()-shaped Ref not yet supported on TPU. | validation | error | jax, pallas, tpu, scalar-ref, smem |
| inline_mgpu_p does not support discharge. | exception | error | jax, pallas, inline-mgpu, discharge, not-implemented |
| Method not recognized | exception | error | jax, scipy, optimize, argument-validation |
| negative dimensions are not allowed, got | validation | error | jax, eye, negative-dimension, valueerror |
| .vjp_bwd should return None or a dict of backward-pass log… | validation | error | jax, autodiff, vjp, api-contract |
| Unimplemented padding mode | exception | error | jnp-pad, unsupported-mode, not-implemented |
| a and b must be 2D, but got | exception | error | jax, pallas, triton, shape, matmul |
| Accumulators are not available on TPU | validation | error | jax, pallas, tpu, hardware, accumulator |
| Argument to to_dlpack must be a jax.Array, got | validation | error | jax, dlpack, type-check, typeguard |
| At most one of batch_group_count and feature_group_count… | validation | error | jax, convolution, grouped-conv, api-misuse |
| Can't instantiate with arguments. | validation | error | jax, pallas, mosaic-gpu, layout, api-misuse, validation |
| clamp requires max.shape == operand.shape or max.shape ==… | validation | error | jax, clamp, shape-mismatch, broadcasting |
| clamp requires min.shape == operand.shape or min.shape ==… | validation | error | jax, clamp, shape-mismatch, broadcasting |
| concatenate dimension out of bounds: dimension | validation | error | jax, concatenate, axis-out-of-bounds, shape-validation |
| Default value must be of type str, got | exception | error | jax, config, typeerror, enum |
| dot_general requires rhs dimension numbers to be… | exception | error | jax, dot-general, dimension-numbers, index-out-of-range |
| Expected a dynamic index or an integer, got | validation | error | jax, mosaic-gpu, barrier, type-mismatch |
| in_specs containing unreduced | exception | error | shard-map, unreduced, sharding-mismatch, jax |
| Mosaic GPU does not yet support AMD ROCm devices. Use… | exception | error | jax, pallas, rocm, amd, gpu, backend, mosaic |
| No valid out swizzle | validation | error | jax, pallas, mosaic-gpu, swizzle, shared-memory, alignment, matmul |
| Offsets have lower rank than tiling | exception | error | jax, mosaic, gpu, tiling, offsets, rank-mismatch |
| Please pass `jax.Array`s with a `NamedSharding` as input to… | exception | error | shard-map, abstract-mesh, named-sharding, jax |
| process_id and num_processes must be nonnegative, with… | validation | error | jax, distributed, rank-validation |
| Remote signal not implemented. | exception | error | pallas, semaphore, remote-signal, not-implemented, jax |
| scan body output must be a pair, got | validation | error | jax, scan, typeerror, return-shape, pytree |
| The dtype of `kv_lens`, `page_indices`, and `cu_q_lens`… | validation | error | jax, pallas, tpu, dtype-validation, paged-attention |
| x.dtype= is not supported, see docstring for supported… | exception | error | jax, spence, dtype, typeerror, dilogarithm |
| Attempting to swap into allocation with `key | exception | error | jax, pallas, mosaic, shared-memory, key-collision, swap |
| bcoo_multiply_sparse: arrays with differing numbers of… | exception | error | jax, sparse, bcoo, not-implemented, dense-dimensions |
| broadcast_in_dim broadcast_dimensions must have length… | validation | error | jax, broadcast-in-dim, shape-validation |
| Cannot swap scalars to VMEM. | exception | error | jax, pallas, tpu, swap, vmem, scalar |
| Cannot tile a transpose | validation | error | jax, mosaic-gpu, transpose, tiling, permutation |
| conv_general_dilated lhs and rhs must have the same number… | validation | error | jax, lax, convolution, shape-mismatch |
| cotangent tree does not match function output, expected | validation | error | jax, linear-transpose, pytree |
| DMA source semaphore must be a Ref. | exception | error | jax, pallas, mosaic, dma, semaphore, ref-vs-array, type-validation |
| Expected but got | validation | error | jax, pallas, shape-mismatch, lowering |
| expected sequence object with len >= 0 or a single integer | exception | error | jax, shape-validation, generator |
| Inferred layout for operand does not match the layout in… | exception | error | jax, mosaic, layout, conflict |
| Invalid argument to dtype | validation | error | jax, dtype, none-argument |
| Invalid trace folder | validation | warning | jax, profiler, perfetto, filesystem |
| need sequence of keys with len > 0 in lexsort | validation | error | jax, lexsort, empty-input, validation, sorting |
| `out_sharding` passed to `broadcast_in_dim` can only… | validation | error | jax, broadcast-in-dim, sharding, gspmd, unreduced |
| pallas_call with a mesh does not support JVP | exception | error | jax, pallas, jvp, autodiff, mesh, spmd, notimplementederror |
| Preinitialized WGMMAAccumulatorRef only supported in… | validation | error | jax, pallas, wgmma, accumulator, api-misuse |
| psend is currently only implemented on GPUs | exception | error | jax, psend, gpu-only, backend |
| Specified which requires a copy since the source device is… | validation | error | jax, dlpack, device-mismatch, copy |
| the first argument to pow must have an inexact dtype (float… | exception | error | jax, lax, pow, dtype-mismatch, type-error |
| the limits of integration in integrate_box_1d must be… | exception | error | jax, scipy, kde, scalar-validation |
| The 'out' argument to jnp.nanvar is not supported. | validation | error | jax, numpy, nanvar, out-argument |
| This operation is implemented for a PjRt-compatible backend… | exception | error | jax, dlpack, ifrt, pjrt, interop |
| `type` argument of array.view() is not supported. | exception | error | jax, view, not-implemented, numpy-compat |
| Unsupported device ID type | exception | error | jax, pallas, device-id, version-mismatch, interpret-mode |
| Unsupported type | exception | error | jax, pallas, mosaic-gpu, aval, type-error |
| A sparse metadata shape mismatch: expected | validation | error | gpu, mosaic, tcgen05, sparsity, shape-mismatch, metadata |
| Array passed to from_dlpack is on unsupported device type | validation | error | jax, dlpack, device-type, interop |
| async done op got , want core.AbstractFuture | exception | error | jax, async-collectives, future, type-error |
| diagonal and off-diagonal values must have same dtype, got | validation | error | jax, linalg, tridiagonal, dtype |
| dtype argument to `maxwell` must be a float dtype, got | validation | error | jax, random, maxwell, dtype, input-validation |
| dtype cannot be None. | validation | error | jax, dtype, none-guard, validation |
| Found multiple equality constraints with the same… | exception | error | jax, shape-polymorphism, constraints, duplicate-key |
| Incorrect output dtype for return value # | exception | error | jax, callback, dtype-mismatch, runtime-validation |
| inf | exception | error | jax, inf, numerical-overflow |
| JAX_SLICE_INDEX has been deprecated. Please use… | console | warning | jax, distributed, environment-variable, deprecation |
| Need to specify backward blocks. | validation | error | jax, pallas, splash-attention, block-sizes, backward |
| put_along_axis arguments 'arr' and 'indices' must have same… | exception | error | jax, put-along-axis, ndim, shape-mismatch |
| ragged_dot_general requires rhs group dimension numbers to… | validation | error | jax, ragged-dot, dimension-numbers, shape-validation |
| select_and_scatter got inconsistent window_strides and… | validation | error | jax, shape-validation, select-and-scatter, pooling |
| shift must be non-negative. | exception | error | jax, pallas, mosaic, roll, argument-validation, shift |
| stop_gradient only works on valid JAX arrays, but input… | validation | error | pytree, jax, pickle, version-mismatch |
| The slice shape must be a multiple of the tile shape. The… | validation | error | mosaic, gpu, slicing, tile-alignment |
| Too many indices: array is | validation | error | jax, indexing, rank, shape-mismatch |
| two-argument mode is available only when mode=='psd' | exception | error | jax, scipy, internal-api, spectral-analysis |
| Unsupported block size type | exception | error | pallas, mosaic-gpu, blockspec, not-implemented |
| Unsupported device id type | exception | error | pallas, device-id, not-implemented, jax |
| Unsupported dtype | exception | error | jax, pallas, mosaic-gpu, integer-pow, complex-dtype, not-implemented |
| Unsupported in/out transforms. In transform | exception | error | jax, mosaic-gpu, transpose, tile-transform, not-implemented |
| Unsupported index type | exception | error | jax, pallas, indexing, type-dispatch, tiling |
| Aval memory space does not match the requested memory space… | exception | error | jax, memory-space, aval |
| Can only infer one dimension | validation | error | jax, mosaic-gpu, memref, reshape, invalid-argument |
| Can only load scalars from SMEM | validation | error | jax, pallas, tpu, smem, shape-mismatch |
| cmp must be scalar. | validation | error | jax, pallas, triton, atomic-cas, shape |
| Default value must be of type | exception | error | jax, config, typeerror, enum |
| dot_general requires lhs batch dimensions to be distinct… | exception | error | jax, dot-general, dimension-numbers, batch-dims |
| expected x and y to have same length | validation | error | jax, numpy, polynomial, polyfit, shape-mismatch |
| in_specs containing unreduced_kind | exception | error | shard-map, unreduced-kind, sharding-mismatch, jax |