Feature/fast prediction

LLNL / MuyGPyS

A fast, pure python implementation of the MuyGPs Gaussian process realization and training algorithm.

Other

25 stars 11 forks source link

Feature/fast prediction #59

Closed alecmdunton closed 1 year ago

alecmdunton commented 2 years ago

Added fast prediction feature to MuyGPs library. This PR includes a new test in tests/jax_correctness.py which ensures the fast predictions generated using Jax and numpy are nearly identical. More tests to be added to mpi_correctness.py, multivariate.py, gp.py, and api.py.