JaxGaussianProcesses

Pinned Repositories

.github
0 0 00
docs
JaxGaussianProcesses docs.
Language:Python0 0 00
GPJax
Gaussian processes in JAX.
Language:Python473 4 17454
gpjax_benchmarks
1 0 00
JaxKern
Kernel functions in JAX.
Language:Python8 2 144
JaxLinOp
Linear Operators in JAX.
Language:Python12 2 220
JaxUtils
Model training utilities in JAX.
Language:Python4 2 142
MOGPJax
Multi-output Gaussian processes in JAX.
Language:Python8 2 30
mytree
"Module pytrees" that cleanly handle parameter trainability and transformations for JAX models.
Language:Python2 2 31
static
0 0 01

JaxGaussianProcesses's Repositories

JaxGaussianProcesses/GPJax
Gaussian processes in JAX.
Language:Python473 4 17454
JaxGaussianProcesses/JaxLinOp
Linear Operators in JAX.
Language:Python12 2 220
JaxGaussianProcesses/JaxKern
Kernel functions in JAX.
Language:Python8 2 144
JaxGaussianProcesses/MOGPJax
Multi-output Gaussian processes in JAX.
Language:Python8 2 30
JaxGaussianProcesses/JaxUtils
Model training utilities in JAX.
Language:Python4 2 142
JaxGaussianProcesses/mytree
"Module pytrees" that cleanly handle parameter trainability and transformations for JAX models.
Language:Python2 2 31
JaxGaussianProcesses/gpjax_benchmarks
1 0 00
JaxGaussianProcesses/.github
0 0 00
JaxGaussianProcesses/docs
JaxGaussianProcesses docs.
Language:Python0 0 00
JaxGaussianProcesses/JaxGaussianProcesses.github.io
JaxGaussianProcesses website.
Language:HTML0 1 00
JaxGaussianProcesses/static
0 0 01