The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (V…

Graph Neural Network Library for PyTorch

A list of all named GANs!

Python Implementation of Reinforcement Learning: An Introduction

Transfer learning / domain adaptation / domain generalization / multi-task learning etc. Papers, codes, datasets, applications, tutorials.-迁移学习

This project aim to reproduce Sora (Open AI T2V model), we wish the open source community contribute to this project.

Enjoy the magic of Diffusion models!

Differentiable ODE solvers with full GPU support and O(1)-memory backpropagation.

PyTorch implementations of deep reinforcement learning algorithms and environments

Graph Convolutional Networks in PyTorch

[EMNLP 2022] An Open Toolkit for Knowledge Graph Extraction and Construction

PyTorch implementation of Advantage Actor Critic (A2C), Proximal Policy Optimization (PPO), Scalable trust-region method for deep reinforcement learning using Kronecker-factored approximation (ACKT…

TorchGeo: datasets, samplers, transforms, and pre-trained models for geospatial data

Pytorch implementation of the Graph Attention Network model by Veličković et. al (2017, https://arxiv.org/abs/1710.10903)

Graphormer is a general-purpose deep learning backbone for molecular modeling.

Simple and easily configurable grid world environments for reinforcement learning

A modular high-level library to train embodied AI agents across a variety of tasks and environments.

PyTorch and TensorFlow implementation of NCP, LTC, and CfC wired neural models

Platform for designing and evaluating Graph Neural Networks (GNN)

Code Repository for Liquid Time-Constant Networks (LTCs)

Simulation of spiking neural networks (SNNs) using PyTorch.

A curated list for Efficient Large Language Models

PyTorch implementation of MAR+DiffLoss https://arxiv.org/abs/2406.11838

Graph Transformer Architecture. Source code for "A Generalization of Transformer Networks to Graphs", DLG-AAAI'21.

Refer to https://github.com/AcutronicRobotics/gym-gazebo2 for the new version

OpenSTL: A Comprehensive Benchmark of Spatio-Temporal Predictive Learning

Pytorch code for ICLR-20 Paper "Learning to Explore using Active Neural SLAM"

Official implementation of our NeurIPS 2023 paper "Augmenting Language Models with Long-Term Memory".

Foundation model for weather & climate