Search awesome repositories

Materials for the Learn PyTorch for Deep Learning: Zero to Mastery course.

Private AI platform for agents, assistants and enterprise search. Built-in Agent Builder, Deep research, Document analysis, Multi-model support, and API connectivity for agents.

Computer Vision Annotation Tool (CVAT) is a leading platform for building high-quality visual datasets for vision AI. It offers open-source, cloud, and enterprise products, as well as labeling services, for image, video, and 3D annotation with AI-assisted labeling, quality assurance, team collaboration, analytics, and developer APIs.

Burn is a next generation tensor library and Deep Learning Framework that doesn't compromise on flexibility, efficiency and portability.

Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet.

RWKV (pronounced RwaKuv) is an RNN with great LLM performance, which can also be directly trained like a GPT transformer (parallelizable). We are at RWKV-7 "Goose". So it's combining the best of RNN and transformer - great performance, linear time, constant space (no kv-cache), fast training, infinite ctx_len, and free sentence embedding.

Nano vLLM

A very simple framework for state-of-the-art Natural Language Processing (NLP)

TensorRT LLM provides users with an easy-to-use Python API to define Large Language Models (LLMs) and supports state-of-the-art optimizations to perform inference efficiently on NVIDIA GPUs. TensorRT LLM also contains components to create Python and C++ runtimes that orchestrate the inference execution in a performant way.

An open source implementation of CLIP.

Open3D: A Modern Library for 3D Data Processing

PyTorch version of Stable Baselines, reliable implementations of reinforcement learning algorithms.

Low-code framework for building custom LLMs, neural networks, and other AI models

A PyTorch-based Speech Toolkit

This repository contains demos I made with the Transformers library by HuggingFace.

🐍 Geometric Computer Vision Library for Spatial AI

An elegant PyTorch deep reinforcement learning library.

LMCache: Supercharge Your LLM with the Fastest KV Cache Layer

Fast and Accurate ML in 3 Lines of Code

🌸 Run LLMs at home, BitTorrent-style. Fine-tuning and inference up to 10x faster than offloading

High-quality single file implementation of Deep Reinforcement Learning algorithms with research-friendly features (PPO, DQN, C51, DDPG, TD3, SAC, PPG)

Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)

ChatRWKV is like ChatGPT but powered by RWKV (100% RNN) language model, and open source.

Containers for machine learning

Swap GPT for any LLM by changing a single line of code. Xinference lets you run open-source, speech, and multimodal models on cloud, on-prem, or your laptop — all through one unified, production-ready inference API.

Deeplake is AI Data Runtime for Agents. It provides serverless postgres with a multimodal datalake, enabling scalable retrieval and training.

Deep universal probabilistic programming with Python and PyTorch

EmotiVoice 😊: a Multi-Voice and Prompt-Controlled TTS Engine

An Extensible Toolkit for Finetuning and Inference of Large Foundation Models. Large Models for All.

AI Toolkit for Healthcare Imaging