mengcz13

PySyft

Perform data science on data that remains in someone else's server

einops

Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)

darts

A python library for user-friendly forecasting and anomaly detection on time series.

interpret

Fit interpretable models. Explain blackbox machine learning.

pytorch-OpCounter

Count the MACs / FLOPs of your PyTorch model.

FedML

FEDML - The unified and scalable ML library for large-scale distributed training, model serving, and federated learning. FEDML Launch, a cross-cloud scheduler, further enables running any AI jobs on any GPU cloud or on-premise cluster. Built on this library, TensorOpera AI (https://TensorOpera.ai) is your generative AI platform at scale.

jittor

Jittor is a high-performance deep learning framework based on JIT compiling and meta-operators.

multiagent-particle-envs

Code for a multi-agent particle environment used in the paper "Multi-Agent Actor-Critic for Mixed Cooperative-Competitive Environments"

tigramite

Tigramite is a python package for causal inference with a focus on time series data. The Tigramite documentation is at

Dassl.pytorch

A PyTorch toolbox for domain generalization, domain adaptation and semi-supervised learning.

floc

This proposal has been replaced by the Topics API.

PyDMD

Python Dynamic Mode Decomposition

builder

Continuous builder and binary build scripts for pytorch

eicu-code

Code and website related to the eICU Collaborative Research Database

AmortizedCausalDiscovery

Code for the paper: Amortized Causal Discovery: Learning to Infer Causal Graphs from Time-Series Data

ZotExpert

Open Zotero PDFs on PDF Expert.

logohunter

Deep learning tool to find brand logos in everyday pictures

FedEM

Official code for "Federated Multi-Task Learning under a Mixture of Distributions" (NeurIPS'21)

General-Purpose-Sound-Recognition-Demo

General purpose sound recognition demo

slurm_gpu_ubuntu

Instructions for setting up a SLURM cluster using Ubuntu 18.04.3 with GPUs.

CausalTransformer

Code for the paper "Causal Transformer for Estimating Counterfactual Outcomes"

XDC

Self-Supervised Learning by Cross-Modal Audio-Video Clustering (NeurIPS 2020)

finite-element-networks

Reference implementation of Finite Element Networks as proposed in "Learning the Dynamics of Physical Systems from Sparse Observations with Finite Element Networks" at ICLR 2022

NeuralLaplace

Neural Laplace: Differentiable Laplace Reconstructions for modelling any time observation with O(1) complexity.

RKN

neural-interaction-detection

Detecting Statistical Interactions from Neural Network Weights

LPGNN

Locally Private Graph Neural Networks (ACM CCS 2021)

FQA

Multi-agent Trajectory Prediction with Fuzzy Query Attention

archipelago

Official Code Repo for the Paper: "How does This Interaction Affect Me? Interpretable Attribution for Feature Interactions", In NeurIPS 2020

DeepLINK

Deep Large-Scale Inference UsingKnockoffs