search

LLNL / MuyGPyS

A fast, pure python implementation of the MuyGPs Gaussian process realization and training algorithm.
Other
23 stars 11 forks source link

Implemention type should be a unified config variable #74

Closed bwpriest closed 1 year ago

bwpriest commented 1 year ago

Instead of several boolean flags muygpys_jax_enabled, muygpys_mpi_enabled, muygpys_pytorch_enabled, etc, we should use a single muygpys_backend variable that specifies which backend to use. Will need to update the _src infrastructure, as well as the CI and docs.

bwpriest commented 1 year ago

This should probably be settable via an environment variable, e.g. MUYGPYS_BACKEND, rather than relying upon source code or absl.

bwpriest commented 1 year ago

addressed with #75