Graphormer is a deep learning package that allows researchers and developers to train custom models for molecule modeling tasks. It aims to accelerate the research and application in AI for molecul…

[ICCV 2021- Oral] Official PyTorch implementation for Generic Attention-model Explainability for Interpreting Bi-Modal and Encoder-Decoder Transformers, a novel method to visualize any Transformer-…

The `rtdl` library + The official implementation of the paper "Revisiting Deep Learning Models for Tabular Data"

It's not just a linter that annoys you!

On Explaining Your Explanations of BERT: An Empirical Study with Sequence Classification

[CVPR 2021] Official PyTorch implementation for Transformer Interpretability Beyond Attention Visualization, a novel method to visualize classifications by Transformer based networks.

PyTorch implementation of Grad-CAM

SnapMix: Semantically Proportional Mixing for Augmenting Fine-grained Data (AAAI 2021)

Spatial Sparse Convolution in PyTorch

Folder / directory structure options and naming conventions for software projects

Public code for implementation and experiments with differentiable decision trees.

A machine learning package for streaming data in Python. The other ancestor of River.

PyTorch fast implementation of "Distilling a Neural Network Into a Soft Decision Tree." Nicholas Frosst, Geoffrey Hinton., 2017.

Latex code for making neural networks diagrams

Pytorch implementation of various Knowledge Distillation (KD) methods.

SpinalNet: Deep Neural Network with Gradual Input

Code release for Fine-Grained Visual Classification via Progressive Multi-Granularity Training of Jigsaw Patches (ECCV2020)

A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.

eForest: Reversible mapping between high-dimensional data and path rule identifiers using trees embedding