albucore0.0.412 mo1 overlayA high-performance image processing library designed to optimize and extend the Albumentations library with specialized functions for advanced image transformations
albumentations2.0.82 mo1 overlayFast and flexible image augmentation library
amd-gaia0.22.010 d1 overlayLightweight agent framework for the edge and AMD Ryzen AI PCs
asteroid-filterbanks0.4.040 d1 overlayAsteroid's filterbanks for PyTorch (STFT, DCT, learnable, ...)
bigcode-eval0_pre2025071522 d1 overlayFramework for evaluating autoregressive code generation language models
bitnet999935 d1 overlayOfficial inference framework for 1-bit LLMs
bitsandbytes0.50.03 d1 overlayk-bit quantization (QLoRA) and 8-bit optimizers for PyTorch
caffe22.13.0-r905 d::gentoo2 overlaysA deep learning framework
caffe2-cuda2.12.025 d1 overlayA deep learning framework (CUDA backend)
caffe2-rocm2.12.025 d1 overlayA deep learning framework (ROCm backend)
clearml1.18.0::gentoo1 overlayAuto-Magical CI/CD to streamline your AI workload
colibri1.1.15 d1 overlayRun GLM-5.2 (744B MoE) on consumer hardware with expert streaming
comfy-aimdo0.4.1022 d1 overlayAI Model Demand Offloading Allocator
pyannote-core6.0.140 d1 overlayData structures for temporal segments with attached labels
pyannote-database6.1.140 d1 overlayReproducible experimental protocols for multimedia databases
pyannote-metrics4.140 d1 overlayReproducible evaluation, diagnostic, and error analysis for speaker diarization
pyannote-pipeline4.0.040 d1 overlayTunable pipelines for hyperparameter optimization (atop optuna)
pyannoteai-sdk0.4.040 d1 overlayOfficial Python SDK for the pyannoteAI hosted diarization API
pytorch2.13.0-r909 d::gentoo2 overlaysTensors and Dynamic neural networks in Python
pytorch-cuda2.12.025 d1 overlayTensors and Dynamic neural networks in Python (CUDA backend)
pytorch-metric-learning2.9.040 d1 overlayLoss functions, samplers, and trainers for metric learning in PyTorch
pytorch-rocm2.12.025 d1 overlayTensors and Dynamic neural networks in Python (ROCm backend)
rapidocr3.9.22 d1 overlayAwesome OCR multiple programing languages toolkits based on ONNXRuntime, OpenVINO, PaddlePaddle and PyTorch
safetensors0.8.022 d::gentoo4 overlaysSimple, safe way to store and distribute tensors
sd-swarmui-web0.9.828 d1 overlayStable Diffusion SwarmUI web application
segment-anything1.024 d1 overlayMeta AI Segment Anything Model (SAM)
sentence-transformers5.6.13 d1 overlaySentence/image embeddings, retrieval, and reranking via Transformers
sentencepiece0.2.1::gentoo1 overlayText tokenizer for Neural Network-based text generation
seqeval1.2.2-r4::gentoo1 overlayPython framework for sequence labeling evaluation
sherpa-onnx1.13.417 d1 overlaySpeech-to-text, TTS, speaker diarization etc. using onnxruntime
sherpa-onnx-bin1.13.420 d1 overlaySpeech-to-text, TTS, speaker diarization etc. using onnxruntime (binary wheels)
spandrel0.4.22 mo1 overlaySpandrel gives your project support for various PyTorch architectures meant for AI Super-Resolution, restoration, and inpainting
stable-diffusion-cpp0.0.66938 h2 overlaysDiffusion model (SD, Flux, Wan, Qwen-Image, ...) inference in pure C/C++
tensorpipe2025.11.05::gentoo1 overlayprovides a tensor-aware channel
tensorzero2026.6.043 d1 overlayTensorZero is an open-source stack for industrial-grade LLM applications
tokenizers0.23.114 mo::gentoo2 overlaysImplementation of today's most used tokenizers
torch-audiomentations0.12.040 d1 overlayFast, differentiable, GPU-friendly audio augmentations for PyTorch
torch-pitch-shift1.2.540 d1 overlayGPU-friendly pitch-shifting for PyTorch tensors
torchaudio2.11.09 d2 overlaysData manipulation and transformation for audio signal processing
torchaudio-cuda2.9.124 d1 overlayAudio processing for PyTorch (CUDA backend)
torchaudio-rocm2.9.124 d1 overlayAudio processing for PyTorch (ROCm backend)
torchcodec0.15.012 d1 overlayDecode/encode video and audio into PyTorch tensors via FFmpeg
torchdata0.11.0::gentoo1 overlayA repo for data loading and utilities on PyTorch domain libraries
torchmetrics1.9.040 d1 overlayMachine learning metrics for distributed PyTorch applications
torchsde0.2.62 mo1 overlayDifferentiable SDE solvers with GPU support and efficient sensitivity analysis
torchvision0.28.05 d::gentoo2 overlaysDatasets, transforms and models to specific to computer vision
torchvision-cuda0.27.025 d1 overlayDatasets, transforms and models for computer vision (CUDA backend)
torchvision-rocm0.27.025 d1 overlayDatasets, transforms and models for computer vision (ROCm backend)
transformers5.14.1-r66610 d::gentoo3 overlaysState-of-the-art Machine Learning for JAX, PyTorch and TensorFlow
trl1.9.05 d1 overlayTrain transformer language models with reinforcement learning (SFT, DPO, GRPO)