Official code Cross-Covariance Image Transformer (XCiT) - View it on GitHub