-
Institute of Automation, Chinese Academy of Acience
- Beijing
Stars
Ray is an AI compute engine. Ray consists of a core distributed runtime and a set of AI Libraries for accelerating ML workloads.
The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (V…
Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch
🍀 Pytorch implementation of various Attention Mechanisms, MLP, Re-parameter, Convolution, which is helpful to further understand papers.⭐⭐⭐
Implementation of DALL-E 2, OpenAI's updated text-to-image synthesis neural network, in Pytorch
Easy-to-use,Modular and Extendible package of deep-learning based CTR models .
PyTorch implementation of MAE https//arxiv.org/abs/2111.06377
Example models using DeepSpeed
High-quality single file implementation of Deep Reinforcement Learning algorithms with research-friendly features (PPO, DQN, C51, DDPG, TD3, SAC, PPG)
Recommendation Algorithm大规模推荐算法库,包含推荐系统经典及最新算法LR、Wide&Deep、DSSM、TDM、MIND、Word2Vec、Bert4Rec、DeepWalk、SSR、AITM,DSIN,SIGN,IPREC、GRU4Rec、Youtube_dnn、NCF、GNN、FM、FFM、DeepFM、DCN、DIN、DIEN、DLRM、MMOE、PLE、ESM…
Modularized Implementation of Deep RL Algorithms in PyTorch
OpenDILab Decision AI Engine. The Most Comprehensive Reinforcement Learning Framework B.P.
【PyTorch】Easy-to-use,Modular and Extendible package of deep-learning based CTR models.
Reinforcement Learning / AI Bots in Card (Poker) Games - Blackjack, Leduc, Texas, DouDizhu, Mahjong, UNO.
An API standard for multi-agent reinforcement learning environments, with popular reference environments and related utilities
Unofficial PyTorch implementation of Masked Autoencoders Are Scalable Vision Learners
A modular, primitive-first, python-first PyTorch library for Reinforcement Learning.
Python Multi-Agent Reinforcement Learning framework
Implementations of IQL, QMIX, VDN, COMA, QTRAN, MAVEN, CommNet, DyMA-CL, and G2ANet on SMAC, the decentralised micromanagement scenario of StarCraft II
A collection of reference environments for offline reinforcement learning
This is the official implementation of Multi-Agent PPO (MAPPO).
An offline deep reinforcement learning library
High-quality single-file implementations of SOTA Offline and Offline-to-Online RL algorithms: AWAC, BC, CQL, DT, EDAC, IQL, SAC-N, TD3+BC, LB-SAC, SPOT, Cal-QL, ReBRAC
The code for 2020 Tencent College Algorithm Contest, and the online result ranks 1st.
One repository is all that is necessary for Multi-agent Reinforcement Learning (MARL)
Code for the paper "Exploration by Random Network Distillation"
A paper summary of image inpainting
Simple and easily configurable 3D FPS-game-like environments for reinforcement learning
The source code for the blog post The 37 Implementation Details of Proximal Policy Optimization