Search awesome repositories

Accessible large language models via k-bit quantization for PyTorch.

Code for the paper "Jukebox: A Generative Model for Music"

tensorboard for pytorch (and chainer, mxnet, numpy, ...)

Background Remover lets you Remove Background from images and video using AI with a simple command line interface that is free and open source.

Stanford NLP Python library for tokenization, sentence segmentation, NER, and parsing of many human languages

Easy Docker setup for Stable Diffusion with user-friendly UI

Aim 💫 — An easy-to-use & supercharged open-source experiment tracker.

A scikit-learn compatible neural network library that wraps PyTorch

FlashInfer: Kernel Library for LLM Serving

MMdnn is a set of tools to help users inter-operate among different deep learning frameworks. E.g. model conversion and visualization. Convert models between Caffe, Keras, MXNet, Tensorflow, CNTK, PyTorch Onnx and CoreML.

A GPU-accelerated library containing highly optimized building blocks and an execution engine for data processing to accelerate deep learning training and inference applications.

Standardized Distributed Generative and Predictive AI Inference Platform for Scalable, Multi-Framework Deployment on Kubernetes

ZenML 🙏: One AI Platform from Pipelines to Agents. https://zenml.io.

Supercharge Your Model Training

Rust bindings for the C++ api of PyTorch.

Probabilistic time series modeling in Python

OpenMMLab's Next Generation Video Understanding Toolbox and Benchmark

Time series forecasting with PyTorch

An Engine-Agnostic Deep Learning Framework in Java

High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.

A repo for distributed training of language models with Reinforcement Learning via Human Feedback (RLHF)

A unified, comprehensive and efficient recommendation library

A repository for storing models that have been inter-converted between various frameworks. Supported frameworks are TensorFlow, PyTorch, ONNX, OpenVINO, TFJS, TFTRT, TensorFlowLite (Float32/16/INT8), EdgeTPU, CoreML.

LLM training code for Databricks foundation models

Open-source evaluation toolkit of large multi-modality models (LMMs), support 220+ LMMs, 80+ benchmarks

Scalable and user friendly neural :brain: forecasting algorithms.

TorchGeo: datasets, samplers, transforms, and pre-trained models for geospatial data

Deepchecks: Tests for Continuous Validation of ML Models & Data. Deepchecks is a holistic open-source solution for all of your AI & ML validation needs, enabling to thoroughly test your data and models from research to production.

A highly efficient implementation of Gaussian Processes in PyTorch

A retargetable MLIR-based machine learning compiler and runtime toolkit.