7 Repos
Techniques for reducing memory usage during model inference by disabling gradient tracking.
Distinguishing note: None of the candidates cover the specific act of deactivating gradient recording for inference efficiency.
Explore 7 awesome GitHub repositories matching artificial intelligence & ml · Inference Memory Optimizations. Refine with filters or upvote what's useful.
This project is a comprehensive collection of educational examples and reference implementations for building vision and language models using PyTorch. It serves as a deep learning tutorial covering the end-to-end process of developing neural networks, from initial architecture definition to final production deployment. The repository provides detailed guides on implementing a wide range of domain-specific models, including convolutional neural networks for object detection and segmentation, as well as transformer and recurrent architectures for natural language processing. It emphasizes gene
Suppresses the recording of intermediate variables during inference to reduce memory consumption.
TabPFN trades additional computation for reduced peak memory usage during model inference.
This project is a neural network extension for Stable Diffusion that provides spatial control and geometric consistency for text-to-image generation. It functions as an image structure controller and conditioning tool, enabling the use of external inputs to guide the layout and geometry of generated imagery. The framework is distinguished by its ability to transform input images into structural guides through various preprocessors. These include the extraction of depth maps, normal maps, and human pose landmarks, as well as the detection of Canny edges, anime lineart, and straight architectur
Offers configurable memory saving modes that trade computation speed for reduced peak memory usage during image generation.
Text2Video-Zero is a text-to-video diffusion model and framework designed to synthesize temporally consistent video sequences from textual prompts. It functions as a zero-shot video generator, repurposing pre-trained image diffusion models to create video content without requiring additional training on video datasets. The system includes a conditional video synthesizer that allows for guided generation using depth, edge, or pose maps to control structural layout and movement. It also provides text-based video editing capabilities to modify the style or content of existing video clips through
Lowers GPU memory requirements during the generation process using frame chunking and token merging.
This project provides a self-hosted server for automatic speech recognition, functioning as a containerized inference engine for the Whisper model. It exposes core transcription and translation capabilities through a standardized web interface, allowing for the integration of speech-to-text services into external applications. The service distinguishes itself by incorporating advanced audio analysis tools, including speaker diarization to attribute text to specific individuals and voice activity detection to filter non-speech segments. It supports automated language detection and provides out
Configures idle timeouts and model caching to manage memory usage and select efficient hardware compute types for inference tasks.
Map-anything is a 3D scene reconstruction framework and neural geometry estimator designed to transform two-dimensional images into metric three-dimensional spatial representations using feed-forward neural networks. It provides a specialized toolkit for predicting camera intrinsics and ray directions from single images without requiring external geometric metadata. The project includes a 3D model benchmarking suite that utilizes a unified model wrapper to standardize outputs from diverse reconstruction models. This allows for consistent evaluation and accuracy measurement across various spat
Reduces GPU memory consumption during inference to allow processing of more views on limited hardware.
Long Llama ist ein Transformer-basiertes Sprachmodell und Fine-Tuning-Framework, das darauf ausgelegt ist, logische Kohärenz über Eingabesequenzen hinweg zu verarbeiten und aufrechtzuerhalten, die Standard-Längenbegrenzungen signifikant überschreiten. Durch die Nutzung einer fokussierten Transformer-Architektur ermöglicht das Projekt Modellen die Handhabung massiver Dokumente oder ganzer Bücher, indem Attention-Schichten darauf trainiert werden, entfernte Token zu verfolgen. Das Framework zeichnet sich durch spezialisierte Attention-Mechanismen aus, die die Verarbeitung von Hunderttausenden von Token ermöglichen. Es integriert speichereffiziente Inferenztechniken, wie Key-Value-Caching und Query-Grouping, die es Benutzern ermöglichen, den Hardware-Ressourcenverbrauch zu konfigurieren und den Rechenaufwand gegen die Anforderungen der Long-Range-Dependency-Verfolgung abzuwägen. Über seine Kern-Modellierungsfähigkeiten hinaus bietet das Projekt Tools zur Anpassung bestehender Open-Source-Modelle an neue Aufgaben. Es unterstützt Fine-Tuning durch kontrastives Lernen, was die Erstellung von Modellen ermöglicht, die zu spezialisiertem Instruction-Following und automatisierter Dokumentenanalyse fähig sind.
Optimizes hardware resource consumption by configuring cache layers and attention query grouping to balance speed against memory.