Composable, differentiable, and simple finite-width and infinite-width NTKs for Gaussian Processes, with cross-functionality with GPyTorch.
library deep-learning torch pytorch neural-networks gaussian-processes kernel-learning neural-tangent-kernel
-
Updated
Jun 18, 2025 - Python