Learn how to design large-scale systems. Prep for the system design interview. Includes Anki flashcards.

Stable Diffusion web UI

Animation engine for explanatory math videos

The uncompromising Python code formatter

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

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

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

🤗 Diffusers: State-of-the-art diffusion models for image and audio generation in PyTorch and FLAX.

Machine Learning From Scratch. Bare bones NumPy implementations of machine learning models and algorithms with a focus on accessibility. Aims to cover everything from linear regression to deep lear…

Image-to-Image Translation in PyTorch

Code for the paper "Language Models are Unsupervised Multitask Learners"

Graph Neural Network Library for PyTorch

A minimal PyTorch re-implementation of the OpenAI GPT (Generative Pretrained Transformer) training

Open standard for machine learning interoperability

🤗 PEFT: State-of-the-art Parameter-Efficient Fine-Tuning.

Datasets, Transforms and Models specific to Computer Vision

Machine learning, in numpy

StyleGAN - Official TensorFlow Implementation

StyleGAN2 - Official TensorFlow Implementation

Code for loralib, an implementation of "LoRA: Low-Rank Adaptation of Large Language Models"

StreamDiffusion: A Pipeline-Level Solution for Real-Time Interactive Generation

The AI developer platform. Use Weights & Biases to train and fine-tune models, and manage models from experimentation to production.

Hydra is a framework for elegantly configuring complex applications

Speech recognition module for Python, supporting several engines and APIs, online and offline.

A unified framework for machine learning with time series

Pytorch implementation of convolutional neural network visualization techniques

An API standard for single-agent reinforcement learning environments, with popular reference environments and related utilities (formerly Gym)

Uniform Manifold Approximation and Projection

Python library for audio and music analysis

Implementation of Graph Convolutional Networks in TensorFlow