oryx.experimental.matching.jax_rewrite.primitive_to_expression

Converts a JAX primitive into a Primitive expression.

prim A jax.core.Primitive to be converted into an expression.

A function that returns an expression when provided operands and parameters.