jax-ml/jax
Documented errors, page 3 of 23. Back to jax-ml/jax
| Code / Message | Type | Severity | Tags |
|---|---|---|---|
| with condition(s), either or functions are expected; got | exception | error | jax, piecewise, list-length-mismatch |
| Input type is incompatible with `preferred_element_type`… | exception | error | jax, dot-general, preferred-element-type, dtype-validation |
| Invalid value received for the layout argument. Expected… | validation | error | jax, layout, type-validation, isinstance-check |
| run_scoped interpret rule does not support collective axes | exception | error | jax, pallas, collectives, interpreter, notimplemented |
| TMEM aliasing only supported for Refs with the same first… | exception | error | jax, pallas, tmem, tcgen05, aliasing, shape-mismatch |
| top_level_all_gather doesn't allow input | validation | error | jax, all-gather, shard-map, resharding |
| Unsupported dot precision | exception | error | jax, pallas, triton, precision, validation |
| When at least one mesh axis of `pred` is in auto mode… | console | warning | jax, distributed, spmd, sharding, error-check |
| Cannot interpret ' ' as a data type.\n\nDid you… | exception | error | jax, shape-validation, dtype, typo-detection |
| Loading from an accumulator is not supported. Use… | validation | error | jax, pallas, tpu, accumulator, matmul |
| ` ` entries must be the same shape | exception | error | jnp-pad, shape-mismatch, numpy |
| One input must be smaller than the other in every dimension. | exception | error | jax, scipy, convolution, shape-mismatch |
| out_specs already specified | validation | error | jax, colocated-python, api-misuse, configuration |
| pallas_call with a mesh does not support batching | exception | error | jax, pallas, vmap, mesh, sharding, not-implemented |
| Missing required keyword argument: 'in_sharding' | validation | error | jax, sharding, missing-argument, pjit |
| `preferred_element_type` must not be narrower than the… | exception | error | jax, dot-general, precision, dtype-validation |
| : Buffers with a memory space of HBM or ANY cannot be… | validation | error | jax, pallas, tpu, hbm, vmem, dma |
| requires 8-, 16-, 32- or 64-bit field width. | exception | error | jax, prng, threefry, bit-width, dtype |
| can only convert to an extended dtype on an array type,but… | exception | error | jax, extended-dtype, abstract-eval, dynamic-shape, typeerror |
| can only convert to extended dtype from an array with… | exception | error | jax, extended-dtype, sharding, partition-spec |
| Invalid sharding for `addupdate`. Ref sharding | validation | error | jax, sharding, spmd, state-primitives |
| Minor dimension of shape must be divisible by packing, got | exception | error | jax, mosaic, tmem, layout, shape-validation |
| Non-debug checks are not supported by the Mosaic GPU… | exception | error | pallas, checkify, mosaic-gpu, not-implemented |
| The length of f along the last axis must be at least 2; got… | exception | error | jax, leslie-matrix, input-validation |
| unreduced in _specs can only be used when the mesh passed… | exception | error | shard-map, partition-spec, mesh, jax, unreduced |
| Cannot convert_element_type from | exception | error | jax, extended-dtype, dtype-conversion, validation |
| must be (1,) | validation | error | jax, pallas, tpu, ragged-attention, shape-validation |
| Only copies transferring a number of bytes divisible by the… | exception | error | mosaic-gpu, pallas, alignment, warpgroup, copy |
| must be greater or equal to given and . | validation | error | jax, pallas, tpu, ragged-attention, kv-cache, capacity-validation |
| `poisson` with method='exact' is only implemented for the… | validation | error | jax, random, poisson, prng, not-implemented, tpu |
| The traceback property was called on | exception | error | jax, concretization, tracer, traceback, debugging |
| BCSR: cannot add out_axis= | exception | error | jax, sparse, bcsr, vmap, out-axes |
| Cannot merge empty list of indexers | exception | error | pallas, internal-api, indexing |
| Riemann zeta function not implemented; pass q != None to… | exception | error | jax, scipy-special, zeta, not-implemented |
| The following ordered effects are not supported for more… | validation | error | jax, effects, multi-device, ordered-effects, pjit, debug-print |
| The Pallas TPU lowering currently requires that rank 1… | validation | error | jax, pallas, tpu, block-shape, rank-1 |
| Checkify does not support batched while-loops… | exception | error | jax, checkify, vmap, while-loop, transformation-order |
| Invalid padding mode | validation | error | jax, lax, conv-transpose, padding |
| key array cannot be converted to boolean. | exception | error | jax, prng, truthiness, bool |
| Mapped away dimension of inputs passed to vmap should be… | validation | error | jax, vmap, sharding, mesh, distributed |
| module ' ' has no attribute | validation | error | jax, lazy-loading, attributeerror, module-structure |
| zero-size arrays not supported in convolutions, got shapes | exception | error | jax, scipy, convolution, empty-array, shape-validation |
| Accumulator Ref must be collective if collective_axis is… | validation | error | jax, pallas, tcgen05, collective-mma, tmem |
| Invalid spec | exception | error | jax, shard-map, partition-spec, manual-mode |
| is not divisible by | validation | error | mosaic-gpu, memref, alignment, sub-byte-types |
| padding argument to conv_general_dilated should be a string… | validation | error | jax, lax, convolution, padding |
| reshape not supported yet | exception | error | |
| String padding is not implemented for transposed… | validation | error | jax, lax, convolution, transposed-conv, padding |
| conv_general_dilated batch_group_count must divide lhs… | validation | error | jax, convolution, shape-validation, grouped-conv |
| Make sure that the axis_name passed to jax.lax.ppermute is… | exception | error | jax, ppermute, mesh, ordering |
| The error code state and the predicate must be on the same… | validation | error | jax, error-checking, sharding, mesh, distributed |
| Unsigned integer dtype | exception | error | jax, pallas, triton, gpu, matmul, dtype |
| cannot cast to | exception | error | jax, triton, pallas, mlir, cast, not-implemented |
| DMA source/destination/semaphore arguments must be Refs. | exception | error | jax, pallas, mosaic, dma, ref-vs-array, type-validation |
| function traced for returned a value of type , which is not… | validation | error | jax, tracing, type-error, return-value, jit |
| invalid argument side= | exception | error | jax, searchsorted, argument-validation |
| Only collective_axes that include all JAX device mesh | exception | error | jax, pallas, tpu, distributed, device-mesh, multimem |
| .abstract_eval() method should return a tuple or a list iff… | validation | error | jax, custom-primitives, abstract-eval, multiple-results, api-contract |
| The cp.async implementation only supports… | exception | error | jax, pallas, mosaic-gpu, gpu-architecture, out-of-bounds, cp-async |
| Unmatched parenthesis in | validation | error | jax, einshape, einsum-notation, parse-error, parentheses |
| Unsupported bitwidth | exception | error | jax, mosaic-gpu, layout-cast, bitwidth, wgmma, unsupported-dtype |
| `a` array must be integer typed | validation | error | jax, numpy, dtype, valueerror, choose |
| can only convert to extended dtype from an array of its… | exception | error | jax, extended-dtype, shape-validation, rank-mismatch |
| Cannot apply with to memref with shape . | validation | error | jax, mosaic-gpu, smem-transforms, swizzle, tiling, shape-mismatch |
| Derivatives not defined for partial eigen decomposition. | validation | error | jax, autodiff, eigh, subset-by-index, gradients |
| Indexer must have integer or boolean type, got indexer with… | validation | error | jax, indexing, dtype |
| Invalid TMEMLayout | validation | error | jax, pallas, mosaic-gpu, tmem-layout, enum, version-mismatch |
| logaddexp2 requires floating-point or complex inputs; got | validation | error | jax, dtype, lax, validation |
| When used within transformed code… | exception | error | jax, prng, stateful-rng, jit, tracing |
| Called with a float0 at position . float0s do not support… | exception | error | jax, float0, autodiff, binop, type-error |
| can only convert to extended dtype from its representation… | exception | error | jax, extended-dtype, dtype-conversion, representation |
| unsupported mode | exception | error | jax, scipy, convolution, invalid-argument-value |
| allow_singular argument of multivariate_normal.logpdf | exception | error | jax, scipy, multivariate-normal, not-implemented, covariance |
| fill_value shape is not broadcast-compatible with batch… | exception | error | jax, nonzero, fill-value, broadcasting |
| No pxla_result_handler for type | error_code | error | jax, pxla, custom-aval, result-handler, extension-api |
| Only collective_axes that include all JAX device mesh axes… | exception | error | jax, pallas, mosaic-gpu, collective-axes, device-mesh, multicast |
| Padding with aliasing not supported. | exception | error | jax, pallas, aliasing, padding, interpreter |
| async_store_smem requires a tiled and swizzled ref | exception | error | jax, pallas, mosaic-gpu, tiled-layout, swizzle, async-store |
| bool() not supported for instances of type | exception | error | jax, config, typeerror, truthiness |
| callbacks do not support specifying sharding inside spmd… | exception | error | jax, spmd, sharding, pure-callback, manual-axes |
| `dimension` must be smaller than the rank of the array. | exception | error | jax, mosaic-gpu, iota, dimension-validation, shape-mismatch |
| invalid value ( ) encountered in ndtri. | exception | error | jax, scipy-special, nan, ndtri, floating-point |
| No Python scalar type for | exception | error | jax, dtype, item, extended-dtype |
| Passing lowering_platforms via jax.export or… | validation | error | jax, abstract-mesh, lowering-platforms, export, aot, pjit |
| tensorinv is only possible when the product of the first… | exception | error | jax, scipy, linalg, tensor, shape-validation |
| Unsupported block dimension type | exception | error | jax, pallas, tpu, block-shape, api-misuse |
| AbstractMesh size: does not match the device assignment size | validation | error | jax, sharding, abstract-mesh, device-assignment, pjit |
| Can only use `unroll` in `fori_loop` if the loop bounds are… | validation | error | jax, fori-loop, unroll, static-shape, traced-values |
| cond_fun must return a boolean scalar, but got output… | validation | error | jax, while-loop, cond-fun, dtype, scalar |
| In order to best JIT compile `rankdata`, we cannot know… | exception | error | jax, scipy, stats, rankdata, jit, not-implemented, nan-policy |
| lam shape must be broadcastable to shape argument; got… | validation | error | jax, random, poisson, broadcasting, shape |
| No support for vmapping over nontrivial slices just yet | exception | error | jax, vmap, batching, not-implemented, slicing |
| two-arguments must have the same rank | exception | error | jax, scipy, csd, rank-mismatch, shape-validation |
| Cannot convert from to | exception | error | jax, mosaic-gpu, layout-conversion, not-implemented, wgmma |
| Cannot determine the ``__name__`` of the caller. | error_code | error | jax, lazy-loading, import-machinery, internal-api |
| Expected source shape to be | exception | error | jax, pallas, mosaic-gpu, shape-mismatch, sharding, gmem-copy |
| Out-of-bounds read of | exception | error | pallas, mosaic-gpu, interpret-mode, out-of-bounds, jax |
| raise_if_error() should not be called within a traced… | validation | error | jax, error-checking, jit, tracer |
| Shape polymorphism is supported for jnp.pad with 'reflect'… | exception | error | jnp-pad, shape-polymorphism, symbolic-dimensions, reflect |
| 'trans' value must be 0, 1, or 2, got | validation | error | jax, lu-solve, invalid-argument, api-misuse |