jax-ml/jax
Documented errors, page 15 of 23. Back to jax-ml/jax
| Code / Message | Type | Severity | Tags |
|---|---|---|---|
| Unexpected value for 'order' argument | exception | error | jax, reshape, invalid-argument, validation |
| Unhandled transforms for semaphore_read | exception | error | pallas, mosaic-gpu, semaphore, not-implemented |
| vary_unreduced_cast input cannot be unreduced across the… | exception | error | jax, named-axes, duplicate-cast |
| is Ref-typed, so its gradient must be accumulated into a… | validation | error | jax, vjp, refs, autodiff |
| WGMMA requires m and n to be multiples of 64 and 8, got | validation | error | jax, mosaic-gpu, wgmma, gpu-kernel, shape-validation |
| A.dtype= is not supported. | validation | error | jax, linalg, dtype, matrix-exponential |
| Barrier can only be indexed with integers or slices, got | validation | error | mosaic-gpu, pallas, barrier, indexing, type-error |
| Bit width must be 32 | exception | error | jax, pallas, tpu, random, prng |
| Cannot apply ' ' padding to empty axis | exception | error | jnp-pad, empty-axis, reflect, wrap |
| Cannot store scalars to VMEM | exception | error | jax, pallas, tpu, vmem, scalar-store |
| convolution requires lhs and rhs ndim to be equal, got | validation | error | jax, convolution, shape-validation, rank-mismatch |
| Custom JVP rule for function must produce a pair (list or… | validation | error | jax, custom-jvp, pytree, autodiff |
| dtype argument to `lognormal` must be a float or complex… | validation | error | jax, random, dtype-validation |
| F32 redux only supported on Blackwell GPUs | validation | error | mosaic-gpu, redux, gpu-architecture, blackwell |
| Init is not implemented | exception | error | jax, collectives, cpu, api-misuse |
| lhs_ragged_dim not found in lhs_noncontracting … | validation | error | jax, ragged-dot-general, dimension-numbers |
| Multiple core support not implemented. | exception | error | pallas, semaphore, multi-core, not-implemented, jax |
| Need at least one array to concatenate. | validation | error | jnp-concatenate, empty-sequence, value-error |
| only 1D box integrations are supported; use… | exception | error | jax, scipy, kde, not-implemented, api-parity |
| Only arguments with shape [..., 1] are supported. | exception | error | jax, triton, pallas, concatenate, shape, not-implemented |
| Only thread-collective allocations are supported in… | exception | error | jax, pallas, run-scoped, collective-axes, warp-specialization |
| preduced input cannot be reduced across the axis_name… | exception | error | jax, sharding, collectives, double-reduction, mesh |
| ragged_dot_general requires | validation | error | jax, ragged-dot-general, index-out-of-range |
| reduce only supported for functions returning a single value | validation | error | jax, ufunc, reduce, api-misuse |
| has no `shape`. | exception | error | jax, attribute-error, shape |
| sparse rule for is not implemented because it would result… | exception | error | jax, sparse, sparsify, lax, not-implemented |
| Unknown boundary option | exception | error | jax, scipy, stft, boundary, invalid-argument-value |
| Unsupported scale type | exception | error | tcgen05, mx-format, matmul, dtype |
| Unsupported wgmma types | validation | error | jax, mosaic-gpu, wgmma, dtype, tensor-core |
| Valid values for indexing are 'xy' and 'ij', got | validation | error | jax, meshgrid, indexing, invalid-argument |
| x.dtype= must be uint32, int32 or float32 | exception | error | jax, pallas, sparsecore, scan-count, dtype |
| accumulate does not allow multiple axes | exception | error | jax, ufunc, accumulate, axis |
| Async copies require the number of bits copied along the… | validation | error | jax, mosaic-gpu, tma, memory-alignment, shape-validation |
| axis_index_groups can only be used with reductions over… | exception | error | jax, collectives, axis-validation |
| Cannot bitcast from ( bits) to ( bits), because * is not… | exception | error | jax, pallas, sparsecore, bitcast, divisibility |
| Cannot lower effectful `reduce_window`. | exception | error | jax, effects, lowering, mlir, reduce-window |
| dtype argument to `geometric` must be an int dtype, got | validation | error | jax, random, dtype-validation, discrete |
| erf_inv_lowering_helper not implemented for | exception | error | jax, pallas, triton, dtype, erfinv |
| Expected A to be a (batched) square matrix, got | validation | error | jax, linalg, matrix-exponential, shape-mismatch |
| Expected the same number of out_ | validation | error | jax, mosaic, mlir, layout, invariant |
| f32 not supported for async atomics | exception | error | jax, mosaic, gpu, atomics, f32, cluster-barrier, not-implemented |
| Factor can only be in one of the reduction, need… | validation | error | jax, sharding, conflicting-config, validation |
| Grid must be specified when using lookahead. | exception | error | jax, pallas, pipeline, lookahead, grid |
| Logical shape cannot be bigger than content shape . | validation | error | jax, pallas, mosaic, shared-memory, shape-mismatch, interpret-mode |
| memref.cast transforms must have identical transforms for… | exception | error | jax, mosaic-gpu, memref, cast, shared-memory, transforms |
| Mesh must be provided for shard_map with checkify. | exception | error | jax, checkify, shard-map, mesh, sharding |
| Number of axis names should match the number of axis_types… | validation | error | jax, mesh, axis-types, shape-mismatch |
| output_is_signed must be non-None if and only if the MLIR… | validation | error | mosaic, gpu, bitcast, type-error, signedness |
| Reassociation is not compatible with tiling , as it causes… | validation | error | jax, mosaic-gpu, collapse-shape, reassociation, tiling |
| stat_length of 0 yields no value for padding | exception | error | jnp-pad, stat-length, statistical-mode |
| TMEM reference must have | validation | error | gpu, mosaic, tcgen05, tensor-memory, shape-mismatch |
| transpose output pytree structure must match that of linear… | exception | error | jax, linear-call, transpose, pytree |
| Unexpected input type for array | exception | error | jax, type-validation, conversion |
| Unsupported layout | exception | error | jax, mosaic, gpu, reduction, layout |
| Unsupported pcast from= | exception | error | jax, pcast, unsupported-transition, named-axes |
| wait_gmem_to_smem is only supported on pre-Hopper GPUs… | validation | error | mosaic-gpu, pallas, gpu, hopper, synchronization |
| argmin and argmax require non-empty reduced dimension… | validation | error | jax, argmax, empty-array, shape-validation |
| axis_index_groups must cover all indices exactly once | exception | error | jax, psum, validation, axis-index-groups |
| bcoo_dynamic_slice: indices must have size mat.ndim= | exception | error | jax, sparse, bcoo, validation, shape-mismatch |
| broadcast_in_dim target broadcast shape must have equal or… | validation | error | jax, broadcast-in-dim, rank-validation |
| Expected 0 < bdim <= n_batch; got | exception | error | jax, sparse, bcoo, vmap, fromdense |
| Expected named tuple, got | validation | error | jax, pytree, type-mismatch, namedtuple |
| f16/bf16 SMEM/multimem atomics only support add, got | exception | error | jax, mosaic, gpu, atomics, f16, bf16, shared-memory, multimem, not-implemented |
| was requested to map its argument along axis , which… | validation | error | jax, vmap, shape, rank, in-axes |
| out argument of | validation | error | jax, ufunc, immutable-arrays, numpy-compat |
| `perm` does not represent a permutation | exception | error | jax, pshuffle, permutation, validation |
| `perm` passed to `jax.lax.ppermute` must be a list or a… | exception | error | jax, ppermute, typeerror, collectives |
| reduced_vary_cast only accepts inputs that are reduced. Got | exception | error | jax, named-axes, manual-parallelism, aval-validation |
| scan got value with no leading axis to scan over | validation | error | jax, scan, shape-inference, scalar-operand, pytree |
| Swizzle is not supported. Only 32, 64 and 128 are accepted. | exception | error | jax, pallas, mosaic-gpu, swizzle, validation |
| Tiles must have a decreasing rank | validation | error | mosaic, fragmented-array, tiling, layout, validation |
| A must be ( , ) matrix A, got output | exception | error | jax, sparse, lobpcg, shape-mismatch |
| Arriving on a collective barrier is not supported in a warp… | validation | error | mosaic-gpu, pallas, barrier, warp-specialization |
| Backend already initialized | exception | error | |
| Barriers are required for TMA GMEM -> SMEM copies | validation | error | jax, mosaic-gpu, tma, barrier, synchronization |
| Cannot register XLA transform | exception | error | jax, xla, pjrt, plugin-capabilities |
| dtype argument to `laplace` must be a float dtype, got | validation | error | jax, random, laplace, dtype, input-validation |
| key cannot be empty | validation | error | jax, lru-cache, input-validation |
| Loading multiple row tiles | exception | error | mosaic, gpu, tcgen05, layout, tensor-memory, not-implemented, jax |
| mask cannot be a block if pointer is not a block | exception | error | jax, triton, pallas, load, mask, shape, validation |
| masked swap with non-32-bit data | validation | error | jax, pallas, tpu, masked-store, dtype |
| must have search dim > 0, got | exception | error | jax, sparse, lobpcg, eigensolver, validation |
| Specified axis_size doesn't match received axis_size . | validation | error | jax, pmap, axis-size, shape-mismatch |
| Storing from register layout | exception | error | mosaic, gpu, tcgen05, layout, tensor-memory, jax |
| Swap only supports slices with stride 1, got | exception | error | jax, pallas, tpu, sparsecore, swap, memory-space |
| The SVD algorithm parameter is not implemented on TPU. | exception | error | jax, tpu, svd, not-implemented, backend-support |
| Too many dynamic shapes in the input. Mosaic currently only… | validation | error | jax, pallas, mosaic, dynamic-shapes, limit-exceeded, lowering |
| Unknown value for mode | exception | error | jax, scipy, internal-api, spectral-analysis |
| Argument `x` to sici must be real-valued. Got dtype | exception | error | jax, sici, complex-numbers, unsupported-dtype |
| {aval_out.dtype} | exception | error | jax, pallas, tpu, dtype, add |
| broadcast_in_dim broadcast_dimensions must not contain… | validation | error | jax, broadcast-in-dim, duplicate-index |
| Can't transpose a TMEM reference. | exception | error | jax, pallas, mosaic-gpu, tmem, transpose |
| convolution dimension_numbers | validation | error | jax, convolution, dimension-numbers, validation |
| expected axis and stride_axis are different. | exception | error | jax, pallas, mosaic, roll, argument-validation |
| Get only supports slices with stride 1, got | exception | error | jax, pallas, tpu, sparsecore, slicing, strides |
| np.reshape order=A is not implemented. | exception | error | jax, reshape, not-implemented, numpy-compat |
| Only thread-collective allocations are supported in… | exception | error | jax, pallas, run-scoped, warp-specialization, collective-axes |
| Please open a feature request! | exception | error | jax, all-to-all, vmap, not-implemented |
| polygamma does not support complex-valued inputs. | validation | error | jax, scipy-special, polygamma, complex-dtype |
| Reductions over not implemented. | exception | error | jax, pallas, tpu, reduction, dtype |