ONNX Model Zoo

enpasos/jax2onnx-models

By enpasos · ↓ 0 · ♥ 0 · Updated 2026-02-28

What's in this repo

This page lists the files in enpasos/jax2onnx-models, including mlp_batched_training_mode.onnx, mlp_batched_training_mode_f64.onnx, mlp_inference_mode.onnx and more.

  • 3461 ONNX models (.onnx files)
  • 3463 files, including 3461 .onnx, 1 .md
  • Last updated 2026-02-28

ONNX files

File Size Download
examples/eqx/mlp_batched_training_mode.onnx — Download
examples/eqx/mlp_batched_training_mode_f64.onnx — Download
examples/eqx/mlp_inference_mode.onnx — Download
examples/eqx/mlp_inference_mode_f64.onnx — Download
examples/eqx/mlp_training_mode.onnx — Download
examples/eqx/mlp_training_mode_f64.onnx — Download
examples/eqx/nn_linear.onnx — Download
examples/eqx/nn_linear_f64.onnx — Download
examples/eqx/simple_linear.onnx — Download
examples/eqx/simple_linear_f64.onnx — Download

Showing 10 of 3461 .onnx files — see the full file list on the right.

How to use

These ONNX files can be loaded with ONNX Runtime. To inspect a model graph, upload the file to the Netron viewer. Download links point to the original files on Hugging Face.

README

jax2onnx-models

🧪 This repository hosts ONNX models generated by the jax2onnx converter as part of its automated pytest suite.

These models are used to:

  • Validate correctness of the ONNX export pipeline
  • Provide public artifacts for visualization and inspection (e.g., with Netron)
  • Enable CI verification of shape/type fidelity, runtime compatibility, and graph structure

📦 Format

All models are stored in the ONNX (Open Neural Network Exchange) format.
They are compatible with:

  • ONNX runtimes (ONNX Runtime, TensorRT, etc.)
  • Netron for visual graph inspection
  • External ONNX validators

Models may include both small unit test graphs and larger architectures like GPT LLMs or Vision Transformers.

🔁 Update Policy

  • This repository is updated regularly with models generated during testing.
  • Only the latest model state is kept — older versions are overwritten as needed.
  • We do not maintain file history to keep the repository lightweight.

📄 License

This repository and all included model files are licensed under the Apache License 2.0.

See LICENSE for details.

Model info

onnx jax jax2onnx model-conversion license:apache-2.0 region:us
↓ 0 ♥ 0

Created: 2025-06-29

Updated: 2026-02-28

View on Hugging Face

Files (3463)

.gitattributes
README.md
examples/eqx/mlp_batched_training_mode.onnx ONNX
examples/eqx/mlp_batched_training_mode_f64.onnx ONNX
examples/eqx/mlp_inference_mode.onnx ONNX
examples/eqx/mlp_inference_mode_f64.onnx ONNX
examples/eqx/mlp_training_mode.onnx ONNX
examples/eqx/mlp_training_mode_f64.onnx ONNX
examples/eqx/nn_linear.onnx ONNX
examples/eqx/nn_linear_f64.onnx ONNX
examples/eqx/simple_linear.onnx ONNX
examples/eqx/simple_linear_f64.onnx ONNX
examples/eqx_dino/attention.onnx ONNX
examples/eqx_dino/attention_core.onnx ONNX
examples/eqx_dino/attention_core_dynamic.onnx ONNX
examples/eqx_dino/attention_dynamic.onnx ONNX
examples/eqx_dino/eqx_dinov3_vit_B14.onnx ONNX
examples/eqx_dino/eqx_dinov3_vit_B14_dynamic.onnx ONNX
examples/eqx_dino/eqx_dinov3_vit_S14.onnx ONNX
examples/eqx_dino/eqx_dinov3_vit_S14_dynamic.onnx ONNX
examples/eqx_dino/eqx_dinov3_vit_S16.onnx ONNX
examples/eqx_dino/eqx_dinov3_vit_S16_dynamic.onnx ONNX
examples/eqx_dino/eqx_dinov3_vit_Ti14.onnx ONNX
examples/eqx_dino/eqx_dinov3_vit_Ti14_dynamic.onnx ONNX
examples/eqx_dino/patch_embed.onnx ONNX
examples/eqx_dino/transformer_block.onnx ONNX
examples/eqx_dino/transformer_block_dynamic.onnx ONNX
examples/eqx_gpt_oss/gpt_oss_attention_block.onnx ONNX
examples/eqx_gpt_oss/gpt_oss_mlp_block.onnx ONNX
examples/eqx_gpt_oss/gpt_oss_rmsnorm.onnx ONNX
examples/eqx_gpt_oss/gpt_oss_rmsnorm_dynamic.onnx ONNX
examples/eqx_gpt_oss/gpt_oss_transformer.onnx ONNX
examples/eqx_gpt_oss/gpt_oss_transformer_block.onnx ONNX
examples/eqx_gpt_oss/gpt_oss_transformer_block_dynamic.onnx ONNX
examples/gpt/gpt.onnx ONNX
examples/gpt/gpt_attention.onnx ONNX
examples/gpt/gpt_block.onnx ONNX
examples/gpt/gpt_block_dynamic.onnx ONNX
examples/gpt/gpt_broadcast_add_dynamic.onnx ONNX
examples/gpt/gpt_broadcast_add_dynamic_dynamic.onnx ONNX
examples/gpt/gpt_broadcast_add_dynamic_dynamic_f64.onnx ONNX
examples/gpt/gpt_broadcast_add_dynamic_f64.onnx ONNX
examples/gpt/gpt_causal_self_attention.onnx ONNX
examples/gpt/gpt_causal_self_attention_dynamic.onnx ONNX
examples/gpt/gpt_dynamic.onnx ONNX
examples/gpt/gpt_embeddings.onnx ONNX
examples/gpt/gpt_embeddings_dynamic.onnx ONNX
examples/gpt/gpt_head.onnx ONNX
examples/gpt/gpt_head_dynamic.onnx ONNX
examples/gpt/gpt_mlp.onnx ONNX
examples/gpt/gpt_mlp_dynamic.onnx ONNX
examples/gpt/gpt_position_embedding.onnx ONNX
examples/gpt/gpt_token_embedding.onnx ONNX
examples/gpt/gpt_token_embedding_dynamic.onnx ONNX
examples/gpt/gpt_transformer_stack.onnx ONNX
examples/gpt/gpt_transformer_stack_dynamic.onnx ONNX
examples/jaxfluids/cfl_timestep_f64.onnx ONNX
examples/jaxfluids/weno_reconstruction_f64.onnx ONNX
examples/jnp/abs_fn.onnx ONNX
examples/jnp/abs_fn_f64.onnx ONNX
examples/jnp/arange_fn.onnx ONNX
examples/jnp/fori_loop_fn.onnx ONNX
examples/jnp/fori_loop_fn_f64.onnx ONNX
examples/jnp/fori_loop_test.onnx ONNX
examples/jnp/fori_loop_test_f64.onnx ONNX
examples/jnp/linspace_fn.onnx ONNX
examples/jnp/scan_fn.onnx ONNX
examples/jnp/select_test_all_options.onnx ONNX
examples/jnp/select_test_default_case.onnx ONNX
examples/jnp/select_test_scalar_select_option_0.onnx ONNX
examples/jnp/select_test_scalar_select_option_1.onnx ONNX
examples/jnp/select_test_scalar_select_option_2.onnx ONNX
examples/jnp/sign_fn.onnx ONNX
examples/jnp/sign_fn_f64.onnx ONNX
examples/jnp/sort_test_basic.onnx ONNX
examples/jnp/where_fn.onnx ONNX
examples/jnp/where_fn_f64.onnx ONNX
examples/jnp/while_loop_fn.onnx ONNX
examples/lax/checkpoint_scalar_f32.onnx ONNX
examples/lax/checkpoint_scalar_f32_f64.onnx ONNX
examples/lax/cond_scatter_add_mul_f64_a.onnx ONNX
examples/lax/cond_scatter_add_mul_f64_b.onnx ONNX
examples/lax/cond_scatter_repro_f64.onnx ONNX
examples/lax/scatter_window_update_f64_example.onnx ONNX
examples/lax/two_times_silu_scalar.onnx ONNX
examples/lax/two_times_silu_scalar_f64.onnx ONNX
examples/linen/simple_cnn_dynamic.onnx ONNX
examples/linen/simple_cnn_static.onnx ONNX
examples/linen/simple_linen_mlp_dynamic.onnx ONNX
examples/linen/simple_linen_mlp_dynamic_f64.onnx ONNX
examples/linen/simple_linen_mlp_sequential_dynamic.onnx ONNX
examples/linen/simple_linen_mlp_sequential_dynamic_f64.onnx ONNX
examples/linen/simple_linen_mlp_sequential_static.onnx ONNX
examples/linen/simple_linen_mlp_sequential_static_f64.onnx ONNX
examples/linen/simple_linen_mlp_static.onnx ONNX
examples/linen/simple_linen_mlp_static_f64.onnx ONNX
examples/linen/simple_linen_mlp_with_call_params.onnx ONNX
examples/linen/simple_linen_mlp_with_call_params_dynamic.onnx ONNX
examples/linen/simple_linen_mlp_with_call_params_dynamic_f64.onnx ONNX
examples/linen/simple_linen_mlp_with_call_params_f64.onnx ONNX
examples/maxtext/maxtext_deepseek-custom.onnx ONNX
examples/maxtext/maxtext_deepseek2-16b.onnx ONNX
examples/maxtext/maxtext_deepseek2-236b.onnx ONNX
examples/maxtext/maxtext_deepseek3-671b-2dfsdp.onnx ONNX
examples/maxtext/maxtext_deepseek3-671b.onnx ONNX
examples/maxtext/maxtext_deepseek3-test.onnx ONNX
examples/maxtext/maxtext_deepseek3-tiny.onnx ONNX
examples/maxtext/maxtext_deepseek3_2-671b.onnx ONNX
examples/maxtext/maxtext_gemma-2b.onnx ONNX
examples/maxtext/maxtext_gemma-7b.onnx ONNX
examples/maxtext/maxtext_gemma2-27b.onnx ONNX
examples/maxtext/maxtext_gemma2-2b.onnx ONNX
examples/maxtext/maxtext_gemma2-9b.onnx ONNX
examples/maxtext/maxtext_gemma3-12b.onnx ONNX
examples/maxtext/maxtext_gemma3-27b.onnx ONNX
examples/maxtext/maxtext_gemma3-4b.onnx ONNX
examples/maxtext/maxtext_gpt3-175b.onnx ONNX
examples/maxtext/maxtext_gpt3-22b.onnx ONNX
examples/maxtext/maxtext_gpt3-52k.onnx ONNX
examples/maxtext/maxtext_gpt3-6b.onnx ONNX
examples/maxtext/maxtext_kimi-k2-1t.onnx ONNX
examples/maxtext/maxtext_llama2-13b.onnx ONNX
examples/maxtext/maxtext_llama2-70b.onnx ONNX
examples/maxtext/maxtext_llama2-7b.onnx ONNX
examples/maxtext/maxtext_llama3-405b.onnx ONNX
examples/maxtext/maxtext_llama3-70b.onnx ONNX
examples/maxtext/maxtext_llama3-8b.onnx ONNX
examples/maxtext/maxtext_llama3_1-405b.onnx ONNX
examples/maxtext/maxtext_llama3_1-70b.onnx ONNX
examples/maxtext/maxtext_llama3_1-8b.onnx ONNX
examples/maxtext/maxtext_llama3_3-70b.onnx ONNX
examples/maxtext/maxtext_mistral-7b.onnx ONNX
examples/maxtext/maxtext_olmo3-32b.onnx ONNX
examples/maxtext/maxtext_olmo3-7b-pt.onnx ONNX
examples/maxtext/maxtext_olmo3-7b.onnx ONNX
examples/maxtext/maxtext_qwen3-0_6b.onnx ONNX
examples/maxtext/maxtext_qwen3-14b.onnx ONNX
examples/maxtext/maxtext_qwen3-235b-a22b.onnx ONNX
examples/maxtext/maxtext_qwen3-30b-a3b.onnx ONNX
examples/maxtext/maxtext_qwen3-32b.onnx ONNX
examples/maxtext/maxtext_qwen3-480b-a35b.onnx ONNX
examples/maxtext/maxtext_qwen3-4b-thinking-2507.onnx ONNX
examples/maxtext/maxtext_qwen3-4b.onnx ONNX
examples/maxtext/maxtext_qwen3-8b.onnx ONNX
examples/maxtext/maxtext_qwen3-next-80b-a3b.onnx ONNX
examples/maxtext/maxtext_qwen3-omni-30b-a3b.onnx ONNX
examples/nnx/depth_to_space_resnet_inputs_outputs_as_nchw.onnx ONNX
examples/nnx/depth_to_space_resnet_inputs_outputs_as_nchw_dynamic_hw.onnx ONNX
examples/nnx/depth_to_space_resnet_scaled_inputs_outputs_as_nchw.onnx ONNX
examples/nnx/depth_to_space_resnet_static.onnx ONNX
examples/nnx/fori_loop_counter.onnx ONNX
examples/nnx/fori_loop_counter_custom_io_names.onnx ONNX
examples/nnx/fori_loop_counter_custom_io_names_f64.onnx ONNX
examples/nnx/fori_loop_counter_custom_io_names_two_inputs.onnx ONNX
examples/nnx/fori_loop_counter_custom_io_names_two_inputs_f64.onnx ONNX
examples/nnx/fori_loop_counter_f64.onnx ONNX
examples/nnx/gru_cell_basic.onnx ONNX
examples/nnx/multihead_attention_2_nnx.onnx ONNX
examples/nnx/multihead_attention_2_nnx_dynamic.onnx ONNX
examples/nnx/multihead_attention_nn.onnx ONNX
examples/nnx/multihead_attention_nn_dynamic.onnx ONNX
examples/nnx/multihead_attention_nnx.onnx ONNX
examples/nnx/multihead_attention_nnx_dynamic.onnx ONNX
examples/nnx/nested_residual_group_static.onnx ONNX
examples/nnx/nested_residual_group_static_nchw.onnx ONNX
examples/nnx/nested_residual_group_with_lead_conv_static.onnx ONNX
examples/nnx/nested_residual_stack_static.onnx ONNX
examples/nnx/nested_residual_stack_static_no_extra_transpose.onnx ONNX
examples/nnx/nested_residual_stack_with_lead_conv_static.onnx ONNX
examples/nnx/resblock_channel_attention_dynamic_hw.onnx ONNX
examples/nnx/resblock_channel_attention_dynamic_hw_nchw.onnx ONNX
examples/nnx/resblock_channel_attention_static.onnx ONNX
examples/nnx/resblock_channel_attention_static_nchw.onnx ONNX
examples/nnx/sequential_double_relu.onnx ONNX
examples/nnx/sequential_double_relu_f64.onnx ONNX
examples/nnx/sequential_nested_with_residual.onnx ONNX
examples/nnx/simple_autoencoder.onnx ONNX
examples/nnx/simple_autoencoder_f64.onnx ONNX
examples/nnx/simple_cnn_dynamic.onnx ONNX
examples/nnx/simple_cnn_static.onnx ONNX
examples/nnx/simple_mlp_dynamic.onnx ONNX
examples/nnx/simple_mlp_dynamic_f64.onnx ONNX
examples/nnx/simple_mlp_static.onnx ONNX
examples/nnx/simple_mlp_static_f64.onnx ONNX
examples/nnx/simple_mlp_with_call_params.onnx ONNX
examples/nnx/simple_mlp_with_call_params_dynamic.onnx ONNX
examples/nnx/simple_mlp_with_call_params_dynamic_f64.onnx ONNX
examples/nnx/simple_mlp_with_call_params_f64.onnx ONNX
examples/nnx/simple_model_clip_nchw_io.onnx ONNX
examples/nnx/simple_model_clip_nhwc.onnx ONNX
examples/nnx/tiny_decoder_with_sequential.onnx ONNX
examples/nnx/tiny_decoder_with_sequential_and_full_dynamic_shapes_dynamic.onnx ONNX
examples/nnx/tiny_decoder_without_sequential.onnx ONNX
examples/nnx_dino/nnx_attention.onnx ONNX
examples/nnx_dino/nnx_attention_core.onnx ONNX
examples/nnx_dino/nnx_attention_core_dynamic.onnx ONNX
examples/nnx_dino/nnx_attention_dynamic.onnx ONNX
examples/nnx_dino/nnx_dinov3_vit_B14.onnx ONNX
examples/nnx_dino/nnx_dinov3_vit_B14_dynamic.onnx ONNX
examples/nnx_dino/nnx_dinov3_vit_S14.onnx ONNX
examples/nnx_dino/nnx_dinov3_vit_S14_dynamic.onnx ONNX
examples/nnx_dino/nnx_dinov3_vit_S16.onnx ONNX
examples/nnx_dino/nnx_dinov3_vit_S16_dynamic.onnx ONNX
examples/nnx_dino/nnx_dinov3_vit_Ti14.onnx ONNX
examples/nnx_dino/nnx_dinov3_vit_Ti14_dynamic.onnx ONNX
examples/nnx_dino/nnx_patch_embed.onnx ONNX
examples/nnx_dino/nnx_transformer_block.onnx ONNX
examples/nnx_dino/nnx_transformer_block_dynamic.onnx ONNX
examples/nnx_gpt_oss/gpt_oss_attention_flax.onnx ONNX
examples/nnx_gpt_oss/gpt_oss_mlp_flax.onnx ONNX
examples/nnx_gpt_oss/gpt_oss_rmsnorm_flax.onnx ONNX
examples/nnx_gpt_oss/gpt_oss_rmsnorm_flax_dynamic.onnx ONNX
examples/nnx_gpt_oss/gpt_oss_rotary_flax.onnx ONNX
examples/nnx_gpt_oss/gpt_oss_sdpa_flax.onnx ONNX
examples/nnx_gpt_oss/gpt_oss_transformer_block_flax.onnx ONNX
examples/nnx_gpt_oss/gpt_oss_transformer_flax.onnx ONNX
examples/onnx_functions/000_one_function_on_outer_layer.onnx ONNX
examples/onnx_functions/000_one_function_on_outer_layer_dynamic.onnx ONNX
examples/onnx_functions/001_one_function_inner.onnx ONNX
examples/onnx_functions/001_one_function_inner_dynamic.onnx ONNX
examples/onnx_functions/002_two_nested_functions.onnx ONNX
examples/onnx_functions/002_two_nested_functions_dynamic.onnx ONNX
examples/onnx_functions/003_two_simple_nested_functions.onnx ONNX
examples/onnx_functions/003_two_simple_nested_functions_dynamic.onnx ONNX
examples/onnx_functions/004_nested_function_plus_component.onnx ONNX
examples/onnx_functions/004_nested_function_plus_component_dynamic.onnx ONNX
examples/onnx_functions/005_nested_function_plus_component.onnx ONNX
examples/onnx_functions/005_nested_function_plus_component_dynamic.onnx ONNX
examples/onnx_functions/006_one_function_outer.onnx ONNX
examples/onnx_functions/006_one_function_outer_dynamic.onnx ONNX
examples/onnx_functions/007_transformer_block.onnx ONNX
examples/onnx_functions/007_transformer_block_dynamic.onnx ONNX
examples/onnx_functions/008_transformer_block.onnx ONNX
examples/onnx_functions/008_transformer_block_dynamic.onnx ONNX
examples/onnx_functions/009_transformer_block.onnx ONNX
examples/onnx_functions/009_transformer_block_dynamic.onnx ONNX
examples/onnx_functions/010_transformer_stack.onnx ONNX
examples/onnx_functions/010_transformer_stack_dynamic.onnx ONNX
examples/onnx_functions/012_vit_conv_embedding.onnx ONNX
examples/onnx_functions/012_vit_conv_embedding_dynamic.onnx ONNX
examples/onnx_functions/013_vit_conv_embedding_with_call_params.onnx ONNX
examples/onnx_functions/013_vit_conv_embedding_with_call_params_dynamic.onnx ONNX
examples/onnx_functions/013_vit_conv_embedding_with_internal_call_params.onnx ONNX
examples/onnx_functions/013_vit_conv_embedding_with_internal_call_params_dynamic.onnx ONNX
examples/onnx_functions/014_one_function_with_input_param_with_default_value.onnx ONNX
examples/onnx_functions/014_one_function_without_input_param_with_default_value.onnx ONNX
examples/onnx_functions/014_one_function_without_input_param_with_default_value_dynamic.onnx ONNX
examples/onnx_functions/015_one_function_with_input_param_without_default_value.onnx ONNX
examples/onnx_functions/015_one_function_with_input_param_without_default_value_dynamic.onnx ONNX
examples/onnx_functions/016_internal_function_with_input_param_with_default_value.onnx ONNX
examples/onnx_functions/016_internal_function_with_input_param_with_default_value_dynamic.onnx ONNX
examples/onnx_functions/017_unique_function_reuse.onnx ONNX
examples/vit/vit_classification_head.onnx ONNX
examples/vit/vit_classification_head_dynamic.onnx ONNX
examples/vit/vit_classification_head_flat.onnx ONNX
examples/vit/vit_classification_head_flat_dynamic.onnx ONNX
examples/vit/vit_concat_cls_token.onnx ONNX
examples/vit/vit_concat_cls_token_dynamic.onnx ONNX
examples/vit/vit_concat_cls_token_flat.onnx ONNX
examples/vit/vit_concat_cls_token_flat_dynamic.onnx ONNX
examples/vit/vit_feed_forward.onnx ONNX
examples/vit/vit_feed_forward_dynamic.onnx ONNX
examples/vit/vit_feed_forward_flat.onnx ONNX
examples/vit/vit_feed_forward_flat_dynamic.onnx ONNX
examples/vit/vit_get_token.onnx ONNX
examples/vit/vit_get_token_dynamic.onnx ONNX
examples/vit/vit_get_token_flat.onnx ONNX
examples/vit/vit_get_token_flat_dynamic.onnx ONNX
examples/vit/vit_mnist_conv_embedding.onnx ONNX
examples/vit/vit_mnist_conv_embedding_dynamic.onnx ONNX
examples/vit/vit_mnist_conv_embedding_flat.onnx ONNX
examples/vit/vit_mnist_conv_embedding_flat_dynamic.onnx ONNX
examples/vit/vit_model_conv_embedding.onnx ONNX
examples/vit/vit_model_conv_embedding_dynamic.onnx ONNX
examples/vit/vit_model_conv_embedding_flat.onnx ONNX
examples/vit/vit_model_conv_embedding_flat_dynamic.onnx ONNX
examples/vit/vit_model_patch_embedding.onnx ONNX
examples/vit/vit_model_patch_embedding_flat.onnx ONNX
examples/vit/vit_model_patch_embedding_flat_dynamic.onnx ONNX
examples/vit/vit_patch_embedding.onnx ONNX
examples/vit/vit_patch_embedding_dynamic.onnx ONNX
examples/vit/vit_patch_embedding_flat.onnx ONNX
examples/vit/vit_patch_embedding_flat_dynamic.onnx ONNX
examples/vit/vit_positional_embedding.onnx ONNX
examples/vit/vit_positional_embedding_dynamic.onnx ONNX
examples/vit/vit_positional_embedding_flat.onnx ONNX
examples/vit/vit_positional_embedding_flat_dynamic.onnx ONNX
examples/vit/vit_transformer_block.onnx ONNX
examples/vit/vit_transformer_block_dynamic.onnx ONNX
examples/vit/vit_transformer_block_flat.onnx ONNX
examples/vit/vit_transformer_block_flat_dynamic.onnx ONNX
examples/vit/vit_transformer_stack.onnx ONNX
examples/vit/vit_transformer_stack_dynamic.onnx ONNX
examples/vit/vit_transformer_stack_flat.onnx ONNX
examples/vit/vit_transformer_stack_flat_dynamic.onnx ONNX
model_with_function.onnx ONNX
my_callable.onnx ONNX
primitives/core/custom_jvp_square.onnx ONNX
primitives/core/custom_jvp_square_f64.onnx ONNX
primitives/core/custom_vjp_square.onnx ONNX
primitives/core/custom_vjp_square_f64.onnx ONNX
primitives/core/dim_as_value.onnx ONNX
primitives/core/dim_as_value_dynamic.onnx ONNX
primitives/core/dim_as_value_dynamic_f64.onnx ONNX
primitives/core/dim_as_value_f64.onnx ONNX
primitives/core/jit_identity.onnx ONNX
primitives/core/jit_identity_f64.onnx ONNX
primitives/dm_pix/depth_to_space_nhwc.onnx ONNX
primitives/dm_pix/space_to_depth_nhwc.onnx ONNX
primitives/eqx/batched_layer_norm.onnx ONNX
primitives/eqx/batched_layer_norm_dynamic.onnx ONNX
primitives/eqx/batched_layer_norm_dynamic_f64.onnx ONNX
primitives/eqx/batched_layer_norm_f64.onnx ONNX
primitives/eqx/eqx_adaptive_avg_pool2d_divisible.onnx ONNX
primitives/eqx/eqx_adaptive_max_pool2d_divisible.onnx ONNX
primitives/eqx/eqx_adaptive_pool_generic_avg_vmap.onnx ONNX
primitives/eqx/eqx_adaptive_pool_generic_avg_vmap_dynamic.onnx ONNX
primitives/eqx/eqx_avg_pool2d_basic.onnx ONNX
primitives/eqx/eqx_avg_pool2d_basic_f64.onnx ONNX
primitives/eqx/eqx_avg_pool2d_batched.onnx ONNX
primitives/eqx/eqx_avg_pool2d_batched_dynamic.onnx ONNX
primitives/eqx/eqx_avg_pool2d_batched_dynamic_f64.onnx ONNX
primitives/eqx/eqx_avg_pool2d_batched_f64.onnx ONNX
primitives/eqx/eqx_batch_norm_inference.onnx ONNX
primitives/eqx/eqx_conv2d_batched_nchw.onnx ONNX
primitives/eqx/eqx_conv2d_nchw.onnx ONNX
primitives/eqx/eqx_dropout_batched_inference.onnx ONNX
primitives/eqx/eqx_dropout_batched_inference_dynamic.onnx ONNX
primitives/eqx/eqx_dropout_batched_inference_dynamic_f64.onnx ONNX
primitives/eqx/eqx_dropout_batched_inference_f64.onnx ONNX
primitives/eqx/eqx_dropout_dynamic_inference.onnx ONNX
primitives/eqx/eqx_dropout_dynamic_inference_f64.onnx ONNX
primitives/eqx/eqx_dropout_inference_mode.onnx ONNX
primitives/eqx/eqx_dropout_inference_mode_f64.onnx ONNX
primitives/eqx/eqx_dropout_training_mode.onnx ONNX
primitives/eqx/eqx_dropout_training_mode_f64.onnx ONNX
primitives/eqx/eqx_embedding_batched_vmap.onnx ONNX
primitives/eqx/eqx_embedding_batched_vmap_dynamic.onnx ONNX
primitives/eqx/eqx_embedding_batched_vmap_dynamic_f64.onnx ONNX
primitives/eqx/eqx_embedding_batched_vmap_f64.onnx ONNX
primitives/eqx/eqx_embedding_scalar_index.onnx ONNX
primitives/eqx/eqx_embedding_scalar_index_f64.onnx ONNX
primitives/eqx/eqx_group_norm_no_affine.onnx ONNX
primitives/eqx/eqx_group_norm_no_affine_f64.onnx ONNX
primitives/eqx/eqx_group_norm_rank3.onnx ONNX
primitives/eqx/eqx_group_norm_rank3_f64.onnx ONNX
primitives/eqx/eqx_gru_cell_basic.onnx ONNX
primitives/eqx/eqx_gru_cell_batched.onnx ONNX
primitives/eqx/eqx_identity_static.onnx ONNX
primitives/eqx/eqx_identity_static_f64.onnx ONNX
primitives/eqx/eqx_identity_symbolic_batch.onnx ONNX
primitives/eqx/eqx_identity_symbolic_batch_dynamic.onnx ONNX
primitives/eqx/eqx_identity_symbolic_batch_dynamic_f64.onnx ONNX
primitives/eqx/eqx_identity_symbolic_batch_f64.onnx ONNX
primitives/eqx/eqx_lambda_relu.onnx ONNX
primitives/eqx/eqx_lambda_relu_f64.onnx ONNX
primitives/eqx/eqx_lambda_sigmoid.onnx ONNX
primitives/eqx/eqx_lambda_sigmoid_f64.onnx ONNX
primitives/eqx/eqx_linear_high_rank.onnx ONNX
primitives/eqx/eqx_linear_high_rank_f64.onnx ONNX
primitives/eqx/eqx_linear_no_bias_symbolic_batch.onnx ONNX
primitives/eqx/eqx_linear_no_bias_symbolic_batch_dynamic.onnx ONNX
primitives/eqx/eqx_linear_no_bias_symbolic_batch_dynamic_f64.onnx ONNX
primitives/eqx/eqx_linear_no_bias_symbolic_batch_f64.onnx ONNX
primitives/eqx/eqx_linear_no_bias_vector.onnx ONNX
primitives/eqx/eqx_linear_no_bias_vector_f64.onnx ONNX
primitives/eqx/eqx_linear_symbolic_batch.onnx ONNX
primitives/eqx/eqx_linear_symbolic_batch_dynamic.onnx ONNX
primitives/eqx/eqx_linear_symbolic_batch_dynamic_f64.onnx ONNX
primitives/eqx/eqx_linear_symbolic_batch_f64.onnx ONNX
primitives/eqx/eqx_linear_vector.onnx ONNX
primitives/eqx/eqx_linear_vector_f64.onnx ONNX
primitives/eqx/eqx_lstm_cell_basic.onnx ONNX
primitives/eqx/eqx_max_pool2d_basic.onnx ONNX
primitives/eqx/eqx_max_pool2d_basic_f64.onnx ONNX
primitives/eqx/eqx_max_pool2d_batched.onnx ONNX
primitives/eqx/eqx_max_pool2d_batched_dynamic.onnx ONNX
primitives/eqx/eqx_max_pool2d_batched_dynamic_f64.onnx ONNX
primitives/eqx/eqx_max_pool2d_batched_f64.onnx ONNX
primitives/eqx/eqx_multihead_attention.onnx ONNX
primitives/eqx/eqx_multihead_attention_core_dynamic.onnx ONNX
primitives/eqx/eqx_multihead_attention_core_static.onnx ONNX
primitives/eqx/eqx_pool_max2d_basic.onnx ONNX
primitives/eqx/eqx_pool_max2d_basic_f64.onnx ONNX
primitives/eqx/eqx_pool_max2d_batched.onnx ONNX
primitives/eqx/eqx_pool_max2d_batched_dynamic.onnx ONNX
primitives/eqx/eqx_pool_max2d_batched_dynamic_f64.onnx ONNX
primitives/eqx/eqx_pool_max2d_batched_f64.onnx ONNX
primitives/eqx/eqx_prelu_channelwise.onnx ONNX
primitives/eqx/eqx_prelu_default.onnx ONNX
primitives/eqx/eqx_prelu_default_dynamic.onnx ONNX
primitives/eqx/eqx_rotary_positional_embedding.onnx ONNX
primitives/eqx/eqx_rotary_positional_embedding_heads.onnx ONNX
primitives/eqx/eqx_sequential_identity_sigmoid.onnx ONNX
primitives/eqx/eqx_sequential_identity_sigmoid_f64.onnx ONNX
primitives/eqx/eqx_sequential_relu_tanh.onnx ONNX
primitives/eqx/eqx_sequential_relu_tanh_dynamic.onnx ONNX
primitives/eqx/eqx_sequential_relu_tanh_dynamic_f64.onnx ONNX
primitives/eqx/eqx_sequential_relu_tanh_f64.onnx ONNX
primitives/eqx/eqx_spectral_norm_linear_inference.onnx ONNX
primitives/eqx/eqx_weight_norm_conv2d.onnx ONNX
primitives/eqx/eqx_weight_norm_linear.onnx ONNX
primitives/eqx/eqx_weight_norm_linear_dynamic.onnx ONNX
primitives/eqx/layer_norm.onnx ONNX
primitives/eqx/layer_norm_f64.onnx ONNX
primitives/eqx/layer_norm_multiaxis.onnx ONNX
primitives/eqx/layer_norm_multiaxis_f64.onnx ONNX
primitives/eqx/layer_norm_no_bias_no_scale.onnx ONNX
primitives/eqx/layer_norm_no_bias_no_scale_f64.onnx ONNX
primitives/eqx/rms_norm_no_affine.onnx ONNX
primitives/eqx/rms_norm_no_affine_f64.onnx ONNX
primitives/eqx/rms_norm_vector.onnx ONNX
primitives/eqx/rms_norm_vector_f64.onnx ONNX
primitives/jax_image/resize_linear.onnx ONNX
primitives/jax_image/resize_linear_opset9_upsample.onnx ONNX
primitives/jax_image/resize_nearest.onnx ONNX
primitives/jax_image/resize_nearest_opset9_upsample.onnx ONNX
primitives/jax_image/resize_nearest_rank3_opset9_upsample.onnx ONNX
primitives/jnp/abs_grad_issue_batch_diff_rules.onnx ONNX
primitives/jnp/abs_vmap_batching.onnx ONNX
primitives/jnp/abs_vmap_batching_f64.onnx ONNX
primitives/jnp/acos_grad_issue_batch_diff_rules.onnx ONNX
primitives/jnp/acos_vmap_batching.onnx ONNX
primitives/jnp/acos_vmap_batching_f64.onnx ONNX
primitives/jnp/acosh_grad_issue_batch_diff_rules.onnx ONNX
primitives/jnp/acosh_vmap_batching.onnx ONNX
primitives/jnp/acosh_vmap_batching_f64.onnx ONNX
primitives/jnp/add.onnx ONNX
primitives/jnp/add_f64.onnx ONNX
primitives/jnp/add_grad_issue_batch_diff_rules.onnx ONNX
primitives/jnp/add_vmap_batching.onnx ONNX
primitives/jnp/add_vmap_batching_f64.onnx ONNX
primitives/jnp/all_vmap_batching.onnx ONNX
primitives/jnp/all_vmap_batching_f64.onnx ONNX
primitives/jnp/amax_vmap_batching.onnx ONNX
primitives/jnp/amax_vmap_batching_f64.onnx ONNX
primitives/jnp/amin_vmap_batching.onnx ONNX
primitives/jnp/amin_vmap_batching_f64.onnx ONNX
primitives/jnp/any_vmap_batching.onnx ONNX
primitives/jnp/any_vmap_batching_f64.onnx ONNX
primitives/jnp/arange_data_dependent_indices.onnx ONNX
primitives/jnp/arange_float_concrete_input_val.onnx ONNX
primitives/jnp/arange_float_concrete_input_val_f64.onnx ONNX
primitives/jnp/arange_start_stop_concrete_input_val.onnx ONNX
primitives/jnp/arange_start_stop_concrete_input_val_f64.onnx ONNX
primitives/jnp/arange_start_stop_step_concrete_input_val.onnx ONNX
primitives/jnp/arange_start_stop_step_concrete_input_val_f64.onnx ONNX
primitives/jnp/arange_static_empty_result_neg_step.onnx ONNX
primitives/jnp/arange_static_empty_result_neg_step_f64.onnx ONNX
primitives/jnp/arange_static_empty_result_pos_step.onnx ONNX
primitives/jnp/arange_static_empty_result_pos_step_f64.onnx ONNX
primitives/jnp/arange_static_float_step_explicit_dtype.onnx ONNX
primitives/jnp/arange_static_float_step_explicit_dtype_f64.onnx ONNX
primitives/jnp/arange_static_float_step_inferred_dtype.onnx ONNX
primitives/jnp/arange_static_float_step_inferred_dtype_f64.onnx ONNX
primitives/jnp/arange_static_large_numbers_int.onnx ONNX
primitives/jnp/arange_static_large_numbers_int_f64.onnx ONNX
primitives/jnp/arange_static_negative_step.onnx ONNX
primitives/jnp/arange_static_negative_step_f64.onnx ONNX
primitives/jnp/arange_static_start_equals_stop.onnx ONNX
primitives/jnp/arange_static_start_equals_stop_f64.onnx ONNX
primitives/jnp/arange_static_start_stop_int.onnx ONNX
primitives/jnp/arange_static_start_stop_int_f64.onnx ONNX
primitives/jnp/arange_static_start_stop_step_int.onnx ONNX
primitives/jnp/arange_static_start_stop_step_int_f64.onnx ONNX
primitives/jnp/arange_static_stop_only_float.onnx ONNX
primitives/jnp/arange_static_stop_only_float_f64.onnx ONNX
primitives/jnp/arange_static_stop_only_int.onnx ONNX
primitives/jnp/arange_static_stop_only_int_f64.onnx ONNX
primitives/jnp/arange_static_stop_zero.onnx ONNX
primitives/jnp/arange_static_stop_zero_f64.onnx ONNX
primitives/jnp/arange_stop_only_concrete_input_val.onnx ONNX
primitives/jnp/arange_stop_only_concrete_input_val_f64.onnx ONNX
primitives/jnp/asin_grad_issue_batch_diff_rules.onnx ONNX
primitives/jnp/asin_vmap_batching.onnx ONNX
primitives/jnp/asin_vmap_batching_f64.onnx ONNX
primitives/jnp/asinh_grad_issue_batch_diff_rules.onnx ONNX
primitives/jnp/asinh_vmap_batching.onnx ONNX
primitives/jnp/asinh_vmap_batching_f64.onnx ONNX
primitives/jnp/atan2_grad_issue_batch_diff_rules.onnx ONNX
primitives/jnp/atan2_vmap_batching.onnx ONNX
primitives/jnp/atan_grad_issue_batch_diff_rules.onnx ONNX
primitives/jnp/atan_vmap_batching.onnx ONNX
primitives/jnp/atan_vmap_batching_f64.onnx ONNX
primitives/jnp/atanh_grad_issue_batch_diff_rules.onnx ONNX
primitives/jnp/atanh_vmap_batching.onnx ONNX
primitives/jnp/atanh_vmap_batching_f64.onnx ONNX
primitives/jnp/basic_mean.onnx ONNX
primitives/jnp/basic_mean_f64.onnx ONNX
primitives/jnp/basic_prod.onnx ONNX
primitives/jnp/basic_prod_f64.onnx ONNX
primitives/jnp/bitwise_and_vmap_batching.onnx ONNX
primitives/jnp/bitwise_and_vmap_batching_f64.onnx ONNX
primitives/jnp/bitwise_left_shift_vmap_batching.onnx ONNX
primitives/jnp/bitwise_left_shift_vmap_batching_f64.onnx ONNX
primitives/jnp/bitwise_not_vmap_batching.onnx ONNX
primitives/jnp/bitwise_not_vmap_batching_f64.onnx ONNX
primitives/jnp/bitwise_or_vmap_batching.onnx ONNX
primitives/jnp/bitwise_or_vmap_batching_f64.onnx ONNX
primitives/jnp/bitwise_right_shift_vmap_batching.onnx ONNX
primitives/jnp/bitwise_right_shift_vmap_batching_f64.onnx ONNX
primitives/jnp/bitwise_xor_vmap_batching.onnx ONNX
primitives/jnp/bitwise_xor_vmap_batching_f64.onnx ONNX
primitives/jnp/ceil_grad_issue_batch_diff_rules.onnx ONNX
primitives/jnp/ceil_vmap_batching.onnx ONNX
primitives/jnp/ceil_vmap_batching_f64.onnx ONNX
primitives/jnp/clip_broadcast_bounds.onnx ONNX
primitives/jnp/clip_f32_scalar_bounds_no_upcast_f64_mode.onnx ONNX
primitives/jnp/clip_grad_issue_batch_diff_rules.onnx ONNX
primitives/jnp/clip_i32_scalar_bounds.onnx ONNX
primitives/jnp/clip_i32_scalar_bounds_f64.onnx ONNX
primitives/jnp/clip_keyword_min_alias.onnx ONNX
primitives/jnp/clip_keyword_min_alias_f64.onnx ONNX
primitives/jnp/clip_only_lower.onnx ONNX
primitives/jnp/clip_only_lower_f64.onnx ONNX
primitives/jnp/clip_only_upper.onnx ONNX
primitives/jnp/clip_only_upper_f64.onnx ONNX
primitives/jnp/compress_grad_issue_batch_diff_rules.onnx ONNX
primitives/jnp/compress_jvp_issue_batch_diff_rules.onnx ONNX
primitives/jnp/concatenate_abstract_middle_dim.onnx ONNX
primitives/jnp/concatenate_abstract_middle_dim_dynamic.onnx ONNX
primitives/jnp/concatenate_abstract_middle_dim_dynamic_f64.onnx ONNX
primitives/jnp/concatenate_abstract_middle_dim_f64.onnx ONNX
primitives/jnp/concatenate_basic.onnx ONNX
primitives/jnp/concatenate_basic_f64.onnx ONNX
primitives/jnp/concatenate_grad_issue191.onnx ONNX
primitives/jnp/concatenate_mixed_dtypes.onnx ONNX
primitives/jnp/concatenate_mixed_dtypes_f64.onnx ONNX
primitives/jnp/concatenate_tile_and_symbolic.onnx ONNX
primitives/jnp/concatenate_tile_and_symbolic_dynamic.onnx ONNX
primitives/jnp/concatenate_tile_and_symbolic_dynamic_f64.onnx ONNX
primitives/jnp/concatenate_tile_and_symbolic_f64.onnx ONNX
primitives/jnp/concatenate_with_explicit_dtype.onnx ONNX
primitives/jnp/concatenate_with_explicit_dtype_casts_inputs.onnx ONNX
primitives/jnp/conj_grad_issue_batch_diff_rules.onnx ONNX
primitives/jnp/conj_vmap_batching.onnx ONNX
primitives/jnp/conj_vmap_batching_f64.onnx ONNX
primitives/jnp/copysign_grad_issue_batch_diff_rules.onnx ONNX
primitives/jnp/copysign_vmap_batching.onnx ONNX
primitives/jnp/copysign_vmap_batching_f64.onnx ONNX
primitives/jnp/cos_grad_issue_batch_diff_rules.onnx ONNX
primitives/jnp/cos_vmap_batching.onnx ONNX
primitives/jnp/cos_vmap_batching_f64.onnx ONNX
primitives/jnp/cosh_grad_issue_batch_diff_rules.onnx ONNX
primitives/jnp/cosh_vmap_batching.onnx ONNX
primitives/jnp/cosh_vmap_batching_f64.onnx ONNX
primitives/jnp/cumsum_axis2_i32.onnx ONNX
primitives/jnp/cumsum_axis2_i32_f64.onnx ONNX
primitives/jnp/cumsum_axis2_reverse_i32.onnx ONNX
primitives/jnp/cumsum_axis2_reverse_i32_f64.onnx ONNX
primitives/jnp/cumsum_vmap_batching.onnx ONNX
primitives/jnp/cumsum_vmap_batching_f64.onnx ONNX
primitives/jnp/divide_grad_issue_batch_diff_rules.onnx ONNX
primitives/jnp/divide_vmap_batching.onnx ONNX
primitives/jnp/divide_vmap_batching_f64.onnx ONNX
primitives/jnp/einsum_attention_logits_batched.onnx ONNX
primitives/jnp/einsum_attention_logits_batched_dynamic.onnx ONNX
primitives/jnp/einsum_attention_logits_batched_dynamic_f64.onnx ONNX
primitives/jnp/einsum_attention_logits_batched_f64.onnx ONNX
primitives/jnp/einsum_attention_logits_batched_rank_mismatch.onnx ONNX
primitives/jnp/einsum_attention_logits_batched_rank_mismatch_f64.onnx ONNX
primitives/jnp/einsum_attention_logits_orig.onnx ONNX
primitives/jnp/einsum_attention_logits_orig_dynamic.onnx ONNX
primitives/jnp/einsum_attention_logits_orig_dynamic_f64.onnx ONNX
primitives/jnp/einsum_attention_logits_orig_f64.onnx ONNX
primitives/jnp/einsum_attention_output_batched.onnx ONNX
primitives/jnp/einsum_attention_output_batched_dynamic.onnx ONNX
primitives/jnp/einsum_attention_output_batched_dynamic_f64.onnx ONNX
primitives/jnp/einsum_attention_output_batched_f64.onnx ONNX
primitives/jnp/einsum_attention_output_orig.onnx ONNX
primitives/jnp/einsum_attention_output_orig_dynamic.onnx ONNX
primitives/jnp/einsum_attention_output_orig_dynamic_f64.onnx ONNX
primitives/jnp/einsum_attention_output_orig_f64.onnx ONNX
primitives/jnp/einsum_batch_transpose.onnx ONNX
primitives/jnp/einsum_batch_transpose_dynamic.onnx ONNX
primitives/jnp/einsum_batch_transpose_dynamic_f64.onnx ONNX
primitives/jnp/einsum_batch_transpose_f64.onnx ONNX
primitives/jnp/einsum_diag.onnx ONNX
primitives/jnp/einsum_diag_f64.onnx ONNX
primitives/jnp/einsum_ellipsis_rank_mismatch.onnx ONNX
primitives/jnp/einsum_ellipsis_rank_mismatch_f64.onnx ONNX
primitives/jnp/einsum_grad_issue_batch_diff_rules.onnx ONNX
primitives/jnp/einsum_matrix_matrix.onnx ONNX
primitives/jnp/einsum_matrix_matrix_dynamic.onnx ONNX
primitives/jnp/einsum_matrix_matrix_dynamic_f64.onnx ONNX
primitives/jnp/einsum_matrix_matrix_f64.onnx ONNX
primitives/jnp/einsum_matrix_vector.onnx ONNX
primitives/jnp/einsum_matrix_vector_f64.onnx ONNX
primitives/jnp/einsum_multi_operand.onnx ONNX
primitives/jnp/einsum_multi_operand_f64.onnx ONNX
primitives/jnp/einsum_sum_reduce.onnx ONNX
primitives/jnp/einsum_sum_reduce_f64.onnx ONNX
primitives/jnp/einsum_transpose.onnx ONNX
primitives/jnp/einsum_transpose_f64.onnx ONNX
primitives/jnp/einsum_vector_dot.onnx ONNX
primitives/jnp/einsum_vector_dot_f64.onnx ONNX
primitives/jnp/exp2_grad_issue_batch_diff_rules.onnx ONNX
primitives/jnp/exp2_vmap_batching.onnx ONNX
primitives/jnp/exp2_vmap_batching_f64.onnx ONNX
primitives/jnp/exp_grad_issue_batch_diff_rules.onnx ONNX
primitives/jnp/exp_vmap_batching.onnx ONNX
primitives/jnp/exp_vmap_batching_f64.onnx ONNX
primitives/jnp/expm1_grad_issue_batch_diff_rules.onnx ONNX
primitives/jnp/expm1_vmap_batching.onnx ONNX
primitives/jnp/expm1_vmap_batching_f64.onnx ONNX
primitives/jnp/fabs_grad_issue_batch_diff_rules.onnx ONNX
primitives/jnp/fabs_vmap_batching.onnx ONNX
primitives/jnp/fabs_vmap_batching_f64.onnx ONNX
primitives/jnp/floor_divide_vmap_batching.onnx ONNX
primitives/jnp/floor_divide_vmap_batching_f64.onnx ONNX
primitives/jnp/floor_grad_issue_batch_diff_rules.onnx ONNX
primitives/jnp/floor_vmap_batching.onnx ONNX
primitives/jnp/floor_vmap_batching_f64.onnx ONNX
primitives/jnp/fmod_vmap_batching.onnx ONNX
primitives/jnp/fmod_vmap_batching_f64.onnx ONNX
primitives/jnp/greater_equal_vmap_batching.onnx ONNX
primitives/jnp/greater_equal_vmap_batching_f64.onnx ONNX
primitives/jnp/greater_vmap_batching.onnx ONNX
primitives/jnp/greater_vmap_batching_f64.onnx ONNX
primitives/jnp/invert_vmap_batching.onnx ONNX
primitives/jnp/invert_vmap_batching_f64.onnx ONNX
primitives/jnp/jnp_abs_basic.onnx ONNX
primitives/jnp/jnp_abs_basic_f64.onnx ONNX
primitives/jnp/jnp_abs_int.onnx ONNX
primitives/jnp/jnp_abs_int_f64.onnx ONNX
primitives/jnp/jnp_abs_matrix.onnx ONNX
primitives/jnp/jnp_abs_matrix_f64.onnx ONNX
primitives/jnp/jnp_abs_vmap_batching.onnx ONNX
primitives/jnp/jnp_abs_vmap_batching_f64.onnx ONNX
primitives/jnp/jnp_acos_basic.onnx ONNX
primitives/jnp/jnp_acos_basic_f64.onnx ONNX
primitives/jnp/jnp_acos_int_promote.onnx ONNX
primitives/jnp/jnp_acos_int_promote_f64.onnx ONNX
primitives/jnp/jnp_acosh_basic.onnx ONNX
primitives/jnp/jnp_acosh_basic_f64.onnx ONNX
primitives/jnp/jnp_acosh_int_promote.onnx ONNX
primitives/jnp/jnp_acosh_int_promote_f64.onnx ONNX
primitives/jnp/jnp_add_broadcast.onnx ONNX
primitives/jnp/jnp_add_broadcast_f64.onnx ONNX
primitives/jnp/jnp_add_vector.onnx ONNX
primitives/jnp/jnp_add_vector_f64.onnx ONNX
primitives/jnp/jnp_all_axis_1.onnx ONNX
primitives/jnp/jnp_all_axis_1_f64.onnx ONNX
primitives/jnp/jnp_all_axis_keepdims.onnx ONNX
primitives/jnp/jnp_all_axis_keepdims_f64.onnx ONNX
primitives/jnp/jnp_all_axis_none.onnx ONNX
primitives/jnp/jnp_all_axis_none_f64.onnx ONNX
primitives/jnp/jnp_all_basic.onnx ONNX
primitives/jnp/jnp_all_basic_f64.onnx ONNX
primitives/jnp/jnp_all_keepdims.onnx ONNX
primitives/jnp/jnp_all_keepdims_f64.onnx ONNX
primitives/jnp/jnp_allclose_false.onnx ONNX
primitives/jnp/jnp_allclose_false_f64.onnx ONNX
primitives/jnp/jnp_allclose_inf_true.onnx ONNX
primitives/jnp/jnp_allclose_inf_true_f64.onnx ONNX
primitives/jnp/jnp_allclose_true.onnx ONNX
primitives/jnp/jnp_allclose_true_f64.onnx ONNX
primitives/jnp/jnp_amax_axis.onnx ONNX
primitives/jnp/jnp_amax_axis_f64.onnx ONNX
primitives/jnp/jnp_amax_basic.onnx ONNX
primitives/jnp/jnp_amax_basic_f64.onnx ONNX
primitives/jnp/jnp_amax_keepdims.onnx ONNX
primitives/jnp/jnp_amax_keepdims_f64.onnx ONNX
primitives/jnp/jnp_amin_axis.onnx ONNX
primitives/jnp/jnp_amin_axis_f64.onnx ONNX
primitives/jnp/jnp_amin_basic.onnx ONNX
primitives/jnp/jnp_amin_basic_f64.onnx ONNX
primitives/jnp/jnp_amin_keepdims.onnx ONNX
primitives/jnp/jnp_amin_keepdims_f64.onnx ONNX
primitives/jnp/jnp_angle_complex64.onnx ONNX
primitives/jnp/jnp_angle_deg_complex64.onnx ONNX
primitives/jnp/jnp_any_axis_1.onnx ONNX
primitives/jnp/jnp_any_axis_1_f64.onnx ONNX
primitives/jnp/jnp_any_axis_keepdims.onnx ONNX
primitives/jnp/jnp_any_axis_keepdims_f64.onnx ONNX
primitives/jnp/jnp_any_axis_none.onnx ONNX
primitives/jnp/jnp_any_axis_none_f64.onnx ONNX
primitives/jnp/jnp_any_basic.onnx ONNX
primitives/jnp/jnp_any_basic_f64.onnx ONNX
primitives/jnp/jnp_any_keepdims.onnx ONNX
primitives/jnp/jnp_any_keepdims_f64.onnx ONNX
primitives/jnp/jnp_arccos.onnx ONNX
primitives/jnp/jnp_arccos_f64.onnx ONNX
primitives/jnp/jnp_arccosh.onnx ONNX
primitives/jnp/jnp_arccosh_f64.onnx ONNX
primitives/jnp/jnp_arcsin.onnx ONNX
primitives/jnp/jnp_arcsin_f64.onnx ONNX
primitives/jnp/jnp_arcsinh.onnx ONNX
primitives/jnp/jnp_arcsinh_f64.onnx ONNX
primitives/jnp/jnp_arctan.onnx ONNX
primitives/jnp/jnp_arctan2_broadcast.onnx ONNX
primitives/jnp/jnp_arctan2_quadrants_and_zero.onnx ONNX
primitives/jnp/jnp_arctan_f64.onnx ONNX
primitives/jnp/jnp_arctanh.onnx ONNX
primitives/jnp/jnp_arctanh_f64.onnx ONNX
primitives/jnp/jnp_argmax_axis0_bool.onnx ONNX
primitives/jnp/jnp_argmax_axis1.onnx ONNX
primitives/jnp/jnp_argmax_axis_1.onnx ONNX
primitives/jnp/jnp_argmax_axis_1_f64.onnx ONNX
primitives/jnp/jnp_argmax_axis_minus_1.onnx ONNX
primitives/jnp/jnp_argmax_axis_minus_1_f64.onnx ONNX
primitives/jnp/jnp_argmax_bool_input.onnx ONNX
primitives/jnp/jnp_argmax_bool_input_f64.onnx ONNX
primitives/jnp/jnp_argmin_axis0.onnx ONNX
primitives/jnp/jnp_argmin_axis1.onnx ONNX
primitives/jnp/jnp_argmin_axis_1.onnx ONNX
primitives/jnp/jnp_argmin_axis_1_f64.onnx ONNX
primitives/jnp/jnp_argmin_axis_minus_1.onnx ONNX
primitives/jnp/jnp_argmin_axis_minus_1_f64.onnx ONNX
primitives/jnp/jnp_argmin_bool_input.onnx ONNX
primitives/jnp/jnp_argmin_bool_input_f64.onnx ONNX
primitives/jnp/jnp_argpartition_basic.onnx ONNX
primitives/jnp/jnp_argpartition_basic_f64.onnx ONNX
primitives/jnp/jnp_argsort.onnx ONNX
primitives/jnp/jnp_argsort_descending.onnx ONNX
primitives/jnp/jnp_argsort_descending_f64.onnx ONNX
primitives/jnp/jnp_argsort_f64.onnx ONNX
primitives/jnp/jnp_argwhere_bool_size4_fill_neg1.onnx ONNX
primitives/jnp/jnp_argwhere_bool_size4_fill_neg1_f64.onnx ONNX
primitives/jnp/jnp_argwhere_int_size3_default_fill.onnx ONNX
primitives/jnp/jnp_argwhere_int_size3_default_fill_f64.onnx ONNX
primitives/jnp/jnp_around.onnx ONNX
primitives/jnp/jnp_around_f64.onnx ONNX
primitives/jnp/jnp_asin_basic.onnx ONNX
primitives/jnp/jnp_asin_basic_f64.onnx ONNX
primitives/jnp/jnp_asin_int_promote.onnx ONNX
primitives/jnp/jnp_asin_int_promote_f64.onnx ONNX
primitives/jnp/jnp_asinh_basic.onnx ONNX
primitives/jnp/jnp_asinh_basic_f64.onnx ONNX
primitives/jnp/jnp_asinh_int_promote.onnx ONNX
primitives/jnp/jnp_asinh_int_promote_f64.onnx ONNX
primitives/jnp/jnp_atan2_broadcast.onnx ONNX
primitives/jnp/jnp_atan2_quadrants.onnx ONNX
primitives/jnp/jnp_atan_basic.onnx ONNX
primitives/jnp/jnp_atan_basic_f64.onnx ONNX
primitives/jnp/jnp_atan_int_promote.onnx ONNX
primitives/jnp/jnp_atan_int_promote_f64.onnx ONNX
primitives/jnp/jnp_atanh_basic.onnx ONNX
primitives/jnp/jnp_atanh_basic_f64.onnx ONNX
primitives/jnp/jnp_atanh_int_promote.onnx ONNX
primitives/jnp/jnp_atanh_int_promote_f64.onnx ONNX
primitives/jnp/jnp_average_axis_keepdims.onnx ONNX
primitives/jnp/jnp_average_axis_keepdims_f64.onnx ONNX
primitives/jnp/jnp_average_basic.onnx ONNX
primitives/jnp/jnp_average_basic_f64.onnx ONNX
primitives/jnp/jnp_bartlett_basic.onnx ONNX
primitives/jnp/jnp_bartlett_basic_f64.onnx ONNX
primitives/jnp/jnp_bartlett_empty.onnx ONNX
primitives/jnp/jnp_bartlett_empty_f64.onnx ONNX
primitives/jnp/jnp_bincount_length.onnx ONNX
primitives/jnp/jnp_bincount_length12.onnx ONNX
primitives/jnp/jnp_bincount_length12_f64.onnx ONNX
primitives/jnp/jnp_bincount_length_f64.onnx ONNX
primitives/jnp/jnp_bincount_neg_and_large.onnx ONNX
primitives/jnp/jnp_bincount_neg_and_large_f64.onnx ONNX
primitives/jnp/jnp_bitwise_and_bool.onnx ONNX
primitives/jnp/jnp_bitwise_and_bool_f64.onnx ONNX
primitives/jnp/jnp_bitwise_and_broadcast.onnx ONNX
primitives/jnp/jnp_bitwise_and_broadcast_f64.onnx ONNX
primitives/jnp/jnp_bitwise_and_i32.onnx ONNX
primitives/jnp/jnp_bitwise_and_i32_f64.onnx ONNX
primitives/jnp/jnp_bitwise_and_int.onnx ONNX
primitives/jnp/jnp_bitwise_and_int_f64.onnx ONNX
primitives/jnp/jnp_bitwise_and_mixed_dtype.onnx ONNX
primitives/jnp/jnp_bitwise_and_mixed_dtype_f64.onnx ONNX
primitives/jnp/jnp_bitwise_count_bool.onnx ONNX
primitives/jnp/jnp_bitwise_count_bool_f64.onnx ONNX
primitives/jnp/jnp_bitwise_count_i32.onnx ONNX
primitives/jnp/jnp_bitwise_count_i32_f64.onnx ONNX
primitives/jnp/jnp_bitwise_left_shift.onnx ONNX
primitives/jnp/jnp_bitwise_left_shift_basic.onnx ONNX
primitives/jnp/jnp_bitwise_left_shift_basic_f64.onnx ONNX
primitives/jnp/jnp_bitwise_left_shift_bool_bool.onnx ONNX
primitives/jnp/jnp_bitwise_left_shift_bool_bool_f64.onnx ONNX
primitives/jnp/jnp_bitwise_left_shift_f64.onnx ONNX
primitives/jnp/jnp_bitwise_not_bool.onnx ONNX
primitives/jnp/jnp_bitwise_not_bool_f64.onnx ONNX
primitives/jnp/jnp_bitwise_not_i32.onnx ONNX
primitives/jnp/jnp_bitwise_not_i32_f64.onnx ONNX
primitives/jnp/jnp_bitwise_not_int.onnx ONNX
primitives/jnp/jnp_bitwise_not_int_f64.onnx ONNX
primitives/jnp/jnp_bitwise_or_bool.onnx ONNX
primitives/jnp/jnp_bitwise_or_bool_f64.onnx ONNX
primitives/jnp/jnp_bitwise_or_broadcast.onnx ONNX
primitives/jnp/jnp_bitwise_or_broadcast_f64.onnx ONNX
primitives/jnp/jnp_bitwise_or_i32.onnx ONNX
primitives/jnp/jnp_bitwise_or_i32_f64.onnx ONNX
primitives/jnp/jnp_bitwise_or_int.onnx ONNX
primitives/jnp/jnp_bitwise_or_int_f64.onnx ONNX
primitives/jnp/jnp_bitwise_or_mixed_dtype.onnx ONNX
primitives/jnp/jnp_bitwise_or_mixed_dtype_f64.onnx ONNX
primitives/jnp/jnp_bitwise_right_shift_bool_bool.onnx ONNX
primitives/jnp/jnp_bitwise_right_shift_bool_bool_f64.onnx ONNX
primitives/jnp/jnp_bitwise_right_shift_i32.onnx ONNX
primitives/jnp/jnp_bitwise_right_shift_i32_f64.onnx ONNX
primitives/jnp/jnp_bitwise_right_shift_signed.onnx ONNX
primitives/jnp/jnp_bitwise_right_shift_signed_f64.onnx ONNX
primitives/jnp/jnp_bitwise_xor_bool.onnx ONNX
primitives/jnp/jnp_bitwise_xor_bool_f64.onnx ONNX
primitives/jnp/jnp_bitwise_xor_broadcast.onnx ONNX
primitives/jnp/jnp_bitwise_xor_broadcast_f64.onnx ONNX
primitives/jnp/jnp_bitwise_xor_i32.onnx ONNX
primitives/jnp/jnp_bitwise_xor_i32_f64.onnx ONNX
primitives/jnp/jnp_bitwise_xor_int.onnx ONNX
primitives/jnp/jnp_bitwise_xor_int_f64.onnx ONNX
primitives/jnp/jnp_bitwise_xor_mixed_dtype.onnx ONNX
primitives/jnp/jnp_bitwise_xor_mixed_dtype_f64.onnx ONNX
primitives/jnp/jnp_blackman_basic.onnx ONNX
primitives/jnp/jnp_block_basic.onnx ONNX
primitives/jnp/jnp_can_cast_safe_false.onnx ONNX
primitives/jnp/jnp_can_cast_safe_false_f64.onnx ONNX
primitives/jnp/jnp_can_cast_safe_true.onnx ONNX
primitives/jnp/jnp_can_cast_safe_true_f64.onnx ONNX
primitives/jnp/jnp_can_cast_unsafe_true.onnx ONNX
primitives/jnp/jnp_can_cast_unsafe_true_f64.onnx ONNX
primitives/jnp/jnp_cbrt.onnx ONNX
primitives/jnp/jnp_cbrt_f64.onnx ONNX
primitives/jnp/jnp_ceil.onnx ONNX
primitives/jnp/jnp_ceil_basic.onnx ONNX
primitives/jnp/jnp_ceil_basic_f64.onnx ONNX
primitives/jnp/jnp_ceil_f64.onnx ONNX
primitives/jnp/jnp_ceil_int_identity.onnx ONNX
primitives/jnp/jnp_ceil_int_identity_f64.onnx ONNX
primitives/jnp/jnp_choose_wrap.onnx ONNX
primitives/jnp/jnp_compress_axis0.onnx ONNX
primitives/jnp/jnp_compress_axis0_f64.onnx ONNX
primitives/jnp/jnp_compress_axis1.onnx ONNX
primitives/jnp/jnp_compress_axis1_f64.onnx ONNX
primitives/jnp/jnp_conj_complex64.onnx ONNX
primitives/jnp/jnp_conj_complex64_f64.onnx ONNX
primitives/jnp/jnp_conj_real.onnx ONNX
primitives/jnp/jnp_conj_real_f64.onnx ONNX
primitives/jnp/jnp_copy.onnx ONNX
primitives/jnp/jnp_copy_f64.onnx ONNX
primitives/jnp/jnp_copysign.onnx ONNX
primitives/jnp/jnp_copysign_basic.onnx ONNX
primitives/jnp/jnp_copysign_basic_f64.onnx ONNX
primitives/jnp/jnp_copysign_broadcast.onnx ONNX
primitives/jnp/jnp_copysign_broadcast_f64.onnx ONNX
primitives/jnp/jnp_copysign_f64.onnx ONNX
primitives/jnp/jnp_copysign_int_promote.onnx ONNX
primitives/jnp/jnp_copysign_int_promote_f64.onnx ONNX
primitives/jnp/jnp_corrcoef_basic.onnx ONNX
primitives/jnp/jnp_cos.onnx ONNX
primitives/jnp/jnp_cos_basic.onnx ONNX
primitives/jnp/jnp_cos_basic_f64.onnx ONNX
primitives/jnp/jnp_cos_f64.onnx ONNX
primitives/jnp/jnp_cos_f64_via_sin.onnx ONNX
primitives/jnp/jnp_cos_int_promote.onnx ONNX
primitives/jnp/jnp_cos_int_promote_f64.onnx ONNX
primitives/jnp/jnp_cosh.onnx ONNX
primitives/jnp/jnp_cosh_basic.onnx ONNX
primitives/jnp/jnp_cosh_basic_f64.onnx ONNX
primitives/jnp/jnp_cosh_f64.onnx ONNX
primitives/jnp/jnp_cosh_int_promote.onnx ONNX
primitives/jnp/jnp_cosh_int_promote_f64.onnx ONNX
primitives/jnp/jnp_count_nonzero_axis_keepdims.onnx ONNX
primitives/jnp/jnp_count_nonzero_axis_keepdims_f64.onnx ONNX
primitives/jnp/jnp_count_nonzero_basic.onnx ONNX
primitives/jnp/jnp_count_nonzero_basic_f64.onnx ONNX
primitives/jnp/jnp_cov_basic.onnx ONNX
primitives/jnp/jnp_cov_basic_f64.onnx ONNX
primitives/jnp/jnp_cross_vectors.onnx ONNX
primitives/jnp/jnp_cross_vectors_f64.onnx ONNX
primitives/jnp/jnp_cumprod_axis1.onnx ONNX
primitives/jnp/jnp_cumprod_axis1_f64.onnx ONNX
primitives/jnp/jnp_cumprod_axis_none_flatten.onnx ONNX
primitives/jnp/jnp_cumprod_axis_none_flatten_f64.onnx ONNX
primitives/jnp/jnp_cumprod_dtype_cast.onnx ONNX
primitives/jnp/jnp_cumprod_dtype_cast_f64.onnx ONNX
primitives/jnp/jnp_cumsum_axis1.onnx ONNX
primitives/jnp/jnp_cumsum_axis1_f64.onnx ONNX
primitives/jnp/jnp_cumsum_reverse_dtype.onnx ONNX
primitives/jnp/jnp_deg2rad.onnx ONNX
primitives/jnp/jnp_deg2rad_f64.onnx ONNX
primitives/jnp/jnp_degrees.onnx ONNX
primitives/jnp/jnp_degrees_f64.onnx ONNX
primitives/jnp/jnp_delete_axis0_index1.onnx ONNX
primitives/jnp/jnp_delete_axis0_index1_f64.onnx ONNX
primitives/jnp/jnp_diag_from_matrix.onnx ONNX
primitives/jnp/jnp_diag_from_matrix_f64.onnx ONNX
primitives/jnp/jnp_diag_from_matrix_k1.onnx ONNX
primitives/jnp/jnp_diag_from_matrix_k1_f64.onnx ONNX
primitives/jnp/jnp_diag_from_vector.onnx ONNX
primitives/jnp/jnp_diag_from_vector_f64.onnx ONNX
primitives/jnp/jnp_diag_vector.onnx ONNX
primitives/jnp/jnp_diag_vector_f64.onnx ONNX
primitives/jnp/jnp_diagonal_2d_basic.onnx ONNX
primitives/jnp/jnp_diagonal_2d_basic_f64.onnx ONNX
primitives/jnp/jnp_diagonal_axes_swapped.onnx ONNX
primitives/jnp/jnp_diagonal_axes_swapped_f64.onnx ONNX
primitives/jnp/jnp_diagonal_basic.onnx ONNX
primitives/jnp/jnp_diagonal_basic_f64.onnx ONNX
primitives/jnp/jnp_diagonal_offset_neg1.onnx ONNX
primitives/jnp/jnp_diagonal_offset_neg1_f64.onnx ONNX
primitives/jnp/jnp_diagonal_vmap_batching.onnx ONNX
primitives/jnp/jnp_diagonal_vmap_batching_f64.onnx ONNX
primitives/jnp/jnp_diff_1d_default.onnx ONNX
primitives/jnp/jnp_diff_1d_default_f64.onnx ONNX
primitives/jnp/jnp_diff_2d_axis0_n2.onnx ONNX
primitives/jnp/jnp_diff_2d_axis0_n2_f64.onnx ONNX
primitives/jnp/jnp_diff_n0_identity.onnx ONNX
primitives/jnp/jnp_diff_n0_identity_f64.onnx ONNX
primitives/jnp/jnp_digitize_basic.onnx ONNX
primitives/jnp/jnp_digitize_basic_f64.onnx ONNX
primitives/jnp/jnp_digitize_matrix.onnx ONNX
primitives/jnp/jnp_digitize_matrix_f64.onnx ONNX
primitives/jnp/jnp_digitize_right_true.onnx ONNX
primitives/jnp/jnp_digitize_right_true_f64.onnx ONNX
primitives/jnp/jnp_divide.onnx ONNX
primitives/jnp/jnp_divide_basic.onnx ONNX
primitives/jnp/jnp_divide_basic_f64.onnx ONNX
primitives/jnp/jnp_divide_broadcast.onnx ONNX
primitives/jnp/jnp_divide_broadcast_f64.onnx ONNX
primitives/jnp/jnp_divide_f64.onnx ONNX
primitives/jnp/jnp_divide_int_promote.onnx ONNX
primitives/jnp/jnp_divide_int_promote_f64.onnx ONNX
primitives/jnp/jnp_divmod_broadcast.onnx ONNX
primitives/jnp/jnp_divmod_broadcast_f64.onnx ONNX
primitives/jnp/jnp_divmod_float.onnx ONNX
primitives/jnp/jnp_divmod_float_f64.onnx ONNX
primitives/jnp/jnp_divmod_int.onnx ONNX
primitives/jnp/jnp_divmod_int_f64.onnx ONNX
primitives/jnp/jnp_dot_matrix_matrix.onnx ONNX
primitives/jnp/jnp_dot_matrix_matrix_f64.onnx ONNX
primitives/jnp/jnp_dot_matrix_vector.onnx ONNX
primitives/jnp/jnp_dot_matrix_vector_f64.onnx ONNX
primitives/jnp/jnp_dot_vector_vector.onnx ONNX
primitives/jnp/jnp_dot_vector_vector_f64.onnx ONNX
primitives/jnp/jnp_empty_2x3.onnx ONNX
primitives/jnp/jnp_empty_2x3_f64.onnx ONNX
primitives/jnp/jnp_empty_int32.onnx ONNX
primitives/jnp/jnp_empty_int32_f64.onnx ONNX
primitives/jnp/jnp_empty_like_basic.onnx ONNX
primitives/jnp/jnp_empty_like_basic_f64.onnx ONNX
primitives/jnp/jnp_empty_like_dtype_override.onnx ONNX
primitives/jnp/jnp_empty_like_dtype_override_f64.onnx ONNX
primitives/jnp/jnp_empty_like_shape_override.onnx ONNX
primitives/jnp/jnp_empty_like_shape_override_f64.onnx ONNX
primitives/jnp/jnp_equal.onnx ONNX
primitives/jnp/jnp_equal_basic.onnx ONNX
primitives/jnp/jnp_equal_basic_f64.onnx ONNX
primitives/jnp/jnp_equal_broadcast.onnx ONNX
primitives/jnp/jnp_equal_broadcast_f64.onnx ONNX
primitives/jnp/jnp_equal_f64.onnx ONNX
primitives/jnp/jnp_equal_mixed_dtype.onnx ONNX
primitives/jnp/jnp_equal_mixed_dtype_f64.onnx ONNX
primitives/jnp/jnp_exp.onnx ONNX
primitives/jnp/jnp_exp2.onnx ONNX
primitives/jnp/jnp_exp2_basic.onnx ONNX
primitives/jnp/jnp_exp2_basic_f64.onnx ONNX
primitives/jnp/jnp_exp2_f64.onnx ONNX
primitives/jnp/jnp_exp2_int_promote.onnx ONNX
primitives/jnp/jnp_exp2_int_promote_f64.onnx ONNX
primitives/jnp/jnp_exp_basic.onnx ONNX
primitives/jnp/jnp_exp_basic_f64.onnx ONNX
primitives/jnp/jnp_exp_f64.onnx ONNX
primitives/jnp/jnp_exp_int_promote.onnx ONNX
primitives/jnp/jnp_exp_int_promote_f64.onnx ONNX
primitives/jnp/jnp_expm1.onnx ONNX
primitives/jnp/jnp_expm1_basic.onnx ONNX
primitives/jnp/jnp_expm1_basic_f64.onnx ONNX
primitives/jnp/jnp_expm1_f64.onnx ONNX
primitives/jnp/jnp_expm1_int_promote.onnx ONNX
primitives/jnp/jnp_expm1_int_promote_f64.onnx ONNX
primitives/jnp/jnp_extract_basic.onnx ONNX
primitives/jnp/jnp_extract_basic_f64.onnx ONNX
primitives/jnp/jnp_extract_flat_condition.onnx ONNX
primitives/jnp/jnp_extract_flat_condition_f64.onnx ONNX
primitives/jnp/jnp_extract_int_condition.onnx ONNX
primitives/jnp/jnp_extract_int_condition_f64.onnx ONNX
primitives/jnp/jnp_eye_rect_k1.onnx ONNX
primitives/jnp/jnp_eye_square.onnx ONNX
primitives/jnp/jnp_fabs.onnx ONNX
primitives/jnp/jnp_fabs_basic.onnx ONNX
primitives/jnp/jnp_fabs_basic_f64.onnx ONNX
primitives/jnp/jnp_fabs_f64.onnx ONNX
primitives/jnp/jnp_fabs_int_promote.onnx ONNX
primitives/jnp/jnp_fabs_int_promote_f64.onnx ONNX
primitives/jnp/jnp_fft_complex128.onnx ONNX
primitives/jnp/jnp_fft_complex64.onnx ONNX
primitives/jnp/jnp_fft_fftfreq_n8.onnx ONNX
primitives/jnp/jnp_fft_fftfreq_n8_f64.onnx ONNX
primitives/jnp/jnp_fft_fftshift_1d.onnx ONNX
primitives/jnp/jnp_fft_fftshift_1d_f64.onnx ONNX
primitives/jnp/jnp_fft_ifftshift_1d.onnx ONNX
primitives/jnp/jnp_fft_ifftshift_1d_f64.onnx ONNX
primitives/jnp/jnp_fft_rfftfreq_n8.onnx ONNX
primitives/jnp/jnp_fft_rfftfreq_n8_f64.onnx ONNX
primitives/jnp/jnp_fill_diagonal_basic.onnx ONNX
primitives/jnp/jnp_fill_diagonal_basic_f64.onnx ONNX
primitives/jnp/jnp_fix.onnx ONNX
primitives/jnp/jnp_fix_f64.onnx ONNX
primitives/jnp/jnp_flip_axes_tuple.onnx ONNX
primitives/jnp/jnp_flip_axes_tuple_f64.onnx ONNX
primitives/jnp/jnp_flip_axis_1.onnx ONNX
primitives/jnp/jnp_flip_axis_1_f64.onnx ONNX
primitives/jnp/jnp_flip_axis_none_2d.onnx ONNX
primitives/jnp/jnp_flip_axis_none_2d_f64.onnx ONNX
primitives/jnp/jnp_flip_grad_issue_batch_diff_rules.onnx ONNX
primitives/jnp/jnp_flip_negative_axis.onnx ONNX
primitives/jnp/jnp_flip_negative_axis_f64.onnx ONNX
primitives/jnp/jnp_flip_vmap_batching.onnx ONNX
primitives/jnp/jnp_flip_vmap_batching_f64.onnx ONNX
primitives/jnp/jnp_float_power_int.onnx ONNX
primitives/jnp/jnp_float_power_int_f64.onnx ONNX
primitives/jnp/jnp_floor.onnx ONNX
primitives/jnp/jnp_floor_basic.onnx ONNX
primitives/jnp/jnp_floor_basic_f64.onnx ONNX
primitives/jnp/jnp_floor_divide.onnx ONNX
primitives/jnp/jnp_floor_divide_broadcast.onnx ONNX
primitives/jnp/jnp_floor_divide_broadcast_f64.onnx ONNX
primitives/jnp/jnp_floor_divide_f64.onnx ONNX
primitives/jnp/jnp_floor_divide_float.onnx ONNX
primitives/jnp/jnp_floor_divide_float_f64.onnx ONNX
primitives/jnp/jnp_floor_divide_int_neg.onnx ONNX
primitives/jnp/jnp_floor_divide_int_neg_f64.onnx ONNX
primitives/jnp/jnp_floor_f64.onnx ONNX
primitives/jnp/jnp_floor_int_identity.onnx ONNX
primitives/jnp/jnp_floor_int_identity_f64.onnx ONNX
primitives/jnp/jnp_fmax.onnx ONNX
primitives/jnp/jnp_fmax_f64.onnx ONNX
primitives/jnp/jnp_fmax_nan_semantics.onnx ONNX
primitives/jnp/jnp_fmin.onnx ONNX
primitives/jnp/jnp_fmin_f64.onnx ONNX
primitives/jnp/jnp_fmin_nan_semantics.onnx ONNX
primitives/jnp/jnp_fmod_broadcast.onnx ONNX
primitives/jnp/jnp_fmod_broadcast_f64.onnx ONNX
primitives/jnp/jnp_fmod_float.onnx ONNX
primitives/jnp/jnp_fmod_float_f64.onnx ONNX
primitives/jnp/jnp_fmod_int.onnx ONNX
primitives/jnp/jnp_fmod_int_f64.onnx ONNX
primitives/jnp/jnp_frexp_basic.onnx ONNX
primitives/jnp/jnp_frexp_basic_f64.onnx ONNX
primitives/jnp/jnp_full_const_scalar.onnx ONNX
primitives/jnp/jnp_full_const_scalar_f64.onnx ONNX
primitives/jnp/jnp_full_expand_row_broadcast.onnx ONNX
primitives/jnp/jnp_full_expand_row_broadcast_f64.onnx ONNX
primitives/jnp/jnp_full_expand_scalar_input.onnx ONNX
primitives/jnp/jnp_full_expand_scalar_input_f64.onnx ONNX
primitives/jnp/jnp_full_like_const.onnx ONNX
primitives/jnp/jnp_full_like_const_f64.onnx ONNX
primitives/jnp/jnp_full_like_dtype_override.onnx ONNX
primitives/jnp/jnp_full_like_dtype_override_f64.onnx ONNX
primitives/jnp/jnp_full_like_fill_input.onnx ONNX
primitives/jnp/jnp_full_like_fill_input_f64.onnx ONNX
primitives/jnp/jnp_full_vmap_batching.onnx ONNX
primitives/jnp/jnp_full_vmap_batching_f64.onnx ONNX
primitives/jnp/jnp_gcd_basic.onnx ONNX
primitives/jnp/jnp_gcd_basic_f64.onnx ONNX
primitives/jnp/jnp_gcd_int32.onnx ONNX
primitives/jnp/jnp_geomspace_scalar.onnx ONNX
primitives/jnp/jnp_geomspace_scalar_f64.onnx ONNX
primitives/jnp/jnp_gradient_1d.onnx ONNX
primitives/jnp/jnp_gradient_1d_f64.onnx ONNX
primitives/jnp/jnp_greater.onnx ONNX
primitives/jnp/jnp_greater_basic.onnx ONNX
primitives/jnp/jnp_greater_basic_f64.onnx ONNX
primitives/jnp/jnp_greater_broadcast.onnx ONNX
primitives/jnp/jnp_greater_broadcast_f64.onnx ONNX
primitives/jnp/jnp_greater_equal.onnx ONNX
primitives/jnp/jnp_greater_equal_basic.onnx ONNX
primitives/jnp/jnp_greater_equal_basic_f64.onnx ONNX
primitives/jnp/jnp_greater_equal_broadcast.onnx ONNX
primitives/jnp/jnp_greater_equal_broadcast_f64.onnx ONNX
primitives/jnp/jnp_greater_equal_f64.onnx ONNX
primitives/jnp/jnp_greater_equal_mixed_dtype.onnx ONNX
primitives/jnp/jnp_greater_equal_mixed_dtype_f64.onnx ONNX
primitives/jnp/jnp_greater_f64.onnx ONNX
primitives/jnp/jnp_greater_mixed_dtype.onnx ONNX
primitives/jnp/jnp_greater_mixed_dtype_f64.onnx ONNX
primitives/jnp/jnp_hamming_basic.onnx ONNX
primitives/jnp/jnp_hanning_basic.onnx ONNX
primitives/jnp/jnp_heaviside.onnx ONNX
primitives/jnp/jnp_heaviside_f64.onnx ONNX
primitives/jnp/jnp_histogram_bin_edges_basic.onnx ONNX
primitives/jnp/jnp_histogram_bin_edges_basic_f64.onnx ONNX
primitives/jnp/jnp_hypot.onnx ONNX
primitives/jnp/jnp_hypot_broadcast.onnx ONNX
primitives/jnp/jnp_hypot_f64.onnx ONNX
primitives/jnp/jnp_i0_basic.onnx ONNX
primitives/jnp/jnp_i0_basic_f64.onnx ONNX
primitives/jnp/jnp_identity_4.onnx ONNX
primitives/jnp/jnp_identity_bool.onnx ONNX
primitives/jnp/jnp_identity_int32.onnx ONNX
primitives/jnp/jnp_ifft_complex64.onnx ONNX
primitives/jnp/jnp_imag_complex64.onnx ONNX
primitives/jnp/jnp_inner_matrix_matrix.onnx ONNX
primitives/jnp/jnp_inner_matrix_matrix_f64.onnx ONNX
primitives/jnp/jnp_inner_matrix_vector.onnx ONNX
primitives/jnp/jnp_inner_matrix_vector_f64.onnx ONNX
primitives/jnp/jnp_inner_vector_matrix.onnx ONNX
primitives/jnp/jnp_inner_vector_matrix_f64.onnx ONNX
primitives/jnp/jnp_inner_vector_vector.onnx ONNX
primitives/jnp/jnp_inner_vector_vector_f64.onnx ONNX
primitives/jnp/jnp_invert_bool.onnx ONNX
primitives/jnp/jnp_invert_bool_f64.onnx ONNX
primitives/jnp/jnp_invert_i32.onnx ONNX
primitives/jnp/jnp_invert_i32_f64.onnx ONNX
primitives/jnp/jnp_invert_int.onnx ONNX
primitives/jnp/jnp_invert_int_f64.onnx ONNX
primitives/jnp/jnp_irfft_complex64.onnx ONNX
primitives/jnp/jnp_iscomplex_complex64.onnx ONNX
primitives/jnp/jnp_iscomplexobj_false.onnx ONNX
primitives/jnp/jnp_iscomplexobj_false_f64.onnx ONNX
primitives/jnp/jnp_iscomplexobj_true.onnx ONNX
primitives/jnp/jnp_iscomplexobj_true_f64.onnx ONNX
primitives/jnp/jnp_isdtype_complex_false.onnx ONNX
primitives/jnp/jnp_isdtype_complex_false_f64.onnx ONNX
primitives/jnp/jnp_isdtype_real_floating_true.onnx ONNX
primitives/jnp/jnp_isdtype_real_floating_true_f64.onnx ONNX
primitives/jnp/jnp_isdtype_tuple_true.onnx ONNX
primitives/jnp/jnp_isdtype_tuple_true_f64.onnx ONNX
primitives/jnp/jnp_isfinite.onnx ONNX
primitives/jnp/jnp_isfinite_basic.onnx ONNX
primitives/jnp/jnp_isfinite_basic_f64.onnx ONNX
primitives/jnp/jnp_isfinite_f64.onnx ONNX
primitives/jnp/jnp_isin_basic.onnx ONNX
primitives/jnp/jnp_isin_basic_f64.onnx ONNX
primitives/jnp/jnp_isin_dynamic_test_elements.onnx ONNX
primitives/jnp/jnp_isin_dynamic_test_elements_f64.onnx ONNX
primitives/jnp/jnp_isin_invert.onnx ONNX
primitives/jnp/jnp_isin_invert_f64.onnx ONNX
primitives/jnp/jnp_isinf.onnx ONNX
primitives/jnp/jnp_isinf_f64.onnx ONNX
primitives/jnp/jnp_isnan.onnx ONNX
primitives/jnp/jnp_isnan_f64.onnx ONNX
primitives/jnp/jnp_isreal_complex64.onnx ONNX
primitives/jnp/jnp_isrealobj_false.onnx ONNX
primitives/jnp/jnp_isrealobj_false_f64.onnx ONNX
primitives/jnp/jnp_isrealobj_true.onnx ONNX
primitives/jnp/jnp_isrealobj_true_f64.onnx ONNX
primitives/jnp/jnp_isscalar_false_vector.onnx ONNX
primitives/jnp/jnp_isscalar_false_vector_f64.onnx ONNX
primitives/jnp/jnp_isscalar_true_numpy_scalar.onnx ONNX
primitives/jnp/jnp_isscalar_true_numpy_scalar_f64.onnx ONNX
primitives/jnp/jnp_isscalar_true_python.onnx ONNX
primitives/jnp/jnp_isscalar_true_python_f64.onnx ONNX
primitives/jnp/jnp_iterable_false_numpy_scalar.onnx ONNX
primitives/jnp/jnp_iterable_false_numpy_scalar_f64.onnx ONNX
primitives/jnp/jnp_iterable_false_scalar.onnx ONNX
primitives/jnp/jnp_iterable_false_scalar_f64.onnx ONNX
primitives/jnp/jnp_iterable_true_vector.onnx ONNX
primitives/jnp/jnp_iterable_true_vector_f64.onnx ONNX
primitives/jnp/jnp_kron_1d.onnx ONNX
primitives/jnp/jnp_kron_1d_f64.onnx ONNX
primitives/jnp/jnp_kron_2d.onnx ONNX
primitives/jnp/jnp_kron_2d_f64.onnx ONNX
primitives/jnp/jnp_lcm_basic.onnx ONNX
primitives/jnp/jnp_lcm_basic_f64.onnx ONNX
primitives/jnp/jnp_ldexp_broadcast.onnx ONNX
primitives/jnp/jnp_ldexp_broadcast_f64.onnx ONNX
primitives/jnp/jnp_ldexp_float_int.onnx ONNX
primitives/jnp/jnp_ldexp_float_int_f64.onnx ONNX
primitives/jnp/jnp_ldexp_int_int.onnx ONNX
primitives/jnp/jnp_ldexp_int_int_f64.onnx ONNX
primitives/jnp/jnp_left_shift.onnx ONNX
primitives/jnp/jnp_left_shift_broadcast.onnx ONNX
primitives/jnp/jnp_left_shift_broadcast_f64.onnx ONNX
primitives/jnp/jnp_left_shift_f64.onnx ONNX
primitives/jnp/jnp_left_shift_int.onnx ONNX
primitives/jnp/jnp_left_shift_int_f64.onnx ONNX
primitives/jnp/jnp_left_shift_mixed_bool_rhs.onnx ONNX
primitives/jnp/jnp_left_shift_mixed_bool_rhs_f64.onnx ONNX
primitives/jnp/jnp_left_shift_unsigned.onnx ONNX
primitives/jnp/jnp_left_shift_unsigned_f64.onnx ONNX
primitives/jnp/jnp_less.onnx ONNX
primitives/jnp/jnp_less_basic.onnx ONNX
primitives/jnp/jnp_less_basic_f64.onnx ONNX
primitives/jnp/jnp_less_broadcast.onnx ONNX
primitives/jnp/jnp_less_broadcast_f64.onnx ONNX
primitives/jnp/jnp_less_equal.onnx ONNX
primitives/jnp/jnp_less_equal_basic.onnx ONNX
primitives/jnp/jnp_less_equal_basic_f64.onnx ONNX
primitives/jnp/jnp_less_equal_broadcast.onnx ONNX
primitives/jnp/jnp_less_equal_broadcast_f64.onnx ONNX
primitives/jnp/jnp_less_equal_f64.onnx ONNX
primitives/jnp/jnp_less_equal_mixed_dtype.onnx ONNX
primitives/jnp/jnp_less_equal_mixed_dtype_f64.onnx ONNX
primitives/jnp/jnp_less_f64.onnx ONNX
primitives/jnp/jnp_less_mixed_dtype.onnx ONNX
primitives/jnp/jnp_less_mixed_dtype_f64.onnx ONNX
primitives/jnp/jnp_linalg_cholesky_basic.onnx ONNX
primitives/jnp/jnp_linalg_cond_basic.onnx ONNX
primitives/jnp/jnp_linalg_eig_basic.onnx ONNX
primitives/jnp/jnp_linalg_eigh_basic.onnx ONNX
primitives/jnp/jnp_linalg_eigvals_basic.onnx ONNX
primitives/jnp/jnp_linalg_eigvalsh_basic.onnx ONNX
primitives/jnp/jnp_linalg_lstsq_basic.onnx ONNX
primitives/jnp/jnp_linalg_matrix_norm_basic.onnx ONNX
primitives/jnp/jnp_linalg_matrix_power_basic.onnx ONNX
primitives/jnp/jnp_linalg_matrix_rank_basic.onnx ONNX
primitives/jnp/jnp_linalg_multi_dot_basic.onnx ONNX
primitives/jnp/jnp_linalg_pinv_basic.onnx ONNX
primitives/jnp/jnp_linalg_qr_basic.onnx ONNX
primitives/jnp/jnp_linalg_svd_basic.onnx ONNX
primitives/jnp/jnp_linalg_svdvals_basic.onnx ONNX
primitives/jnp/jnp_linalg_tensordot_basic.onnx ONNX
primitives/jnp/jnp_linalg_trace_basic.onnx ONNX
primitives/jnp/jnp_linalg_vecdot_basic.onnx ONNX
primitives/jnp/jnp_linalg_vector_norm_basic.onnx ONNX
primitives/jnp/jnp_log.onnx ONNX
primitives/jnp/jnp_log10.onnx ONNX
primitives/jnp/jnp_log10_f64.onnx ONNX
primitives/jnp/jnp_log1p.onnx ONNX
primitives/jnp/jnp_log1p_f64.onnx ONNX
primitives/jnp/jnp_log2.onnx ONNX
primitives/jnp/jnp_log2_f64.onnx ONNX
primitives/jnp/jnp_log_basic.onnx ONNX
primitives/jnp/jnp_log_basic_f64.onnx ONNX
primitives/jnp/jnp_log_f64.onnx ONNX
primitives/jnp/jnp_log_int_promote.onnx ONNX
primitives/jnp/jnp_log_int_promote_f64.onnx ONNX
primitives/jnp/jnp_logaddexp.onnx ONNX
primitives/jnp/jnp_logaddexp2.onnx ONNX
primitives/jnp/jnp_logaddexp2_f64.onnx ONNX
primitives/jnp/jnp_logaddexp_f64.onnx ONNX
primitives/jnp/jnp_logical_and_bool.onnx ONNX
primitives/jnp/jnp_logical_and_bool_f64.onnx ONNX
primitives/jnp/jnp_logical_and_numeric.onnx ONNX
primitives/jnp/jnp_logical_and_numeric_f64.onnx ONNX
primitives/jnp/jnp_logical_not_bool.onnx ONNX
primitives/jnp/jnp_logical_not_bool_f64.onnx ONNX
primitives/jnp/jnp_logical_not_numeric.onnx ONNX
primitives/jnp/jnp_logical_not_numeric_f64.onnx ONNX
primitives/jnp/jnp_logical_or_bool.onnx ONNX
primitives/jnp/jnp_logical_or_bool_f64.onnx ONNX
primitives/jnp/jnp_logical_or_numeric.onnx ONNX
primitives/jnp/jnp_logical_or_numeric_f64.onnx ONNX
primitives/jnp/jnp_logical_xor_bool.onnx ONNX
primitives/jnp/jnp_logical_xor_bool_f64.onnx ONNX
primitives/jnp/jnp_logical_xor_numeric.onnx ONNX
primitives/jnp/jnp_logical_xor_numeric_f64.onnx ONNX
primitives/jnp/jnp_logspace_basic.onnx ONNX
primitives/jnp/jnp_matrix_transpose_2d.onnx ONNX
primitives/jnp/jnp_matrix_transpose_2d_f64.onnx ONNX
primitives/jnp/jnp_matrix_transpose_3d.onnx ONNX
primitives/jnp/jnp_matrix_transpose_3d_f64.onnx ONNX
primitives/jnp/jnp_matrix_transpose_vmap_batching.onnx ONNX
primitives/jnp/jnp_matrix_transpose_vmap_batching_f64.onnx ONNX
primitives/jnp/jnp_matvec_basic.onnx ONNX
primitives/jnp/jnp_max_axis.onnx ONNX
primitives/jnp/jnp_max_axis_f64.onnx ONNX
primitives/jnp/jnp_max_basic.onnx ONNX
primitives/jnp/jnp_max_basic_f64.onnx ONNX
primitives/jnp/jnp_max_keepdims.onnx ONNX
primitives/jnp/jnp_max_keepdims_f64.onnx ONNX
primitives/jnp/jnp_maximum.onnx ONNX
primitives/jnp/jnp_maximum_basic.onnx ONNX
primitives/jnp/jnp_maximum_basic_f64.onnx ONNX
primitives/jnp/jnp_maximum_broadcast.onnx ONNX
primitives/jnp/jnp_maximum_broadcast_f64.onnx ONNX
primitives/jnp/jnp_maximum_broadcast_scalar.onnx ONNX
primitives/jnp/jnp_maximum_broadcast_scalar_f64.onnx ONNX
primitives/jnp/jnp_maximum_f64.onnx ONNX
primitives/jnp/jnp_mean_axis.onnx ONNX
primitives/jnp/jnp_mean_axis_f64.onnx ONNX
primitives/jnp/jnp_mean_basic.onnx ONNX
primitives/jnp/jnp_mean_basic_f64.onnx ONNX
primitives/jnp/jnp_median_basic.onnx ONNX
primitives/jnp/jnp_min_axis.onnx ONNX
primitives/jnp/jnp_min_axis_f64.onnx ONNX
primitives/jnp/jnp_min_basic.onnx ONNX
primitives/jnp/jnp_min_basic_f64.onnx ONNX
primitives/jnp/jnp_min_keepdims.onnx ONNX
primitives/jnp/jnp_min_keepdims_f64.onnx ONNX
primitives/jnp/jnp_minimum.onnx ONNX
primitives/jnp/jnp_minimum_basic.onnx ONNX
primitives/jnp/jnp_minimum_basic_f64.onnx ONNX
primitives/jnp/jnp_minimum_broadcast.onnx ONNX
primitives/jnp/jnp_minimum_broadcast_f64.onnx ONNX
primitives/jnp/jnp_minimum_broadcast_scalar.onnx ONNX
primitives/jnp/jnp_minimum_broadcast_scalar_f64.onnx ONNX
primitives/jnp/jnp_minimum_f64.onnx ONNX
primitives/jnp/jnp_mod.onnx ONNX
primitives/jnp/jnp_mod_f64.onnx ONNX
primitives/jnp/jnp_modf_float.onnx ONNX
primitives/jnp/jnp_modf_float_f64.onnx ONNX
primitives/jnp/jnp_modf_int_input.onnx ONNX
primitives/jnp/jnp_modf_int_input_f64.onnx ONNX
primitives/jnp/jnp_modf_matrix.onnx ONNX
primitives/jnp/jnp_modf_matrix_f64.onnx ONNX
primitives/jnp/jnp_moveaxis_0_to_2.onnx ONNX
primitives/jnp/jnp_moveaxis_0_to_2_f64.onnx ONNX
primitives/jnp/jnp_moveaxis_2d.onnx ONNX
primitives/jnp/jnp_moveaxis_2d_f64.onnx ONNX
primitives/jnp/jnp_moveaxis_3d_tuple.onnx ONNX
primitives/jnp/jnp_moveaxis_3d_tuple_f64.onnx ONNX
primitives/jnp/jnp_moveaxis_negative_axis.onnx ONNX
primitives/jnp/jnp_moveaxis_negative_axis_f64.onnx ONNX
primitives/jnp/jnp_moveaxis_tuple.onnx ONNX
primitives/jnp/jnp_moveaxis_tuple_f64.onnx ONNX
primitives/jnp/jnp_moveaxis_vmap_batching.onnx ONNX
primitives/jnp/jnp_moveaxis_vmap_batching_f64.onnx ONNX
primitives/jnp/jnp_multiply.onnx ONNX
primitives/jnp/jnp_multiply_broadcast.onnx ONNX
primitives/jnp/jnp_multiply_broadcast_f64.onnx ONNX
primitives/jnp/jnp_multiply_f64.onnx ONNX
primitives/jnp/jnp_nan_to_num_custom_values.onnx ONNX
primitives/jnp/jnp_nan_to_num_custom_values_f64.onnx ONNX
primitives/jnp/jnp_nan_to_num_default.onnx ONNX
primitives/jnp/jnp_nan_to_num_default_f64.onnx ONNX
primitives/jnp/jnp_nan_to_num_int_passthrough.onnx ONNX
primitives/jnp/jnp_nan_to_num_int_passthrough_f64.onnx ONNX
primitives/jnp/jnp_nanargmax_basic.onnx ONNX
primitives/jnp/jnp_nanargmin_basic.onnx ONNX
primitives/jnp/jnp_nancumsum_basic.onnx ONNX
primitives/jnp/jnp_nanmax_basic.onnx ONNX
primitives/jnp/jnp_nanmean_basic.onnx ONNX
primitives/jnp/jnp_nanmedian_basic.onnx ONNX
primitives/jnp/jnp_nanmin_basic.onnx ONNX
primitives/jnp/jnp_nanpercentile_basic.onnx ONNX
primitives/jnp/jnp_nanprod_basic.onnx ONNX
primitives/jnp/jnp_nanquantile_basic.onnx ONNX
primitives/jnp/jnp_nanstd_basic.onnx ONNX
primitives/jnp/jnp_nansum_basic.onnx ONNX
primitives/jnp/jnp_nanvar_basic.onnx ONNX
primitives/jnp/jnp_ndarray_at_set_basic.onnx ONNX
primitives/jnp/jnp_ndim_matrix.onnx ONNX
primitives/jnp/jnp_ndim_matrix_f64.onnx ONNX
primitives/jnp/jnp_ndim_rank3.onnx ONNX
primitives/jnp/jnp_ndim_rank3_f64.onnx ONNX
primitives/jnp/jnp_ndim_scalar.onnx ONNX
primitives/jnp/jnp_ndim_scalar_f64.onnx ONNX
primitives/jnp/jnp_negative.onnx ONNX
primitives/jnp/jnp_negative_f64.onnx ONNX
primitives/jnp/jnp_nextafter.onnx ONNX
primitives/jnp/jnp_nextafter_special_values.onnx ONNX
primitives/jnp/jnp_nonzero_bool_size3.onnx ONNX
primitives/jnp/jnp_nonzero_bool_size3_f64.onnx ONNX
primitives/jnp/jnp_nonzero_size4_default_fill.onnx ONNX
primitives/jnp/jnp_nonzero_size4_default_fill_f64.onnx ONNX
primitives/jnp/jnp_nonzero_size5_fill_neg1.onnx ONNX
primitives/jnp/jnp_nonzero_size5_fill_neg1_f64.onnx ONNX
primitives/jnp/jnp_not_equal.onnx ONNX
primitives/jnp/jnp_not_equal_broadcast.onnx ONNX
primitives/jnp/jnp_not_equal_broadcast_f64.onnx ONNX
primitives/jnp/jnp_not_equal_f64.onnx ONNX
primitives/jnp/jnp_ones_2x3.onnx ONNX
primitives/jnp/jnp_ones_2x3_f64.onnx ONNX
primitives/jnp/jnp_ones_bool.onnx ONNX
primitives/jnp/jnp_ones_bool_f64.onnx ONNX
primitives/jnp/jnp_ones_like_basic.onnx ONNX
primitives/jnp/jnp_ones_like_basic_f64.onnx ONNX
primitives/jnp/jnp_ones_like_dtype_override.onnx ONNX
primitives/jnp/jnp_ones_like_dtype_override_f64.onnx ONNX
primitives/jnp/jnp_pad_const_1d.onnx ONNX
primitives/jnp/jnp_pad_const_1d_f64.onnx ONNX
primitives/jnp/jnp_pad_const_2d.onnx ONNX
primitives/jnp/jnp_pad_const_2d_f64.onnx ONNX
primitives/jnp/jnp_pad_constant_1d.onnx ONNX
primitives/jnp/jnp_pad_constant_1d_f64.onnx ONNX
primitives/jnp/jnp_pad_constant_2d.onnx ONNX
primitives/jnp/jnp_pad_constant_2d_f64.onnx ONNX
primitives/jnp/jnp_pad_constant_2d_tuple.onnx ONNX
primitives/jnp/jnp_pad_constant_2d_tuple_f64.onnx ONNX
primitives/jnp/jnp_pad_constant_vmap_batching.onnx ONNX
primitives/jnp/jnp_pad_constant_vmap_batching_f64.onnx ONNX
primitives/jnp/jnp_partition_basic.onnx ONNX
primitives/jnp/jnp_percentile_basic.onnx ONNX
primitives/jnp/jnp_permute_dims_2d.onnx ONNX
primitives/jnp/jnp_permute_dims_2d_f64.onnx ONNX
primitives/jnp/jnp_permute_dims_3d.onnx ONNX
primitives/jnp/jnp_permute_dims_3d_f64.onnx ONNX
primitives/jnp/jnp_permute_dims_vmap_batching.onnx ONNX
primitives/jnp/jnp_permute_dims_vmap_batching_f64.onnx ONNX
primitives/jnp/jnp_polyadd_basic.onnx ONNX
primitives/jnp/jnp_polyder_basic.onnx ONNX
primitives/jnp/jnp_polydiv_basic.onnx ONNX
primitives/jnp/jnp_polyint_basic.onnx ONNX
primitives/jnp/jnp_polysub_basic.onnx ONNX
primitives/jnp/jnp_positive.onnx ONNX
primitives/jnp/jnp_positive_f64.onnx ONNX
primitives/jnp/jnp_pow_vector.onnx ONNX
primitives/jnp/jnp_pow_vector_f64.onnx ONNX
primitives/jnp/jnp_power_vector.onnx ONNX
primitives/jnp/jnp_power_vector_f64.onnx ONNX
primitives/jnp/jnp_prod_axis.onnx ONNX
primitives/jnp/jnp_prod_axis_f64.onnx ONNX
primitives/jnp/jnp_prod_basic.onnx ONNX
primitives/jnp/jnp_prod_basic_f64.onnx ONNX
primitives/jnp/jnp_prod_keepdims.onnx ONNX
primitives/jnp/jnp_prod_keepdims_f64.onnx ONNX
primitives/jnp/jnp_ptp_axis.onnx ONNX
primitives/jnp/jnp_ptp_axis_f64.onnx ONNX
primitives/jnp/jnp_ptp_basic.onnx ONNX
primitives/jnp/jnp_ptp_basic_f64.onnx ONNX
primitives/jnp/jnp_ptp_keepdims.onnx ONNX
primitives/jnp/jnp_ptp_keepdims_f64.onnx ONNX
primitives/jnp/jnp_put_along_axis_basic.onnx ONNX
primitives/jnp/jnp_put_basic.onnx ONNX
primitives/jnp/jnp_quantile_basic.onnx ONNX
primitives/jnp/jnp_rad2deg.onnx ONNX
primitives/jnp/jnp_rad2deg_f64.onnx ONNX
primitives/jnp/jnp_radians.onnx ONNX
primitives/jnp/jnp_radians_f64.onnx ONNX
primitives/jnp/jnp_ravel_2d.onnx ONNX
primitives/jnp/jnp_ravel_2d_f64.onnx ONNX
primitives/jnp/jnp_ravel_3d.onnx ONNX
primitives/jnp/jnp_ravel_3d_f64.onnx ONNX
primitives/jnp/jnp_ravel_multi_index_wrap.onnx ONNX
primitives/jnp/jnp_ravel_scalar.onnx ONNX
primitives/jnp/jnp_ravel_scalar_f64.onnx ONNX
primitives/jnp/jnp_real_complex64.onnx ONNX
primitives/jnp/jnp_reciprocal.onnx ONNX
primitives/jnp/jnp_reciprocal_f64.onnx ONNX
primitives/jnp/jnp_remainder.onnx ONNX
primitives/jnp/jnp_remainder_f64.onnx ONNX
primitives/jnp/jnp_rfft_float32.onnx ONNX
primitives/jnp/jnp_right_shift_broadcast.onnx ONNX
primitives/jnp/jnp_right_shift_broadcast_f64.onnx ONNX
primitives/jnp/jnp_right_shift_i32.onnx ONNX
primitives/jnp/jnp_right_shift_i32_f64.onnx ONNX
primitives/jnp/jnp_right_shift_mixed_bool_rhs.onnx ONNX
primitives/jnp/jnp_right_shift_mixed_bool_rhs_f64.onnx ONNX
primitives/jnp/jnp_right_shift_signed_arithmetic.onnx ONNX
primitives/jnp/jnp_right_shift_signed_arithmetic_f64.onnx ONNX
primitives/jnp/jnp_right_shift_u32.onnx ONNX
primitives/jnp/jnp_right_shift_u32_f64.onnx ONNX
primitives/jnp/jnp_right_shift_unsigned_logical.onnx ONNX
primitives/jnp/jnp_right_shift_unsigned_logical_f64.onnx ONNX
primitives/jnp/jnp_rint.onnx ONNX
primitives/jnp/jnp_rint_f64.onnx ONNX
primitives/jnp/jnp_roll_basic.onnx ONNX
primitives/jnp/jnp_rot90_basic.onnx ONNX
primitives/jnp/jnp_round.onnx ONNX
primitives/jnp/jnp_round_f64.onnx ONNX
primitives/jnp/jnp_sign.onnx ONNX
primitives/jnp/jnp_sign_basic.onnx ONNX
primitives/jnp/jnp_sign_basic_f64.onnx ONNX
primitives/jnp/jnp_sign_f64.onnx ONNX
primitives/jnp/jnp_sign_int.onnx ONNX
primitives/jnp/jnp_sign_int_f64.onnx ONNX
primitives/jnp/jnp_signbit_float.onnx ONNX
primitives/jnp/jnp_signbit_float_f64.onnx ONNX
primitives/jnp/jnp_signbit_int.onnx ONNX
primitives/jnp/jnp_signbit_int_f64.onnx ONNX
primitives/jnp/jnp_sin.onnx ONNX
primitives/jnp/jnp_sin_basic.onnx ONNX
primitives/jnp/jnp_sin_basic_f64.onnx ONNX
primitives/jnp/jnp_sin_f64.onnx ONNX
primitives/jnp/jnp_sin_int_promote.onnx ONNX
primitives/jnp/jnp_sin_int_promote_f64.onnx ONNX
primitives/jnp/jnp_sinc_basic.onnx ONNX
primitives/jnp/jnp_sinh.onnx ONNX
primitives/jnp/jnp_sinh_basic.onnx ONNX
primitives/jnp/jnp_sinh_basic_f64.onnx ONNX
primitives/jnp/jnp_sinh_f64.onnx ONNX
primitives/jnp/jnp_sinh_int_promote.onnx ONNX
primitives/jnp/jnp_sinh_int_promote_f64.onnx ONNX
primitives/jnp/jnp_size_all.onnx ONNX
primitives/jnp/jnp_size_all_f64.onnx ONNX
primitives/jnp/jnp_size_axis.onnx ONNX
primitives/jnp/jnp_size_axis_f64.onnx ONNX
primitives/jnp/jnp_size_axis_tuple.onnx ONNX
primitives/jnp/jnp_size_axis_tuple_f64.onnx ONNX
primitives/jnp/jnp_size_dynamic.onnx ONNX
primitives/jnp/jnp_size_dynamic_dynamic.onnx ONNX
primitives/jnp/jnp_size_dynamic_dynamic_f64.onnx ONNX
primitives/jnp/jnp_size_dynamic_f64.onnx ONNX
primitives/jnp/jnp_sqrt.onnx ONNX
primitives/jnp/jnp_sqrt_basic.onnx ONNX
primitives/jnp/jnp_sqrt_basic_f64.onnx ONNX
primitives/jnp/jnp_sqrt_f64.onnx ONNX
primitives/jnp/jnp_sqrt_int_promote.onnx ONNX
primitives/jnp/jnp_sqrt_int_promote_f64.onnx ONNX
primitives/jnp/jnp_sqrt_reduce_sum_square_axis1.onnx ONNX
primitives/jnp/jnp_sqrt_reduce_sum_square_axis1_f64.onnx ONNX
primitives/jnp/jnp_square.onnx ONNX
primitives/jnp/jnp_square_f64.onnx ONNX
primitives/jnp/jnp_stack_axis0.onnx ONNX
primitives/jnp/jnp_stack_axis0_f64.onnx ONNX
primitives/jnp/jnp_stack_axis1.onnx ONNX
primitives/jnp/jnp_stack_axis1_f64.onnx ONNX
primitives/jnp/jnp_stack_negative_axis.onnx ONNX
primitives/jnp/jnp_stack_negative_axis_f64.onnx ONNX
primitives/jnp/jnp_stack_scalars.onnx ONNX
primitives/jnp/jnp_stack_scalars_f64.onnx ONNX
primitives/jnp/jnp_std_basic.onnx ONNX
primitives/jnp/jnp_subtract.onnx ONNX
primitives/jnp/jnp_subtract_broadcast.onnx ONNX
primitives/jnp/jnp_subtract_broadcast_f64.onnx ONNX
primitives/jnp/jnp_subtract_f64.onnx ONNX
primitives/jnp/jnp_sum_axis.onnx ONNX
primitives/jnp/jnp_sum_axis_f64.onnx ONNX
primitives/jnp/jnp_sum_basic.onnx ONNX
primitives/jnp/jnp_sum_basic_f64.onnx ONNX
primitives/jnp/jnp_sum_dtype_cast.onnx ONNX
primitives/jnp/jnp_sum_dtype_cast_f64.onnx ONNX
primitives/jnp/jnp_sum_int8_promote.onnx ONNX
primitives/jnp/jnp_sum_keepdims.onnx ONNX
primitives/jnp/jnp_sum_keepdims_f64.onnx ONNX
primitives/jnp/jnp_swapaxes_2d.onnx ONNX
primitives/jnp/jnp_swapaxes_2d_f64.onnx ONNX
primitives/jnp/jnp_swapaxes_3d.onnx ONNX
primitives/jnp/jnp_swapaxes_3d_f64.onnx ONNX
primitives/jnp/jnp_swapaxes_negative_axis.onnx ONNX
primitives/jnp/jnp_swapaxes_negative_axis_f64.onnx ONNX
primitives/jnp/jnp_swapaxes_vmap_batching.onnx ONNX
primitives/jnp/jnp_swapaxes_vmap_batching_f64.onnx ONNX
primitives/jnp/jnp_tan.onnx ONNX
primitives/jnp/jnp_tan_basic.onnx ONNX
primitives/jnp/jnp_tan_basic_f64.onnx ONNX
primitives/jnp/jnp_tan_int_promote.onnx ONNX
primitives/jnp/jnp_tan_int_promote_f64.onnx ONNX
primitives/jnp/jnp_tanh.onnx ONNX
primitives/jnp/jnp_tanh_basic.onnx ONNX
primitives/jnp/jnp_tanh_basic_f64.onnx ONNX
primitives/jnp/jnp_tanh_f64.onnx ONNX
primitives/jnp/jnp_tanh_int_promote.onnx ONNX
primitives/jnp/jnp_tanh_int_promote_f64.onnx ONNX
primitives/jnp/jnp_tensordot_basic.onnx ONNX
primitives/jnp/jnp_tile_basic.onnx ONNX
primitives/jnp/jnp_tile_basic_f64.onnx ONNX
primitives/jnp/jnp_tile_pad_rank.onnx ONNX
primitives/jnp/jnp_tile_pad_rank_f64.onnx ONNX
primitives/jnp/jnp_tile_scalar_repeats.onnx ONNX
primitives/jnp/jnp_tile_scalar_repeats_f64.onnx ONNX
primitives/jnp/jnp_tile_symbolic.onnx ONNX
primitives/jnp/jnp_tile_symbolic_dynamic.onnx ONNX
primitives/jnp/jnp_tile_symbolic_dynamic_f64.onnx ONNX
primitives/jnp/jnp_tile_symbolic_f64.onnx ONNX
primitives/jnp/jnp_trace_basic.onnx ONNX
primitives/jnp/jnp_trapezoid_basic.onnx ONNX
primitives/jnp/jnp_tri_basic.onnx ONNX
primitives/jnp/jnp_tril_negative_k.onnx ONNX
primitives/jnp/jnp_tril_negative_k_f64.onnx ONNX
primitives/jnp/jnp_trilu_vmap_batching.onnx ONNX
primitives/jnp/jnp_trilu_vmap_batching_f64.onnx ONNX
primitives/jnp/jnp_triu_basic.onnx ONNX
primitives/jnp/jnp_triu_basic_f64.onnx ONNX
primitives/jnp/jnp_triu_symbolic_batch.onnx ONNX
primitives/jnp/jnp_triu_symbolic_batch_dynamic.onnx ONNX
primitives/jnp/jnp_triu_symbolic_batch_dynamic_f64.onnx ONNX
primitives/jnp/jnp_triu_symbolic_batch_f64.onnx ONNX
primitives/jnp/jnp_true_divide.onnx ONNX
primitives/jnp/jnp_true_divide_broadcast.onnx ONNX
primitives/jnp/jnp_true_divide_broadcast_f64.onnx ONNX
primitives/jnp/jnp_true_divide_f64.onnx ONNX
primitives/jnp/jnp_trunc.onnx ONNX
primitives/jnp/jnp_trunc_f64.onnx ONNX
primitives/jnp/jnp_unique_f32_size_fill.onnx ONNX
primitives/jnp/jnp_unique_i32_size_fill.onnx ONNX
primitives/jnp/jnp_unique_symbolic_size_fill.onnx ONNX
primitives/jnp/jnp_unique_symbolic_size_fill_dynamic.onnx ONNX
primitives/jnp/jnp_unravel_index_basic.onnx ONNX
primitives/jnp/jnp_unravel_index_basic_f64.onnx ONNX
primitives/jnp/jnp_var_basic.onnx ONNX
primitives/jnp/jnp_vdot_basic.onnx ONNX
primitives/jnp/jnp_vecdot_basic.onnx ONNX
primitives/jnp/jnp_vecmat_basic.onnx ONNX
primitives/jnp/jnp_vectorize_basic.onnx ONNX
primitives/jnp/jnp_where_basic.onnx ONNX
primitives/jnp/jnp_where_basic_f64.onnx ONNX
primitives/jnp/jnp_where_broadcast.onnx ONNX
primitives/jnp/jnp_where_broadcast_f64.onnx ONNX
primitives/jnp/jnp_where_scalar_else.onnx ONNX
primitives/jnp/jnp_where_scalar_else_f64.onnx ONNX
primitives/jnp/jnp_zeros_2x3.onnx ONNX
primitives/jnp/jnp_zeros_2x3_f64.onnx ONNX
primitives/jnp/jnp_zeros_int32.onnx ONNX
primitives/jnp/jnp_zeros_int32_f64.onnx ONNX
primitives/jnp/jnp_zeros_like_basic.onnx ONNX
primitives/jnp/jnp_zeros_like_basic_f64.onnx ONNX
primitives/jnp/jnp_zeros_like_dtype_override.onnx ONNX
primitives/jnp/jnp_zeros_like_dtype_override_f64.onnx ONNX
primitives/jnp/left_shift_vmap_batching.onnx ONNX
primitives/jnp/left_shift_vmap_batching_f64.onnx ONNX
primitives/jnp/less_equal_vmap_batching.onnx ONNX
primitives/jnp/less_equal_vmap_batching_f64.onnx ONNX
primitives/jnp/less_vmap_batching.onnx ONNX
primitives/jnp/less_vmap_batching_f64.onnx ONNX
primitives/jnp/linalg_det_2x2.onnx ONNX
primitives/jnp/linalg_det_2x2_f64.onnx ONNX
primitives/jnp/linalg_det_batched.onnx ONNX
primitives/jnp/linalg_det_batched_f64.onnx ONNX
primitives/jnp/linalg_norm_global_default.onnx ONNX
primitives/jnp/linalg_norm_global_fro.onnx ONNX
primitives/jnp/linalg_norm_global_l1.onnx ONNX
primitives/jnp/linalg_norm_global_l2.onnx ONNX
primitives/jnp/linalg_norm_global_ord1.onnx ONNX
primitives/jnp/linalg_norm_global_ord1_nokeepdims.onnx ONNX
primitives/jnp/linspace_basic_f32.onnx ONNX
primitives/jnp/linspace_basic_f32_f64.onnx ONNX
primitives/jnp/linspace_endpoint_false_i32.onnx ONNX
primitives/jnp/linspace_endpoint_false_i32_f64.onnx ONNX
primitives/jnp/linspace_num_one.onnx ONNX
primitives/jnp/linspace_num_one_f64.onnx ONNX
primitives/jnp/linspace_num_zero.onnx ONNX
primitives/jnp/linspace_num_zero_f64.onnx ONNX
primitives/jnp/linspace_static_basic.onnx ONNX
primitives/jnp/linspace_static_basic_f64.onnx ONNX
primitives/jnp/linspace_static_endpoint_false.onnx ONNX
primitives/jnp/linspace_static_endpoint_false_f64.onnx ONNX
primitives/jnp/linspace_static_int_inputs_default_dtype.onnx ONNX
primitives/jnp/linspace_static_int_inputs_default_dtype_f64.onnx ONNX
primitives/jnp/linspace_static_num_0.onnx ONNX
primitives/jnp/linspace_static_num_0_f64.onnx ONNX
primitives/jnp/linspace_static_num_1.onnx ONNX
primitives/jnp/linspace_static_num_1_f64.onnx ONNX
primitives/jnp/linspace_vmap_batching.onnx ONNX
primitives/jnp/linspace_vmap_batching_f64.onnx ONNX
primitives/jnp/log_grad_issue_batch_diff_rules.onnx ONNX
primitives/jnp/log_vmap_batching.onnx ONNX
primitives/jnp/log_vmap_batching_f64.onnx ONNX
primitives/jnp/matmul_1d.onnx ONNX
primitives/jnp/matmul_1d_2d.onnx ONNX
primitives/jnp/matmul_1d_2d_f64.onnx ONNX
primitives/jnp/matmul_1d_f64.onnx ONNX
primitives/jnp/matmul_2d.onnx ONNX
primitives/jnp/matmul_2d_1d.onnx ONNX
primitives/jnp/matmul_2d_1d_f64.onnx ONNX
primitives/jnp/matmul_2d_f64.onnx ONNX
primitives/jnp/matmul_3d.onnx ONNX
primitives/jnp/matmul_3d_f64.onnx ONNX
primitives/jnp/matmul_complex64.onnx ONNX
primitives/jnp/matmul_complex64_f64.onnx ONNX
primitives/jnp/matmul_dynamic.onnx ONNX
primitives/jnp/matmul_dynamic_a.onnx ONNX
primitives/jnp/matmul_dynamic_a_dynamic.onnx ONNX
primitives/jnp/matmul_dynamic_a_dynamic_f64.onnx ONNX
primitives/jnp/matmul_dynamic_a_f64.onnx ONNX
primitives/jnp/matmul_dynamic_dynamic.onnx ONNX
primitives/jnp/matmul_dynamic_dynamic_f64.onnx ONNX
primitives/jnp/matmul_dynamic_f64.onnx ONNX
primitives/jnp/matmul_grad_issue_batch_diff_rules.onnx ONNX
primitives/jnp/matmul_vmap_batching.onnx ONNX
primitives/jnp/matmul_vmap_batching_f64.onnx ONNX
primitives/jnp/max_vmap_batching.onnx ONNX
primitives/jnp/max_vmap_batching_f64.onnx ONNX
primitives/jnp/mean_grad_issue_batch_diff_rules.onnx ONNX
primitives/jnp/mean_with_axis.onnx ONNX
primitives/jnp/mean_with_axis_f64.onnx ONNX
primitives/jnp/mean_with_keepdims.onnx ONNX
primitives/jnp/mean_with_keepdims_f64.onnx ONNX
primitives/jnp/min_vmap_batching.onnx ONNX
primitives/jnp/min_vmap_batching_f64.onnx ONNX
primitives/jnp/outer.onnx ONNX
primitives/jnp/outer_f64.onnx ONNX
primitives/jnp/outer_grad_issue_batch_diff_rules.onnx ONNX
primitives/jnp/outer_vector.onnx ONNX
primitives/jnp/outer_vector_f64.onnx ONNX
primitives/jnp/outer_vmap_batching.onnx ONNX
primitives/jnp/outer_vmap_batching_f64.onnx ONNX
primitives/jnp/pow_grad_issue_batch_diff_rules.onnx ONNX
primitives/jnp/pow_jnp_pow.onnx ONNX
primitives/jnp/pow_jnp_pow_f64.onnx ONNX
primitives/jnp/pow_jnp_power.onnx ONNX
primitives/jnp/pow_jnp_power_f64.onnx ONNX
primitives/jnp/pow_vmap_batching.onnx ONNX
primitives/jnp/pow_vmap_batching_f64.onnx ONNX
primitives/jnp/power_grad_issue_batch_diff_rules.onnx ONNX
primitives/jnp/power_vmap_batching.onnx ONNX
primitives/jnp/power_vmap_batching_f64.onnx ONNX
primitives/jnp/prod_grad_issue_batch_diff_rules.onnx ONNX
primitives/jnp/prod_vmap_batching.onnx ONNX
primitives/jnp/prod_vmap_batching_f64.onnx ONNX
primitives/jnp/prod_with_axis.onnx ONNX
primitives/jnp/prod_with_axis_f64.onnx ONNX
primitives/jnp/prod_with_keepdims.onnx ONNX
primitives/jnp/prod_with_keepdims_f64.onnx ONNX
primitives/jnp/reshape_1.onnx ONNX
primitives/jnp/reshape_1_f64.onnx ONNX
primitives/jnp/reshape_2.onnx ONNX
primitives/jnp/reshape_2_f64.onnx ONNX
primitives/jnp/reshape_3.onnx ONNX
primitives/jnp/reshape_3_f64.onnx ONNX
primitives/jnp/reshape_4.onnx ONNX
primitives/jnp/reshape_4_dynamic.onnx ONNX
primitives/jnp/reshape_4_dynamic_f64.onnx ONNX
primitives/jnp/reshape_4_f64.onnx ONNX
primitives/jnp/reshape_basic.onnx ONNX
primitives/jnp/reshape_basic_f64.onnx ONNX
primitives/jnp/reshape_cnn.onnx ONNX
primitives/jnp/reshape_cnn_dynamic.onnx ONNX
primitives/jnp/reshape_cnn_dynamic_f64.onnx ONNX
primitives/jnp/reshape_cnn_f64.onnx ONNX
primitives/jnp/reshape_flatten_static_uses_flatten.onnx ONNX
primitives/jnp/reshape_flatten_static_uses_flatten_f64.onnx ONNX
primitives/jnp/reshape_from_scalar.onnx ONNX
primitives/jnp/reshape_from_scalar_f64.onnx ONNX
primitives/jnp/reshape_grad_issue_batch_diff_rules.onnx ONNX
primitives/jnp/reshape_infer.onnx ONNX
primitives/jnp/reshape_infer_f64.onnx ONNX
primitives/jnp/reshape_symbolic_flatten.onnx ONNX
primitives/jnp/reshape_symbolic_flatten_dynamic.onnx ONNX
primitives/jnp/reshape_symbolic_flatten_dynamic_f64.onnx ONNX
primitives/jnp/reshape_symbolic_flatten_f64.onnx ONNX
primitives/jnp/reshape_to_scalar.onnx ONNX
primitives/jnp/reshape_to_scalar_f64.onnx ONNX
primitives/jnp/reshape_valid_flatten_trailing.onnx ONNX
primitives/jnp/reshape_valid_flatten_trailing_f64.onnx ONNX
primitives/jnp/reshape_vmap_batching_issue_144.onnx ONNX
primitives/jnp/reshape_vmap_batching_issue_144_f64.onnx ONNX
primitives/jnp/reshape_with_target_shape_from_symbolic_dim_computation.onnx ONNX
primitives/jnp/reshape_with_target_shape_from_symbolic_dim_computation_f64.onnx ONNX
primitives/jnp/right_shift_vmap_batching.onnx ONNX
primitives/jnp/right_shift_vmap_batching_f64.onnx ONNX
primitives/jnp/select_basic.onnx ONNX
primitives/jnp/select_basic_f64.onnx ONNX
primitives/jnp/select_broadcast.onnx ONNX
primitives/jnp/select_broadcast_f64.onnx ONNX
primitives/jnp/select_gpt2_attention_mask.onnx ONNX
primitives/jnp/select_gpt2_attention_mask_dynamic.onnx ONNX
primitives/jnp/select_gpt2_attention_mask_dynamic_f64.onnx ONNX
primitives/jnp/select_gpt2_attention_mask_f64.onnx ONNX
primitives/jnp/select_grad_issue_batch_diff_rules.onnx ONNX
primitives/jnp/select_simple.onnx ONNX
primitives/jnp/select_simple_f64.onnx ONNX
primitives/jnp/select_vmap_batching.onnx ONNX
primitives/jnp/select_vmap_batching_f64.onnx ONNX
primitives/jnp/shape_basic.onnx ONNX
primitives/jnp/shape_basic_f64.onnx ONNX
primitives/jnp/shape_dynamic.onnx ONNX
primitives/jnp/shape_dynamic_dynamic.onnx ONNX
primitives/jnp/shape_dynamic_dynamic_f64.onnx ONNX
primitives/jnp/shape_dynamic_f64.onnx ONNX
primitives/jnp/shape_vmap_batching.onnx ONNX
primitives/jnp/shape_vmap_batching_f64.onnx ONNX
primitives/jnp/sign_grad_issue_batch_diff_rules.onnx ONNX
primitives/jnp/sign_vmap_batching.onnx ONNX
primitives/jnp/sign_vmap_batching_f64.onnx ONNX
primitives/jnp/sin_grad_issue_batch_diff_rules.onnx ONNX
primitives/jnp/sin_vmap_batching.onnx ONNX
primitives/jnp/sin_vmap_batching_f64.onnx ONNX
primitives/jnp/sinh_grad_issue_batch_diff_rules.onnx ONNX
primitives/jnp/sinh_vmap_batching.onnx ONNX
primitives/jnp/sinh_vmap_batching_f64.onnx ONNX
primitives/jnp/sort_1d.onnx ONNX
primitives/jnp/sort_1d_f64.onnx ONNX
primitives/jnp/sort_2d_axis0.onnx ONNX
primitives/jnp/sort_2d_axis0_f64.onnx ONNX
primitives/jnp/sort_basic.onnx ONNX
primitives/jnp/sort_basic_f64.onnx ONNX
primitives/jnp/sort_vmap_batching.onnx ONNX
primitives/jnp/sort_vmap_batching_f64.onnx ONNX
primitives/jnp/split_by_indices.onnx ONNX
primitives/jnp/split_by_indices_f64.onnx ONNX
primitives/jnp/split_by_indices_symbolic.onnx ONNX
primitives/jnp/split_by_indices_symbolic_dynamic.onnx ONNX
primitives/jnp/split_by_indices_symbolic_dynamic_f64.onnx ONNX
primitives/jnp/split_by_indices_symbolic_f64.onnx ONNX
primitives/jnp/split_by_sections.onnx ONNX
primitives/jnp/split_by_sections_f64.onnx ONNX
primitives/jnp/split_grad_issue_batch_diff_rules.onnx ONNX
primitives/jnp/split_indices_numpy.onnx ONNX
primitives/jnp/split_indices_numpy_f64.onnx ONNX
primitives/jnp/split_sections.onnx ONNX
primitives/jnp/split_sections_f64.onnx ONNX
primitives/jnp/sqrt_grad_issue_batch_diff_rules.onnx ONNX
primitives/jnp/sqrt_vmap_batching.onnx ONNX
primitives/jnp/sqrt_vmap_batching_f64.onnx ONNX
primitives/jnp/squeeze_all_dims.onnx ONNX
primitives/jnp/squeeze_all_dims_f64.onnx ONNX
primitives/jnp/squeeze_dynamic_and_negative_axis.onnx ONNX
primitives/jnp/squeeze_dynamic_and_negative_axis_dynamic.onnx ONNX
primitives/jnp/squeeze_dynamic_and_negative_axis_dynamic_f64.onnx ONNX
primitives/jnp/squeeze_dynamic_and_negative_axis_f64.onnx ONNX
primitives/jnp/squeeze_dynamic_batch.onnx ONNX
primitives/jnp/squeeze_dynamic_batch_dynamic.onnx ONNX
primitives/jnp/squeeze_dynamic_batch_dynamic_f64.onnx ONNX
primitives/jnp/squeeze_dynamic_batch_f64.onnx ONNX
primitives/jnp/squeeze_grad_issue_batch_diff_rules.onnx ONNX
primitives/jnp/squeeze_multiple_dims.onnx ONNX
primitives/jnp/squeeze_multiple_dims_f64.onnx ONNX
primitives/jnp/squeeze_negative_axis.onnx ONNX
primitives/jnp/squeeze_negative_axis_f64.onnx ONNX
primitives/jnp/squeeze_negative_axis_tuple.onnx ONNX
primitives/jnp/squeeze_negative_axis_tuple_f64.onnx ONNX
primitives/jnp/squeeze_single_dim.onnx ONNX
primitives/jnp/squeeze_single_dim_f64.onnx ONNX
primitives/jnp/squeeze_vit_output.onnx ONNX
primitives/jnp/squeeze_vit_output_f64.onnx ONNX
primitives/jnp/squeeze_vmap_batching.onnx ONNX
primitives/jnp/squeeze_vmap_batching_f64.onnx ONNX
primitives/jnp/stack_axis_0.onnx ONNX
primitives/jnp/stack_axis_0_f64.onnx ONNX
primitives/jnp/stack_axis_1.onnx ONNX
primitives/jnp/stack_axis_1_f64.onnx ONNX
primitives/jnp/stack_grad_issue_batch_diff_rules.onnx ONNX
primitives/jnp/stack_negative_axis.onnx ONNX
primitives/jnp/stack_negative_axis_f64.onnx ONNX
primitives/jnp/stack_scalars.onnx ONNX
primitives/jnp/stack_scalars_f64.onnx ONNX
primitives/jnp/sum_grad_issue_batch_diff_rules.onnx ONNX
primitives/jnp/sum_vmap_batching.onnx ONNX
primitives/jnp/sum_vmap_batching_f64.onnx ONNX
primitives/jnp/take_basic_axis1.onnx ONNX
primitives/jnp/take_basic_axis1_f64.onnx ONNX
primitives/jnp/take_data_dependent_indices.onnx ONNX
primitives/jnp/take_grad_issue_batch_diff_rules.onnx ONNX
primitives/jnp/take_vmap_batching.onnx ONNX
primitives/jnp/take_vmap_batching_f64.onnx ONNX
primitives/jnp/tan_grad_issue_batch_diff_rules.onnx ONNX
primitives/jnp/tan_vmap_batching.onnx ONNX
primitives/jnp/tan_vmap_batching_f64.onnx ONNX
primitives/jnp/tanh_grad_issue_batch_diff_rules.onnx ONNX
primitives/jnp/tanh_vmap_batching.onnx ONNX
primitives/jnp/tanh_vmap_batching_f64.onnx ONNX
primitives/jnp/tile_a.onnx ONNX
primitives/jnp/tile_a_f64.onnx ONNX
primitives/jnp/tile_b.onnx ONNX
primitives/jnp/tile_b_f64.onnx ONNX
primitives/jnp/tile_c.onnx ONNX
primitives/jnp/tile_c_f64.onnx ONNX
primitives/jnp/tile_d.onnx ONNX
primitives/jnp/tile_d_f64.onnx ONNX
primitives/jnp/tile_dynamic_input.onnx ONNX
primitives/jnp/tile_dynamic_input_dynamic.onnx ONNX
primitives/jnp/tile_dynamic_input_dynamic_f64.onnx ONNX
primitives/jnp/tile_dynamic_input_f64.onnx ONNX
primitives/jnp/tile_dynamic_input_static.onnx ONNX
primitives/jnp/tile_dynamic_input_static_f64.onnx ONNX
primitives/jnp/tile_grad_issue_batch_diff_rules.onnx ONNX
primitives/jnp/tile_pad.onnx ONNX
primitives/jnp/tile_pad_f64.onnx ONNX
primitives/jnp/tile_param_symbolic.onnx ONNX
primitives/jnp/tile_param_symbolic_dynamic.onnx ONNX
primitives/jnp/tile_param_symbolic_dynamic_f64.onnx ONNX
primitives/jnp/tile_param_symbolic_f64.onnx ONNX
primitives/jnp/tile_repeats.onnx ONNX
primitives/jnp/tile_repeats_f64.onnx ONNX
primitives/jnp/tile_vmap_batching.onnx ONNX
primitives/jnp/tile_vmap_batching_f64.onnx ONNX
primitives/jnp/tile_with_symbolic_repeats.onnx ONNX
primitives/jnp/tile_with_symbolic_repeats_dynamic.onnx ONNX
primitives/jnp/tile_with_symbolic_repeats_dynamic_f64.onnx ONNX
primitives/jnp/tile_with_symbolic_repeats_f64.onnx ONNX
primitives/jnp/tile_with_symbolic_repeats_static.onnx ONNX
primitives/jnp/tile_with_symbolic_repeats_static_f64.onnx ONNX
primitives/jnp/transpose_3d.onnx ONNX
primitives/jnp/transpose_3d_f64.onnx ONNX
primitives/jnp/transpose_4d.onnx ONNX
primitives/jnp/transpose_4d_f64.onnx ONNX
primitives/jnp/transpose_basic.onnx ONNX
primitives/jnp/transpose_basic_f64.onnx ONNX
primitives/jnp/transpose_grad_issue_batch_diff_rules.onnx ONNX
primitives/jnp/transpose_high_dim.onnx ONNX
primitives/jnp/transpose_high_dim_f64.onnx ONNX
primitives/jnp/transpose_no_axes.onnx ONNX
primitives/jnp/transpose_no_axes_f64.onnx ONNX
primitives/jnp/transpose_reverse.onnx ONNX
primitives/jnp/transpose_reverse_default.onnx ONNX
primitives/jnp/transpose_reverse_default_f64.onnx ONNX
primitives/jnp/transpose_reverse_f64.onnx ONNX
primitives/jnp/transpose_square_matrix.onnx ONNX
primitives/jnp/transpose_square_matrix_f64.onnx ONNX
primitives/jnp/transpose_vmap_batching.onnx ONNX
primitives/jnp/transpose_vmap_batching_f64.onnx ONNX
primitives/jnp/unstack_axis_0.onnx ONNX
primitives/jnp/unstack_axis_0_f64.onnx ONNX
primitives/jnp/unstack_axis_1.onnx ONNX
primitives/jnp/unstack_axis_1_f64.onnx ONNX
primitives/jnp/unstack_negative_axis.onnx ONNX
primitives/jnp/unstack_negative_axis_f64.onnx ONNX
primitives/jnp/unstack_vmap_batching.onnx ONNX
primitives/jnp/unstack_vmap_batching_f64.onnx ONNX
primitives/jnp/where_A.onnx ONNX
primitives/jnp/where_A_f64.onnx ONNX
primitives/jnp/where_B.onnx ONNX
primitives/jnp/where_B_f64.onnx ONNX
primitives/jnp/where_broadcast.onnx ONNX
primitives/jnp/where_broadcast_f64.onnx ONNX
primitives/jnp/where_dtype_mismatch_f64_vs_i32_promote.onnx ONNX
primitives/jnp/where_gpt_mask_scores_literal_else.onnx ONNX
primitives/jnp/where_gpt_mask_scores_literal_else_dynamic.onnx ONNX
primitives/jnp/where_gpt_mask_scores_literal_else_dynamic_f64.onnx ONNX
primitives/jnp/where_gpt_mask_scores_literal_else_f64.onnx ONNX
primitives/jnp/where_gpt_mask_scores_scalar_else.onnx ONNX
primitives/jnp/where_gpt_mask_scores_scalar_else_dynamic.onnx ONNX
primitives/jnp/where_gpt_mask_scores_scalar_else_dynamic_f64.onnx ONNX
primitives/jnp/where_gpt_mask_scores_scalar_else_f64.onnx ONNX
primitives/jnp/where_grad_issue_batch_diff_rules.onnx ONNX
primitives/jnp/where_int_condition_cast.onnx ONNX
primitives/jnp/where_int_condition_cast_f64.onnx ONNX
primitives/jnp/where_jax_int_literals_broadcast_f64_mode.onnx ONNX
primitives/jnp/where_literal_else_pyfloat.onnx ONNX
primitives/jnp/where_literal_else_pyfloat_f64.onnx ONNX
primitives/jnp/where_multidim_condition_scalar_branches_broadcast.onnx ONNX
primitives/jnp/where_multidim_condition_scalar_branches_broadcast_f64.onnx ONNX
primitives/jnp/where_simple.onnx ONNX
primitives/jnp/where_simple_f64.onnx ONNX
primitives/jnp/where_vmap_batching.onnx ONNX
primitives/jnp/where_vmap_batching_f64.onnx ONNX
primitives/lax/abs.onnx ONNX
primitives/lax/abs_f64.onnx ONNX
primitives/lax/acos.onnx ONNX
primitives/lax/acos_f64.onnx ONNX
primitives/lax/acosh.onnx ONNX
primitives/lax/acosh_f64.onnx ONNX
primitives/lax/add.onnx ONNX
primitives/lax/add_any_via_jvp_on_mul.onnx ONNX
primitives/lax/add_any_via_jvp_on_mul_f64.onnx ONNX
primitives/lax/add_complex64.onnx ONNX
primitives/lax/add_complex64_f64.onnx ONNX
primitives/lax/add_const.onnx ONNX
primitives/lax/add_const_f64.onnx ONNX
primitives/lax/add_f64.onnx ONNX
primitives/lax/and_bool.onnx ONNX
primitives/lax/and_bool_f64.onnx ONNX
primitives/lax/and_int.onnx ONNX
primitives/lax/and_int_f64.onnx ONNX
primitives/lax/approx_max_k_matrix.onnx ONNX
primitives/lax/approx_max_k_matrix_f64.onnx ONNX
primitives/lax/approx_min_k_axis0.onnx ONNX
primitives/lax/approx_min_k_axis0_f64.onnx ONNX
primitives/lax/argmax_boolean_input_axis0_specific_values.onnx ONNX
primitives/lax/argmax_boolean_input_axis0_specific_values_f64.onnx ONNX
primitives/lax/argmax_boolean_input_axis1_specific_values.onnx ONNX
primitives/lax/argmax_boolean_input_axis1_specific_values_f64.onnx ONNX
primitives/lax/argmax_boolean_random_input_axis0.onnx ONNX
primitives/lax/argmax_boolean_random_input_axis0_f64.onnx ONNX
primitives/lax/argmax_float_axis0.onnx ONNX
primitives/lax/argmax_float_axis0_f64.onnx ONNX
primitives/lax/argmax_float_axis1.onnx ONNX
primitives/lax/argmax_float_axis1_f64.onnx ONNX
primitives/lax/argmin_test1.onnx ONNX
primitives/lax/argmin_test1_f64.onnx ONNX
primitives/lax/argmin_test2.onnx ONNX
primitives/lax/argmin_test2_f64.onnx ONNX
primitives/lax/asin.onnx ONNX
primitives/lax/asin_f64.onnx ONNX
primitives/lax/asinh.onnx ONNX
primitives/lax/asinh_f64.onnx ONNX
primitives/lax/atan.onnx ONNX
primitives/lax/atan2_broadcast.onnx ONNX
primitives/lax/atan2_quadrants_and_zero.onnx ONNX
primitives/lax/atan2_quadrants_and_zero_f64.onnx ONNX
primitives/lax/atan2_smoke.onnx ONNX
primitives/lax/atan_f64.onnx ONNX
primitives/lax/atanh.onnx ONNX
primitives/lax/atanh_f64.onnx ONNX
primitives/lax/bessel_i0e_basic.onnx ONNX
primitives/lax/bessel_i0e_basic_f64.onnx ONNX
primitives/lax/bessel_i1e_basic.onnx ONNX
primitives/lax/bessel_i1e_basic_f64.onnx ONNX
primitives/lax/betainc_basic.onnx ONNX
primitives/lax/betainc_edge_x.onnx ONNX
primitives/lax/bitcast_scalar_f32_to_i32.onnx ONNX
primitives/lax/bitcast_scalar_f32_to_i32_f64.onnx ONNX
primitives/lax/bitcast_tensor_i32_to_f32.onnx ONNX
primitives/lax/bitcast_tensor_i32_to_f32_f64.onnx ONNX
primitives/lax/bitwise_not_bool.onnx ONNX
primitives/lax/bitwise_not_bool_f64.onnx ONNX
primitives/lax/bitwise_not_i32.onnx ONNX
primitives/lax/bitwise_not_i32_f64.onnx ONNX
primitives/lax/broadcast_in_dim.onnx ONNX
primitives/lax/broadcast_in_dim_2d_to_3d.onnx ONNX
primitives/lax/broadcast_in_dim_2d_to_3d_f64.onnx ONNX
primitives/lax/broadcast_in_dim_batch.onnx ONNX
primitives/lax/broadcast_in_dim_batch_dynamic.onnx ONNX
primitives/lax/broadcast_in_dim_batch_dynamic_f64.onnx ONNX
primitives/lax/broadcast_in_dim_batch_f64.onnx ONNX
primitives/lax/broadcast_in_dim_dynamic_B.onnx ONNX
primitives/lax/broadcast_in_dim_dynamic_B_dynamic.onnx ONNX
primitives/lax/broadcast_in_dim_dynamic_B_dynamic_f64.onnx ONNX
primitives/lax/broadcast_in_dim_dynamic_B_f64.onnx ONNX
primitives/lax/broadcast_in_dim_f64.onnx ONNX
primitives/lax/broadcast_in_dim_scalar.onnx ONNX
primitives/lax/broadcast_in_dim_scalar_f64.onnx ONNX
primitives/lax/broadcasted_iota.onnx ONNX
primitives/lax/broadcasted_iota_f64.onnx ONNX
primitives/lax/cbrt.onnx ONNX
primitives/lax/cbrt_f64.onnx ONNX
primitives/lax/ceil.onnx ONNX
primitives/lax/ceil_f64.onnx ONNX
primitives/lax/cholesky_batched_2x3x3.onnx ONNX
primitives/lax/cholesky_batched_2x3x3_f64.onnx ONNX
primitives/lax/cholesky_diagonal.onnx ONNX
primitives/lax/cholesky_diagonal_f64.onnx ONNX
primitives/lax/cholesky_spd_3x3.onnx ONNX
primitives/lax/cholesky_spd_3x3_f64.onnx ONNX
primitives/lax/cholesky_update_identity.onnx ONNX
primitives/lax/cholesky_update_identity_f64.onnx ONNX
primitives/lax/cholesky_update_upper_3x3.onnx ONNX
primitives/lax/cholesky_update_upper_3x3_f64.onnx ONNX
primitives/lax/clamp_i32_scalar_bounds.onnx ONNX
primitives/lax/clamp_i32_scalar_bounds_f64.onnx ONNX
primitives/lax/clamp_pyint_bounds_promote_to_x_dtype.onnx ONNX
primitives/lax/clamp_pyint_bounds_promote_to_x_dtype_f64.onnx ONNX
primitives/lax/clamp_scalar_float_bounds_match_x.onnx ONNX
primitives/lax/clamp_scalar_float_bounds_match_x_f64.onnx ONNX
primitives/lax/clamp_vector_bounds_match.onnx ONNX
primitives/lax/clz_i32.onnx ONNX
primitives/lax/clz_i32_f64.onnx ONNX
primitives/lax/clz_u8.onnx ONNX
primitives/lax/clz_u8_f64.onnx ONNX
primitives/lax/complex_array.onnx ONNX
primitives/lax/complex_array_f64.onnx ONNX
primitives/lax/complex_double_precision.onnx ONNX
primitives/lax/complex_scalar.onnx ONNX
primitives/lax/complex_scalar_f64.onnx ONNX
primitives/lax/concatenate.onnx ONNX
primitives/lax/concatenate_3d.onnx ONNX
primitives/lax/concatenate_3d_f64.onnx ONNX
primitives/lax/concatenate_axis0.onnx ONNX
primitives/lax/concatenate_axis0_f64.onnx ONNX
primitives/lax/concatenate_axis1.onnx ONNX
primitives/lax/concatenate_axis1_dynamic.onnx ONNX
primitives/lax/concatenate_axis1_dynamic_f64.onnx ONNX
primitives/lax/concatenate_axis1_f64.onnx ONNX
primitives/lax/concatenate_f64.onnx ONNX
primitives/lax/concatenate_internal_int32_then_cast_to_f32_zeroarg.onnx ONNX
primitives/lax/cond_internal_constant_f64.onnx ONNX
primitives/lax/cond_multiple_operands_in_tuple.onnx ONNX
primitives/lax/cond_multiple_operands_in_tuple_f64.onnx ONNX
primitives/lax/cond_my_new_complex_scenario.onnx ONNX
primitives/lax/cond_my_new_complex_scenario_f64.onnx ONNX
primitives/lax/cond_nested_conditional.onnx ONNX
primitives/lax/cond_nested_conditional_f64.onnx ONNX
primitives/lax/cond_passthrough_identity.onnx ONNX
primitives/lax/cond_passthrough_identity_f64.onnx ONNX
primitives/lax/cond_scalar.onnx ONNX
primitives/lax/cond_scalar_f64.onnx ONNX
primitives/lax/cond_variables.onnx ONNX
primitives/lax/cond_variables_f64.onnx ONNX
primitives/lax/cond_with_scatter.onnx ONNX
primitives/lax/cond_with_scatter_f64.onnx ONNX
primitives/lax/conj_complex64.onnx ONNX
primitives/lax/conj_complex64_f64.onnx ONNX
primitives/lax/conj_real.onnx ONNX
primitives/lax/conj_real_f64.onnx ONNX
primitives/lax/conv.onnx ONNX
primitives/lax/conv2.onnx ONNX
primitives/lax/conv_complex128_grouped.onnx ONNX
primitives/lax/conv_complex64.onnx ONNX
primitives/lax/conv_complex64_nhwc.onnx ONNX
primitives/lax/conv_general_dilated_nhwc_output.onnx ONNX
primitives/lax/conv_nchw.onnx ONNX
primitives/lax/conv_nhwc.onnx ONNX
primitives/lax/convert_element_type.onnx ONNX
primitives/lax/convert_element_type_f64.onnx ONNX
primitives/lax/copy_float32_array.onnx ONNX
primitives/lax/copy_int64_scalar.onnx ONNX
primitives/lax/cos.onnx ONNX
primitives/lax/cos_f64.onnx ONNX
primitives/lax/cosh.onnx ONNX
primitives/lax/cosh_f64.onnx ONNX
primitives/lax/cumlogsumexp_axis1.onnx ONNX
primitives/lax/cumlogsumexp_axis1_f64.onnx ONNX
primitives/lax/cumlogsumexp_reverse_last_axis.onnx ONNX
primitives/lax/cumlogsumexp_reverse_last_axis_f64.onnx ONNX
primitives/lax/cummax_axis1.onnx ONNX
primitives/lax/cummax_axis1_f64.onnx ONNX
primitives/lax/cummax_reverse_last_axis.onnx ONNX
primitives/lax/cummax_reverse_last_axis_f64.onnx ONNX
primitives/lax/cummin_axis1.onnx ONNX
primitives/lax/cummin_axis1_f64.onnx ONNX
primitives/lax/cummin_reverse_last_axis.onnx ONNX
primitives/lax/cummin_reverse_last_axis_f64.onnx ONNX
primitives/lax/cumprod_f32_axism1_reverse.onnx ONNX
primitives/lax/cumprod_f32_axism1_reverse_f64.onnx ONNX
primitives/lax/cumprod_i32_axis2.onnx ONNX
primitives/lax/cumprod_i32_axis2_f64.onnx ONNX
primitives/lax/cumsum_f32_axism1_reverse.onnx ONNX
primitives/lax/cumsum_f32_axism1_reverse_f64.onnx ONNX
primitives/lax/cumsum_i32_axis2.onnx ONNX
primitives/lax/cumsum_i32_axis2_f64.onnx ONNX
primitives/lax/custom_linear_solve_via_matvec.onnx ONNX
primitives/lax/custom_linear_solve_via_matvec_f64.onnx ONNX
primitives/lax/device_put_array.onnx ONNX
primitives/lax/device_put_array_f64.onnx ONNX
primitives/lax/device_put_scalar.onnx ONNX
primitives/lax/device_put_scalar_f64.onnx ONNX
primitives/lax/digamma_mixed.onnx ONNX
primitives/lax/digamma_mixed_f64.onnx ONNX
primitives/lax/digamma_positive.onnx ONNX
primitives/lax/digamma_positive_f64.onnx ONNX
primitives/lax/div.onnx ONNX
primitives/lax/div_add_half_f64_no_mean.onnx ONNX
primitives/lax/div_add_half_fuses_to_mean.onnx ONNX
primitives/lax/div_add_half_fuses_to_mean_f64.onnx ONNX
primitives/lax/div_add_half_fuses_to_mean_symbolic.onnx ONNX
primitives/lax/div_add_half_fuses_to_mean_symbolic_dynamic.onnx ONNX
primitives/lax/div_add_third_no_mean.onnx ONNX
primitives/lax/div_complex64.onnx ONNX
primitives/lax/div_complex64_f64.onnx ONNX
primitives/lax/div_const.onnx ONNX
primitives/lax/div_const_f64.onnx ONNX
primitives/lax/div_f64.onnx ONNX
primitives/lax/div_lpnorm_l1_axis1.onnx ONNX
primitives/lax/div_lpnorm_l2_axis1.onnx ONNX
primitives/lax/div_lpnorm_l2_axis2.onnx ONNX
primitives/lax/div_sqrt_of_norm_no_lpnorm_fusion.onnx ONNX
primitives/lax/dot_batched_double_contract.onnx ONNX
primitives/lax/dot_batched_double_contract_f64.onnx ONNX
primitives/lax/dot_contract_inner_lhs_with_middle_rhs.onnx ONNX
primitives/lax/dot_contract_inner_lhs_with_middle_rhs_f64.onnx ONNX
primitives/lax/dot_contract_min.onnx ONNX
primitives/lax/dot_contract_min_f64.onnx ONNX
primitives/lax/dot_contract_nm.onnx ONNX
primitives/lax/dot_contract_nm_f64.onnx ONNX
primitives/lax/dot_double_contract.onnx ONNX
primitives/lax/dot_double_contract_f64.onnx ONNX
primitives/lax/dot_full_contract_scalar.onnx ONNX
primitives/lax/dot_full_contract_scalar_f64.onnx ONNX
primitives/lax/dot_general.onnx ONNX
primitives/lax/dot_general_complex_matmul.onnx ONNX
primitives/lax/dot_general_complex_matmul_f64.onnx ONNX
primitives/lax/dot_general_f64.onnx ONNX
primitives/lax/dot_general_lhs1_rhs1.onnx ONNX
primitives/lax/dot_general_lhs1_rhs1_f64.onnx ONNX
primitives/lax/dot_highrank_batch.onnx ONNX
primitives/lax/dot_highrank_batch_f64.onnx ONNX
primitives/lax/dot_outer_product.onnx ONNX
primitives/lax/dot_outer_product_f64.onnx ONNX
primitives/lax/dus_1d_block_update.onnx ONNX
primitives/lax/dus_1d_block_update_f64.onnx ONNX
primitives/lax/dus_1d_scalar_update.onnx ONNX
primitives/lax/dus_1d_scalar_update_f64.onnx ONNX
primitives/lax/dus_2d_block_update.onnx ONNX
primitives/lax/dus_2d_block_update_f64.onnx ONNX
primitives/lax/dus_3d_block_update.onnx ONNX
primitives/lax/dus_3d_block_update_f64.onnx ONNX
primitives/lax/dus_4d_block_update.onnx ONNX
primitives/lax/dus_4d_block_update_f64.onnx ONNX
primitives/lax/dus_tensorscatter_axis1_opset24.onnx ONNX
primitives/lax/dynamic_slice_2d.onnx ONNX
primitives/lax/dynamic_slice_2d_f64.onnx ONNX
primitives/lax/dynamic_slice_3d.onnx ONNX
primitives/lax/dynamic_slice_3d_f64.onnx ONNX
primitives/lax/dynamic_slice_test1.onnx ONNX
primitives/lax/dynamic_slice_test1_f64.onnx ONNX
primitives/lax/dynamic_slice_vit_like.onnx ONNX
primitives/lax/dynamic_slice_vit_like_dynamic.onnx ONNX
primitives/lax/dynamic_slice_vit_like_dynamic_f64.onnx ONNX
primitives/lax/dynamic_slice_vit_like_f64.onnx ONNX
primitives/lax/eig_1x1_default.onnx ONNX
primitives/lax/eig_1x1_default_f64.onnx ONNX
primitives/lax/eig_1x1_full.onnx ONNX
primitives/lax/eig_1x1_full_f64.onnx ONNX
primitives/lax/eig_1x1_left_only.onnx ONNX
primitives/lax/eig_1x1_left_only_f64.onnx ONNX
primitives/lax/eig_1x1_right_only.onnx ONNX
primitives/lax/eig_1x1_right_only_f64.onnx ONNX
primitives/lax/eig_1x1_values_only.onnx ONNX
primitives/lax/eig_1x1_values_only_f64.onnx ONNX
primitives/lax/eig_2x2_values_only_complex128.onnx ONNX
primitives/lax/eig_2x2_values_only_real.onnx ONNX
primitives/lax/eig_2x2_values_only_real_f64.onnx ONNX
primitives/lax/eigh_1x1.onnx ONNX
primitives/lax/eigh_1x1_f64.onnx ONNX
primitives/lax/eigh_2x2_lower_false.onnx ONNX
primitives/lax/eigh_2x2_lower_false_f64.onnx ONNX
primitives/lax/eigh_2x2_lower_true.onnx ONNX
primitives/lax/eigh_2x2_lower_true_f64.onnx ONNX
primitives/lax/eigh_2x2_subset_top1.onnx ONNX
primitives/lax/eigh_2x2_subset_top1_f64.onnx ONNX
primitives/lax/eq.onnx ONNX
primitives/lax/eq_f64.onnx ONNX
primitives/lax/erf.onnx ONNX
primitives/lax/erf_inv_matrix.onnx ONNX
primitives/lax/erf_inv_matrix_f64.onnx ONNX
primitives/lax/erf_inv_midrange.onnx ONNX
primitives/lax/erf_inv_midrange_f64.onnx ONNX
primitives/lax/erfc.onnx ONNX
primitives/lax/erfc_f64.onnx ONNX
primitives/lax/exp.onnx ONNX
primitives/lax/exp2.onnx ONNX
primitives/lax/exp2_f64.onnx ONNX
primitives/lax/exp_f64.onnx ONNX
primitives/lax/expm1.onnx ONNX
primitives/lax/expm1_f64.onnx ONNX
primitives/lax/fft_complex128_1d.onnx ONNX
primitives/lax/fft_complex128_len8.onnx ONNX
primitives/lax/fft_complex64_1d.onnx ONNX
primitives/lax/fft_complex64_batch.onnx ONNX
primitives/lax/fft_complex64_len8.onnx ONNX
primitives/lax/floor.onnx ONNX
primitives/lax/floor_f64.onnx ONNX
primitives/lax/fori_loop_counter.onnx ONNX
primitives/lax/fori_loop_counter_f64.onnx ONNX
primitives/lax/fori_loop_example.onnx ONNX
primitives/lax/fori_loop_example_f64.onnx ONNX
primitives/lax/fori_loop_test.onnx ONNX
primitives/lax/fori_loop_test_f64.onnx ONNX
primitives/lax/fori_loop_vector.onnx ONNX
primitives/lax/fori_loop_vector_f64.onnx ONNX
primitives/lax/fori_loop_zero.onnx ONNX
primitives/lax/fori_loop_zero_f64.onnx ONNX
primitives/lax/gather_dynamic_batch_simple_index.onnx ONNX
primitives/lax/gather_dynamic_batch_simple_index_dynamic.onnx ONNX
primitives/lax/gather_dynamic_batch_simple_index_dynamic_f64.onnx ONNX
primitives/lax/gather_dynamic_batch_simple_index_f64.onnx ONNX
primitives/lax/gather_f64_data_i32_indices_cast_and_output_is_f64.onnx ONNX
primitives/lax/gather_f64_data_i64_indices_output_is_f64.onnx ONNX
primitives/lax/gather_static.onnx ONNX
primitives/lax/gather_static_f64.onnx ONNX
primitives/lax/gather_trig_where_pipeline_f64_indices_i32.onnx ONNX
primitives/lax/gather_trig_where_pipeline_f64_indices_i64.onnx ONNX
primitives/lax/greater_equal.onnx ONNX
primitives/lax/greater_equal_f64.onnx ONNX
primitives/lax/gt.onnx ONNX
primitives/lax/gt_f64.onnx ONNX
primitives/lax/hessenberg_diagonal.onnx ONNX
primitives/lax/hessenberg_diagonal_f64.onnx ONNX
primitives/lax/hessenberg_square_4x4.onnx ONNX
primitives/lax/hessenberg_square_4x4_f64.onnx ONNX
primitives/lax/householder_product_basic.onnx ONNX
primitives/lax/householder_product_basic_f64.onnx ONNX
primitives/lax/householder_product_k0.onnx ONNX
primitives/lax/householder_product_k0_f64.onnx ONNX
primitives/lax/ifft_complex128_batch.onnx ONNX
primitives/lax/ifft_complex64_1d.onnx ONNX
primitives/lax/ifft_complex64_len8.onnx ONNX
primitives/lax/igamma_basic.onnx ONNX
primitives/lax/igamma_grad_a_basic.onnx ONNX
primitives/lax/igamma_zero_x.onnx ONNX
primitives/lax/igammac_basic.onnx ONNX
primitives/lax/igammac_zero_x.onnx ONNX
primitives/lax/imag_complex64_input.onnx ONNX
primitives/lax/imag_complex64_input_f64.onnx ONNX
primitives/lax/integer_pow.onnx ONNX
primitives/lax/integer_pow_f64.onnx ONNX
primitives/lax/integer_pow_reciprocal.onnx ONNX
primitives/lax/integer_pow_reciprocal_f64.onnx ONNX
primitives/lax/iota_float32.onnx ONNX
primitives/lax/iota_float32_f64.onnx ONNX
primitives/lax/iota_int32.onnx ONNX
primitives/lax/iota_int32_f64.onnx ONNX
primitives/lax/irfft_complex128_len8.onnx ONNX
primitives/lax/irfft_complex64_1d.onnx ONNX
primitives/lax/is_finite_vec.onnx ONNX
primitives/lax/is_finite_vec_f64.onnx ONNX
primitives/lax/lax_logistic_basic.onnx ONNX
primitives/lax/lax_logistic_basic_f64.onnx ONNX
primitives/lax/lax_split_equal_parts.onnx ONNX
primitives/lax/lax_split_equal_parts_f64.onnx ONNX
primitives/lax/lax_split_unequal_parts.onnx ONNX
primitives/lax/lax_split_unequal_parts_f64.onnx ONNX
primitives/lax/lax_squeeze_multiple_axes.onnx ONNX
primitives/lax/lax_squeeze_multiple_axes_f64.onnx ONNX
primitives/lax/lax_squeeze_no_op_empty_dims.onnx ONNX
primitives/lax/lax_squeeze_no_op_empty_dims_f64.onnx ONNX
primitives/lax/lax_squeeze_problem_case_input_squeeze_all_dims_explicitly.onnx ONNX
primitives/lax/lax_squeeze_problem_case_input_squeeze_all_dims_explicitly_f64.onnx ONNX
primitives/lax/lax_squeeze_problem_case_input_squeeze_axes_0_2.onnx ONNX
primitives/lax/lax_squeeze_problem_case_input_squeeze_axes_0_2_f64.onnx ONNX
primitives/lax/lax_squeeze_problem_case_input_squeeze_only_axis_0.onnx ONNX
primitives/lax/lax_squeeze_problem_case_input_squeeze_only_axis_0_f64.onnx ONNX
primitives/lax/lax_squeeze_specific_axis_0.onnx ONNX
primitives/lax/lax_squeeze_specific_axis_0_f64.onnx ONNX
primitives/lax/less_equal.onnx ONNX
primitives/lax/less_equal_f64.onnx ONNX
primitives/lax/lgamma_negative_noninteger.onnx ONNX
primitives/lax/lgamma_negative_noninteger_f64.onnx ONNX
primitives/lax/lgamma_positive.onnx ONNX
primitives/lax/lgamma_positive_f64.onnx ONNX
primitives/lax/log.onnx ONNX
primitives/lax/log1p.onnx ONNX
primitives/lax/log1p_f64.onnx ONNX
primitives/lax/log_f64.onnx ONNX
primitives/lax/log_of_reduce_sum_axis1.onnx ONNX
primitives/lax/log_of_reduce_sum_axis1_f64.onnx ONNX
primitives/lax/log_of_reduce_sum_exp_axis1.onnx ONNX
primitives/lax/log_of_reduce_sum_exp_axis1_f64.onnx ONNX
primitives/lax/lt.onnx ONNX
primitives/lax/lt_f64.onnx ONNX
primitives/lax/lu_pivots_to_permutation_basic.onnx ONNX
primitives/lax/lu_pivots_to_permutation_basic_f64.onnx ONNX
primitives/lax/lu_pivots_to_permutation_batched.onnx ONNX
primitives/lax/lu_pivots_to_permutation_batched_f64.onnx ONNX
primitives/lax/lu_pivots_to_permutation_identity.onnx ONNX
primitives/lax/lu_pivots_to_permutation_identity_f64.onnx ONNX
primitives/lax/lu_rectangular_4x3.onnx ONNX
primitives/lax/lu_rectangular_4x3_f64.onnx ONNX
primitives/lax/lu_square_3x3.onnx ONNX
primitives/lax/lu_square_3x3_f64.onnx ONNX
primitives/lax/max.onnx ONNX
primitives/lax/max_f64.onnx ONNX
primitives/lax/min_test1.onnx ONNX
primitives/lax/min_test1_f64.onnx ONNX
primitives/lax/mul_complex128.onnx ONNX
primitives/lax/mul_complex64.onnx ONNX
primitives/lax/mul_pyfloat_promotes_to_array_dtype_f64.onnx ONNX
primitives/lax/mul_scalar_broadcast_promote_to_f64.onnx ONNX
primitives/lax/mul_test1.onnx ONNX
primitives/lax/mul_test1_f64.onnx ONNX
primitives/lax/mul_test2.onnx ONNX
primitives/lax/mul_test2_f64.onnx ONNX
primitives/lax/ne.onnx ONNX
primitives/lax/ne_f64.onnx ONNX
primitives/lax/neg.onnx ONNX
primitives/lax/neg_f64.onnx ONNX
primitives/lax/nextafter_f32_basic.onnx ONNX
primitives/lax/nextafter_f32_basic_f64.onnx ONNX
primitives/lax/nextafter_f32_nan_inf.onnx ONNX
primitives/lax/nextafter_f32_nan_inf_f64.onnx ONNX
primitives/lax/nextafter_special_values.onnx ONNX
primitives/lax/nextafter_vector.onnx ONNX
primitives/lax/optimization_barrier_single.onnx ONNX
primitives/lax/optimization_barrier_single_f64.onnx ONNX
primitives/lax/optimization_barrier_tuple.onnx ONNX
primitives/lax/optimization_barrier_tuple_f64.onnx ONNX
primitives/lax/or_bool_vec.onnx ONNX
primitives/lax/or_bool_vec_f64.onnx ONNX
primitives/lax/or_int_vec.onnx ONNX
primitives/lax/or_int_vec_f64.onnx ONNX
primitives/lax/pad_const_1d.onnx ONNX
primitives/lax/pad_const_1d_f64.onnx ONNX
primitives/lax/pad_const_2d.onnx ONNX
primitives/lax/pad_const_2d_cval.onnx ONNX
primitives/lax/pad_const_2d_cval_f64.onnx ONNX
primitives/lax/pad_const_2d_f64.onnx ONNX
primitives/lax/pad_inside_nested_scan_smoke_f64.onnx ONNX
primitives/lax/pad_inside_scan_smoke_f64.onnx ONNX
primitives/lax/pjit_inline_mul.onnx ONNX
primitives/lax/pjit_inline_mul_f64.onnx ONNX
primitives/lax/pjit_inline_tuple.onnx ONNX
primitives/lax/pjit_inline_tuple_f64.onnx ONNX
primitives/lax/polygamma_orders.onnx ONNX
primitives/lax/polygamma_zero_order.onnx ONNX
primitives/lax/population_count_i32.onnx ONNX
primitives/lax/population_count_i32_f64.onnx ONNX
primitives/lax/population_count_u8.onnx ONNX
primitives/lax/population_count_u8_f64.onnx ONNX
primitives/lax/pow_basic.onnx ONNX
primitives/lax/pow_basic_f64.onnx ONNX
primitives/lax/pow_lax.onnx ONNX
primitives/lax/pow_lax_f64.onnx ONNX
primitives/lax/qr_full_tall.onnx ONNX
primitives/lax/qr_full_tall_f64.onnx ONNX
primitives/lax/qr_full_wide.onnx ONNX
primitives/lax/qr_full_wide_f64.onnx ONNX
primitives/lax/qr_reduced_tall.onnx ONNX
primitives/lax/qr_reduced_tall_f64.onnx ONNX
primitives/lax/qr_reduced_wide.onnx ONNX
primitives/lax/qr_reduced_wide_f64.onnx ONNX
primitives/lax/real_complex64_input.onnx ONNX
primitives/lax/real_complex64_input_f64.onnx ONNX
primitives/lax/reduce_and_all_true.onnx ONNX
primitives/lax/reduce_and_all_true_f64.onnx ONNX
primitives/lax/reduce_and_keepdims.onnx ONNX
primitives/lax/reduce_and_keepdims_f64.onnx ONNX
primitives/lax/reduce_and_one_false.onnx ONNX
primitives/lax/reduce_and_one_false_f64.onnx ONNX
primitives/lax/reduce_max.onnx ONNX
primitives/lax/reduce_max_allaxes.onnx ONNX
primitives/lax/reduce_max_allaxes_f64.onnx ONNX
primitives/lax/reduce_max_axes_input.onnx ONNX
primitives/lax/reduce_max_axes_input_f64.onnx ONNX
primitives/lax/reduce_max_f64.onnx ONNX
primitives/lax/reduce_max_keepdims.onnx ONNX
primitives/lax/reduce_max_keepdims_f64.onnx ONNX
primitives/lax/reduce_max_lambda.onnx ONNX
primitives/lax/reduce_max_lambda_f64.onnx ONNX
primitives/lax/reduce_min.onnx ONNX
primitives/lax/reduce_min_allaxes.onnx ONNX
primitives/lax/reduce_min_allaxes_f64.onnx ONNX
primitives/lax/reduce_min_f64.onnx ONNX
primitives/lax/reduce_min_keepdims.onnx ONNX
primitives/lax/reduce_min_keepdims_f64.onnx ONNX
primitives/lax/reduce_min_lambda.onnx ONNX
primitives/lax/reduce_min_lambda_f64.onnx ONNX
primitives/lax/reduce_or_all_false.onnx ONNX
primitives/lax/reduce_or_all_false_f64.onnx ONNX
primitives/lax/reduce_or_keepdims.onnx ONNX
primitives/lax/reduce_or_keepdims_f64.onnx ONNX
primitives/lax/reduce_or_one_true.onnx ONNX
primitives/lax/reduce_or_one_true_f64.onnx ONNX
primitives/lax/reduce_precision_mantissa10.onnx ONNX
primitives/lax/reduce_precision_underflow_overflow.onnx ONNX
primitives/lax/reduce_prod.onnx ONNX
primitives/lax/reduce_prod_allaxes.onnx ONNX
primitives/lax/reduce_prod_allaxes_f64.onnx ONNX
primitives/lax/reduce_prod_dtype.onnx ONNX
primitives/lax/reduce_prod_dtype_f64.onnx ONNX
primitives/lax/reduce_prod_dtype_f64_f64.onnx ONNX
primitives/lax/reduce_prod_f64.onnx ONNX
primitives/lax/reduce_prod_keepdims.onnx ONNX
primitives/lax/reduce_prod_keepdims_f64.onnx ONNX
primitives/lax/reduce_sum.onnx ONNX
primitives/lax/reduce_sum_allaxes.onnx ONNX
primitives/lax/reduce_sum_allaxes_f64.onnx ONNX
primitives/lax/reduce_sum_dtype.onnx ONNX
primitives/lax/reduce_sum_dtype_f64.onnx ONNX
primitives/lax/reduce_sum_dtype_f64_f64.onnx ONNX
primitives/lax/reduce_sum_f64.onnx ONNX
primitives/lax/reduce_sum_keepdims.onnx ONNX
primitives/lax/reduce_sum_keepdims_f64.onnx ONNX
primitives/lax/reduce_sum_of_abs_axis1.onnx ONNX
primitives/lax/reduce_sum_of_abs_axis1_f64.onnx ONNX
primitives/lax/reduce_sum_of_square_axis1.onnx ONNX
primitives/lax/reduce_sum_of_square_axis1_f64.onnx ONNX
primitives/lax/reduce_window_add_lambda.onnx ONNX
primitives/lax/reduce_window_max_lambda.onnx ONNX
primitives/lax/reduce_window_max_primitive.onnx ONNX
primitives/lax/reduce_window_min_lambda_stride.onnx ONNX
primitives/lax/reduce_window_sum_abs_lppool_dilated_opset23.onnx ONNX
primitives/lax/reduce_window_sum_abs_lppool_opset23.onnx ONNX
primitives/lax/reduce_window_sum_base_dilation.onnx ONNX
primitives/lax/reduce_window_sum_int32.onnx ONNX
primitives/lax/reduce_window_sum_same_padding.onnx ONNX
primitives/lax/reduce_window_sum_stride_dilate.onnx ONNX
primitives/lax/reduce_window_sum_valid.onnx ONNX
primitives/lax/reduce_xor_all_false.onnx ONNX
primitives/lax/reduce_xor_all_false_f64.onnx ONNX
primitives/lax/reduce_xor_keepdims.onnx ONNX
primitives/lax/reduce_xor_keepdims_f64.onnx ONNX
primitives/lax/reduce_xor_one_true.onnx ONNX
primitives/lax/reduce_xor_one_true_f64.onnx ONNX
primitives/lax/reduce_xor_two_true.onnx ONNX
primitives/lax/reduce_xor_two_true_f64.onnx ONNX
primitives/lax/rem_float.onnx ONNX
primitives/lax/rem_float_f64.onnx ONNX
primitives/lax/rem_float_neg.onnx ONNX
primitives/lax/rem_float_neg_f64.onnx ONNX
primitives/lax/rem_int.onnx ONNX
primitives/lax/rem_int_f64.onnx ONNX
primitives/lax/rem_int_neg.onnx ONNX
primitives/lax/rem_int_neg_f64.onnx ONNX
primitives/lax/remat2_scalar_sin_chain.onnx ONNX
primitives/lax/remat2_scalar_sin_chain_f64.onnx ONNX
primitives/lax/remat2_tuple_passthrough.onnx ONNX
primitives/lax/remat2_tuple_passthrough_f64.onnx ONNX
primitives/lax/reshape.onnx ONNX
primitives/lax/reshape_after_transpose_folds_const_shape.onnx ONNX
primitives/lax/reshape_after_transpose_folds_const_shape_f64.onnx ONNX
primitives/lax/reshape_f64.onnx ONNX
primitives/lax/reshape_flatten_trailing_folds_const_shape.onnx ONNX
primitives/lax/reshape_flatten_trailing_folds_const_shape_f64.onnx ONNX
primitives/lax/reshape_merge_symbolic_with_static_and_check_name.onnx ONNX
primitives/lax/reshape_merge_symbolic_with_static_and_check_name_dynamic.onnx ONNX
primitives/lax/reshape_valid_flatten_trailing.onnx ONNX
primitives/lax/reshape_valid_flatten_trailing_f64.onnx ONNX
primitives/lax/reshape_valid_squeeze_middle_dim_from_problematic_source.onnx ONNX
primitives/lax/reshape_valid_squeeze_middle_dim_from_problematic_source_f64.onnx ONNX
primitives/lax/reshape_with_inferred_dimension_from_input.onnx ONNX
primitives/lax/reshape_with_inferred_dimension_from_input_dynamic.onnx ONNX
primitives/lax/reshape_with_inferred_dimension_from_input_dynamic_dynamic.onnx ONNX
primitives/lax/reshape_with_inferred_dimension_from_input_dynamic_dynamic_f64.onnx ONNX
primitives/lax/reshape_with_inferred_dimension_from_input_dynamic_f64.onnx ONNX
primitives/lax/reshape_with_inferred_dimension_from_input_f64.onnx ONNX
primitives/lax/reshape_with_target_shape_from_symbolic_dim_computation.onnx ONNX
primitives/lax/reshape_with_target_shape_from_symbolic_dim_computation_f64.onnx ONNX
primitives/lax/rev_matrix_axes01.onnx ONNX
primitives/lax/rev_matrix_axes01_f64.onnx ONNX
primitives/lax/rev_vector.onnx ONNX
primitives/lax/rev_vector_f64.onnx ONNX
primitives/lax/rfft_real32_1d.onnx ONNX
primitives/lax/rfft_real64_len8.onnx ONNX
primitives/lax/rng_bit_generator_u32.onnx ONNX
primitives/lax/rng_uniform_f32.onnx ONNX
primitives/lax/round.onnx ONNX
primitives/lax/round_f64.onnx ONNX
primitives/lax/rsqrt.onnx ONNX
primitives/lax/rsqrt_f64.onnx ONNX
primitives/lax/scan_captured_scalar.onnx ONNX
primitives/lax/scan_captured_scalar_f64.onnx ONNX
primitives/lax/scan_captured_scalar_with_xs.onnx ONNX
primitives/lax/scan_captured_vector_with_xs_f64.onnx ONNX
primitives/lax/scan_carry_only.onnx ONNX
primitives/lax/scan_carry_only_f64.onnx ONNX
primitives/lax/scan_cumsum.onnx ONNX
primitives/lax/scan_cumsum_f64.onnx ONNX
primitives/lax/scan_fn.onnx ONNX
primitives/lax/scan_fn_f64.onnx ONNX
primitives/lax/scan_identity_slice_helper.onnx ONNX
primitives/lax/scan_identity_slice_helper_f64.onnx ONNX
primitives/lax/scan_jit_no_xs.onnx ONNX
primitives/lax/scan_jit_no_xs_f64.onnx ONNX
primitives/lax/scan_matrix_carry_multidim_xs.onnx ONNX
primitives/lax/scan_matrix_carry_multidim_xs_f64.onnx ONNX
primitives/lax/scan_multiple_carry.onnx ONNX
primitives/lax/scan_multiple_carry_f64.onnx ONNX
primitives/lax/scan_multiple_sequences.onnx ONNX
primitives/lax/scan_multiple_sequences_f64.onnx ONNX
primitives/lax/scan_nested_len_mismatch.onnx ONNX
primitives/lax/scan_nested_len_mismatch_f64.onnx ONNX
primitives/lax/scan_no_xs.onnx ONNX
primitives/lax/scan_no_xs_f64.onnx ONNX
primitives/lax/scan_rank0_sequence_vectorized.onnx ONNX
primitives/lax/scan_rank0_sequence_vectorized_f64.onnx ONNX
primitives/lax/scan_two_diff_lengths.onnx ONNX
primitives/lax/scan_two_diff_lengths_broadcast.onnx ONNX
primitives/lax/scan_two_diff_lengths_broadcast_f64.onnx ONNX
primitives/lax/scan_two_diff_lengths_f64.onnx ONNX
primitives/lax/scan_two_diff_lengths_with_broadcast.onnx ONNX
primitives/lax/scatter_add_batch_updates_1d_operand.onnx ONNX
primitives/lax/scatter_add_batch_updates_1d_operand_f64.onnx ONNX
primitives/lax/scatter_add_depth2_depth2_helper_regression.onnx ONNX
primitives/lax/scatter_add_fluids_pattern_updates_5_4_1_1.onnx ONNX
primitives/lax/scatter_add_fp64_dtype_mismatch.onnx ONNX
primitives/lax/scatter_add_in_cond_float64.onnx ONNX
primitives/lax/scatter_add_mismatched_window_dims_from_user_report.onnx ONNX
primitives/lax/scatter_add_mismatched_window_dims_from_user_report2.onnx ONNX
primitives/lax/scatter_add_mismatched_window_dims_from_user_report3.onnx ONNX
primitives/lax/scatter_add_scalar.onnx ONNX
primitives/lax/scatter_add_scalar_f64.onnx ONNX
primitives/lax/scatter_add_simple_1d.onnx ONNX
primitives/lax/scatter_add_simple_1d_f64.onnx ONNX
primitives/lax/scatter_add_vector.onnx ONNX
primitives/lax/scatter_add_vector_f64.onnx ONNX
primitives/lax/scatter_add_window_2d_operand_1d_indices.onnx ONNX
primitives/lax/scatter_add_window_2d_operand_1d_indices_f64.onnx ONNX
primitives/lax/scatter_clip_2d_window_at_edge.onnx ONNX
primitives/lax/scatter_correct_axis_determination.onnx ONNX
primitives/lax/scatter_correct_axis_determination_f64.onnx ONNX
primitives/lax/scatter_depth2_fp64_type_mismatch.onnx ONNX
primitives/lax/scatter_depth2_mixed_dtypes_fp_mismatch.onnx ONNX
primitives/lax/scatter_depth2_mixed_dtypes_fp_mismatch_f64.onnx ONNX
primitives/lax/scatter_elements_set_vector_promise_in_bounds.onnx ONNX
primitives/lax/scatter_elements_set_vector_promise_in_bounds_f64.onnx ONNX
primitives/lax/scatter_from_user_warning_shapes_valid_jax.onnx ONNX
primitives/lax/scatter_from_user_warning_shapes_valid_jax_f64.onnx ONNX
primitives/lax/scatter_max_batch_updates_1d_operand.onnx ONNX
primitives/lax/scatter_max_batch_updates_1d_operand_f64.onnx ONNX
primitives/lax/scatter_max_depth2_helper_regression_fp64.onnx ONNX
primitives/lax/scatter_max_fp64_dtype_path_check.onnx ONNX
primitives/lax/scatter_max_simple_1d.onnx ONNX
primitives/lax/scatter_max_simple_1d_f64.onnx ONNX
primitives/lax/scatter_max_window_2d_operand_1d_indices.onnx ONNX
primitives/lax/scatter_max_window_2d_operand_1d_indices_f64.onnx ONNX
primitives/lax/scatter_min_batch_updates_1d_operand.onnx ONNX
primitives/lax/scatter_min_batch_updates_1d_operand_f64.onnx ONNX
primitives/lax/scatter_min_depth2_helper_regression_fp64.onnx ONNX
primitives/lax/scatter_min_fp64_dtype_path_check.onnx ONNX
primitives/lax/scatter_min_simple_1d.onnx ONNX
primitives/lax/scatter_min_simple_1d_f64.onnx ONNX
primitives/lax/scatter_min_window_2d_operand_1d_indices.onnx ONNX
primitives/lax/scatter_min_window_2d_operand_1d_indices_f64.onnx ONNX
primitives/lax/scatter_mul_batch_updates_1d_operand.onnx ONNX
primitives/lax/scatter_mul_batch_updates_1d_operand_f64.onnx ONNX
primitives/lax/scatter_mul_fluids_pattern_updates_5_4_1_1.onnx ONNX
primitives/lax/scatter_mul_in_cond_float64.onnx ONNX
primitives/lax/scatter_mul_mismatched_window_dims_from_user_report.onnx ONNX
primitives/lax/scatter_mul_mismatched_window_dims_from_user_report2.onnx ONNX
primitives/lax/scatter_mul_mismatched_window_dims_from_user_report3.onnx ONNX
primitives/lax/scatter_mul_simple_1d.onnx ONNX
primitives/lax/scatter_mul_simple_1d_f64.onnx ONNX
primitives/lax/scatter_mul_window_2d_operand_1d_indices.onnx ONNX
primitives/lax/scatter_mul_window_2d_operand_1d_indices_f64.onnx ONNX
primitives/lax/scatter_set_axis0.onnx ONNX
primitives/lax/scatter_set_axis0_f64.onnx ONNX
primitives/lax/scatter_set_middle.onnx ONNX
primitives/lax/scatter_set_middle_f64.onnx ONNX
primitives/lax/scatter_set_single.onnx ONNX
primitives/lax/scatter_set_single_f64.onnx ONNX
primitives/lax/scatter_set_vector.onnx ONNX
primitives/lax/scatter_set_vector_f64.onnx ONNX
primitives/lax/scatter_set_vector_promise_in_bounds_unique.onnx ONNX
primitives/lax/scatter_set_vector_promise_in_bounds_unique_f64.onnx ONNX
primitives/lax/scatter_simple_2d_window_out_of_bounds.onnx ONNX
primitives/lax/scatter_static_slice_set_f64.onnx ONNX
primitives/lax/scatter_sub_simple_1d.onnx ONNX
primitives/lax/scatter_sub_simple_1d_f64.onnx ONNX
primitives/lax/scatter_sub_window_2d_operand_1d_indices.onnx ONNX
primitives/lax/scatter_sub_window_2d_operand_1d_indices_f64.onnx ONNX
primitives/lax/scatter_updates_slice_needed_axis0.onnx ONNX
primitives/lax/scatter_updates_slice_needed_axis0_f64.onnx ONNX
primitives/lax/scatter_user_error_scenario_precise.onnx ONNX
primitives/lax/scatter_user_error_scenario_precise_f64.onnx ONNX
primitives/lax/scatter_window_update_depth3_shapes_ok.onnx ONNX
primitives/lax/scatter_window_update_f64.onnx ONNX
primitives/lax/schur_1x1_default.onnx ONNX
primitives/lax/schur_1x1_default_f64.onnx ONNX
primitives/lax/schur_1x1_no_vectors.onnx ONNX
primitives/lax/schur_1x1_no_vectors_f64.onnx ONNX
primitives/lax/select_basic.onnx ONNX
primitives/lax/select_basic_f64.onnx ONNX
primitives/lax/select_mask_scores_tensor_else.onnx ONNX
primitives/lax/select_mask_scores_tensor_else_dynamic.onnx ONNX
primitives/lax/select_mask_scores_tensor_else_dynamic_f64.onnx ONNX
primitives/lax/select_mask_scores_tensor_else_f64.onnx ONNX
primitives/lax/select_n_bool_predicate_scalar_broadcast.onnx ONNX
primitives/lax/select_n_bool_predicate_scalar_broadcast_f64.onnx ONNX
primitives/lax/select_n_bool_predicate_two_cases_float.onnx ONNX
primitives/lax/select_n_bool_predicate_two_cases_float_f64.onnx ONNX
primitives/lax/select_n_bool_predicate_two_cases_int.onnx ONNX
primitives/lax/select_n_bool_predicate_two_cases_int_f64.onnx ONNX
primitives/lax/select_n_int_indices_four_cases.onnx ONNX
primitives/lax/select_n_int_indices_four_cases_f64.onnx ONNX
primitives/lax/select_n_int_indices_three_cases.onnx ONNX
primitives/lax/select_n_int_indices_three_cases_f64.onnx ONNX
primitives/lax/select_simple.onnx ONNX
primitives/lax/select_simple_f64.onnx ONNX
primitives/lax/shard_map_inline_add.onnx ONNX
primitives/lax/shard_map_inline_add_f64.onnx ONNX
primitives/lax/shift_left_scalar.onnx ONNX
primitives/lax/shift_left_scalar_f64.onnx ONNX
primitives/lax/shift_left_vec.onnx ONNX
primitives/lax/shift_left_vec_f64.onnx ONNX
primitives/lax/shift_right_arithmetic_scalar.onnx ONNX
primitives/lax/shift_right_arithmetic_scalar_f64.onnx ONNX
primitives/lax/shift_right_arithmetic_vec.onnx ONNX
primitives/lax/shift_right_arithmetic_vec_f64.onnx ONNX
primitives/lax/shift_right_logical_scalar.onnx ONNX
primitives/lax/shift_right_logical_scalar_f64.onnx ONNX
primitives/lax/shift_right_logical_vec.onnx ONNX
primitives/lax/shift_right_logical_vec_f64.onnx ONNX
primitives/lax/sign.onnx ONNX
primitives/lax/sign_f64.onnx ONNX
primitives/lax/sin.onnx ONNX
primitives/lax/sin_f64.onnx ONNX
primitives/lax/sinh.onnx ONNX
primitives/lax/sinh_f64.onnx ONNX
primitives/lax/slice_3d_none_strides.onnx ONNX
primitives/lax/slice_3d_none_strides_f64.onnx ONNX
primitives/lax/slice_scan_axis_drop.onnx ONNX
primitives/lax/slice_scan_axis_drop_f64.onnx ONNX
primitives/lax/slice_test1.onnx ONNX
primitives/lax/slice_test1_f64.onnx ONNX
primitives/lax/sort_1d.onnx ONNX
primitives/lax/sort_1d_f64.onnx ONNX
primitives/lax/sort_2d.onnx ONNX
primitives/lax/sort_2d_f64.onnx ONNX
primitives/lax/sqrt.onnx ONNX
primitives/lax/sqrt_f64.onnx ONNX
primitives/lax/sqrt_reduce_sum_square_axis1.onnx ONNX
primitives/lax/sqrt_reduce_sum_square_axis1_f64.onnx ONNX
primitives/lax/square.onnx ONNX
primitives/lax/square_f64.onnx ONNX
primitives/lax/squeeze_all_unit_dims_default.onnx ONNX
primitives/lax/squeeze_all_unit_dims_default_f64.onnx ONNX
primitives/lax/squeeze_single_axis.onnx ONNX
primitives/lax/squeeze_single_axis_f64.onnx ONNX
primitives/lax/stop_gradient.onnx ONNX
primitives/lax/stop_gradient_basic.onnx ONNX
primitives/lax/stop_gradient_basic_f64.onnx ONNX
primitives/lax/stop_gradient_f64.onnx ONNX
primitives/lax/sub_complex64.onnx ONNX
primitives/lax/sub_complex64_f64.onnx ONNX
primitives/lax/sub_const.onnx ONNX
primitives/lax/sub_const_f64.onnx ONNX
primitives/lax/sub_test1.onnx ONNX
primitives/lax/sub_test1_f64.onnx ONNX
primitives/lax/sub_test2.onnx ONNX
primitives/lax/sub_test2_f64.onnx ONNX
primitives/lax/svd_1x1_default.onnx ONNX
primitives/lax/svd_1x1_default_f64.onnx ONNX
primitives/lax/svd_1x1_values_only.onnx ONNX
primitives/lax/svd_1x1_values_only_f64.onnx ONNX
primitives/lax/svd_2x2_values_only.onnx ONNX
primitives/lax/svd_2x2_values_only_f64.onnx ONNX
primitives/lax/svd_2x4_values_only.onnx ONNX
primitives/lax/svd_2x4_values_only_f64.onnx ONNX
primitives/lax/svd_3x2_values_only.onnx ONNX
primitives/lax/svd_3x2_values_only_f64.onnx ONNX
primitives/lax/symmetric_product_alpha_beta.onnx ONNX
primitives/lax/symmetric_product_alpha_beta_f64.onnx ONNX
primitives/lax/symmetric_product_default.onnx ONNX
primitives/lax/symmetric_product_default_f64.onnx ONNX
primitives/lax/tan.onnx ONNX
primitives/lax/tan_f64.onnx ONNX
primitives/lax/tanh.onnx ONNX
primitives/lax/tanh_f64.onnx ONNX
primitives/lax/top_k_last_axis.onnx ONNX
primitives/lax/top_k_last_axis_f64.onnx ONNX
primitives/lax/top_k_matrix.onnx ONNX
primitives/lax/top_k_matrix_f64.onnx ONNX
primitives/lax/transpose_3d.onnx ONNX
primitives/lax/transpose_3d_f64.onnx ONNX
primitives/lax/transpose_4d.onnx ONNX
primitives/lax/transpose_4d_f64.onnx ONNX
primitives/lax/transpose_basic.onnx ONNX
primitives/lax/transpose_basic_f64.onnx ONNX
primitives/lax/transpose_nhwc_to_nchw.onnx ONNX
primitives/lax/transpose_nhwc_to_nchw_f64.onnx ONNX
primitives/lax/transpose_no_axes.onnx ONNX
primitives/lax/transpose_no_axes_f64.onnx ONNX
primitives/lax/transpose_reverse.onnx ONNX
primitives/lax/transpose_reverse_f64.onnx ONNX
primitives/lax/transpose_square_matrix.onnx ONNX
primitives/lax/transpose_square_matrix_f64.onnx ONNX
primitives/lax/triangular_solve_batched_unit_diag.onnx ONNX
primitives/lax/triangular_solve_left_basic.onnx ONNX
primitives/lax/triangular_solve_left_basic_f64.onnx ONNX
primitives/lax/tridiagonal_2x2_lower_false.onnx ONNX
primitives/lax/tridiagonal_2x2_lower_false_f64.onnx ONNX
primitives/lax/tridiagonal_2x2_lower_true.onnx ONNX
primitives/lax/tridiagonal_2x2_lower_true_f64.onnx ONNX
primitives/lax/tridiagonal_solve_multi_rhs.onnx ONNX
primitives/lax/tridiagonal_solve_multi_rhs_f64.onnx ONNX
primitives/lax/tridiagonal_solve_single_rhs.onnx ONNX
primitives/lax/tridiagonal_solve_single_rhs_f64.onnx ONNX
primitives/lax/while_loop_4d_and_scalar_state.onnx ONNX
primitives/lax/while_loop_4d_and_scalar_state_f64.onnx ONNX
primitives/lax/while_loop_basic.onnx ONNX
primitives/lax/while_loop_captured_tracer.onnx ONNX
primitives/lax/while_loop_closure_topo.onnx ONNX
primitives/lax/while_loop_cnn_scalar_state_bug.onnx ONNX
primitives/lax/while_loop_cnn_scalar_state_bug_f64.onnx ONNX
primitives/lax/while_loop_counter.onnx ONNX
primitives/lax/while_loop_counter_f64.onnx ONNX
primitives/lax/while_loop_f64.onnx ONNX
primitives/lax/while_loop_mixed_rank.onnx ONNX
primitives/lax/while_loop_multi_state_f32.onnx ONNX
primitives/lax/while_loop_multi_state_f64.onnx ONNX
primitives/lax/while_loop_nnx_repro.onnx ONNX
primitives/lax/while_loop_nnx_repro_f64.onnx ONNX
primitives/lax/while_loop_no_loop_output_reused_as_input.onnx ONNX
primitives/lax/while_loop_renamed_passthrough.onnx ONNX
primitives/lax/while_loop_tracer_passthrough.onnx ONNX
primitives/lax/while_loop_two_state.onnx ONNX
primitives/lax/while_loop_vector.onnx ONNX
primitives/lax/while_loop_vector_f64.onnx ONNX
primitives/lax/while_loop_with_closure.onnx ONNX
primitives/lax/while_loop_with_closure_f64.onnx ONNX
primitives/lax/while_loop_with_scalar_state.onnx ONNX
primitives/lax/while_scalar_counter.onnx ONNX
primitives/lax/while_scalar_counter_f64.onnx ONNX
primitives/lax/while_tuple_state.onnx ONNX
primitives/lax/while_tuple_state_f64.onnx ONNX
primitives/lax/xor_bool_vec.onnx ONNX
primitives/lax/xor_bool_vec_f64.onnx ONNX
primitives/lax/xor_int_vec.onnx ONNX
primitives/lax/xor_int_vec_f64.onnx ONNX
primitives/lax/zeta_broadcast.onnx ONNX
primitives/lax/zeta_positive.onnx ONNX
primitives/linen/activation_glu_basic.onnx ONNX
primitives/linen/activation_glu_basic_f64.onnx ONNX
primitives/linen/activation_hard_sigmoid_basic.onnx ONNX
primitives/linen/activation_hard_sigmoid_basic_f64.onnx ONNX
primitives/linen/activation_hard_silu_basic.onnx ONNX
primitives/linen/activation_hard_silu_basic_f64.onnx ONNX
primitives/linen/activation_hard_swish_basic.onnx ONNX
primitives/linen/activation_hard_swish_basic_f64.onnx ONNX
primitives/linen/activation_hard_tanh_basic.onnx ONNX
primitives/linen/activation_hard_tanh_basic_f64.onnx ONNX
primitives/linen/activation_log_sigmoid_basic.onnx ONNX
primitives/linen/activation_log_sigmoid_basic_f64.onnx ONNX
primitives/linen/activation_log_softmax_basic.onnx ONNX
primitives/linen/activation_log_softmax_basic_f64.onnx ONNX
primitives/linen/activation_normalize_basic.onnx ONNX
primitives/linen/activation_normalize_basic_f64.onnx ONNX
primitives/linen/activation_one_hot_basic.onnx ONNX
primitives/linen/activation_one_hot_basic_f64.onnx ONNX
primitives/linen/activation_relu6_basic.onnx ONNX
primitives/linen/activation_relu6_basic_f64.onnx ONNX
primitives/linen/activation_silu_basic.onnx ONNX
primitives/linen/activation_silu_basic_f64.onnx ONNX
primitives/linen/activation_swish_basic.onnx ONNX
primitives/linen/activation_swish_basic_f64.onnx ONNX
primitives/linen/activation_tanh_basic.onnx ONNX
primitives/linen/activation_tanh_basic_f64.onnx ONNX
primitives/linen/avg_pool.onnx ONNX
primitives/linen/avg_pool_count_include_pad_false.onnx ONNX
primitives/linen/avg_pool_count_include_pad_false_dynamic.onnx ONNX
primitives/linen/avg_pool_default_padding.onnx ONNX
primitives/linen/avg_pool_default_padding_dynamic.onnx ONNX
primitives/linen/avg_pool_dynamic.onnx ONNX
primitives/linen/avg_pool_global_window.onnx ONNX
primitives/linen/avg_pool_global_window_dynamic.onnx ONNX
primitives/linen/avg_pool_same_padding.onnx ONNX
primitives/linen/avg_pool_same_padding_dynamic.onnx ONNX
primitives/linen/avg_pool_stride1.onnx ONNX
primitives/linen/avg_pool_stride1_dynamic.onnx ONNX
primitives/linen/avg_pool_stride_none.onnx ONNX
primitives/linen/avg_pool_stride_none_dynamic.onnx ONNX
primitives/linen/avg_pool_win3x3_stride2.onnx ONNX
primitives/linen/avg_pool_win3x3_stride2_dynamic.onnx ONNX
primitives/linen/batch_norm_3d.onnx ONNX
primitives/linen/batch_norm_3d_dynamic.onnx ONNX
primitives/linen/batch_norm_4d.onnx ONNX
primitives/linen/batch_norm_4d_dynamic.onnx ONNX
primitives/linen/batch_norm_4d_no_bias_no_scale.onnx ONNX
primitives/linen/batch_norm_4d_no_bias_no_scale_dynamic.onnx ONNX
primitives/linen/batch_norm_bias_no_scale.onnx ONNX
primitives/linen/batch_norm_bias_no_scale_dynamic.onnx ONNX
primitives/linen/batch_norm_bias_scale.onnx ONNX
primitives/linen/batch_norm_bias_scale_dynamic.onnx ONNX
primitives/linen/batch_norm_no_bias_no_scale.onnx ONNX
primitives/linen/batch_norm_no_bias_no_scale_dynamic.onnx ONNX
primitives/linen/batch_norm_no_bias_scale.onnx ONNX
primitives/linen/batch_norm_no_bias_scale_dynamic.onnx ONNX
primitives/linen/bidirectional_basic.onnx ONNX
primitives/linen/bidirectional_basic_dynamic.onnx ONNX
primitives/linen/bidirectional_basic_dynamic_f64.onnx ONNX
primitives/linen/bidirectional_basic_f64.onnx ONNX
primitives/linen/conv_basic.onnx ONNX
primitives/linen/conv_basic_dynamic.onnx ONNX
primitives/linen/conv_local_same.onnx ONNX
primitives/linen/conv_local_valid.onnx ONNX
primitives/linen/conv_lstm_cell_basic.onnx ONNX
primitives/linen/conv_lstm_cell_basic_dynamic.onnx ONNX
primitives/linen/conv_no_bias.onnx ONNX
primitives/linen/conv_stride.onnx ONNX
primitives/linen/conv_transpose_basic.onnx ONNX
primitives/linen/conv_transpose_basic_dynamic.onnx ONNX
primitives/linen/conv_transpose_valid_stride.onnx ONNX
primitives/linen/dense_basic.onnx ONNX
primitives/linen/dense_basic_dynamic.onnx ONNX
primitives/linen/dense_basic_dynamic_f64.onnx ONNX
primitives/linen/dense_basic_f64.onnx ONNX
primitives/linen/dense_general_basic.onnx ONNX
primitives/linen/dense_general_basic_dynamic.onnx ONNX
primitives/linen/dense_general_basic_dynamic_f64.onnx ONNX
primitives/linen/dense_general_basic_f64.onnx ONNX
primitives/linen/dense_general_contract_last_two.onnx ONNX
primitives/linen/dense_general_contract_last_two_f64.onnx ONNX
primitives/linen/dense_general_dynamic_batch_dynamic.onnx ONNX
primitives/linen/dense_general_dynamic_batch_dynamic_f64.onnx ONNX
primitives/linen/dense_general_multi_out.onnx ONNX
primitives/linen/dense_general_multi_out_f64.onnx ONNX
primitives/linen/dense_general_no_bias.onnx ONNX
primitives/linen/dense_general_no_bias_f64.onnx ONNX
primitives/linen/dense_high_rank_dynamic_dynamic.onnx ONNX
primitives/linen/dense_high_rank_dynamic_dynamic_f64.onnx ONNX
primitives/linen/dense_high_rank_no_bias.onnx ONNX
primitives/linen/dense_high_rank_no_bias_f64.onnx ONNX
primitives/linen/dense_high_rank_static.onnx ONNX
primitives/linen/dense_high_rank_static_f64.onnx ONNX
primitives/linen/dense_no_bias.onnx ONNX
primitives/linen/dense_no_bias_dynamic.onnx ONNX
primitives/linen/dense_no_bias_dynamic_f64.onnx ONNX
primitives/linen/dense_no_bias_f64.onnx ONNX
primitives/linen/dot_product_attention_basic.onnx ONNX
primitives/linen/dot_product_attention_basic_f64.onnx ONNX
primitives/linen/dot_product_attention_weights_basic.onnx ONNX
primitives/linen/dot_product_attention_weights_basic_f64.onnx ONNX
primitives/linen/dropout_call_params.onnx ONNX
primitives/linen/dropout_call_params_dynamic.onnx ONNX
primitives/linen/dropout_call_params_dynamic_f64.onnx ONNX
primitives/linen/dropout_call_params_f64.onnx ONNX
primitives/linen/dropout_init_params.onnx ONNX
primitives/linen/dropout_init_params_dynamic.onnx ONNX
primitives/linen/dropout_init_params_dynamic_f64.onnx ONNX
primitives/linen/dropout_init_params_f64.onnx ONNX
primitives/linen/einsum_no_bias.onnx ONNX
primitives/linen/einsum_no_bias_f64.onnx ONNX
primitives/linen/einsum_with_bias.onnx ONNX
primitives/linen/einsum_with_bias_f64.onnx ONNX
primitives/linen/group_norm_no_bias_no_scale.onnx ONNX
primitives/linen/group_norm_no_bias_no_scale_dynamic.onnx ONNX
primitives/linen/group_norm_rank2.onnx ONNX
primitives/linen/group_norm_rank2_dynamic.onnx ONNX
primitives/linen/group_norm_rank4.onnx ONNX
primitives/linen/gru_cell_basic.onnx ONNX
primitives/linen/gru_cell_basic_dynamic.onnx ONNX
primitives/linen/gru_cell_basic_dynamic_f64.onnx ONNX
primitives/linen/gru_cell_basic_f64.onnx ONNX
primitives/linen/instance_norm_rank2.onnx ONNX
primitives/linen/instance_norm_rank2_dynamic.onnx ONNX
primitives/linen/instance_norm_rank4.onnx ONNX
primitives/linen/instance_norm_rank4_dynamic.onnx ONNX
primitives/linen/layer_norm.onnx ONNX
primitives/linen/layer_norm_default_epsilon.onnx ONNX
primitives/linen/layer_norm_default_epsilon_dynamic.onnx ONNX
primitives/linen/layer_norm_dynamic.onnx ONNX
primitives/linen/layer_norm_multiaxis.onnx ONNX
primitives/linen/layer_norm_multiaxis_dynamic.onnx ONNX
primitives/linen/layer_norm_no_bias_no_scale.onnx ONNX
primitives/linen/layer_norm_no_bias_no_scale_dynamic.onnx ONNX
primitives/linen/linen_prelu_custom_slope.onnx ONNX
primitives/linen/linen_prelu_default.onnx ONNX
primitives/linen/linen_prelu_default_dynamic.onnx ONNX
primitives/linen/lstm_cell_basic.onnx ONNX
primitives/linen/lstm_cell_basic_dynamic.onnx ONNX
primitives/linen/lstm_cell_basic_dynamic_f64.onnx ONNX
primitives/linen/lstm_cell_basic_f64.onnx ONNX
primitives/linen/make_attention_mask_basic.onnx ONNX
primitives/linen/make_attention_mask_basic_f64.onnx ONNX
primitives/linen/make_causal_mask_basic.onnx ONNX
primitives/linen/make_causal_mask_basic_f64.onnx ONNX
primitives/linen/max_pool.onnx ONNX
primitives/linen/max_pool_basic.onnx ONNX
primitives/linen/max_pool_global_window.onnx ONNX
primitives/linen/max_pool_global_window_dynamic.onnx ONNX
primitives/linen/max_pool_same.onnx ONNX
primitives/linen/max_pool_same_dynamic.onnx ONNX
primitives/linen/max_pool_same_padding.onnx ONNX
primitives/linen/mgu_cell_basic.onnx ONNX
primitives/linen/mgu_cell_basic_dynamic.onnx ONNX
primitives/linen/mgu_cell_basic_dynamic_f64.onnx ONNX
primitives/linen/mgu_cell_basic_f64.onnx ONNX
primitives/linen/min_pool.onnx ONNX
primitives/linen/min_pool_basic.onnx ONNX
primitives/linen/min_pool_same.onnx ONNX
primitives/linen/min_pool_same_dynamic.onnx ONNX
primitives/linen/min_pool_same_padding.onnx ONNX
primitives/linen/multi_head_attention_basic.onnx ONNX
primitives/linen/multi_head_attention_basic_dynamic.onnx ONNX
primitives/linen/multi_head_attention_basic_dynamic_f64.onnx ONNX
primitives/linen/multi_head_attention_basic_f64.onnx ONNX
primitives/linen/multi_head_attention_no_bias.onnx ONNX
primitives/linen/multi_head_attention_no_bias_dynamic.onnx ONNX
primitives/linen/multi_head_attention_no_bias_dynamic_f64.onnx ONNX
primitives/linen/multi_head_attention_no_bias_f64.onnx ONNX
primitives/linen/multi_head_dot_product_attention_basic.onnx ONNX
primitives/linen/multi_head_dot_product_attention_basic_dynamic.onnx ONNX
primitives/linen/multi_head_dot_product_attention_basic_dynamic_f64.onnx ONNX
primitives/linen/multi_head_dot_product_attention_basic_f64.onnx ONNX
primitives/linen/multi_head_dot_product_attention_no_bias.onnx ONNX
primitives/linen/multi_head_dot_product_attention_no_bias_dynamic.onnx ONNX
primitives/linen/multi_head_dot_product_attention_no_bias_dynamic_f64.onnx ONNX
primitives/linen/multi_head_dot_product_attention_no_bias_f64.onnx ONNX
primitives/linen/optimized_lstm_cell_basic.onnx ONNX
primitives/linen/optimized_lstm_cell_basic_dynamic.onnx ONNX
primitives/linen/optimized_lstm_cell_basic_dynamic_f64.onnx ONNX
primitives/linen/optimized_lstm_cell_basic_f64.onnx ONNX
primitives/linen/pool_max_basic.onnx ONNX
primitives/linen/pool_max_basic_dynamic.onnx ONNX
primitives/linen/pool_min_basic.onnx ONNX
primitives/linen/pool_min_basic_dynamic.onnx ONNX
primitives/linen/pool_sum_basic.onnx ONNX
primitives/linen/pool_sum_basic_dynamic.onnx ONNX
primitives/linen/positional_embedding.onnx ONNX
primitives/linen/positional_embedding_dynamic.onnx ONNX
primitives/linen/positional_embedding_dynamic_f64.onnx ONNX
primitives/linen/positional_embedding_f64.onnx ONNX
primitives/linen/rms_norm_4d_dynamic.onnx ONNX
primitives/linen/rms_norm_4d_dynamic_dynamic.onnx ONNX
primitives/linen/rms_norm_basic.onnx ONNX
primitives/linen/rms_norm_use_scale_false.onnx ONNX
primitives/linen/rnn_basic.onnx ONNX
primitives/linen/rnn_basic_dynamic.onnx ONNX
primitives/linen/rnn_basic_dynamic_f64.onnx ONNX
primitives/linen/rnn_basic_f64.onnx ONNX
primitives/linen/self_attention_basic.onnx ONNX
primitives/linen/self_attention_basic_dynamic.onnx ONNX
primitives/linen/self_attention_basic_dynamic_f64.onnx ONNX
primitives/linen/self_attention_basic_f64.onnx ONNX
primitives/linen/self_attention_no_bias.onnx ONNX
primitives/linen/self_attention_no_bias_dynamic.onnx ONNX
primitives/linen/self_attention_no_bias_dynamic_f64.onnx ONNX
primitives/linen/self_attention_no_bias_f64.onnx ONNX
primitives/linen/simple_cell_basic.onnx ONNX
primitives/linen/simple_cell_basic_dynamic.onnx ONNX
primitives/linen/simple_cell_basic_dynamic_f64.onnx ONNX
primitives/linen/simple_cell_basic_f64.onnx ONNX
primitives/linen/spectral_norm_dense.onnx ONNX
primitives/linen/spectral_norm_dense_dynamic.onnx ONNX
primitives/linen/token_embedding.onnx ONNX
primitives/linen/token_embedding_dynamic.onnx ONNX
primitives/linen/token_embedding_dynamic_f64.onnx ONNX
primitives/linen/token_embedding_f64.onnx ONNX
primitives/linen/weight_norm_dense.onnx ONNX
primitives/linen/weight_norm_dense_dynamic.onnx ONNX
primitives/nn/celu_grad_issue_batch_diff_rules.onnx ONNX
primitives/nn/dpa_axis1.onnx ONNX
primitives/nn/dpa_axis1_f64.onnx ONNX
primitives/nn/dpa_basic.onnx ONNX
primitives/nn/dpa_basic_f64.onnx ONNX
primitives/nn/dpa_batch1_seq2.onnx ONNX
primitives/nn/dpa_batch1_seq2_f64.onnx ONNX
primitives/nn/dpa_batch4_seq16.onnx ONNX
primitives/nn/dpa_batch4_seq16_f64.onnx ONNX
primitives/nn/dpa_batch8_seq4.onnx ONNX
primitives/nn/dpa_batch8_seq4_f64.onnx ONNX
primitives/nn/dpa_diff_heads_embed.onnx ONNX
primitives/nn/dpa_diff_heads_embed_f64.onnx ONNX
primitives/nn/dpa_float64.onnx ONNX
primitives/nn/dpa_heads1_embed4.onnx ONNX
primitives/nn/dpa_heads1_embed4_f64.onnx ONNX
primitives/nn/dpa_heads8_embed8.onnx ONNX
primitives/nn/dpa_heads8_embed8_f64.onnx ONNX
primitives/nn/dpa_mask_none.onnx ONNX
primitives/nn/dpa_mostly_false.onnx ONNX
primitives/nn/dpa_mostly_false_f64.onnx ONNX
primitives/nn/dpa_one_false.onnx ONNX
primitives/nn/dpa_one_false_f64.onnx ONNX
primitives/nn/dpa_positional_bias_mask.onnx ONNX
primitives/nn/dpa_positional_bias_mask_f64.onnx ONNX
primitives/nn/dpa_tiny_mask_all_valid.onnx ONNX
primitives/nn/dpa_tiny_mask_all_valid_f64.onnx ONNX
primitives/nn/dpa_tiny_mask_mixed.onnx ONNX
primitives/nn/dpa_tiny_mask_mixed_f64.onnx ONNX
primitives/nn/dpa_vmap_tnh_issue190.onnx ONNX
primitives/nn/dpa_with_causal_mask.onnx ONNX
primitives/nn/dpa_with_causal_mask_f64.onnx ONNX
primitives/nn/dpa_with_local_window_mask.onnx ONNX
primitives/nn/dpa_with_local_window_mask_f64.onnx ONNX
primitives/nn/dpa_with_padding_mask.onnx ONNX
primitives/nn/dpa_with_padding_mask_f64.onnx ONNX
primitives/nn/dpa_with_tensor_mask.onnx ONNX
primitives/nn/dpa_with_tensor_mask_f64.onnx ONNX
primitives/nn/elu_grad_issue_batch_diff_rules.onnx ONNX
primitives/nn/flax_dense_like_init.onnx ONNX
primitives/nn/gelu_grad_issue_batch_diff_rules.onnx ONNX
primitives/nn/initializer.onnx ONNX
primitives/nn/jaxnn_celu.onnx ONNX
primitives/nn/jaxnn_celu_1.onnx ONNX
primitives/nn/jaxnn_celu_alpha_custom.onnx ONNX
primitives/nn/jaxnn_celu_alpha_custom_dynamic.onnx ONNX
primitives/nn/jaxnn_celu_alpha_default.onnx ONNX
primitives/nn/jaxnn_elu.onnx ONNX
primitives/nn/jaxnn_elu_1.onnx ONNX
primitives/nn/jaxnn_elu_custom_alpha.onnx ONNX
primitives/nn/jaxnn_elu_custom_alpha_dynamic.onnx ONNX
primitives/nn/jaxnn_elu_default.onnx ONNX
primitives/nn/jaxnn_gelu.onnx ONNX
primitives/nn/jaxnn_gelu_1.onnx ONNX
primitives/nn/jaxnn_gelu_approx.onnx ONNX
primitives/nn/jaxnn_gelu_exact.onnx ONNX
primitives/nn/jaxnn_gelu_tanh.onnx ONNX
primitives/nn/jaxnn_gelu_tanh_dynamic.onnx ONNX
primitives/nn/jaxnn_glu_axis1_dynamic.onnx ONNX
primitives/nn/jaxnn_glu_axis1_dynamic_dynamic.onnx ONNX
primitives/nn/jaxnn_glu_axis1_dynamic_dynamic_f64.onnx ONNX
primitives/nn/jaxnn_glu_axis1_dynamic_f64.onnx ONNX
primitives/nn/jaxnn_glu_axis_last.onnx ONNX
primitives/nn/jaxnn_glu_axis_last_f64.onnx ONNX
primitives/nn/jaxnn_hard_sigmoid.onnx ONNX
primitives/nn/jaxnn_hard_sigmoid_dynamic.onnx ONNX
primitives/nn/jaxnn_hard_sigmoid_dynamic_dynamic.onnx ONNX
primitives/nn/jaxnn_hard_sigmoid_dynamic_dynamic_f64.onnx ONNX
primitives/nn/jaxnn_hard_sigmoid_dynamic_f64.onnx ONNX
primitives/nn/jaxnn_hard_sigmoid_f64.onnx ONNX
primitives/nn/jaxnn_hard_swish.onnx ONNX
primitives/nn/jaxnn_hard_swish_dynamic.onnx ONNX
primitives/nn/jaxnn_hard_swish_dynamic_dynamic.onnx ONNX
primitives/nn/jaxnn_hard_swish_dynamic_dynamic_f64.onnx ONNX
primitives/nn/jaxnn_hard_swish_dynamic_f64.onnx ONNX
primitives/nn/jaxnn_hard_swish_f64.onnx ONNX
primitives/nn/jaxnn_hard_tanh.onnx ONNX
primitives/nn/jaxnn_hard_tanh_dynamic.onnx ONNX
primitives/nn/jaxnn_hard_tanh_dynamic_dynamic.onnx ONNX
primitives/nn/jaxnn_hard_tanh_dynamic_dynamic_f64.onnx ONNX
primitives/nn/jaxnn_hard_tanh_dynamic_f64.onnx ONNX
primitives/nn/jaxnn_hard_tanh_f64.onnx ONNX
primitives/nn/jaxnn_hardmax_axis0.onnx ONNX
primitives/nn/jaxnn_hardmax_default_axis.onnx ONNX
primitives/nn/jaxnn_hardmax_dynamic.onnx ONNX
primitives/nn/jaxnn_hardmax_dynamic_dynamic.onnx ONNX
primitives/nn/jaxnn_hardmax_vmap_batching.onnx ONNX
primitives/nn/jaxnn_identity.onnx ONNX
primitives/nn/jaxnn_identity_1.onnx ONNX
primitives/nn/jaxnn_identity_1_f64.onnx ONNX
primitives/nn/jaxnn_identity_basic.onnx ONNX
primitives/nn/jaxnn_identity_basic_f64.onnx ONNX
primitives/nn/jaxnn_identity_dynamic.onnx ONNX
primitives/nn/jaxnn_identity_dynamic_dynamic.onnx ONNX
primitives/nn/jaxnn_identity_dynamic_dynamic_f64.onnx ONNX
primitives/nn/jaxnn_identity_dynamic_f64.onnx ONNX
primitives/nn/jaxnn_identity_f64.onnx ONNX
primitives/nn/jaxnn_leaky_relu.onnx ONNX
primitives/nn/jaxnn_leaky_relu_1.onnx ONNX
primitives/nn/jaxnn_leaky_relu_custom.onnx ONNX
primitives/nn/jaxnn_leaky_relu_default.onnx ONNX
primitives/nn/jaxnn_leaky_relu_default_dynamic.onnx ONNX
primitives/nn/jaxnn_log1mexp_basic.onnx ONNX
primitives/nn/jaxnn_log1mexp_basic_f64.onnx ONNX
primitives/nn/jaxnn_log1mexp_dynamic.onnx ONNX
primitives/nn/jaxnn_log1mexp_dynamic_dynamic.onnx ONNX
primitives/nn/jaxnn_log1mexp_dynamic_dynamic_f64.onnx ONNX
primitives/nn/jaxnn_log1mexp_dynamic_f64.onnx ONNX
primitives/nn/jaxnn_log_sigmoid.onnx ONNX
primitives/nn/jaxnn_log_sigmoid_dynamic.onnx ONNX
primitives/nn/jaxnn_log_sigmoid_dynamic_dynamic.onnx ONNX
primitives/nn/jaxnn_log_sigmoid_dynamic_dynamic_f64.onnx ONNX
primitives/nn/jaxnn_log_sigmoid_dynamic_f64.onnx ONNX
primitives/nn/jaxnn_log_sigmoid_f64.onnx ONNX
primitives/nn/jaxnn_log_softmax_axis0.onnx ONNX
primitives/nn/jaxnn_log_softmax_axis0_f64.onnx ONNX
primitives/nn/jaxnn_log_softmax_axis_last_3d.onnx ONNX
primitives/nn/jaxnn_log_softmax_axis_last_3d_f64.onnx ONNX
primitives/nn/jaxnn_log_softmax_default_axis.onnx ONNX
primitives/nn/jaxnn_log_softmax_default_axis_dynamic.onnx ONNX
primitives/nn/jaxnn_log_softmax_default_axis_dynamic_f64.onnx ONNX
primitives/nn/jaxnn_log_softmax_default_axis_f64.onnx ONNX
primitives/nn/jaxnn_logmeanexp_axis1.onnx ONNX
primitives/nn/jaxnn_logmeanexp_axis12_keepdims.onnx ONNX
primitives/nn/jaxnn_logmeanexp_axis12_keepdims_f64.onnx ONNX
primitives/nn/jaxnn_logmeanexp_axis1_f64.onnx ONNX
primitives/nn/jaxnn_logmeanexp_dynamic_static_reduction_axis.onnx ONNX
primitives/nn/jaxnn_logmeanexp_dynamic_static_reduction_axis_dynamic.onnx ONNX
primitives/nn/jaxnn_logmeanexp_dynamic_static_reduction_axis_dynamic_f64.onnx ONNX
primitives/nn/jaxnn_logmeanexp_dynamic_static_reduction_axis_f64.onnx ONNX
primitives/nn/jaxnn_logsumexp_axis1.onnx ONNX
primitives/nn/jaxnn_logsumexp_axis12_keepdims.onnx ONNX
primitives/nn/jaxnn_logsumexp_axis12_keepdims_f64.onnx ONNX
primitives/nn/jaxnn_logsumexp_axis1_f64.onnx ONNX
primitives/nn/jaxnn_mish.onnx ONNX
primitives/nn/jaxnn_mish_1.onnx ONNX
primitives/nn/jaxnn_mish_basic.onnx ONNX
primitives/nn/jaxnn_one_hot_axis0.onnx ONNX
primitives/nn/jaxnn_one_hot_axis0_f64.onnx ONNX
primitives/nn/jaxnn_one_hot_default.onnx ONNX
primitives/nn/jaxnn_one_hot_default_f64.onnx ONNX
primitives/nn/jaxnn_one_hot_float16.onnx ONNX
primitives/nn/jaxnn_relu.onnx ONNX
primitives/nn/jaxnn_relu6.onnx ONNX
primitives/nn/jaxnn_relu6_dynamic.onnx ONNX
primitives/nn/jaxnn_relu6_dynamic_dynamic.onnx ONNX
primitives/nn/jaxnn_relu6_dynamic_dynamic_f64.onnx ONNX
primitives/nn/jaxnn_relu6_dynamic_f64.onnx ONNX
primitives/nn/jaxnn_relu6_f64.onnx ONNX
primitives/nn/jaxnn_relu_1.onnx ONNX
primitives/nn/jaxnn_relu_1_f64.onnx ONNX
primitives/nn/jaxnn_relu_basic.onnx ONNX
primitives/nn/jaxnn_relu_basic_f64.onnx ONNX
primitives/nn/jaxnn_relu_dynamic.onnx ONNX
primitives/nn/jaxnn_relu_dynamic_dynamic.onnx ONNX
primitives/nn/jaxnn_relu_dynamic_dynamic_f64.onnx ONNX
primitives/nn/jaxnn_relu_dynamic_f64.onnx ONNX
primitives/nn/jaxnn_relu_f64.onnx ONNX
primitives/nn/jaxnn_scaled_dot_general_basic.onnx ONNX
primitives/nn/jaxnn_scaled_dot_general_basic_f64.onnx ONNX
primitives/nn/jaxnn_scaled_dot_general_batched.onnx ONNX
primitives/nn/jaxnn_scaled_dot_general_batched_f64.onnx ONNX
primitives/nn/jaxnn_scaled_matmul_basic.onnx ONNX
primitives/nn/jaxnn_scaled_matmul_basic_f64.onnx ONNX
primitives/nn/jaxnn_selu.onnx ONNX
primitives/nn/jaxnn_selu_1.onnx ONNX
primitives/nn/jaxnn_selu_basic.onnx ONNX
primitives/nn/jaxnn_selu_basic_dynamic.onnx ONNX
primitives/nn/jaxnn_sigmoid.onnx ONNX
primitives/nn/jaxnn_sigmoid_1.onnx ONNX
primitives/nn/jaxnn_sigmoid_1_f64.onnx ONNX
primitives/nn/jaxnn_sigmoid_f64.onnx ONNX
primitives/nn/jaxnn_silu.onnx ONNX
primitives/nn/jaxnn_silu_dynamic.onnx ONNX
primitives/nn/jaxnn_silu_dynamic_dynamic.onnx ONNX
primitives/nn/jaxnn_silu_dynamic_dynamic_f64.onnx ONNX
primitives/nn/jaxnn_silu_dynamic_f64.onnx ONNX
primitives/nn/jaxnn_silu_f64.onnx ONNX
primitives/nn/jaxnn_soft_sign.onnx ONNX
primitives/nn/jaxnn_soft_sign_1.onnx ONNX
primitives/nn/jaxnn_softplus.onnx ONNX
primitives/nn/jaxnn_softplus_1.onnx ONNX
primitives/nn/jaxnn_softplus_basic.onnx ONNX
primitives/nn/jaxnn_softsign_basic.onnx ONNX
primitives/nn/jaxnn_sparse_plus.onnx ONNX
primitives/nn/jaxnn_sparse_plus_dynamic.onnx ONNX
primitives/nn/jaxnn_sparse_plus_dynamic_dynamic.onnx ONNX
primitives/nn/jaxnn_sparse_plus_dynamic_dynamic_f64.onnx ONNX
primitives/nn/jaxnn_sparse_plus_dynamic_f64.onnx ONNX
primitives/nn/jaxnn_sparse_plus_f64.onnx ONNX
primitives/nn/jaxnn_sparse_sigmoid.onnx ONNX
primitives/nn/jaxnn_sparse_sigmoid_dynamic.onnx ONNX
primitives/nn/jaxnn_sparse_sigmoid_dynamic_dynamic.onnx ONNX
primitives/nn/jaxnn_sparse_sigmoid_dynamic_dynamic_f64.onnx ONNX
primitives/nn/jaxnn_sparse_sigmoid_dynamic_f64.onnx ONNX
primitives/nn/jaxnn_sparse_sigmoid_f64.onnx ONNX
primitives/nn/jaxnn_squareplus_broadcast_b.onnx ONNX
primitives/nn/jaxnn_squareplus_broadcast_b_f64.onnx ONNX
primitives/nn/jaxnn_squareplus_default_b.onnx ONNX
primitives/nn/jaxnn_squareplus_default_b_f64.onnx ONNX
primitives/nn/jaxnn_standardize_axis_last_eps0.onnx ONNX
primitives/nn/jaxnn_standardize_axis_tuple_eps0.onnx ONNX
primitives/nn/jaxnn_standardize_axis_tuple_eps0_dynamic.onnx ONNX
primitives/nn/jaxnn_swish.onnx ONNX
primitives/nn/jaxnn_swish_alias.onnx ONNX
primitives/nn/jaxnn_swish_alias_f64.onnx ONNX
primitives/nn/jaxnn_swish_dynamic.onnx ONNX
primitives/nn/jaxnn_swish_dynamic_dynamic.onnx ONNX
primitives/nn/jaxnn_tanh.onnx ONNX
primitives/nn/jaxnn_tanh_dynamic.onnx ONNX
primitives/nn/jaxnn_tanh_dynamic_dynamic.onnx ONNX
primitives/nn/jaxnn_tanh_dynamic_dynamic_f64.onnx ONNX
primitives/nn/jaxnn_tanh_dynamic_f64.onnx ONNX
primitives/nn/jaxnn_tanh_f64.onnx ONNX
primitives/nn/jaxnn_thresholded_relu_custom.onnx ONNX
primitives/nn/jaxnn_thresholded_relu_custom_dynamic.onnx ONNX
primitives/nn/jaxnn_thresholded_relu_default.onnx ONNX
primitives/nn/jaxscipy_logsumexp_axis_last.onnx ONNX
primitives/nn/jaxscipy_logsumexp_axis_last_dynamic.onnx ONNX
primitives/nn/jaxscipy_logsumexp_axis_last_dynamic_f64.onnx ONNX
primitives/nn/jaxscipy_logsumexp_axis_last_f64.onnx ONNX
primitives/nn/leaky_relu_grad_issue_batch_diff_rules.onnx ONNX
primitives/nn/log_softmax_grad_issue_batch_diff_rules.onnx ONNX
primitives/nn/logsumexp_grad_issue_batch_diff_rules.onnx ONNX
primitives/nn/mish_grad_issue_batch_diff_rules.onnx ONNX
primitives/nn/random_truncated_normal_positional.onnx ONNX
primitives/nn/relu_grad_issue_batch_diff_rules.onnx ONNX
primitives/nn/selu_grad_issue_batch_diff_rules.onnx ONNX
primitives/nn/sigmoid_grad_issue_batch_diff_rules.onnx ONNX
primitives/nn/silu_grad_issue_batch_diff_rules.onnx ONNX
primitives/nn/silu_opset24_smoke.onnx ONNX
primitives/nn/softmax.onnx ONNX
primitives/nn/softmax_2d.onnx ONNX
primitives/nn/softmax_2d_f64.onnx ONNX
primitives/nn/softmax_3d.onnx ONNX
primitives/nn/softmax_3d_f64.onnx ONNX
primitives/nn/softmax_f64.onnx ONNX
primitives/nn/softmax_grad_issue_batch_diff_rules.onnx ONNX
primitives/nn/softmax_mask_where.onnx ONNX
primitives/nn/softmax_mask_where_f64.onnx ONNX
primitives/nn/softplus_grad_issue_batch_diff_rules.onnx ONNX
primitives/nn/softsign_grad_issue_batch_diff_rules.onnx ONNX
primitives/nn/standardize_grad_issue_batch_diff_rules.onnx ONNX
primitives/nnx/avg_pool.onnx ONNX
primitives/nnx/avg_pool_count_include_pad_false.onnx ONNX
primitives/nnx/avg_pool_count_include_pad_false_dynamic.onnx ONNX
primitives/nnx/avg_pool_default_padding.onnx ONNX
primitives/nnx/avg_pool_default_padding_dynamic.onnx ONNX
primitives/nnx/avg_pool_dynamic.onnx ONNX
primitives/nnx/avg_pool_global_window.onnx ONNX
primitives/nnx/avg_pool_global_window_dynamic.onnx ONNX
primitives/nnx/avg_pool_same_padding.onnx ONNX
primitives/nnx/avg_pool_same_padding_dynamic.onnx ONNX
primitives/nnx/avg_pool_stride1.onnx ONNX
primitives/nnx/avg_pool_stride1_dynamic.onnx ONNX
primitives/nnx/avg_pool_stride_none.onnx ONNX
primitives/nnx/avg_pool_stride_none_dynamic.onnx ONNX
primitives/nnx/avg_pool_win3x3_stride2.onnx ONNX
primitives/nnx/avg_pool_win3x3_stride2_dynamic.onnx ONNX
primitives/nnx/batch_norm_3d.onnx ONNX
primitives/nnx/batch_norm_3d_dynamic.onnx ONNX
primitives/nnx/batch_norm_4d.onnx ONNX
primitives/nnx/batch_norm_4d_dynamic.onnx ONNX
primitives/nnx/batch_norm_4d_no_bias_no_scale.onnx ONNX
primitives/nnx/batch_norm_4d_no_bias_no_scale_dynamic.onnx ONNX
primitives/nnx/batch_norm_bias_no_scale.onnx ONNX
primitives/nnx/batch_norm_bias_no_scale_dynamic.onnx ONNX
primitives/nnx/batch_norm_bias_scale.onnx ONNX
primitives/nnx/batch_norm_bias_scale_dynamic.onnx ONNX
primitives/nnx/batch_norm_no_bias_no_scale.onnx ONNX
primitives/nnx/batch_norm_no_bias_no_scale_dynamic.onnx ONNX
primitives/nnx/batch_norm_no_bias_scale.onnx ONNX
primitives/nnx/batch_norm_no_bias_scale_dynamic.onnx ONNX
primitives/nnx/combine_masks_two.onnx ONNX
primitives/nnx/combine_masks_with_none.onnx ONNX
primitives/nnx/conv_1d.onnx ONNX
primitives/nnx/conv_1d_complex_on_4d.onnx ONNX
primitives/nnx/conv_1d_dilation.onnx ONNX
primitives/nnx/conv_1d_group_conv_more_dims.onnx ONNX
primitives/nnx/conv_1d_group_on_higher_dim.onnx ONNX
primitives/nnx/conv_1d_high_dilation_on_3d.onnx ONNX
primitives/nnx/conv_1d_kernel_1.onnx ONNX
primitives/nnx/conv_1d_large_kernel.onnx ONNX
primitives/nnx/conv_1d_large_kernel_on_4d.onnx ONNX
primitives/nnx/conv_1d_more_1d_inputs.onnx ONNX
primitives/nnx/conv_1d_more_2d_inputs.onnx ONNX
primitives/nnx/conv_1d_same_padding_on_3d.onnx ONNX
primitives/nnx/conv_1d_stride_dilation.onnx ONNX
primitives/nnx/conv_1d_unit_group_on_multi_dim.onnx ONNX
primitives/nnx/conv_1d_wide_input.onnx ONNX
primitives/nnx/conv_2d_asymmetric_dilation.onnx ONNX
primitives/nnx/conv_2d_asymmetric_kernel.onnx ONNX
primitives/nnx/conv_2d_asymmetric_on_5d.onnx ONNX
primitives/nnx/conv_2d_asymmetric_stride.onnx ONNX
primitives/nnx/conv_2d_complex_on_5d.onnx ONNX
primitives/nnx/conv_2d_depthwise.onnx ONNX
primitives/nnx/conv_2d_group_conv.onnx ONNX
primitives/nnx/conv_2d_group_stride_dilation.onnx ONNX
primitives/nnx/conv_2d_kernel_1x1.onnx ONNX
primitives/nnx/conv_2d_large_dilation.onnx ONNX
primitives/nnx/conv_2d_large_stride.onnx ONNX
primitives/nnx/conv_2d_many_channels.onnx ONNX
primitives/nnx/conv_2d_mixed_params.onnx ONNX
primitives/nnx/conv_2d_same_padding_mixed_dilation.onnx ONNX
primitives/nnx/conv_2d_small_input.onnx ONNX
primitives/nnx/conv_3d_asymmetric.onnx ONNX
primitives/nnx/conv_3d_basic.onnx ONNX
primitives/nnx/conv_3d_dilation.onnx ONNX
primitives/nnx/conv_3d_group_complex.onnx ONNX
primitives/nnx/conv_3d_stride.onnx ONNX
primitives/nnx/conv_basic_bias.onnx ONNX
primitives/nnx/conv_basic_bias_2.onnx ONNX
primitives/nnx/conv_basic_bias_3.onnx ONNX
primitives/nnx/conv_basic_bias_dynamic.onnx ONNX
primitives/nnx/conv_different_kernel.onnx ONNX
primitives/nnx/conv_float64.onnx ONNX
primitives/nnx/conv_large_batch.onnx ONNX
primitives/nnx/conv_no_bias.onnx ONNX
primitives/nnx/conv_no_bias_dynamic.onnx ONNX
primitives/nnx/conv_single_batch.onnx ONNX
primitives/nnx/conv_stride1.onnx ONNX
primitives/nnx/conv_stride2.onnx ONNX
primitives/nnx/conv_stride2_bias.onnx ONNX
primitives/nnx/conv_valid_padding.onnx ONNX
primitives/nnx/dpa_basic.onnx ONNX
primitives/nnx/dpa_basic_f64.onnx ONNX
primitives/nnx/dpa_with_bias.onnx ONNX
primitives/nnx/dpa_with_causal_mask.onnx ONNX
primitives/nnx/dpa_with_causal_mask_f64.onnx ONNX
primitives/nnx/dpa_with_mask_and_bias.onnx ONNX
primitives/nnx/dpa_with_tensor_mask.onnx ONNX
primitives/nnx/dropout_call_params.onnx ONNX
primitives/nnx/dropout_call_params_dynamic.onnx ONNX
primitives/nnx/dropout_call_params_dynamic_f64.onnx ONNX
primitives/nnx/dropout_call_params_f64.onnx ONNX
primitives/nnx/dropout_init_params.onnx ONNX
primitives/nnx/dropout_init_params_dynamic.onnx ONNX
primitives/nnx/dropout_init_params_dynamic_f64.onnx ONNX
primitives/nnx/dropout_init_params_f64.onnx ONNX
primitives/nnx/dynamic_batch_and_feature_dims_dynamic.onnx ONNX
primitives/nnx/einsum_module_no_bias.onnx ONNX
primitives/nnx/einsum_module_no_bias_f64.onnx ONNX
primitives/nnx/einsum_module_with_bias.onnx ONNX
primitives/nnx/einsum_module_with_bias_f64.onnx ONNX
primitives/nnx/elu.onnx ONNX
primitives/nnx/elu_alpha.onnx ONNX
primitives/nnx/elu_default.onnx ONNX
primitives/nnx/elu_default_dynamic.onnx ONNX
primitives/nnx/flip_sequences_batch_major_no_lengths.onnx ONNX
primitives/nnx/flip_sequences_batch_major_no_lengths_dynamic.onnx ONNX
primitives/nnx/flip_sequences_batch_major_no_lengths_dynamic_f64.onnx ONNX
primitives/nnx/flip_sequences_batch_major_no_lengths_f64.onnx ONNX
primitives/nnx/flip_sequences_batch_major_with_lengths.onnx ONNX
primitives/nnx/flip_sequences_batch_major_with_lengths_f64.onnx ONNX
primitives/nnx/flip_sequences_time_major_with_lengths.onnx ONNX
primitives/nnx/flip_sequences_time_major_with_lengths_f64.onnx ONNX
primitives/nnx/gelu.onnx ONNX
primitives/nnx/gelu_1.onnx ONNX
primitives/nnx/gelu_2.onnx ONNX
primitives/nnx/gelu_2_f64.onnx ONNX
primitives/nnx/gelu_3.onnx ONNX
primitives/nnx/gelu_3_dynamic.onnx ONNX
primitives/nnx/gelu_3_dynamic_f64.onnx ONNX
primitives/nnx/gelu_3_f64.onnx ONNX
primitives/nnx/gelu_4.onnx ONNX
primitives/nnx/gelu_4_f64.onnx ONNX
primitives/nnx/gelu_5.onnx ONNX
primitives/nnx/gelu_5_dynamic.onnx ONNX
primitives/nnx/gelu_5_dynamic_f64.onnx ONNX
primitives/nnx/gelu_5_f64.onnx ONNX
primitives/nnx/glu_axis_1.onnx ONNX
primitives/nnx/glu_axis_1_dynamic.onnx ONNX
primitives/nnx/glu_axis_1_dynamic_f64.onnx ONNX
primitives/nnx/glu_axis_1_f64.onnx ONNX
primitives/nnx/glu_last_axis.onnx ONNX
primitives/nnx/glu_last_axis_f64.onnx ONNX
primitives/nnx/group_norm.onnx ONNX
primitives/nnx/group_norm_bias_no_scale.onnx ONNX
primitives/nnx/group_norm_bias_no_scale_dynamic.onnx ONNX
primitives/nnx/group_norm_bias_scale.onnx ONNX
primitives/nnx/group_norm_bias_scale_dynamic.onnx ONNX
primitives/nnx/group_norm_no_bias.onnx ONNX
primitives/nnx/group_norm_no_bias_dynamic.onnx ONNX
primitives/nnx/group_norm_no_bias_no_scale.onnx ONNX
primitives/nnx/group_norm_no_bias_no_scale_dynamic.onnx ONNX
primitives/nnx/group_norm_no_bias_scale.onnx ONNX
primitives/nnx/group_norm_no_bias_scale_dynamic.onnx ONNX
primitives/nnx/group_norm_no_scale.onnx ONNX
primitives/nnx/group_norm_no_scale_dynamic.onnx ONNX
primitives/nnx/group_norm_rank2.onnx ONNX
primitives/nnx/group_norm_rank2_dynamic.onnx ONNX
primitives/nnx/group_norm_rank4.onnx ONNX
primitives/nnx/hard_tanh_basic.onnx ONNX
primitives/nnx/hard_tanh_basic_dynamic.onnx ONNX
primitives/nnx/hard_tanh_basic_dynamic_f64.onnx ONNX
primitives/nnx/hard_tanh_basic_f64.onnx ONNX
primitives/nnx/layer_norm.onnx ONNX
primitives/nnx/layer_norm_bias_no_scale.onnx ONNX
primitives/nnx/layer_norm_bias_no_scale_dynamic.onnx ONNX
primitives/nnx/layer_norm_bias_scale.onnx ONNX
primitives/nnx/layer_norm_bias_scale_dynamic.onnx ONNX
primitives/nnx/layer_norm_dynamic.onnx ONNX
primitives/nnx/layer_norm_multiaxis.onnx ONNX
primitives/nnx/layer_norm_multiaxis_dynamic.onnx ONNX
primitives/nnx/layer_norm_negative_axis_no_div.onnx ONNX
primitives/nnx/layer_norm_negative_axis_no_div_dynamic.onnx ONNX
primitives/nnx/layer_norm_no_bias_no_scale.onnx ONNX
primitives/nnx/layer_norm_no_bias_no_scale_dynamic.onnx ONNX
primitives/nnx/layer_norm_no_bias_scale.onnx ONNX
primitives/nnx/layer_norm_no_bias_scale_dynamic.onnx ONNX
primitives/nnx/layer_norm_symbolic_batch.onnx ONNX
primitives/nnx/layer_norm_symbolic_batch_dynamic.onnx ONNX
primitives/nnx/layer_norm_symbolic_batch_seq10_feat3.onnx ONNX
primitives/nnx/layer_norm_symbolic_batch_seq10_feat3_2.onnx ONNX
primitives/nnx/layer_norm_symbolic_batch_seq10_feat3_2_dynamic.onnx ONNX
primitives/nnx/layer_norm_symbolic_batch_seq10_feat3_dynamic.onnx ONNX
primitives/nnx/leaky_relu.onnx ONNX
primitives/nnx/leaky_relu_custom.onnx ONNX
primitives/nnx/leaky_relu_default.onnx ONNX
primitives/nnx/leaky_relu_default_dynamic.onnx ONNX
primitives/nnx/linear_general.onnx ONNX
primitives/nnx/linear_general_2.onnx ONNX
primitives/nnx/linear_general_3.onnx ONNX
primitives/nnx/linear_general_4.onnx ONNX
primitives/nnx/linear_general_abstract_eval_axes.onnx ONNX
primitives/nnx/linear_general_abstract_eval_axes_pair.onnx ONNX
primitives/nnx/linear_general_dynamic.onnx ONNX
primitives/nnx/linear_general_merge_symbolic_dim_dynamic.onnx ONNX
primitives/nnx/linear_high_rank_dynamic.onnx ONNX
primitives/nnx/linear_high_rank_dynamic_f64.onnx ONNX
primitives/nnx/linear_high_rank_no_bias.onnx ONNX
primitives/nnx/linear_high_rank_no_bias_dynamic.onnx ONNX
primitives/nnx/linear_high_rank_no_bias_dynamic_f64.onnx ONNX
primitives/nnx/linear_high_rank_no_bias_f64.onnx ONNX
primitives/nnx/linear_high_rank_static.onnx ONNX
primitives/nnx/linear_high_rank_static_f64.onnx ONNX
primitives/nnx/linear_merge_symbolic_dim_dynamic.onnx ONNX
primitives/nnx/linear_no_bias.onnx ONNX
primitives/nnx/linear_no_bias_dynamic.onnx ONNX
primitives/nnx/linear_no_bias_dynamic_f64.onnx ONNX
primitives/nnx/linear_no_bias_f64.onnx ONNX
primitives/nnx/linear_symbolic_batch.onnx ONNX
primitives/nnx/linear_symbolic_batch_dynamic.onnx ONNX
primitives/nnx/linear_symbolic_batch_dynamic_f64.onnx ONNX
primitives/nnx/linear_symbolic_batch_f64.onnx ONNX
primitives/nnx/log_sigmoid_basic.onnx ONNX
primitives/nnx/log_sigmoid_basic_dynamic.onnx ONNX
primitives/nnx/log_sigmoid_basic_dynamic_f64.onnx ONNX
primitives/nnx/log_sigmoid_basic_f64.onnx ONNX
primitives/nnx/log_softmax.onnx ONNX
primitives/nnx/log_softmax_axis0.onnx ONNX
primitives/nnx/log_softmax_axis0_f64.onnx ONNX
primitives/nnx/log_softmax_default_axis.onnx ONNX
primitives/nnx/log_softmax_default_axis_dynamic.onnx ONNX
primitives/nnx/log_softmax_default_axis_dynamic_f64.onnx ONNX
primitives/nnx/log_softmax_default_axis_f64.onnx ONNX
primitives/nnx/log_softmax_f64.onnx ONNX
primitives/nnx/lora_basic.onnx ONNX
primitives/nnx/lora_basic_dynamic.onnx ONNX
primitives/nnx/lora_basic_dynamic_f64.onnx ONNX
primitives/nnx/lora_basic_f64.onnx ONNX
primitives/nnx/lora_linear_basic.onnx ONNX
primitives/nnx/lora_linear_basic_dynamic.onnx ONNX
primitives/nnx/lora_linear_basic_dynamic_f64.onnx ONNX
primitives/nnx/lora_linear_basic_f64.onnx ONNX
primitives/nnx/lora_linear_high_rank.onnx ONNX
primitives/nnx/lora_linear_high_rank_dynamic.onnx ONNX
primitives/nnx/lora_linear_high_rank_dynamic_f64.onnx ONNX
primitives/nnx/lora_linear_high_rank_f64.onnx ONNX
primitives/nnx/lora_static.onnx ONNX
primitives/nnx/lora_static_f64.onnx ONNX
primitives/nnx/max_pool.onnx ONNX
primitives/nnx/max_pool_basic.onnx ONNX
primitives/nnx/max_pool_global_window.onnx ONNX
primitives/nnx/max_pool_global_window_dynamic.onnx ONNX
primitives/nnx/max_pool_same.onnx ONNX
primitives/nnx/max_pool_same_dynamic.onnx ONNX
primitives/nnx/max_pool_same_padding.onnx ONNX
primitives/nnx/positional_embedding.onnx ONNX
primitives/nnx/positional_embedding_dynamic.onnx ONNX
primitives/nnx/positional_embedding_dynamic_f64.onnx ONNX
primitives/nnx/positional_embedding_f64.onnx ONNX
primitives/nnx/prelu_custom_slope.onnx ONNX
primitives/nnx/prelu_custom_slope_dynamic.onnx ONNX
primitives/nnx/prelu_default.onnx ONNX
primitives/nnx/relu_1d.onnx ONNX
primitives/nnx/relu_1d_f64.onnx ONNX
primitives/nnx/relu_4d.onnx ONNX
primitives/nnx/relu_4d_dynamic.onnx ONNX
primitives/nnx/relu_4d_dynamic_f64.onnx ONNX
primitives/nnx/relu_4d_f64.onnx ONNX
primitives/nnx/rms_norm_4d_dynamic.onnx ONNX
primitives/nnx/rms_norm_4d_dynamic_dynamic.onnx ONNX
primitives/nnx/rms_norm_4d_dynamic_no_scale.onnx ONNX
primitives/nnx/rms_norm_4d_dynamic_no_scale_dynamic.onnx ONNX
primitives/nnx/rms_norm_basic.onnx ONNX
primitives/nnx/rms_norm_use_scale_false.onnx ONNX
primitives/nnx/sigmoid.onnx ONNX
primitives/nnx/sigmoid_dynamic.onnx ONNX
primitives/nnx/sigmoid_dynamic_f64.onnx ONNX
primitives/nnx/sigmoid_f64.onnx ONNX
primitives/nnx/softmax.onnx ONNX
primitives/nnx/softmax_dynamic.onnx ONNX
primitives/nnx/softmax_dynamic_f64.onnx ONNX
primitives/nnx/softmax_f64.onnx ONNX
primitives/nnx/softplus.onnx ONNX
primitives/nnx/tanh.onnx ONNX
primitives/nnx/tanh_f64.onnx ONNX
primitives/nnx/token_embedding.onnx ONNX
primitives/nnx/token_embedding_dynamic.onnx ONNX
primitives/nnx/token_embedding_dynamic_f64.onnx ONNX
primitives/nnx/token_embedding_f64.onnx ONNX
primitives/random/bernoulli_scalar_prob_shape.onnx ONNX
primitives/random/bernoulli_tensor_prob.onnx ONNX
primitives/random/random_bits_uint32.onnx ONNX
primitives/random/random_bits_uint32_f64.onnx ONNX
primitives/random/random_categorical_logits_batch.onnx ONNX
primitives/random/random_categorical_logits_batch_opset23.onnx ONNX
primitives/random/random_categorical_logits_rank3.onnx ONNX
primitives/random/random_categorical_logits_rank3_opset23.onnx ONNX
primitives/random/random_categorical_logits_symbolic_batch.onnx ONNX
primitives/random/random_categorical_logits_symbolic_batch_dynamic.onnx ONNX
primitives/random/random_categorical_logits_symbolic_batch_opset23_dynamic.onnx ONNX
primitives/random/random_fold_in_passthrough.onnx ONNX
primitives/random/random_fold_in_passthrough_f64.onnx ONNX
primitives/random/random_normal_f32_2x3.onnx ONNX
primitives/random/random_seed_basic.onnx ONNX
primitives/random/random_seed_basic_f64.onnx ONNX
tmp_bernoulli.onnx ONNX
tmp_compress_jvp.onnx ONNX
tmp_cumprod_opset21.onnx ONNX
tmp_cumprod_opset22.onnx ONNX
tmp_cumprod_opset23.onnx ONNX
tmp_cumprod_opset24.onnx ONNX
tmp_logsumexp.onnx ONNX

Related models

More from enpasos

No other models by this author.

More models

No task type declared.

Popular models