WynMew

AIX360

Interpretability and explainability of data and machine learning models

ChatRWKV

ChatRWKV is like ChatGPT but powered by RWKV (100% RNN) language model, and open source.

ColossalAI

Making large AI models cheaper, faster and more accessible

crest

Repo for CReST: A Class-Rebalancing Self-Training Framework for Imbalanced Semi-Supervised Learning

determined

Determined: Deep Learning Training Platform

eigenlearning

codebase for Neural Tangent Kernel Eigenvalues Accurately Predict Generalization

fairseq

Facebook AI Research Sequence-to-Sequence Toolkit written in Python.

fast_pytorch_kmeans

This is a pytorch implementation of k-means clustering algorithm

Gradient-Free-Optimizers

Simple and reliable optimization with local, global, population-based and sequential techniques in numerical discrete search spaces.

imbalanced-regression

[ICML 2021, Long Talk] Delving into Deep Imbalanced Regression

LDAM-DRW

[NeurIPS 2019] Learning Imbalanced Datasets with Label-Distribution-Aware Margin Loss

MNBVC

MNBVC(Massive Never-ending BT Vast Chinese corpus)超大规模中文语料集。对标chatGPT训练的40T数据。MNBVC数据集不但包括主流文化，也包括各个小众文化甚至火星文的数据。MNBVC数据集包括新闻、作文、小说、书籍、杂志、论文、台词、帖子、wiki、古诗、歌词、商品介绍、笑话、糗事、聊天记录等一切形式的纯文本中文数据。

nerf-pytorch

A PyTorch implementation of NeRF (Neural Radiance Fields) that reproduces the results.

nevergrad

A Python toolbox for performing gradient-free optimization

node

Neural Oblivious Decision Ensembles for Deep Learning on Tabular Data

non-parametric-transformers

Code for "Self-Attention Between Datapoints: Going Beyond Individual Input-Output Pairs in Deep Learning"

OpenFE

OpenFE: automated feature generation with expert-level performance

parametric-robustness-evaluation

Code for paper "Evaluating Robustness to Dataset Shift via Parametric Robustness Sets"

pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

pytorch-minimize

Newton and Quasi-Newton optimization with PyTorch

pytorch-pretrained-BERT

A PyTorch implementation of Google AI's BERT model provided with Google's pre-trained models, examples and utilities.

pytorch_tabular

A standard framework for modelling Deep Learning Models for tabular data

RepVGG

RepVGG: Making VGG-style ConvNets Great Again

rulefit

Python implementation of the rulefit algorithm

stock-embeddings

Code relating to the paper - Stock Embeddings: Learning Distributed Representations for Financial Assets

tabnet

PyTorch implementation of TabNet paper : https://arxiv.org/pdf/1908.07442.pdf

TEDVAE

tianshou

An elegant PyTorch deep reinforcement learning library.

vit-pytorch

Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch

what-if-tool

Source code/webpage/demos for the What-If Tool