# This file is auto-generated. Do not modify manually.
plugin_registry = {
    "nnx.avg_pool": "jax2onnx.converter.plugins.flax.nnx.avg_pool",
    "nnx.batch_norm": "jax2onnx.converter.plugins.flax.nnx.batch_norm",
    "nnx.conv": "jax2onnx.converter.plugins.flax.nnx.conv",
    "nnx.dot_product_attention": "jax2onnx.converter.plugins.flax.nnx.dot_product_attention",
    "nnx.dropout": "jax2onnx.converter.plugins.flax.nnx.dropout",
    "nnx.elu": "jax2onnx.converter.plugins.flax.nnx.elu",
    "nnx.gelu": "jax2onnx.converter.plugins.flax.nnx.gelu",
    "nnx.layer_norm": "jax2onnx.converter.plugins.flax.nnx.layer_norm",
    "nnx.leaky_relu": "jax2onnx.converter.plugins.flax.nnx.leaky_relu",
    "nnx.linear": "jax2onnx.converter.plugins.flax.nnx.linear",
    "nnx.linear_general": "jax2onnx.converter.plugins.flax.nnx.linear_general",
    "nnx.log_softmax": "jax2onnx.converter.plugins.flax.nnx.log_softmax",
    "nnx.max_pool": "jax2onnx.converter.plugins.flax.nnx.max_pool",
    "nnx.relu": "jax2onnx.converter.plugins.flax.nnx.relu",
    "nnx.sigmoid": "jax2onnx.converter.plugins.flax.nnx.sigmoid",
    "nnx.softmax": "jax2onnx.converter.plugins.flax.nnx.softmax",
    "nnx.softplus": "jax2onnx.converter.plugins.flax.nnx.softplus",
    "nnx.tanh": "jax2onnx.converter.plugins.flax.nnx.tanh",
    "add": "jax2onnx.converter.plugins.jax.lax.add",
    "argmax": "jax2onnx.converter.plugins.jax.lax.argmax",
    "argmin": "jax2onnx.converter.plugins.jax.lax.argmin",
    "broadcast_in_dim": "jax2onnx.converter.plugins.jax.lax.broadcast_in_dim",
    "concatenate": "jax2onnx.converter.plugins.jax.lax.concatenate",
    "conv_general_dilated": "jax2onnx.converter.plugins.jax.lax.conv",
    "convert_element_type": "jax2onnx.converter.plugins.jax.lax.convert_element_type",
    "div": "jax2onnx.converter.plugins.jax.lax.div",
    "dot_general": "jax2onnx.converter.plugins.jax.lax.dot_general",
    "dynamic_slice": "jax2onnx.converter.plugins.jax.lax.dynamic_slice",
    "eq": "jax2onnx.converter.plugins.jax.lax.eq",
    "exp": "jax2onnx.converter.plugins.jax.lax.exp",
    "gather": "jax2onnx.converter.plugins.jax.lax.gather",
    "gt": "jax2onnx.converter.plugins.jax.lax.gt",
    "integer_pow": "jax2onnx.converter.plugins.jax.lax.integer_pow",
    "log": "jax2onnx.converter.plugins.jax.lax.log",
    "lt": "jax2onnx.converter.plugins.jax.lax.lt",
    "max": "jax2onnx.converter.plugins.jax.lax.max",
    "min": "jax2onnx.converter.plugins.jax.lax.min",
    "mul": "jax2onnx.converter.plugins.jax.lax.mul",
    "ne": "jax2onnx.converter.plugins.jax.lax.ne",
    "neg": "jax2onnx.converter.plugins.jax.lax.neg",
    "reduce_max": "jax2onnx.converter.plugins.jax.lax.reduce_max",
    "reduce_min": "jax2onnx.converter.plugins.jax.lax.reduce_min",
    "reduce_sum": "jax2onnx.converter.plugins.jax.lax.reduce_sum",
    "reshape": "jax2onnx.converter.plugins.jax.lax.reshape",
    "slice": "jax2onnx.converter.plugins.jax.lax.slice",
    "sort": "jax2onnx.converter.plugins.jax.lax.sort",
    "sqrt": "jax2onnx.converter.plugins.jax.lax.sqrt",
    "square": "jax2onnx.converter.plugins.jax.lax.square",
    "squeeze": "jax2onnx.converter.plugins.jax.lax.squeeze",
    "stop_gradient": "jax2onnx.converter.plugins.jax.lax.stop_gradient",
    "sub": "jax2onnx.converter.plugins.jax.lax.sub",
    "tanh": "jax2onnx.converter.plugins.jax.lax.tanh",
    "transpose": "jax2onnx.converter.plugins.jax.lax.transpose",
    "nn.softmax": "jax2onnx.converter.plugins.jax.nn.softmax",
    "jnp.add": "jax2onnx.converter.plugins.jax.numpy.add",
    "jnp.concat": "jax2onnx.converter.plugins.jax.numpy.concat",
    "jnp.einsum": "jax2onnx.converter.plugins.jax.numpy.einsum",
    "jnp.matmul": "jax2onnx.converter.plugins.jax.numpy.matmul",
    "jnp.reshape": "jax2onnx.converter.plugins.jax.numpy.reshape",
    "jnp.squeeze": "jax2onnx.converter.plugins.jax.numpy.squeeze",
    "jnp.tile": "jax2onnx.converter.plugins.jax.numpy.tile",
    "jnp.transpose": "jax2onnx.converter.plugins.jax.numpy.transpose",
    "random_gamma": "jax2onnx.converter.plugins.jax.random.random_gamma",
}
