A library for graph deep learning research
A high-performance distributed training framework
Mars is a tensor-based unified framework for large-scale data
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
A minimal PyTorch re-implementation of the OpenAI GPT
Code for a multi-agent particle environment used in a paper
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
OpenMMLab Video Perception Toolbox
An open-source NLP research library, built on PyTorch
Implementation of BEVFormer, a camera-only framework
Web mining module for Python, with tools for scraping
A flexible deep learning framework
A commented and documented implementation of MuZero
Deep learning PyTorch library for time series forecasting
Accelerated deep learning R&D
An Open-Source Framework for Prompt-Learning
TensorFlowOnSpark brings TensorFlow programs to Apache Spark clusters