jax-ml/jax
Documented errors, page 17 of 23. Back to jax-ml/jax
| Code / Message | Type | Severity | Tags |
|---|---|---|---|
| Expected 'output' to be either 'real' or 'complex', got | validation | error | jax, linalg, schur, argument-validation |
| f16/bf16 only supports add, min, max atomics, got | exception | error | jax, mosaic, gpu, atomics, f16, bf16, not-implemented |
| gaussian_kde does not support complex data | exception | error | jax, scipy, kde, complex-dtype |
| Get does not support masked scalar loads | exception | error | jax, pallas, tpu, sparsecore, masked-load |
| Input array shapes must match: Z | exception | error | jax, rsf2csf, shape-mismatch |
| mask.dtype= is not a boolean dtype | exception | error | jax, pallas, sparsecore, scan-count, mask-dtype |
| Mesh requires the ndim of its first argument (`devices`) to… | validation | error | jax, mesh, shape-mismatch, input-validation |
| No input was provided to the clip function. | exception | error | jax, clip, missing-argument |
| Only a single tiling transform is supported when collapsing… | exception | error | jax, mosaic-gpu, collapse-shape, tile-transform, not-implemented |
| Only WGMMA layouts supported in WGMMAAccumulator | validation | error | jax, mosaic-gpu, wgmma, fragmented-array, layout |
| Pytree serialization too large to deserialize. | exception | error | jax, pytree, protobuf, size-limit |
| rel_entr does not support complex-valued inputs. | validation | error | jax, scipy-special, relative-entropy, complex-dtype |
| Slice along axis is out of bounds for shape | validation | error | jax, mosaic-gpu, memref, slice, index-out-of-bounds |
| Unsupported block dimension type | exception | error | jax, pallas, blockspec, type-validation |
| Weights shape must match 'a' shape when axis is None. | validation | error | jax, numpy, quantile, weights, shape-mismatch |
| zero-length scan is not supported in disable_jit() mode… | validation | error | jax, scan, disable-jit, zero-length, edge-case |
| dynamic_slice: only scalar indices allowed. Got index of… | validation | error | jax, indexing, scalar-index, dynamic-slice |
| expected group_sizes to have shape | validation | error | jax, ragged-dot-general, group-sizes, shape-mismatch |
| expected rhs group dimension size to be | validation | error | jax, ragged-dot, shape-mismatch, moe |
| expn does not support complex-valued inputs. | exception | error | jax, expn, complex-numbers, unsupported-dtype, exponential-integral |
| for vmap support, subclass | validation | error | jax, vmap, batching, custom-primitive |
| Found multiple SwizzleTransformAttr | exception | error | jax, mosaic, gpu, swizzle, transforms |
| Indices must not be empty | exception | error | jax, pallas, sparsecore, scatter, empty-indices |
| last two dimensions of LU decomposition must be equal, got… | validation | error | jax, lu-solve, shape-validation, square-matrix |
| mulhi operands must have the same dtype, got | validation | error | jax, dtype, mulhi, type-mismatch |
| Only the TMA implementation supports reductions | validation | error | jax, mosaic-gpu, tma, reduction, unsupported-operation |
| scan got no values to scan over and `length` not provided. | validation | error | jax, scan, missing-length, argument-validation |
| should have a different subcore axis name from the… | validation | error | jax, pallas, tpu, sparsecore, axis-naming |
| the only valid string value of `right` is 'extrapolate'… | exception | error | jax, interp, invalid-argument, sentinel-value |
| the rematted computation's closure contains a mutable array… | validation | error | pytree, jax, registry, compose |
| unknown method | exception | error | jax, scipy, stats, rankdata, argument-validation, method |
| Unsigned integer dtype | exception | error | jax, pallas, tpu, dot-general, unsigned-integer, dtype |
| Wrong number of pads for spatial dimensions | validation | error | jax, numpy-reference, padding, shape-validation, convolution |
| array() takes at most 5 positional arguments but | exception | error | jax, argument-validation, numpy-compat |
| arrive_expect_tx is only supported on Hopper+ hardware | validation | error | jax, mosaic-gpu, barrier, tma, gpu-architecture, hopper |
| dtype must be specified. | validation | error | jax, dtype, argument-validation |
| Expected int32 input, but got | validation | error | jax, dtype, splash-attention, tpu, int64 |
| Expected the same number of in/out transforms, but got | validation | error | jax, mosaic-gpu, collapse-shape, transforms, count-mismatch |
| Jacobi implementation is not supported on CPU | validation | error | jax, eigh, jacobi, cpu, backend, not-implemented |
| JAX requires ml_dtypes version 0.5 or newer; installed… | validation | critical | jax, dependency, version, ml-dtypes |
| Multiple ':' separators for option in… | exception | error | |
| Only 32-bit types supported | validation | error | mosaic-gpu, redux, dtype, hardware-limit |
| only real valued inputs supported for rfft | validation | error | jax, fft, dtype, api-misuse |
| Only single axis reduction supported | exception | error | jax, pallas, tpu, argmax, reduction |
| Only stack along the last dimension is supported in Triton. | exception | error | jax, triton, pallas, stack, axis, not-implemented |
| position is out of range for clock | exception | error | jax, pallas, mosaic, vector-clock, race-detection, config |
| RHS must be an SMEM Ref. | validation | error | jax, pallas, tcgen05, smem, memory-space |
| start and end must fit in int32 | validation | error | jax, pallas, triton, int32-overflow, arange, large-tensors |
| take_along_axis indices must be 1D if axis=None, got shape | validation | error | jax, take-along-axis, shape-mismatch, axis |
| top_k is not compatible with complex inputs. | validation | error | jax, top-k, complex-dtype, value-error, sorting |
| transpose permutation isn't a permutation of operand… | exception | error | jax, sparse, bcoo, transpose, permutation |
| Unimplemented group_offset support. | exception | error | jax, ragged-dot, autodiff, not-implemented |
| Unsupported transforms | exception | error | jax, pallas, tcgen05, fp8, scaling, not-implemented |
| accumulate only supported for functions returning a single… | exception | error | jax, ufunc, accumulate, api-misuse |
| is out of bounds for array of | validation | error | jax, mosaic-gpu, validation, axis-bounds |
| Cannot determine dtype of | validation | error | jax, dtype, array-like, inference-failure |
| cuDNN is not detected. | exception | critical | jax, cudnn, installation, environment, missing-dependency |
| custom_gradient function used with with_logs=True must… | exception | error | jax, custom-gradient, typeerror, vjp |
| custom_vjp symbolic_zeros support with shard_map is not… | exception | error | shard-map, custom-vjp, symbolic-zeros, autodiff, jax |
| `device_id`s must be an int32 value, but got | exception | error | pallas, semaphore, dtype-validation, device-id, jax |
| device_put with explicit device not allowed within… | validation | error | shard-map, device-put, placement, jax |
| dot_general requires lhs contracting dimensions to be… | exception | error | jax, dot-general, dimension-numbers, contraction |
| hankel: r must be at least 1-dimensional, got a scalar. | exception | error | jax, hankel, input-validation, shape-error |
| incompatible numbers of samples and aweights | exception | error | jax, covariance, weights, length-mismatch |
| inline_mgpu_p only supports only SomeLayout and RefType arg… | validation | error | jax, pallas, mosaic-gpu, inline-mgpu, type-validation |
| Input 'Z' must be square. | exception | error | jax, rsf2csf, schur-form, shape-validation |
| integer np.round not implemented for decimals < 0 | exception | error | jax, round, integer-dtype, notimplementederror |
| Invalid , for | validation | error | jax, sparse, bcoo, random, validation |
| `linearize_from_jvp` is a pair of rules, not a single rule… | validation | error | jax, api-misuse, unpacking |
| Only single-thread arrival is supported on pre-Hopper… | validation | error | jax, mosaic-gpu, barrier, gpu-architecture, hopper |
| other requires mask to be provided | exception | error | jax, triton, pallas, load, mask, validation |
| Output layouts do not match condition layouts | validation | error | mosaic, gpu, scf, while-loop, layout-mismatch |
| Per-operand dot precision unsupported | exception | error | jax, pallas, tpu, dot-general, precision |
| primal and tangent arguments to jax.jvp must be tuples or… | validation | error | jax, jvp, autodiff, api-misuse, type-error |
| reduced_vary_cast input cannot be varying across the… | exception | error | jax, named-axes, manual-parallelism |
| should have the same core axis name and number of cores as… | validation | error | jax, pallas, tpu, sparsecore, mesh-compat |
| The product of the major dimensions must be a multiple of | validation | error | jax, pallas, tpu, accumulator, shape, alignment |
| the `static_argnums` argument to `jax.checkpoint` /… | validation | error | pytree, unflatten, jax, leaf-count |
| The two inputs to `nextafter` must have the same dtype, but… | validation | error | jax, pallas, triton, dtype, nextafter |
| Unsupported element type for atomic stores | exception | error | jax, mosaic, gpu, atomics, dtype, not-implemented |
| {vec_bitwidth} | exception | error | mosaic, gpu, ptx, packed-instruction |
| vmapped away explicit mesh axis cannot appear in shard_map… | validation | error | jax, vmap, shard-map, explicit-mesh |
| Cannot provide both 'factory' and 'options' when… | exception | error | |
| Compound factors should be one level, nested brackets are… | validation | error | jax, sharding, compound-factor, rule-syntax |
| Custom VJP bwd rule must produce a tuple but got . | exception | error | jax, custom-vjp, return-type |
| Dynamic indexing not supported in GPU interpret mode | exception | error | pallas, mosaic-gpu, interpret-mode, dynamic-indexing |
| Ellipsis can only be used at the beginning of a dimension | exception | error | jax, shardy, sharding-rule, ellipsis |
| fill_value must be a scalar or a tuple of length | exception | error | jax, nonzero, fill-value, scalar-validation |
| List arity mismatch: != ; list: . | validation | error | jax, pytree, arity-mismatch, list |
| MMA barrier must have orders_tensor_core set to True. | validation | error | jax, pallas, tcgen05, barrier, pipelining |
| out_sharding passed to top_level_all_gather cannot be | validation | error | jax, shard-map, all-gather, sharding-type |
| pmap out_axes must be an int, None, or (nested) container… | validation | error | jax, pmap, out-axes, type-validation |
| pmapped function has static_broadcasted_argnums= | validation | error | jax, pmap, static-argnums, argument-count |
| Replica groups must be equally sized | exception | error | jax, all-to-all, collectives, validation |
| Sparse metadata format not implemented for | validation | error | jax, pallas, mosaic-gpu, sparse, tcgen05, dtype, not-implemented |
| The 'out' argument to jnp.outer is not supported. | validation | error | jax, outer, immutable-arrays, numpy-compat |
| vmapped away explicit mesh axis cannot appear in shard_map… | validation | error | jax, vmap, shard-map, explicit-mesh, out-specs |
| Accumulator must be a TMEM Ref. | validation | error | jax, pallas, tcgen05, tmem, memory-space |
| `alpha` must be one-dimensional; got alpha.shape= | exception | error | jax, scipy, dirichlet, shape-validation |
| `axis_names` argument of shard_map should be of type… | exception | error | jax, shard-map, axis-names, type-error |