20 Repos
Strategies for splitting model parameters across multiple hardware devices to handle large-scale models.
Distinct from Model Weight Management: Focuses on the distribution of weights across hardware for execution, distinct from general weight management/storage.
Explore 20 awesome GitHub repositories matching artificial intelligence & ml · Weight Distribution. Refine with filters or upvote what's useful.
Llama is a large language model runtime and inference engine designed to load and execute autoregressive transformer models. It enables the generation of natural language text completions from prompts using pretrained weights. The system features multi-GPU model parallelism, which distributes model weights and workloads across multiple graphics processors to support larger parameter counts. It also incorporates a content safety filter that uses classifiers to intercept and block unsafe inputs or outputs during the inference process. The project covers broad capabilities in distributed model
Distributes model weights across multiple graphics processors to support models larger than a single GPU's memory.
This project is a comprehensive Chinese translation of a technical deep learning textbook, providing an educational resource on the theory and implementation of neural networks. It functions as a collaborative technical translation project designed to make complex academic AI literature accessible to non-English speakers. The project utilizes a community-driven translation model that integrates external suggestions and pull requests to refine linguistic accuracy and reduce bias. It employs standardized terminology mapping to ensure a uniform vocabulary throughout the translated content. To i
Details the use of moving averages to maintain lagged model parameters for improved training stability.
Sglang is a high-performance inference engine and serving system designed for large language and multimodal models. It provides a programmable interface for orchestrating complex generation workflows, enabling developers to coordinate multi-turn dialogues, tool invocations, and reasoning chains through a domain-specific language. The platform is built to support production-scale deployments, offering an OpenAI-compatible API that allows for integration with existing application ecosystems. The system distinguishes itself through a disaggregated architecture that separates compute-intensive pr
Splits model parameters across multiple devices to enable the execution of large models that exceed the memory capacity of a single hardware unit.
MiniCPM-V is a multimodal large language model and vision-language system designed for complex visual and linguistic understanding. It functions as an on-device AI model, providing the capacity to process text, images, and video as a compact neural network. The project is specifically developed as an edge AI framework, utilizing quantization and weight sharding to run on memory-constrained mobile chipsets. This allows for the deployment of multimodal intelligence directly on mobile operating systems for local inference. Its capabilities cover multimodal content analysis of high-resolution im
Splits model layers across multiple graphics processors to enable the execution of large networks on memory-constrained hardware.
This project is a comprehensive framework and toolkit for developing, optimizing, and deploying transformer-based models across multimodal, document intelligence, and natural language processing tasks. It provides a unified neural architecture that processes text, vision, audio, and document layout data through a shared set of weights, enabling researchers and developers to build foundational models that align cross-modal representations. The platform distinguishes itself through advanced training and inference strategies designed for large-scale deep learning. It incorporates specialized mec
Provides distributed weight partitioning strategies to handle memory-intensive computations across multiple processors during large-scale model development.
TensorFlow.js is a JavaScript machine learning library used for training and deploying models in web browsers and server-side environments. It functions as a browser-based model trainer, a WebAssembly inference engine, and a WebGPU accelerated tensor library for low-level linear algebra. The project also includes a model converter to transform Python-based models into optimized formats for JavaScript execution. The library distinguishes itself through a pluggable backend architecture that allows mathematical operations to be executed via CPU, WebGL, or WebGPU. It supports the conversion of Py
Splits large model weights into smaller binary shards to optimize browser caching and parallel loading.
This library provides a comprehensive framework for fine-tuning, aligning, and distilling transformer-based language models. It serves as a toolkit for adapting models to specialized domains through supervised learning, while offering advanced methodologies to improve output quality and reasoning capabilities. The project distinguishes itself through specialized alignment and optimization techniques, including direct preference optimization and reinforcement learning, which allow models to be tuned against human preferences without complex reward modeling. It further supports training efficie
Maintains bias-corrected moving averages of model weights to ensure training stability and faster convergence.
This project is a comprehensive engineering framework and technical reference for managing, scaling, and optimizing distributed machine learning infrastructure. It provides a suite of methodologies and diagnostic tools designed to support large-scale model training and inference on high-performance computing clusters. The project distinguishes itself through a specialized diagnostic toolkit and infrastructure optimization suite that addresses the complexities of multi-node environments. It enables precise control over cluster resources, including hardware maintenance, network topology configu
Distributes model weights and computations across multiple hardware devices to accommodate large-scale models.
ChatGLM2-6B is an open-weight large language model designed for natural language conversations and text generation in both English and Chinese. It functions as a bilingual chat model capable of processing and maintaining coherence across text sequences up to 32K tokens. The model is optimized for local deployment through precision quantization, which reduces memory requirements to allow execution on consumer-grade hardware. It supports distributing model weights across multiple graphics cards to handle parameters that exceed the memory of a single device. The project covers capabilities for
Distributes model weights across multiple graphics cards to handle large-scale parameter sets.
Qwen3-Coder is a specialized large language model designed for software development, technical reasoning, and automated code synthesis. Built on transformer-based sequence modeling, it functions as a multilingual programming assistant capable of generating, completing, and debugging source code across more than one hundred programming languages. The model distinguishes itself through its capacity to process and maintain logical coherence across massive datasets, supporting context windows of up to one million tokens. This allows for repository-scale reasoning, enabling the model to analyze co
Provides open-weight files to facilitate local execution and integration into various machine learning frameworks.
This project is a collection of pre-trained machine learning models and conversion pipelines designed for running inference directly in the browser using TensorFlow.js. It provides a library of ready-to-use models for computer vision, audio classification, and natural language processing tasks. The suite includes specialized tools for transforming Python-based Keras models into JSON formats compatible with web environments. It enables the deployment of these models by fetching architectures and weight shards via HTTP for client-side execution. The project covers a broad range of capabilities
Splits large model weights into smaller binary files to optimize loading speed and memory usage during initialization.
Petals is a decentralized framework and inference engine for running large language models across a peer-to-peer network. It enables the execution of models that exceed the memory of any single machine by splitting computations and model layers across a collaborative swarm of GPUs. The system functions as a collaborative compute network where participants share local GPU resources and host model weights. It supports distributed prompt-tuning to adapt massive models to specific tasks and allows for the establishment of private compute swarms to process sensitive data within restricted, trusted
Implements strategies for splitting and hosting model parameters across multiple GPU devices in a distributed swarm.
TensorFlow.js is a JavaScript machine learning library and browser-based runtime used to build, train, and execute models. It functions as a WebGL accelerated tensor engine, providing a foundation for high-performance linear algebra operations and an automatic differentiation framework for computing gradients. The project distinguishes itself through its ability to run machine learning directly in web environments, supporting both client-side inference and browser-based training. It enables the deployment of Python-based models by converting Keras or TensorFlow models into compatible formats
Retrieves large model parameters in fragmented binary shards to optimize network transfer and memory.
SlowFast is a PyTorch video understanding framework and spatiotemporal neural network library. It serves as a toolset for video action recognition, enabling the training and evaluation of models designed to classify complex activities and objects within video sequences. The framework is distinguished by its use of dual-pathway spatiotemporal sampling to capture both slow and fast motions. It supports self-supervised video learning for pre-training models on unlabeled data and employs multigrid spatiotemporal training to optimize learning across multiple spatial and temporal resolutions. The
Provides weighted temporal aggregation to combine features across different time scales into a single video representation.
This project provides a comprehensive technical guide and framework for engineering large-scale machine learning systems. It covers the full lifecycle of model development, focusing on the infrastructure and computational principles required to build, train, and serve generative AI models across distributed GPU clusters. The repository distinguishes itself by offering deep-dive tutorials and implementation strategies for complex system challenges. It emphasizes high-performance architectural primitives, such as collective communication orchestration, distributed tensor sharding, and static gr
Provides strategies for calculating importance weights to balance bias and variance in reinforcement learning.
PlugNPlay-Modules is a collection of reusable PyTorch computer vision modules and deep learning architectural components. It provides a library of standardized building blocks for constructing neural networks, focusing on attention mechanisms, signal processing layers, and feature fusion modules. The project is distinguished by its extensive variety of attention primitives, covering spatial, channel, and temporal weighting, as well as specialized variants like deformable, frequency-enhanced, and linear-complexity attention. It also implements advanced signal processing tools within the neural
Fuses two sets of input features using multi-scale dilated convolutions and attention for change detection.
UniAD ist ein einheitliches Deep-Learning-Framework für autonomes Fahren, das Wahrnehmung, Vorhersage und Planung in einem einzigen End-to-End-Modell integriert. Es fungiert als neuronale Netzwerkarchitektur, die rohe Sensordaten direkt in Fahrtrajektorien und Bewegungspläne übersetzt. Dieses Projekt dient als Forschungsimplementierung eines planungsorientierten Ansatzes, der Module für Belegung, Kartierung und Objektverfolgung gemeinsam trainiert. Es nutzt ein Multi-Task-Wahrnehmungs-Framework, um die gesamte Fahrleistung zu optimieren. Das System deckt ein breites Fähigkeitsspektrum ab, einschließlich End-to-End-Fahr-Pipelines, Optimierung der Fahrzeugbewegung und Aggregation visueller Merkmale. Es koordiniert verschiedene Aufgaben des autonomen Fahrens, um den gesamten Fahrprozess in einem einzigen Trainingszyklus zu verfeinern.
Combines visual information across multiple video frames to create stable representations for tracking and mapping.
mmtracking is a PyTorch video perception framework designed for training and deploying computer vision models that analyze sequential image data. It provides specialized tools for multi-object tracking, video instance segmentation, and a configuration-driven system for managing deep learning models. The project utilizes a deep learning model registry and a configuration-driven pipeline to swap model backbones and detectors without modifying the core codebase. This modular approach allows for the development of custom perception architectures by combining various components and configurations.
Aggregates visual features across sequential frames to maintain object identity and consistency in video streams.
LightLLM is a high-performance serving framework for deploying and executing large language models. It functions as a multi-GPU inference engine and server capable of handling dense architectures, mixture-of-experts designs, and multimodal models that process both text and images. The system is distinguished by its specialized support for Mixture-of-Experts models using expert parallelism and fused kernels. It implements structured text generation through deterministic state machines and pushdown automata to enforce precise output formats. To optimize throughput, the framework employs specula
Splits model weights for embedding and transformer layers across multiple devices based on tensor parallelism.
This project is a deep learning computer vision library designed for video action recognition. It provides a framework for training and evaluating neural networks that identify and categorize human activities within recorded footage by processing temporal sequences of frames. The library focuses on the implementation of three-dimensional neural network architectures, specifically utilizing three-dimensional convolutional layers to capture both spatial and temporal patterns. By aggregating features across consecutive frame sequences, the models learn to represent the evolution of actions over
Combines spatial information from individual frames into a unified representation to model the evolution of actions over time.