ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

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.