Build Deep Learning Models in Scala

Storch provides GPU accelerated tensor operations, automatic differentiation, and a neural network API for building and training machine learning models.

Get the Best of PyTorch & Scala

Storch aims to be close to the original PyTorch API, while still leveraging Scala's powerful type system for safer tensor operations.

Powered by LibTorch & JavaCPP

Storch is based on LibTorch, the C++ library underlying PyTorch. JVM bindings are provided by JavaCPP for seamless interop with native code & CUDA support.

Torch by Mailtoanton / CC BY-SA 3.0