Jasier

gpt-2

Code for the paper "Language Models are Unsupervised Multitask Learners"

jukebox

Code for the paper "Jukebox: A Generative Model for Music"

GPT2-Chinese

Chinese version of GPT2 training code, using BERT tokenizer.

denoising-diffusion-pytorch

Implementation of Denoising Diffusion Probabilistic Model in Pytorch

scikit-opt

Genetic Algorithm, Particle Swarm Optimization, Simulated Annealing, Ant Colony Optimization Algorithm,Immune Algorithm, Artificial Fish Swarm Algorithm, Differential Evolution and TSP(Traveling salesman)

NUWA

A unified 3D Transformer Pipeline for visual synthesis

decision-transformer

Official codebase for Decision Transformer: Reinforcement Learning via Sequence Modeling.

GeneticAlgorithmPython

Source code of PyGAD, a Python 3 library for building the genetic algorithm and training machine learning algorithms (Keras & PyTorch).

CogView

Text-to-Image generation. The repo for NeurIPS 2021 paper "CogView: Mastering Text-to-Image Generation via Transformers".

vq-vae-2-pytorch

Implementation of Generating Diverse High-Fidelity Images with VQ-VAE-2 in PyTorch

multilingual-t5

Transformers4Rec

Transformers4Rec is a flexible and efficient library for sequential and session-based recommendation and works with PyTorch.

VideoGPT

ProteinMPNN

Code for the ProteinMPNN paper

CLIPasso

OpenHGNN

This is an open-source toolkit for Heterogeneous Graph Neural Network(OpenHGNN) based on DGL.

tab-transformer-pytorch

Implementation of TabTransformer, attention network for tabular data, in Pytorch

sklearn2pmml

Python library for converting Scikit-Learn pipelines to PMML

sig-mlops

CDF SIG MLOps

byt5

GraphMAE

GraphMAE: Self-Supervised Masked Graph Autoencoders in KDD'22

RepMLP

RepMLPNet: Hierarchical Vision MLP with Re-parameterized Locality (CVPR 2022)

DVAE

Official implementation of Dynamical VAEs

res-mlp-pytorch

Implementation of ResMLP, an all MLP solution to image classification, in Pytorch

jpmml-lightgbm

Java library and command-line application for converting LightGBM models to PMML

vimpac

gpt-j-6b

ILSVRC2012

training imagenet2012 with pytorch

wave2vec-recognize-docker

Wave2vec 2.0 Recognize pipeline

jpmml-catboost

Java library and command-line application for converting CatBoost models to PMML