Simplest working implementation of Stylegan2
Differentiable SDE solvers with GPU support and efficient sensitivity
High quality, fast, modular reference implementation of SSD in PyTorch
Deep learning optimization library making distributed training easy
Tensor Learning in Python
Real-Time High-Resolution Background Matting
Easy-to-use LLM fine-tuning framework (LLaMA-2, BLOOM, Falcon
CPT: A Pre-Trained Unbalanced Transformer
Data loaders and abstractions for text and NLP
Low-latency REST API for serving text-embeddings
Standardized Serverless ML Inference Platform on Kubernetes
Ray Aviary - evaluate multiple LLMs easily
Chinese LLaMA & Alpaca large language model + local CPU/GPU training
MII makes low-latency and high-throughput inference possible
Trainable, memory-efficient, and GPU-friendly PyTorch reproduction
Basaran, an open-source alternative to the OpenAI text completion API
PyTorch extensions for fast R&D prototyping and Kaggle farming
Open Source Differentiable Computer Vision Library
A set of Docker images for training and serving models in TensorFlow
Nexa SDK is a comprehensive toolkit for supporting ONNX and GGML
A Pythonic framework to simplify AI service building
Implementation of Denoising Diffusion Probabilistic Model in Pytorch
A library for accelerating Transformer models on NVIDIA GPUs
JAX-based neural network library
Core ML tools contain supporting tools for Core ML model conversion