Lists (13)
Stars
All Algorithms implemented in Python
🤗 Transformers: the model-definition framework for state-of-the-art machine learning models in text, vision, audio, and multimodal models, for both inference and training.
🤗 Diffusers: State-of-the-art diffusion models for image, video, and audio generation in PyTorch.
Fast and memory-efficient exact attention
Graph Neural Network Library for PyTorch
Repository to track the progress in Natural Language Processing (NLP), including the datasets and the current state-of-the-art for the most common NLP tasks.
Open standard for machine learning interoperability
Generate 3D objects conditioned on text or images
Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.
🐍 Geometric Computer Vision Library for Spatial AI
Hackable and optimized Transformers building blocks, supporting a composable construction.
Running large language models on a single GPU for throughput-oriented scenarios.
Implementation of RLHF (Reinforcement Learning with Human Feedback) on top of the PaLM architecture. Basically ChatGPT but with PaLM
Framework agnostic sliced/tiled inference + interactive ui + error analysis plots
Count the MACs / FLOPs of your PyTorch model.
[ICLR 2024 Oral] Generative Gaussian Splatting for Efficient 3D Content Creation
NVIDIA's Deep Imagination Team's PyTorch Library
OpenMMLab Video Perception Toolbox. It supports Video Object Detection (VID), Multiple Object Tracking (MOT), Single Object Tracking (SOT), Video Instance Segmentation (VIS) with a unified framework.
FCOS: Fully Convolutional One-Stage Object Detection (ICCV'19)
PennyLane is an open-source quantum software platform for quantum computing, quantum machine learning, and quantum chemistry. Create meaningful quantum algorithms, from inspiration to implementation.
PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT
A PyTorch Library for Meta-learning Research