2021-06-26 00:36:01 +03:00
|
|
|
# Module kmath-noa
|
|
|
|
|
|
2021-07-11 21:01:50 +03:00
|
|
|
A Bayesian computation library over
|
2021-06-26 00:36:01 +03:00
|
|
|
[NOA](https://github.com/grinisrit/noa.git)
|
2021-07-11 21:01:50 +03:00
|
|
|
together with relevant functionality from
|
2021-06-26 00:36:01 +03:00
|
|
|
[LibTorch](https://pytorch.org/cppdocs).
|
|
|
|
|
|
2021-07-11 21:01:50 +03:00
|
|
|
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`.
|
2021-06-26 00:36:01 +03:00
|
|
|
|
2021-06-27 18:28:28 +03:00
|
|
|
## Installation
|
|
|
|
|
|
2021-07-12 23:12:02 +03:00
|
|
|
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](https://gcc.gnu.org/) toolchain. For `GPU` kernels, we require a compatible
|
2021-06-27 18:28:28 +03:00
|
|
|
[CUDA](https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html)
|
|
|
|
|
installation. If you are on Windows, we recommend setting up
|
|
|
|
|
everything on [WSL](https://docs.nvidia.com/cuda/wsl-user-guide/index.html).
|
