Stars
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 o…
Pyserini is a Python toolkit for reproducible information retrieval research with sparse and dense representations.
Causal Discovery in Python. It also includes (conditional) independence tests and score functions.
DAGs with NO TEARS: Continuous Optimization for Structure Learning
FedScale is a scalable and extensible open-source federated learning (FL) platform.
Python implementation of elastic-net regularized generalized linear models
Distributionally robust neural networks for group shifts
Lightweight Python library for in-memory matrix completion.
OpenDataVal: a Unified Benchmark for Data Valuation in Python (NeurIPS 2023)
Supporting code for the paper "Finding Influential Training Samples for Gradient Boosted Decision Trees"
PyTorch implementation of SCAFFOLD (Stochastic Controlled Averaging for Federated Learning, ICML 2020).
Implementation of SCAFFOLD: Stochastic Controlled Averaging for Federated Learning
A simulator for single-cell expression data guided by gene regulatory networks
Code for "Federated Accelerated Stochastic Gradient Descent" (NeurIPS 2020)
Implementation of EF and EF21 methods and their comparison on several tasks