Implementation of new kernels and a backend move to using a PyTree in place of a Chex dataclass. What's Changed More kernels by @thomaspinder in https://github.com/JaxGaussianProcesses/GPJax/pull/148 Full Changelog: https://github.com/JaxGaussianProcesses/GPJax/compare/v0.5.2...v0.5.4