TorchKM: A GPU-Oriented Library for Kernel Learning and Model Selection

Yikai Zhang, Gaoxiang Jia, Jie Ding +1

TorchKM is an open-source library for kernel machines, including support vector machines, kernel logistic regression, and kernel quantile regression, with GPU acceleration. The lib…