kmath/kmath-noa
2021-07-12 21:12:02 +01:00
..
src optimiser 2021-07-12 20:39:29 +01:00
build.gradle.kts dependencies fixed 2021-07-12 21:12:02 +01:00
README.md dependencies fixed 2021-07-12 21:12:02 +01:00

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.