aidanscannell/MoGPJaxE

Mixtures of Gaussian Process Experts in JAX with custom state management.

Python