The easiest way to use deep metric learning in your application. Modular, flexible, and extensible. Written in PyTorch.

🎓Leetcode solutions in Python 📚

Generative Adversarial Transformers

[CVPR 2022] StyleSwin: Transformer-based GAN for High-resolution Image Generation

PyTorch implementation of the InfoNCE loss for self-supervised learning.

matrix factorization in PyTorch

Code for the paper: "On the Bottleneck of Graph Neural Networks and Its Practical Implications"

Official PyTorch BIVA implementation (BIVA: A Very Deep Hierarchy of Latent Variables for Generative Modeling)

Information Bottlenecks for Attribution

Unpaired Image-to-Image Translation with Transformer-based GANs in PyTorch [WIP]

The codebase for "Group-wise Contrastive Learning for Neural Dialogue Generation" (Cai et al., Findings of EMNLP 2020)

Official code for "The Group Loss for Deep Metric Learning" paper (ECCV 2020)

Official source code repository for the ICML 2021 paper "Hierarchical VAEs Know What They Don't Know"

FOD-Net: A Deep Learning Method for Fiber Orientation Distribution Angular Super Resolution

This is the repository for OPNested, OPNested+GMM, and OPNested-ComBat from the manuscript entitled "Improved generalized ComBat methods for harmonization of radiomic features" by Horng et al., pub…