oryx.experimental.matching.jax_rewrite.primitive_to_expression

Stay organized with collections Save and categorize content based on your preferences.

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.