pith. sign in

Auto-Vectorizing TensorFlow Graphs: Jacobians, Auto-Batching And Beyond

1 Pith paper cite this work. Polarity classification is still indexing.

1 Pith paper citing it
abstract

We propose a static loop vectorization optimization on top of high level dataflow IR used by frameworks like TensorFlow. A new statically vectorized parallel-for abstraction is provided on top of TensorFlow, and used for applications ranging from auto-batching and per-example gradients, to jacobian computation, optimized map functions and input pipeline optimization. We report huge speedups compared to both loop based implementations, as well as run-time batching adopted by the DyNet framework.

fields

math.ST 1

years

2026 1

verdicts

CONDITIONAL 1

representative citing papers

citing papers explorer

Showing 1 of 1 citing paper.