Google I/O is a wrap! Catch up on TensorFlow sessions View sessions

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.