Minimal Gaussian process library in JAX with a simple (custom) approach to state management.
Primary LanguagePython