#pytorch

🐍 Geometric Computer Vision Library for Spatial AI

Swap GPT for any LLM by changing a single line of code. Xinference lets you run open-source, speech, and multimodal models on cloud, on-prem, or your laptop — all through one unified, production-ready inference API.

Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.

cuDNN Frontend is NVIDIA's modern, open-source entry point to the cuDNN library and a growing collection of high-performance open-source kernels.

A library for training and deploying machine learning models on Amazon SageMaker

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

DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.

Flower: A Friendly Federated AI Framework

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

Pretrained model hub for Keras 3.

Bayesian optimization in PyTorch

We write your reusable computer vision tools. 💜

sbi is a Python package for simulation-based inference, designed to meet the needs of both researchers and practitioners. Whether you need fine-grained control or an easy-to-use interface, sbi has you covered.

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

Multilingual neural TTS (6 languages: JA/EN/ZH/ES/FR/PT, code supports SV) — C++, C#, Rust, Go, Python, npm (WASM). VITS + Prosody, streaming, CUDA/CoreML/DirectML. pip install piper-plus | npm install piper-plus | cargo install piper-plus-cli

FlagGems is an operator library for large language models implemented in the Triton Language.

🦀 Low-level 3D Computer Vision library in Rust

A python library for self-supervised learning on images.

Low-code framework for building custom LLMs, neural networks, and other AI models

Time series forecasting with PyTorch

Burn is a next generation tensor library and Deep Learning Framework that doesn't compromise on flexibility, efficiency and portability.

Streamlining reinforcement learning with RLOps. State-of-the-art RL algorithms and tools, with 10x faster training through evolutionary hyperparameter optimization.

Brevitas: neural network quantization in PyTorch

Image augmentation for computer vision. AGPL-3.0-only or commercial licensing.

Probabilistic time series modeling in Python

DeepInverse: a PyTorch library for solving imaging inverse problems using deep learning

Machine learning on FPGAs using HLS

🚀 Accelerate inference and training of 🤗 Transformers, Diffusers, TIMM and Sentence Transformers with easy to use hardware optimization tools

Open security scanner and self-hosted control plane for AI, MCP, and cloud. One evidence model — run scans in your environment, centralize findings, govern in your VPC.

Pytorch domain library for recommendation systems