Provides reusable PyTorch modules and utilities to build, train and scale Graph Neural Networks — includes many implemented GNN layers, benchmark datasets, minibatch/sampling loaders, and support for large-scale, heterogeneous, temporal and point-cloud graphs.
Swaps faces in images and videos using deep learning, offering tools to extract faces, train generative models, and convert media via CLI or GUI for research, VFX, and ethical experimentation.
Provides composable function transformations for Python+NumPy—automatic differentiation, JIT compilation, and vectorization—while using XLA to run and scale numerical programs on GPUs/TPUs for ML and scientific computing.
Web and desktop/mobile WebUI for generating, editing, captioning and processing images and videos with Stable Diffusion and many diffusion models. Key features include automatic model download, SDNQ on-the-fly quantization for VRAM savings, balanced CPU/GPU offload, multi-backend GPU support, and built-in captioning/tagging/upscaling workflows.
Web UI to train and run retrieval-based voice conversion models from small datasets (≈10 minutes), featuring top-1 feature retrieval to avoid timbre leakage, model fusion, real-time conversion, vocal separation, and multi-hardware support.
Provides a self-hosted inference engine that serves all models an agent needs—embeddings, retrieval/reranking, document-to-markdown OCR, structured extraction, content-safety scoring, and LLM generation—through an OpenAI-compatible API. Bundles a 100+ model catalog, SDKs, and production deployment tooling.
Local WebUI for video and audio dubbing: download YouTube, isolate vocals, transcribe with Whisper, translate into 100+ languages, and generate multilingual TTS or zero-shot voice clones. Uses Gradio, yt-dlp, Demucs, Whisper/Faster-Whisper, F5/E2/CosyVoice and Edge-TTS; Windows-focused with optional Azure integration.
Generates and edits full-length music locally using a hybrid language-model planner and a diffusion transformer. Provides text- and audio-driven song synthesis plus selective editing (cover generation, repainting, vocal→BGM), metadata control and lightweight LoRA personalization, and runs on consumer GPUs/CPUs.
Provides 6,000 runnable, operator-level PyTorch tasks for training and evaluating CUDA kernel generation models; each sample includes executable code, operator descriptors, and provenance tags, with execution-driven filtering to ensure reproducibility and contamination control.
Runs local speech-to-text inference for a wide range of ASR model families using GGUF models on the ggml runtime. Supports Metal, Vulkan, and CUDA GPU backends plus a tinyBLAS-accelerated CPU path, prebuilt GGUFs on Hugging Face, and a quantization tool.
Local-first voice workflows for cloning, multi-engine TTS/ASR, video dubbing, dictation, transcription and audiobook production across 646 languages. Desktop app with a local OpenAI-compatible API, engine catalogue (TTS/ASR/LLM), and explicit opt-ins for remote features to keep audio and projects on-device.
Provides high-performance CUDA/CUTLASS kernels implementing Kimi Delta Attention (KDA), accelerating KDA prefill on SM90+ (Hopper) GPUs. Integrates as a drop-in backend for flash-linear-attention, supports native variable-length batching, and targets K=V=128; requires CUDA 12.9+/PyTorch 2.4+.