pytorch: Python-пакет для тензорных вычислений на GPU и глубокого обучения с автодифференцированием

Чем полезно

Даёт тензоры как в NumPy с ускорением на GPU и гибкие динамические нейросети, которые можно менять прямо во время выполнения кода без пересборки графа.

Главные возможности

• Тензоры на CPU и GPU, совместимые по духу с NumPy
• Динамические нейросети на основе tape-based autograd
• Интеграция с NumPy, SciPy и Cython, расширения на C++

Для кого

Исследователи и инженеры, которые занимаются глубоким обучением на Python

Как начать

pip install torch (команды для Conda и pip, на pytorch.org)