GitHub repository: JaxGaussianProcesses/JaxUtils
Primary language: Python
Description:  Model training utilities in JAX.