Layer 6 Labs

T-Fixup

Code for the ICML'20 paper "Improving Transformer Optimization Through Better Initialization"

GSS

Code for the NeurIPS'19 paper "Guided Similarity Separation for Image Retrieval"

HGCF

Code for the WWW'21 paper "HGCF: Hyperbolic Graph Convolution Networks for Collaborative Filtering"

SGG-Seq2Seq

Code for the ICCV'21 paper "Context-aware Scene Graph Generation with Seq2Seq Transformers"

EGT

Code for the CVPR'19 paper "Explore-Exploit Graph Traversal for Image Retrieval"

ASL

Code for CVPR'21 paper "Weakly Supervised Action Selection Learning in Video"

TAFA

Code for the RecSys'20 paper "TAFA: Two-headed Attention Fused Autoencoder for Context-Aware Recommendations"

CMLMC

Code for the ICLR'22 paper "Improving Non-Autoregressive Translation Models Without Distillation"

MCL

Code for the WWW'22 paper "MCL: Mixed-Centric Loss for Collaborative Filtering"

CNNEmbed

rectangular-flows

Code for the NeurIPS'21 paper "Rectangular Flows for Manifold Learning"

CEF

Code for the NeurIPS'21 paper "Tractable Density Estimation on Learned Manifolds with Conformal Embedding Flows"

semi-supervised-egt

Code for CVPR'19 paper Semi-Supervised Exploration for Image Retrieval

CAE

Code for the ICLR'21 paper "C-Learning: Horizon-Aware Cumulative Accessibility Estimation"

denoising_dgms

transformers-CLIPScripting

🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.