kmath/kmath-noa
2021-07-12 17:18:02 +01:00
..
src data for jit modules 2021-07-12 17:18:02 +01:00
build.gradle.kts minor corrections 2021-07-09 08:16:33 +01:00
README.md readme updates 2021-07-11 19:01:50 +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

Currently, to build native artifacts, 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.

To install the library, simply publish the modules kmath-tensors and kmath-noa locally:

./gradlew -q :kmath-tensors:publishToMavenLocal :kmath-noa:publishToMavenLocal