A library for graph deep learning research
A method to increase the speed and lower the memory footprint
A high-performance distributed training framework
Implementation of model parallel autoregressive transformers on GPUs
Sequence-to-sequence framework, focused on Neural Machine Translation
A Machine Learning Framework for Time Series Intelligence
Massively Parallel Deep Reinforcement Learning
Code release for ConvNeXt V2 model
Code for a multi-agent particle environment used in a paper
A minimal PyTorch re-implementation of the OpenAI GPT
A text generation library with pre-trained language models github.com
Build and deploy machine learning microservices
Chinese-language edition of Dive into Deep Learning
An API Oriented Open-source Python Framework for Unsupervised Learning
A flexible Federated Learning Framework based on PyTorch
Module for automatic summarization of text documents and HTML pages
OpenMMLab Video Perception Toolbox
An open-source NLP research library, built on PyTorch
Implementation of BEVFormer, a camera-only framework
A flexible deep learning framework
Deep learning PyTorch library for time series forecasting
An Open-Source Framework for Prompt-Learning
Accelerated deep learning R&D
TensorFlowOnSpark brings TensorFlow programs to Apache Spark clusters
A flexible and efficient library for deep learning