Borealis AI

advertorch

A Toolbox for Adversarial Robustness Research

noise_flow

Noise Flow: Noise Modeling with Conditional Normalizing Flows

private-data-generation

A toolbox for differentially private data generation

scaleformer

SLAPS-GNN

PyTorch code of "SLAPS: Self-Supervision Improves Structure Learning for Graph Neural Networks"

de-simple

Diachronic Embedding for Temporal Knowledge Graph Completion

ranksim-imbalanced-regression

[ICML 2022] RankSim: Ranking Similarity Regularization for Deep Imbalanced Regression

MMoEEx-MTL

PyTorch Implementation of the Multi-gate Mixture-of-Experts with Exclusivity (MMoEEx)

TSC-Disc-Proto

Discriminative Prototypes learned by Dynamic Time Warping (DTW) for Time Series Classification (TSC)

mtmfrl

Multi Type Mean Field Reinforcement Learning

DT-Fixup

Optimizing Deeper Transformers on Small Datasets https://arxiv.org/abs/2012.15355

PROVIDE

PROVIDE: A Probabilistic Framework for Unsupervised Video Decomposition (UAI 2021)

code-gen-TAE

Code generation from natural language with less prior and more monolingual data

continuous-latent-process-flows

Code, data, and pre-trained models for the paper "Continuous Latent Process Flows" (NeurIPS 2021)

efficient-vit-training

PyTorch code of "Training a Vision Transformer from scratch in less than 24 hours with 1 GPU" (HiTY workshop at Neurips 2022)

ssl-for-timeseries

Self Supervised Learning for Time Series Using Similarity Distillation

latent-bottlenecked-anp

nflow-cdf-approximations

Official implementation of "Efficient CDF Approximations for Normalizing Flows"

StayPositive

towards-better-sel-cls

ConR

Contrastive Regularizer

group-feature-importance

Group feature importance

ProbForest

Differentiable relaxations of tree-based models.

tree-cross-attention

monotonicity-mixup

Code of "Not Too Close and Not Too Far: Enforcing Monotonicity Requires Penalizing The Right Points"

raps

Code for the paper "Causal Bandits without Graph Learning"

robust-gan

On Minimax Optimality of GANs for Robust Mean Estimation

dcf

DynaShare-MTL

PyTorch Implementation of DynaShare: Task and Instance Conditioned Parameter Sharing for Multi-Task Learning

sasrec-ccql

PyTorch code of "Robust Reinforcement Learning Objectives for Sequential Recommender Systems"