Add support for "parallel for" abstraction, optimized via graph rewrite. This
enables applications like auto-batching, jacobians, per-example gradients. PiperOrigin-RevId: 203026617
Loading
Please sign in to comment
enables applications like auto-batching, jacobians, per-example gradients. PiperOrigin-RevId: 203026617