PyTorch library of curated Transformer models and their components
Audiocraft is a library for audio processing and generation
Stable Diffusion with Core ML on Apple Silicon
A Customizable Image-to-Video Model based on HunyuanVideo
Democratizing Deep-Learning for Drug Discovery, Quantum Chemistry, etc
Python toolbox to create adversarial examples
Deep and online learning with spiking neural networks in Python
Build high-performance AI models with modular building blocks
Fault-tolerant, highly scalable GPU orchestration
High quality, fast, modular reference implementation of SSD in PyTorch
An Efficient and Easy-to-use Federated Learning Framework
Implementation of Recurrent Interface Network (RIN)
MMEditing is a low-level vision toolbox based on PyTorch
A tool for learning vector representations of words and entities
An open source object detection toolbox based on PyTorch
Transformers4Rec is a flexible and efficient library
Library of self-supervised methods for visual representation
Implementation of DALL-E 2, OpenAI's updated text-to-image synthesis
Implementation of MusicLM music generation model in Pytorch
Embed images and sentences into fixed-length vectors
Differentiable SDE solvers with GPU support and efficient sensitivity
Unofficial Parallel WaveGAN
The PyTorch-based audio source separation toolkit for researchers
Database system for building simpler and faster AI-powered application
textgen, Text Generation models