Amanda2024

ToM2C

The offcial implementation of "ToM2C: Target-oriented Multi-agent Communication and Cooperation with Theory of Mind" (ICLR 2022) .

relay-policy-learning

hok_env

Honor of Kings AI Open Environment of Tencent

CORRO

CORRO code

Metaworld

Collections of robotics environments geared towards benchmarking multi-task and meta reinforcement learning

unmas

the source code of UNMAS: Multi-Agent Reinforcement Learning for Unshaped Cooperative Scenarios

Multi-Agent-Reinforcement-Learning-papers

Multi-Agent Reinforcement Learning (MARL) papers

maximum_entropy_population_based_training

Maximum Entropy Population Based Training for Zero-Shot Human-AI Coordination

Chinese-Text-Classification-Pytorch

中文文本分类，TextCNN，TextRNN，FastText，TextRCNN，BiLSTM_Attention，DPCNN，Transformer，基于pytorch，开箱即用。

vgae_pytorch

This repository implements variational graph auto encoder by Thomas Kipf.

MM-cVAE

Implementation of the moment matching contrastive VAE

C-GMVAE

C-GMVAE: Gaussian Mixture VAE with Contrastive Learning for Multi-Label Classification

Contrastve-VAE

Contrastive-VAE is a model that combine the traditional Variational Auto-Encoder and contrastive learning techniques.

contrastive_vae

Contrastive Variational Autoencoders

VAE-CVAE-MNIST

Variational Autoencoder and Conditional Variational Autoencoder on MNIST in PyTorch

info-nce-pytorch

PyTorch implementation of the InfoNCE loss for self-supervised learning.

FOCAL-ICLR

Code for FOCAL Paper Published at ICLR 2021

babyai

BabyAI platform. A testbed for training agents to understand and execute language commands.

Transformer-MM-Explainability

[ICCV 2021- Oral] Official PyTorch implementation for Generic Attention-model Explainability for Interpreting Bi-Modal and Encoder-Decoder Transformers, a novel method to visualize any Transformer-based network. Including examples for DETR, VQA.

ShapleyExplanationNetworks

Implementation of the paper "Shapley Explanation Networks"

rl_waypoint_mrta

multi-agent-PPO-on-SMAC

Implementations of MAPPO and IPPO on SMAC, the multi-agent StarCraft environment.

PyTorch-RL

DQN, Reinforce, A2C (Reference: https://github.com/pytorch)

attention-learn-to-route

Attention based model for learning to solve different routing problems

AI-Paper-Collector

MLNLP社区用来更好进行论文搜索的工具。Fully-automated scripts for collecting AI-related papers

mini-AlphaStar

(JAIR'2022) A mini-scale reproduction code of the AlphaStar program. Note: the original AlphaStar is the AI proposed by DeepMind to play StarCraft II. JAIR = Journal of Artificial Intelligence Research.

human_aware_rl

Code for "On the Utility of Learning about Humans for Human-AI Coordination"

Paper-Writing-Tips

MLNLP社区用来帮助大家避免论文投稿小错误的整理仓库。 Paper Writing Tips

Sectar

Self-Consistent Trajectory Autoencoder: Hierarchical Reinforcement Learning with Trajectory Embeddings

SIDE

Codes for the paper "SIDE: State Inference for Partially Observable Cooperative Multi-Agent Reinforcement Learning"