kmath/kmath-noa/README.md
2021-07-12 21:12:02 +01:00

902 B

Module kmath-noa

A Bayesian computation library over NOA together with relevant functionality from LibTorch.

Our aim is to cover a wide set of applications from particle physics simulations to deep learning. In fact, we support any differentiable program written on top of AutoGrad & ATen.

Installation

To install the library, you can simply publish to the local Maven repository:

./gradlew -q :kmath-noa:publishToMavenLocal

This will fetch and build native artifacts as well. Currently, we support only the GNU toolchain. For GPU kernels, we require a compatible CUDA installation. If you are on Windows, we recommend setting up everything on WSL.