torch-tangent
PublicComposable, differentiable, and simple finite-width and infinite-width NTKs for Gaussian Processes, with cross-functionality with GPyTorch.
Composable, differentiable, and simple finite-width and infinite-width NTKs for Gaussian Processes, with cross-functionality with GPyTorch.