I've just finished the first draft of tensorflow integration for KMath (using graph lazy computation of course).
https://github.com/mipt-npm/kmath/tree/feature/tensorflow. Comments are welcome (especially from
@zaleslaw). I plan to add more operations as extensions, direct operation builder (use tensorflow graph directly with some improvements from a special scope in algebra). Compilation of MST to graph (expressions and preliminary optimizations).