Skip to content

PyTORCH, TensorComprehension

laurentopia edited this page May 24, 2019 · 1 revision

Python torch. Dynamic neural networks, don't need compilation, high speed. Favored for RL https://github.com/pytorch/pytorch/

Tensor Cmprehension: compiler from math language to high performance code. Uses a sort of automated benchmark and some genetic algo to improve performance at compile time. https://github.com/facebookresearch/TensorComprehensions

Clone this wiki locally