Tensors and Dynamic neural networks in Python with strong GPU acceleration

Models and examples built with TensorFlow

Deep Learning for humans

The simplest, fastest repository for training/finetuning medium-sized GPTs.

OpenAssistant is a chat-based assistant that understands tasks, can interact with third-party systems, and retrieve information dynamically to do so.

Streamlit — A faster way to build and share data apps.

Ray is an AI compute engine. Ray consists of a core distributed runtime and a set of AI Libraries for accelerating ML workloads.

The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (V…

Let us control diffusion models!

State-of-the-art 2D and 3D Face Analysis Project

Python sample codes for robotics algorithms.

A generative world for general-purpose robotics & embodied AI learning.

Graph Neural Network Library for PyTorch

[NeurIPS'23 Oral] Visual Instruction Tuning (LLaVA) built towards GPT-4V level capabilities and beyond.

Large-scale Self-supervised Pre-training Across Tasks, Languages, and Modalities

GUI for ChatGPT API and many LLMs. Supports agents, file-based QA, GPT finetuning and query with web search. All with a neat UI.

This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows".

Image Polygonal Annotation with Python (polygon, rectangle, circle, line, point and image-level flag annotation).

Mamba SSM architecture

Annotate better with CVAT, the industry-leading data engine for machine learning. Used and trusted by teams at any scale, for data of any scale.

3D Reconstruction Software

AutoML library for deep learning

Robust Video Matting in PyTorch, TensorFlow, TensorFlow.js, ONNX, CoreML!

A fast, scalable, high performance Gradient Boosting on Decision Trees library, used for ranking, classification, regression and other machine learning tasks for Python, R, Java, C++. Supports comp…

An elegant PyTorch deep reinforcement learning library.

Uniform Manifold Approximation and Projection

Keras code and weights files for popular deep learning models.

PySlowFast: video understanding codebase from FAIR for reproducing state-of-the-art video models.

Python Stream Processing

Track-Anything is a flexible and interactive tool for video object tracking and segmentation, based on Segment Anything, XMem, and E2FGVI.