Code for "On Feature Collapse and Deep Kernel Learning for Single Forward Pass Uncertainty". - View it on GitHub