search

JaxGaussianProcesses / JaxKern

Kernel functions in JAX.
MIT License
7 stars 3 forks source link

Fix typing and expose compute class #54

Closed thomaspinder closed 1 year ago

thomaspinder commented 1 year ago

This PR cleans up the typing, cleans up the Graph Kernel object and exposes the compute_class.

Resolves Issues JaxGaussianProcesses/GPJax#214 and JaxGaussianProcesses/GPJax#215