ONNX Operator Coverage

This matrix starts from the official ONNX operator index and shows which operators are referenced in jax2onnx/plugins.

• Metadata: at least one onnx__<Op>.html operator link was found in plugin metadata.
• Lowering: at least one builder.<Op>(...) call was found in plugin code.
• Potential JAX Ops: candidate JAX APIs that naturally map to the ONNX operator.

• Source index: ONNX 1.21.0 documentation (https://onnx.ai/onnx/operators/index.html)
• Operators in index: 200
• Operators referenced in plugins: 156
• Coverage: 78.0%
• Potential JAX Ops lists candidate JAX entry points for each operator.
• Extra plugin names not present in index: 6

ONNX Operator	In Plugins	Metadata	Lowering	Plugin Modules	Potential JAX Ops
Abs	✅	✅	✅	jax/lax/abs jax/lax/bessel_i0e jax/lax/bessel_i1e jax/lax/cbrt jax/lax/lgamma jax/lax/lu ... +6	Covered via existing plugin primitives.
Acos	✅	✅	✅	jax/lax/acos jax/numpy/acos	jax.lax.acos jax.numpy.arccos
Acosh	✅	✅	✅	jax/lax/acosh jax/numpy/acosh	jax.lax.acosh jax.numpy.arccosh
Add	✅	✅	✅	equinox/eqx/nn/recurrent equinox/eqx/nn/rms_norm equinox/eqx/nn/rotary_positional_embedding flax/linen/dot_product_attention flax/linen/einsum flax/linen/multi_head_attention ... +60	Covered via existing plugin primitives.
AffineGrid	➖	➖	➖	➖	No clear direct primitive; likely composite JAX expression.
And	✅	✅	✅	flax/nnx/combine_masks jax/lax/and jax/lax/betainc jax/lax/clz jax/lax/igamma jax/lax/nextafter ... +4	Covered via existing plugin primitives.
ArgMax	✅	✅	✅	jax/lax/_arg_utils jax/lax/argmax jax/lax/lu jax/numpy/argmax jax/numpy/composite_metadata_batch3 jax/random/categorical	Covered via existing plugin primitives.
ArgMin	✅	✅	✅	jax/lax/_arg_utils jax/lax/argmin jax/numpy/argmin jax/numpy/composite_metadata_batch3	Covered via existing plugin primitives.
Asin	✅	✅	✅	jax/lax/asin jax/numpy/asin	jax.lax.asin jax.numpy.arcsin
Asinh	✅	✅	✅	jax/lax/asinh jax/numpy/asinh	jax.lax.asinh jax.numpy.arcsinh
Atan	✅	✅	✅	jax/lax/atan jax/lax/atan2 jax/numpy/atan jax/numpy/atan2	jax.lax.atan jax.numpy.arctan
Atanh	✅	✅	✅	jax/lax/atanh jax/numpy/atanh	jax.lax.atanh jax.numpy.arctanh
Attention	✅	✅	✅	equinox/eqx/nn/multihead_attention flax/nnx/dot_product_attention	Covered via existing plugin primitives.
AveragePool	✅	✅	✅	equinox/eqx/nn/adaptive_pool equinox/eqx/nn/avg_pool flax/linen/avg_pool flax/linen/pool flax/nnx/avg_pool	Covered via existing plugin primitives.
BatchNormalization	✅	✅	✅	equinox/eqx/nn/batch_norm flax/linen/batch_norm flax/nnx/batch_norm	Covered via existing plugin primitives.
Bernoulli	✅	✅	✅	jax/random/bernoulli	Covered via existing plugin primitives.
BitCast	✅	✅	✅	jax/lax/bitcast_convert_type	Covered via existing plugin primitives.
BitShift	✅	✅	✅	jax/lax/clz jax/lax/population_count jax/lax/shift_left jax/lax/shift_right_arithmetic jax/lax/shift_right_logical jax/numpy/bitwise_left_shift ... +3	Covered via existing plugin primitives.
BitwiseAnd	✅	✅	✅	jax/lax/and jax/lax/clz jax/lax/population_count jax/numpy/bitwise_and	Covered via existing plugin primitives.
BitwiseNot	✅	✅	✅	jax/lax/bitwise_not jax/numpy/bitwise_not jax/numpy/invert	Covered via existing plugin primitives.
BitwiseOr	✅	✅	✅	jax/lax/clz jax/lax/or jax/lax/shift_right_arithmetic jax/numpy/bitwise_or jax/numpy/right_shift	Covered via existing plugin primitives.
BitwiseXor	✅	✅	✅	jax/lax/xor jax/numpy/bitwise_xor	Covered via existing plugin primitives.
BlackmanWindow	✅	✅	✅	jax/numpy/windows	Covered via existing plugin primitives.
Cast	✅	✅	✅	_complex_utils equinox/eqx/nn/embedding flax/linen/dot_product_attention flax/nnx/combine_masks flax/nnx/dot_product_attention flax/nnx/embed ... +65	Covered via existing plugin primitives.
CastLike	✅	✅	✅	_utils flax/linen/conv flax/linen/dense_general flax/nnx/conv flax/nnx/linear flax/nnx/linear_general ... +1	Covered via existing plugin primitives.
Ceil	✅	✅	✅	jax/lax/ceil jax/numpy/ceil	jax.lax.ceil jax.numpy.ceil
Celu	✅	✅	✅	jax/nn/celu	Covered via existing plugin primitives.
CenterCropPad	➖	➖	➖	➖	No clear direct primitive; likely composite JAX expression.
Clip	✅	✅	✅	flax/nnx/hard_tanh jax/numpy/clip	Covered via existing plugin primitives.
Col2Im	➖	➖	➖	➖	No clear direct primitive; likely composite JAX expression.
Compress	✅	✅	✅	jax/numpy/compress	Covered via existing plugin primitives.
Concat	✅	✅	✅	_axis0_utils _complex_utils equinox/eqx/nn/linear equinox/eqx/nn/multihead_attention equinox/eqx/nn/rotary_positional_embedding flax/linen/dense ... +32	Covered via existing plugin primitives.
ConcatFromSequence	➖	➖	➖	➖	No clear direct primitive; likely composite JAX expression.
Constant	✅	✅	✅	flax/nnx/dropout jax/numpy/windows	Covered via existing plugin primitives.
ConstantOfShape	✅	✅	✅	jax/numpy/eye jax/numpy/full jax/numpy/ones jax/numpy/zeros	Covered via existing plugin primitives.
Conv	✅	✅	✅	equinox/eqx/nn/conv flax/linen/conv flax/linen/conv_local flax/linen/recurrent flax/nnx/conv jax/lax/conv ... +2	Covered via existing plugin primitives.
ConvInteger	➖	➖	➖	➖	No clear direct primitive; likely composite JAX expression.
ConvTranspose	✅	✅	✅	flax/linen/conv_transpose jax/lax/conv	Covered via existing plugin primitives.
Cos	✅	✅	✅	jax/lax/cos jax/lax/digamma jax/numpy/cos	Covered via existing plugin primitives.
Cosh	✅	✅	✅	jax/lax/cosh jax/numpy/cosh	Covered via existing plugin primitives.
CumProd	✅	✅	✅	jax/lax/cumprod jax/numpy/cumprod	Covered via existing plugin primitives.
CumSum	✅	✅	✅	jax/lax/cumlogsumexp jax/lax/cumsum jax/numpy/composite_metadata_batch3 jax/numpy/cumsum	Covered via existing plugin primitives.
DFT	✅	✅	✅	jax/lax/fft jax/numpy/fft	Covered via existing plugin primitives.
DeformConv	➖	➖	➖	➖	No clear direct primitive; likely composite JAX expression.
DepthToSpace	✅	✅	✅	dm_pix/depth_to_space	Covered via existing plugin primitives.
DequantizeLinear	➖	➖	➖	➖	No clear direct primitive; likely composite JAX expression.
Det	✅	✅	✅	jax/numpy/linalg_det	Covered via existing plugin primitives.
Div	✅	✅	✅	equinox/eqx/nn/rms_norm equinox/eqx/nn/spectral_norm equinox/eqx/nn/weight_norm flax/linen/dot_product_attention flax/linen/spectral_norm flax/linen/weight_norm ... +37	Covered via existing plugin primitives.
Dropout	✅	✅	✅	equinox/eqx/nn/dropout flax/linen/dropout flax/nnx/dot_product_attention flax/nnx/dropout	Covered via existing plugin primitives.
DynamicQuantizeLinear	➖	➖	➖	➖	No clear direct primitive; likely composite JAX expression.
Einsum	✅	✅	✅	flax/linen/einsum flax/nnx/einsum jax/lax/dot_general jax/nn/scaled_dot_general jax/numpy/einsum	Covered via existing plugin primitives.
Elu	✅	✅	✅	flax/nnx/elu jax/nn/elu	Covered via existing plugin primitives.
Equal	✅	✅	✅	jax/lax/_reduce_utils jax/lax/atan2 jax/lax/betainc jax/lax/clz jax/lax/eigh jax/lax/eq ... +15	Covered via existing plugin primitives.
Erf	✅	✅	✅	jax/lax/erf jax/lax/erf_inv jax/lax/erfc	Covered via existing plugin primitives.
Exp	✅	✅	✅	jax/lax/bessel_i0e jax/lax/bessel_i1e jax/lax/betainc jax/lax/cosh jax/lax/cumlogsumexp jax/lax/erf_inv ... +11	Covered via existing plugin primitives.
Expand	✅	✅	✅	_axis0_utils jax/lax/broadcast_in_dim jax/lax/dynamic_update_slice jax/lax/gather jax/lax/iota jax/lax/scan ... +3	Covered via existing plugin primitives.
EyeLike	✅	✅	✅	jax/numpy/eye	Covered via existing plugin primitives.
Flatten	✅	✅	✅	jax/numpy/reshape	Covered via existing plugin primitives.
Floor	✅	✅	✅	jax/lax/floor jax/lax/nextafter jax/lax/polygamma jax/lax/reduce_precision jax/numpy/floor jax/numpy/floor_divide ... +1	Covered via existing plugin primitives.
GRU	➖	➖	➖	➖	No clear direct primitive; likely composite JAX expression.
Gather	✅	✅	✅	_complex_utils equinox/eqx/nn/embedding equinox/eqx/nn/multihead_attention flax/linen/embed flax/nnx/conv flax/nnx/embed ... +31	Covered via existing plugin primitives.
GatherElements	✅	✅	✅	flax/nnx/flip_sequences jax/lax/sort	Covered via existing plugin primitives.
GatherND	✅	✅	✅	jax/lax/gather	Covered via existing plugin primitives.
Gelu	✅	✅	✅	flax/nnx/gelu jax/nn/gelu	Covered via existing plugin primitives.
Gemm	✅	✅	✅	equinox/eqx/nn/linear equinox/eqx/nn/multihead_attention equinox/eqx/nn/recurrent flax/linen/conv_local flax/linen/dense flax/linen/dense_general ... +8	Covered via existing plugin primitives.
GlobalAveragePool	✅	✅	✅	flax/linen/avg_pool flax/nnx/avg_pool	Covered via existing plugin primitives.
GlobalLpPool	✅	✅	✅	jax/numpy/linalg_norm	Covered via existing plugin primitives.
GlobalMaxPool	✅	✅	✅	flax/linen/max_pool flax/nnx/max_pool	Covered via existing plugin primitives.
Greater	✅	✅	✅	jax/lax/atan2 jax/lax/betainc jax/lax/digamma jax/lax/gt jax/lax/igamma jax/lax/nextafter ... +6	Covered via existing plugin primitives.
GreaterOrEqual	✅	✅	✅	jax/lax/ge jax/numpy/greater_equal	Covered via existing plugin primitives.
GridSample	➖	➖	➖	➖	No clear direct primitive; likely composite JAX expression.
GroupNormalization	✅	✅	✅	equinox/eqx/nn/group_norm flax/linen/group_norm flax/linen/instance_norm flax/nnx/group_norm	Covered via existing plugin primitives.
HammingWindow	✅	✅	✅	jax/numpy/windows	Covered via existing plugin primitives.
HannWindow	✅	✅	✅	jax/numpy/windows	Covered via existing plugin primitives.
HardSigmoid	✅	✅	✅	jax/nn/hard_sigmoid	Covered via existing plugin primitives.
HardSwish	✅	✅	✅	jax/nn/hard_swish	Covered via existing plugin primitives.
Hardmax	✅	✅	✅	jax/nn/hardmax	Covered via existing plugin primitives.
Identity	✅	✅	✅	equinox/eqx/nn/identity equinox/eqx/nn/rms_norm equinox/eqx/nn/sequential jax/core/name jax/lax/_control_flow_utils jax/lax/_cum_extrema ... +45	Covered via existing plugin primitives.
If	✅	✅	✅	jax/lax/cond	Covered via existing plugin primitives.
ImageDecoder	➖	➖	➖	➖	No clear direct primitive; likely composite JAX expression.
InstanceNormalization	✅	✅	✅	flax/linen/instance_norm	Covered via existing plugin primitives.
IsInf	✅	✅	✅	jax/lax/is_finite jax/lax/nextafter jax/numpy/isfinite	Covered via existing plugin primitives.
IsNaN	✅	✅	✅	jax/lax/is_finite jax/lax/nextafter jax/numpy/composite_metadata_batch2 jax/numpy/composite_metadata_batch3 jax/numpy/isfinite	Covered via existing plugin primitives.
LRN	➖	➖	➖	➖	No clear direct primitive; likely composite JAX expression.
LSTM	➖	➖	➖	➖	No clear direct primitive; likely composite JAX expression.
LayerNormalization	✅	✅	✅	equinox/eqx/nn/layer_norm flax/linen/layer_norm flax/nnx/layer_norm	Covered via existing plugin primitives.
LeakyRelu	✅	✅	✅	flax/nnx/leaky_relu jax/nn/leaky_relu	Covered via existing plugin primitives.
Less	✅	✅	✅	flax/linen/dot_product_attention jax/lax/atan2 jax/lax/bessel_i0e jax/lax/bessel_i1e jax/lax/betainc jax/lax/digamma ... +18	Covered via existing plugin primitives.
LessOrEqual	✅	✅	✅	jax/lax/eigh jax/lax/le jax/numpy/composite_metadata_batch2 jax/numpy/less_equal	Covered via existing plugin primitives.
Log	✅	✅	✅	jax/lax/cumlogsumexp jax/lax/digamma jax/lax/erf_inv jax/lax/lgamma jax/lax/log jax/lax/log1p ... +7	Covered via existing plugin primitives.
LogSoftmax	✅	✅	✅	flax/nnx/log_softmax jax/nn/log_softmax	Covered via existing plugin primitives.
Loop	✅	✅	✅	flax/linen/recurrent jax/lax/_control_flow_utils jax/lax/fori_loop jax/lax/scan jax/lax/while_loop jax/numpy/composite_metadata_batch6	Covered via existing plugin primitives.
LpNormalization	✅	✅	✅	jax/lax/div	Covered via existing plugin primitives.
LpPool	✅	✅	✅	jax/lax/reduce_window_sum	Covered via existing plugin primitives.
MatMul	✅	✅	✅	equinox/eqx/nn/multihead_attention equinox/eqx/nn/spectral_norm flax/linen/dot_product_attention flax/linen/multi_head_attention flax/linen/multi_head_dot_product_attention flax/linen/self_attention ... +18	Covered via existing plugin primitives.
MatMulInteger	➖	➖	➖	➖	No clear direct primitive; likely composite JAX expression.
Max	✅	✅	✅	jax/lax/clamp jax/lax/eig jax/lax/igamma jax/lax/max jax/lax/shift_right_arithmetic jax/lax/svd ... +5	Covered via existing plugin primitives.
MaxPool	✅	✅	✅	equinox/eqx/nn/adaptive_pool equinox/eqx/nn/max_pool equinox/eqx/nn/pool flax/linen/max_pool flax/linen/min_pool flax/linen/pool ... +6	Covered via existing plugin primitives.
MaxRoiPool	➖	➖	➖	➖	No clear direct primitive; likely composite JAX expression.
MaxUnpool	➖	➖	➖	➖	No clear direct primitive; likely composite JAX expression.
Mean	✅	✅	✅	jax/lax/div	Covered via existing plugin primitives.
MeanVarianceNormalization	✅	✅	✅	jax/nn/standardize	Covered via existing plugin primitives.
MelWeightMatrix	➖	➖	➖	➖	No clear direct primitive; likely composite JAX expression.
Min	✅	✅	✅	jax/lax/clamp jax/lax/igamma jax/lax/min jax/lax/shift_right_arithmetic jax/nn/hard_tanh jax/nn/relu6 ... +3	Covered via existing plugin primitives.
Mish	✅	✅	✅	jax/nn/mish	Covered via existing plugin primitives.
Mod	✅	✅	✅	jax/lax/_reduce_utils jax/lax/reduce_xor jax/lax/rem jax/numpy/composite_metadata_batch2 jax/numpy/composite_metadata_batch3 jax/numpy/fmod	Covered via existing plugin primitives.
Mul	✅	✅	✅	_complex_utils equinox/eqx/nn/multihead_attention equinox/eqx/nn/recurrent equinox/eqx/nn/rms_norm equinox/eqx/nn/rotary_positional_embedding equinox/eqx/nn/weight_norm ... +67	Covered via existing plugin primitives.
Multinomial	✅	✅	✅	jax/random/categorical	Covered via existing plugin primitives.
Neg	✅	✅	✅	_complex_utils equinox/eqx/nn/rotary_positional_embedding flax/linen/min_pool flax/linen/pool flax/nnx/log_sigmoid jax/lax/_cum_extrema ... +16	Covered via existing plugin primitives.
NegativeLogLikelihoodLoss	➖	➖	➖	➖	No clear direct primitive; likely composite JAX expression.
NonMaxSuppression	➖	➖	➖	➖	No clear direct primitive; likely composite JAX expression.
NonZero	✅	✅	✅	jax/lax/scatter jax/lax/scatter_utils	Covered via existing plugin primitives.
Not	✅	✅	✅	equinox/eqx/nn/dropout flax/nnx/dot_product_attention flax/nnx/dropout jax/lax/betainc jax/lax/bitwise_not jax/lax/igamma ... +9	Covered via existing plugin primitives.
OneHot	✅	✅	✅	jax/nn/one_hot	Covered via existing plugin primitives.
Optional	➖	➖	➖	➖	No clear direct primitive; likely composite JAX expression.
OptionalGetElement	➖	➖	➖	➖	No clear direct primitive; likely composite JAX expression.
OptionalHasElement	➖	➖	➖	➖	No clear direct primitive; likely composite JAX expression.
Or	✅	✅	✅	jax/lax/atan2 jax/lax/betainc jax/lax/is_finite jax/lax/nextafter jax/lax/or jax/numpy/bitwise_or ... +1	Covered via existing plugin primitives.
PRelu	✅	✅	✅	equinox/eqx/nn/prelu flax/linen/prelu flax/nnx/prelu	Covered via existing plugin primitives.
Pad	✅	✅	✅	_axis0_utils jax/lax/pad jax/lax/reduce_window_sum jax/numpy/composite_metadata_batch4 jax/numpy/pad	Covered via existing plugin primitives.
Pow	✅	✅	✅	equinox/eqx/nn/rms_norm flax/nnx/rms_norm jax/lax/betainc jax/lax/cbrt jax/lax/exp2 jax/lax/igamma ... +9	Covered via existing plugin primitives.
QLinearConv	➖	➖	➖	➖	No clear direct primitive; likely composite JAX expression.
QLinearMatMul	➖	➖	➖	➖	No clear direct primitive; likely composite JAX expression.
QuantizeLinear	➖	➖	➖	➖	No clear direct primitive; likely composite JAX expression.
RMSNormalization	✅	✅	✅	equinox/eqx/nn/rms_norm flax/linen/rms_norm flax/nnx/rms_norm	Covered via existing plugin primitives.
RNN	➖	➖	➖	➖	No clear direct primitive; likely composite JAX expression.
RandomNormal	✅	✅	✅	jax/random/normal	Covered via existing plugin primitives.
RandomNormalLike	✅	✅	✅	jax/random/normal	Covered via existing plugin primitives.
RandomUniform	✅	✅	✅	jax/lax/rng_bit_generator jax/lax/rng_uniform jax/random/random_bits	Covered via existing plugin primitives.
RandomUniformLike	✅	✅	✅	jax/random/categorical jax/random/random_bits	Covered via existing plugin primitives.
Range	✅	✅	✅	jax/lax/dynamic_update_slice jax/lax/gather jax/lax/iota jax/lax/rev jax/numpy/arange jax/numpy/composite_metadata ... +3	Covered via existing plugin primitives.
Reciprocal	✅	✅	✅	jax/lax/integer_pow	jax.numpy.reciprocal jax.lax.integer_pow(x, -1) 1.0 / x
ReduceL1	✅	✅	✅	jax/lax/_reduce_utils jax/lax/reduce_sum jax/numpy/sum	jax.numpy.linalg.norm(ord=1) jax.numpy.sum(jax.numpy.abs(x), axis=...)
ReduceL2	✅	✅	✅	jax/lax/_reduce_utils jax/lax/sqrt jax/numpy/sqrt	jax.numpy.linalg.norm(ord=2) jax.numpy.sqrt(jax.numpy.sum(x * x, axis=...))
ReduceLogSum	✅	✅	✅	jax/lax/_reduce_utils jax/lax/log jax/numpy/log	jax.numpy.log(jax.numpy.sum(x, axis=...))
ReduceLogSumExp	✅	✅	✅	jax/lax/_reduce_utils jax/lax/log jax/nn/logmeanexp jax/nn/logsumexp jax/numpy/log	jax.nn.logsumexp jax.scipy.special.logsumexp
ReduceMax	✅	✅	✅	jax/lax/_reduce_utils jax/lax/reduce jax/lax/reduce_max jax/lax/reduce_or jax/numpy/amax jax/numpy/any ... +3	Covered via existing plugin primitives.
ReduceMean	✅	✅	✅	equinox/eqx/nn/rms_norm flax/nnx/rms_norm jax/numpy/composite_metadata jax/numpy/composite_metadata_batch2 jax/numpy/composite_metadata_batch5 jax/numpy/mean	Covered via existing plugin primitives.
ReduceMin	✅	✅	✅	jax/lax/_reduce_utils jax/lax/reduce jax/lax/reduce_and jax/lax/reduce_min jax/numpy/all jax/numpy/amin ... +3	Covered via existing plugin primitives.
ReduceProd	✅	✅	✅	flax/nnx/conv jax/lax/_cum_extrema jax/lax/_reduce_utils jax/lax/dynamic_update_slice jax/lax/reduce_prod jax/numpy/composite_metadata_batch2 ... +2	Covered via existing plugin primitives.
ReduceSum	✅	✅	✅	flax/linen/dot_product_attention jax/lax/_reduce_utils jax/lax/reduce_sum jax/lax/reduce_xor jax/lax/svd jax/numpy/composite_metadata_batch2 ... +3	Covered via existing plugin primitives.
ReduceSumSquare	✅	✅	✅	jax/lax/_reduce_utils jax/lax/reduce_sum jax/numpy/sum	jax.numpy.sum(jax.numpy.square(x), axis=...)
RegexFullMatch	➖	➖	➖	➖	No clear direct primitive; likely composite JAX expression.
Relu	✅	✅	✅	equinox/eqx/nn/lambda equinox/eqx/nn/sequential flax/nnx/relu jax/nn/relu	Covered via existing plugin primitives.
Reshape	✅	✅	✅	equinox/eqx/nn/linear equinox/eqx/nn/multihead_attention flax/linen/conv flax/linen/conv_local flax/linen/dense flax/linen/dense_general ... +31	Covered via existing plugin primitives.
Resize	✅	✅	✅	jax/image/resize	Covered via existing plugin primitives.
ReverseSequence	✅	✅	➖	jax/numpy/composite_metadata_batch2	Covered via existing plugin primitives.
RoiAlign	➖	➖	➖	➖	No clear direct primitive; likely composite JAX expression.
RotaryEmbedding	✅	✅	✅	equinox/eqx/nn/rotary_positional_embedding	Covered via existing plugin primitives.
Round	✅	✅	✅	jax/lax/polygamma jax/lax/reduce_precision jax/lax/round	jax.lax.round jax.numpy.round
STFT	➖	➖	➖	➖	No clear direct primitive; likely composite JAX expression.
Scan	➖	➖	➖	➖	No clear direct primitive; likely composite JAX expression.
Scatter	✅	✅	✅	jax/lax/scatter jax/lax/scatter_utils	Covered via existing plugin primitives.
ScatterElements	✅	✅	✅	jax/lax/lu jax/lax/lu_pivots_to_permutation jax/lax/scatter jax/lax/scatter_utils	Covered via existing plugin primitives.
ScatterND	✅	✅	✅	jax/lax/cholesky jax/lax/cholesky_update jax/lax/dynamic_update_slice jax/lax/hessenberg jax/lax/householder_product jax/lax/lu ... +10	Covered via existing plugin primitives.
Selu	✅	✅	✅	jax/nn/selu	Covered via existing plugin primitives.
SequenceAt	➖	➖	➖	➖	No clear direct primitive; likely composite JAX expression.
SequenceConstruct	➖	➖	➖	➖	No clear direct primitive; likely composite JAX expression.
SequenceEmpty	➖	➖	➖	➖	No clear direct primitive; likely composite JAX expression.
SequenceErase	➖	➖	➖	➖	No clear direct primitive; likely composite JAX expression.
SequenceInsert	➖	➖	➖	➖	No clear direct primitive; likely composite JAX expression.
SequenceLength	➖	➖	➖	➖	No clear direct primitive; likely composite JAX expression.
SequenceMap	➖	➖	➖	➖	No clear direct primitive; likely composite JAX expression.
Shape	✅	✅	✅	_axis0_utils equinox/eqx/nn/linear equinox/eqx/nn/multihead_attention flax/linen/dense flax/linen/dense_general flax/nnx/conv ... +13	Covered via existing plugin primitives.
Shrink	➖	➖	➖	➖	No clear direct primitive; likely composite JAX expression.
Sigmoid	✅	✅	✅	equinox/eqx/nn/lambda equinox/eqx/nn/recurrent equinox/eqx/nn/sequential flax/linen/recurrent flax/nnx/glu flax/nnx/sigmoid ... +5	Covered via existing plugin primitives.
Sign	✅	✅	✅	jax/lax/bessel_i1e jax/lax/cbrt jax/lax/eig jax/lax/erf_inv jax/lax/nextafter jax/lax/reduce_precision ... +3	Covered via existing plugin primitives.
Sin	✅	✅	✅	jax/lax/cos jax/lax/digamma jax/lax/lgamma jax/lax/sin jax/numpy/composite_metadata_batch2 jax/numpy/cos ... +2	Covered via existing plugin primitives.
Sinh	✅	✅	✅	jax/lax/sinh jax/numpy/sinh	Covered via existing plugin primitives.
Size	✅	✅	✅	jax/numpy/size	Covered via existing plugin primitives.
Slice	✅	✅	✅	_axis0_utils equinox/eqx/nn/linear equinox/eqx/nn/multihead_attention flax/linen/conv_transpose flax/linen/dense flax/linen/dense_general ... +19	Covered via existing plugin primitives.
Softmax	✅	✅	✅	equinox/eqx/nn/multihead_attention flax/linen/dot_product_attention flax/linen/multi_head_attention flax/linen/multi_head_dot_product_attention flax/linen/self_attention flax/nnx/dot_product_attention ... +4	Covered via existing plugin primitives.
SoftmaxCrossEntropyLoss	➖	➖	➖	➖	No clear direct primitive; likely composite JAX expression.
Softplus	✅	✅	✅	flax/nnx/log_sigmoid flax/nnx/softplus jax/nn/softplus	Covered via existing plugin primitives.
Softsign	✅	✅	✅	jax/nn/softsign	Covered via existing plugin primitives.
SpaceToDepth	✅	✅	✅	dm_pix/space_to_depth	Covered via existing plugin primitives.
Split	✅	✅	✅	equinox/eqx/nn/rotary_positional_embedding flax/nnx/glu jax/lax/split jax/nn/glu jax/numpy/split jax/numpy/unstack	Covered via existing plugin primitives.
SplitToSequence	➖	➖	➖	➖	No clear direct primitive; likely composite JAX expression.
Sqrt	✅	✅	✅	equinox/eqx/nn/rms_norm flax/nnx/rms_norm jax/lax/bessel_i0e jax/lax/bessel_i1e jax/lax/cholesky jax/lax/cholesky_update ... +12	Covered via existing plugin primitives.
Squeeze	✅	✅	✅	dm_pix/depth_to_space dm_pix/space_to_depth equinox/eqx/nn/adaptive_pool equinox/eqx/nn/avg_pool equinox/eqx/nn/conv equinox/eqx/nn/max_pool ... +14	Covered via existing plugin primitives.
StringConcat	➖	➖	➖	➖	No clear direct primitive; likely composite JAX expression.
StringNormalizer	➖	➖	➖	➖	No clear direct primitive; likely composite JAX expression.
StringSplit	➖	➖	➖	➖	No clear direct primitive; likely composite JAX expression.
Sub	✅	✅	✅	jax/lax/atan2 jax/lax/betainc jax/lax/cholesky jax/lax/cholesky_update jax/lax/conv jax/lax/digamma ... +36	Covered via existing plugin primitives.
Sum	✅	✅	✅	jax/lax/add_any	Covered via existing plugin primitives.
Swish	✅	✅	✅	jax/nn/silu	Covered via existing plugin primitives.
Tan	✅	✅	✅	jax/lax/tan jax/numpy/tan	jax.lax.tan jax.numpy.tan
Tanh	✅	✅	✅	equinox/eqx/nn/lambda equinox/eqx/nn/recurrent equinox/eqx/nn/sequential flax/linen/recurrent flax/nnx/tanh jax/lax/tanh ... +2	Covered via existing plugin primitives.
TensorScatter	✅	✅	✅	jax/lax/dynamic_update_slice	Covered via existing plugin primitives.
TfIdfVectorizer	➖	➖	➖	➖	No clear direct primitive; likely composite JAX expression.
ThresholdedRelu	✅	✅	✅	jax/nn/thresholded_relu	Covered via existing plugin primitives.
Tile	✅	✅	✅	jax/nn/scaled_matmul jax/numpy/tile	Covered via existing plugin primitives.
TopK	✅	✅	✅	jax/lax/approx_top_k jax/lax/sort jax/lax/top_k jax/numpy/composite_metadata_batch2 jax/numpy/sort	Covered via existing plugin primitives.
Transpose	✅	✅	✅	dm_pix/depth_to_space dm_pix/space_to_depth equinox/eqx/nn/multihead_attention flax/linen/avg_pool flax/linen/conv flax/linen/conv_local ... +34	Covered via existing plugin primitives.
Trilu	✅	✅	✅	jax/numpy/trilu	Covered via existing plugin primitives.
Unique	✅	✅	✅	jax/numpy/unique	Covered via existing plugin primitives.
Unsqueeze	✅	✅	✅	_complex_utils dm_pix/depth_to_space dm_pix/space_to_depth equinox/eqx/nn/adaptive_pool equinox/eqx/nn/avg_pool equinox/eqx/nn/conv ... +27	Covered via existing plugin primitives.
Upsample	✅	✅	✅	jax/image/resize	Covered via existing plugin primitives.
Where	✅	✅	✅	flax/linen/dot_product_attention flax/nnx/dot_product_attention jax/lax/atan2 jax/lax/bessel_i0e jax/lax/bessel_i1e jax/lax/betainc ... +27	Covered via existing plugin primitives.
Xor	✅	✅	✅	jax/lax/xor jax/numpy/bitwise_xor jax/numpy/floor_divide	Covered via existing plugin primitives.

Plugin Names Not In Index

This section lists operator-like names found in plugins that do not appear in the ONNX index page.

Name Found in Plugins	Metadata	Lowering	Plugin Modules	Potential JAX Ops
Cholesky	✅	➖	jax/numpy/composite_metadata_batch4	Internal alias/helper; map manually if needed.
Complex	➖	✅	_complex_utils	jax.lax.complex jax.numpy.asarray(..., dtype=complex64/128)
Eig	✅	➖	jax/numpy/composite_metadata_batch5	Internal alias/helper; map manually if needed.
Eigh	✅	➖	jax/numpy/composite_metadata_batch5	Internal alias/helper; map manually if needed.
QR	✅	➖	jax/numpy/composite_metadata_batch4	Internal alias/helper; map manually if needed.
SVD	✅	➖	jax/numpy/composite_metadata_batch4 jax/numpy/composite_metadata_batch5	Internal alias/helper; map manually if needed.