-
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
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
low-cost, high-efficiency, easy-to-implement