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