jax-ml/jax
Documented errors, page 16 of 23. Back to jax-ml/jax
| Code / Message | Type | Severity | Tags |
|---|---|---|---|
| run_scoped discharge does not support collective_axes yet. | exception | error | pallas, run-scoped, collectives, not-implemented, jax |
| select_and_gather_add tangents and operand shapes must… | validation | error | jax, autodiff, vjp, pooling, shape-validation |
| series_order must be <= 30. | validation | error | jax, scipy-special, log-ndtr, argument-validation |
| sharding_rule callable must produce either an… | exception | error | jax, shardy, sharding-rule, custom-partitioning |
| Strides have lower rank than tiling | exception | error | jax, mosaic, gpu, tiling, strides, rank-mismatch |
| The 'diff_n' argument to jax.scipy.special.sph_harm_y is… | exception | error | jax, scipy, spherical-harmonics, not-implemented, version-mismatch |
| Unsupported accumulator dtype | exception | error | tcgen05, matmul, mosaic, gpu, dtype |
| unsupported cache modifier | exception | error | jax, triton, pallas, load, cache-modifier, validation |
| `addupdate` must be called on `Ref` types | validation | error | jax, state-primitives, type-error, addupdate |
| custom_jvp inputs marked with nondiff_argnums must be… | exception | error | jax, custom-jvp, tracer-error, nondiff-argnums |
| dawsn does not support complex-valued inputs. | validation | error | jax, scipy-special, dawson, complex-dtype |
| dynamic_slice: unrecognized index | validation | error | jax, indexing, internal, dynamic-slice |
| Expected pad_width to have shape | exception | error | jnp-pad, pad-width, shape-validation |
| Expected strategy to be IndexingStrategy; got | exception | error | jax, internal, indexing-strategy, type-error |
| must form a tree prefix of the corresponding values (up to… | validation | error | jax, pytree, arity |
| Invalid pipeline stage | validation | error | jax, xla, compiler-passes, validation |
| kl_div does not support complex-valued inputs. | validation | error | jax, scipy-special, kl-divergence, complex-dtype |
| m has more than 2 dimensions | exception | error | jax, statistics, covariance, ndim-validation |
| N mismatch: != | exception | error | jax, mosaic, mma, shape-mismatch |
| offset must be an integer, got | validation | error | jax, lax, tri, integer-required, type-error |
| pmap requires at least one argument with a mapped axis. | validation | error | jax, pmap, in-axes, axis-size |
| ragged_all_to_all send_sizes must be integer type. | exception | error | jax, ragged-all-to-all, dtype |
| ragged_dot_general requires exactly one rhs group dimension… | validation | error | jax, ragged-dot, shape-validation, moe |
| SMEM store does not support masks | validation | error | jax, pallas, tpu, smem, masked-store |
| The device specification passed to to_dlpack contains an… | exception | error | jax, dlpack, device-type, buffer-protocol |
| TPU generation is not supported | exception | error | tpu, hardware-generation, version-mismatch, jax |
| x.dtype= is not supported. | exception | error | jax, scipy-special, log-ndtr, dtype-validation |
| Atomic stores not supported for splat layout | exception | error | jax, mosaic-gpu, atomic, splat-layout, store |
| complex input not supported. | validation | error | jax, scipy-special, bessel, complex-dtype |
| Custom VJP bwd rule was registered with defvjp_with_logs… | exception | error | jax, custom-vjp, logging, type-validation |
| __dlpack__ device only supported for CPU, GPU and TPU… | exception | error | jax, dlpack, backend, unsupported-platform, interop |
| Ellipsis can't be used in a compound factor | exception | error | jax, shardy, sharding-rule, ellipsis |
| float64 dot requires M>=16, N>=8, K>=16 per warp tile… | exception | error | jax, pallas, triton, float64, tile-shape |
| full must be called with scalar fill_value, got… | validation | error | jax, lax, type-error, scalar-required, shape-validation |
| Fused load-reduce is not supported for this layout | exception | error | mosaic, gpu, tcgen05, reduction, layout, jax |
| Mesh has cores, but the current TPU chip has only… | exception | error | jax, pallas, tpu, sparsecore, hardware, resource-limits |
| method argument to `chisquare` must be one of | validation | error | jax, random, chisquare, method, input-validation |
| new enum value must be in | validation | error | jax, config, valueerror, enum |
| Only concatenate along the last dimension is supported. | exception | error | jax, triton, pallas, concatenate, axis, not-implemented |
| Only packing supported for bitwidth , but got TMEM packing… | exception | error | mosaic, gpu, tcgen05, packing, not-implemented, jax |
| has no `manual_axis_type`. | exception | error | jax, axis-type, attribute-error |
| TracerIntegerConversionError | exception | error | jax, tracer, integer-conversion, indexing, jit |
| Unknown FFT type | validation | error | jax, fft, api-misuse, invalid-enum-value |
| Unsupported scope | validation | error | jax, mosaic-gpu, barrier, scope, invalid-argument |
| `x` must have either the same number of entries as `alpha`… | exception | error | jax, scipy, dirichlet, shape-validation |
| A single Mosaic subkernel cannot contain multiple core… | exception | error | tpu, pallas, core-parallel, subkernel, validation |
| batch_dims must be None or satisfy 0 < dim < n_batch. Got | exception | error | jax, sparse, bcoo, vmap, batching |
| bcoo_slice: invalid indices. Got | exception | error | jax, sparse, bcoo, index-out-of-bounds, slice |
| should be a multiple of | exception | error | jax, pallas, tpu, splash-attention, block-size |
| Brackets should contain at least two factors | validation | error | jax, sharding, compound-factor, rule-syntax |
| Cannot concatenate arrays with shapes that differ in… | validation | error | jax, concatenate, shape-mismatch, shape-validation |
| dtype argument to `weibull_min` must be a float dtype, got | validation | error | jax, random, dtype-validation |
| dynamic_slice with partial slices does not support… | validation | error | jax, sharding, dynamic-slice, multi-device |
| dynamically changing the bandwidth method is not supported | exception | error | jax, scipy, kde, not-implemented, api-parity |
| Expected fill values, got | exception | error | jax, nonzero, fill-value, argument-validation |
| Expected SMEM but got | validation | error | jax, mosaic-gpu, smem, memory-space, gpu-clusters |
| f16/bf16 not supported for async atomics | exception | error | jax, mosaic, gpu, atomics, f16, bf16, cluster-barrier, not-implemented |
| indices and arr must have the same number of dimensions; | validation | error | jax, take-along-axis, ndim, shape-mismatch |
| Lookahead is not supported for XLA pipeline emitter… | validation | error | jax, pallas, tpu, pipelining, lookahead |
| Need at least one array to concatenate | validation | error | jax, mosaic-gpu, validation, empty-input |
| Only f16 WGMMA supports transposes | validation | error | jax, mosaic-gpu, wgmma, transpose, dtype |
| Scan with extensive argument(s) is not supported. Found… | validation | error | jax, pallas, triton, scan, control-flow |
| Shape of weights must be consistent with shape of a along… | validation | error | jax, average, weights, shape-mismatch |
| smap axis_name should be a `str` or a `Hashable`, but got | exception | error | jax, shard-map, smap, axis-name, distributed |
| The cp.async implementation does not support user-defined… | exception | error | jax, pallas, mosaic-gpu, gpu-architecture, predicate, cp-async, not-implemented |
| Unrecognized method . The two valid methods are either… | exception | error | jax, scipy, linalg, sylvester, argument-validation |
| Unsupported range type | exception | error | jax, pallas, mosaic, indexing, unsupported-type |
| axis is out of bounds for array of shape | validation | error | jax, lax, cumsum, axis-out-of-bounds |
| Cannot on a non-semaphore Ref | exception | error | pallas, semaphore, type-validation, jax |
| collective_id has to be unspecified or None when not using… | exception | error | tpu, pallas, collective-id, barrier, invalid-argument |
| conv_general_dilated window and window_strides must have… | validation | error | jax, convolution, strides, shape-validation |
| Default value must be of type float, got | exception | error | jax, config, typeerror, float |
| dynamic_slice: unrecognized index | validation | error | jax, indexing, unsupported-index, dynamic-slice |
| Grid must consist of Python integers and JAX Arrays | exception | error | jax, pallas, grid, validation |
| Index in subset_by_index[1] exceeds matrix size. | exception | error | jax, tpu, svd, linalg, bounds-check |
| M mismatch: != | exception | error | jax, mosaic, mma, shape-mismatch |
| Missing required keyword argument: 'in_layouts' | validation | error | jax, layout, missing-argument |
| MMA with element type | validation | error | jax, mosaic, gpu, block-scaling, dtype, mma, tcgen05, mxfp |
| argument of ManualAxisType should of type `frozenset` or… | validation | error | jax, sharding, type-error, constructor-validation |
| ndmin cannot be greater than object's ndims for string… | exception | error | jax, string-dtype, shape-validation |
| new enum value must be an instance of | validation | error | jax, config, typeerror, enum |
| new float config value must be None or of type float, got | validation | error | jax, config, valueerror, float |
| Pallas top_k only supports float32 and bfloat16, got | exception | error | jax, pallas, tpu, top-k, dtype |
| Partitioned callback not implemented on | exception | error | jax, callback, tpu, sharding, backend-support |
| shape mismatch | exception | error | jax, sparse, shape-validation, bcoo |
| The dtype of `lengths` must be int32. Got | validation | error | jax, pallas, tpu, paged-attention, dtype-validation |
| The iteration bound corresponding to the core-parallel… | exception | error | tpu, pallas, core-parallel, subkernel, consistency |
| TMEM cannot be sliced along rows | exception | error | jax, mosaic, tmem, slicing, unsupported-operation |
| TransformedRef must have been indexed before passing into… | exception | error | pallas, refs, jaxpr-call, validation, jax |
| Transforms are not yet implemented for warpgroup semantics | exception | error | jax, pallas, mosaic-gpu, transforms, warpgroup |
| Unrecognized | exception | error | jax, scipy, signal, internal-api, argument-validation |
| x.shape= != mask.shape= | exception | error | jax, pallas, sparsecore, scan-count, shape-mismatch |
| Zero-dimensional arrays cannot be concatenated. | validation | error | jnp-concatenate, wrong-argument-form |
| zero-size array to reduction operation | exception | error | jax, ufunc, reduce, empty-array |
| b matrix must have rank >= 1, got shape | validation | error | jax, lu-solve, shape-validation, rank |
| Block size is not a multiple of | validation | error | profiling, mosaic, gpu, cuda-launch-config |
| Brackets are not balanced in rule | validation | error | jax, sharding, unbalanced-parens, rule-syntax |
| Custom VJP bwd rule must produce a tuple of length equal to… | exception | error | jax, custom-vjp, arity-mismatch |
| DLPack is supported for PjRt-compatible backends only. | exception | error | jax, dlpack, ifrt, pjrt, interop |
| Eager shard_map cannot return a `jax.Ref`. Please wrap your… | validation | error | shard-map, ref, jit-required, jax |