Search awesome repositories

Containers for machine learning

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

Accelerate local LLM inference and finetuning (LLaMA, Mistral, ChatGLM, Qwen, DeepSeek, Mixtral, Gemma, Phi, MiniCPM, Qwen-VL, MiniCPM-V, etc.) on Intel XPU (e.g., local PC with iGPU and NPU, discrete GPU such as Arc, Flex and Max); seamlessly integrate with llama.cpp, Ollama, HuggingFace, LangChain, LlamaIndex, vLLM, DeepSpeed, Axolotl, etc.

AI Toolkit for Healthcare Imaging

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

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

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

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

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

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

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

FlashInfer: Kernel Library for LLM Serving

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

A scikit-learn compatible neural network library that wraps PyTorch

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

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.

A programmable Mixture-of-Models router for heterogeneous LLM inference

Fine-tune LLMs from one YAML. Layer streaming trains an 8B model on a 4 GB laptop GPU.

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

Rust bindings for the C++ api of PyTorch.

Supercharge Your Model Training

Probabilistic time series modeling in Python

Implement a reasoning LLM in PyTorch from scratch, step by step

OpenMMLab's Next Generation Video Understanding Toolbox and Benchmark

An easy-to-use LLMs quantization package with user-friendly apis, based on GPTQ algorithm.

Time series forecasting with PyTorch

An Engine-Agnostic Deep Learning Framework in Java