pytorch

Multi-Task Deep Neural Networks for Natural Language Understanding

Fast and accurate human pose estimation in PyTorch. Contains implementation of "Real-time 2D Multi-Person Pose Estimation on CPU: Lightweight OpenPose" paper.

An open source library and framework for deep learning on satellite and aerial imagery.

Basic Utilities for PyTorch Natural Language Processing (NLP)

Controllable and fast Text-to-Speech for over 7000 languages!

开源社区第一个能下载、能运行的中文 LLaMA2 模型！

Distributed AI Model Training and LLM Fine-Tuning on Kubernetes

Reformer, the efficient Transformer, in Pytorch

EasyNLP: A Comprehensive and Easy-to-use NLP Toolkit

PyTorch implementation of adversarial attacks [torchattacks]

MoBA: Mixture of Block Attention for Long-Context LLMs

Convert images of LaTex math equations into LaTex code.

A platform for reproducible world model research and evaluation

The official repo for [NeurIPS'22] "ViTPose: Simple Vision Transformer Baselines for Human Pose Estimation" and [TPAMI'23] "ViTPose++: Vision Transformer for Generic Body Pose Estimation"

C++ Implementation of PyTorch Tutorials for Everyone

A sketch extractor for anime/illustration.

Machine learning on FPGAs using HLS

MII makes low-latency and high-throughput inference possible, powered by DeepSpeed.

Training RNNs as Fast as CNNs (https://arxiv.org/abs/1709.02755)

SpikingJelly is an open-source deep learning framework for Spiking Neural Network (SNN) based on PyTorch.