awesome-repositories.com
Blog
awesome-repositories.com

Descoperă cele mai bune repository-uri open source cu căutare AI.

ExploreazăCăutări recomandateAlternative open-sourceSoftware self-hostedBlogHartă site
ProiectDespreCum realizăm clasamentulPresăServer MCP
LegalConfidențialitateTermeni
© 2026 Bringes Technology SRL·VAT RO45896025·hello@awesome-repositories.com
·

8 repository-uri

Awesome GitHub RepositoriesAutomatic Batch Size Optimization

Mechanisms that automatically determine the most efficient batch size based on the hardware and model architecture.

Distinct from Batch Size Tuning: Focuses on automatic optimization of batch size for throughput, rather than manual tuning of hyperparameters.

Explore 8 awesome GitHub repositories matching artificial intelligence & ml · Automatic Batch Size Optimization. Refine with filters or upvote what's useful.

Awesome Automatic Batch Size Optimization GitHub Repositories

Găsește cele mai bune repo-uri cu AI.Vom căuta cele mai potrivite repository-uri folosind AI.
  • openvinotoolkit/openvinoAvatar openvinotoolkit

    openvinotoolkit/openvino

    10,414Vezi pe GitHub↗

    OpenVINO is an AI inference engine and model serving platform designed to execute optimized deep learning models across CPUs, GPUs, and NPUs through a unified API. It includes a model optimization toolkit for converting, quantizing, and compressing models from various frameworks, alongside a specialized generative AI runtime for large language models. The project distinguishes itself through a plugin-based hardware acceleration layer that maps neural network operations to vendor-specific drivers. It features advanced execution mechanisms such as continuous batching, speculative decoding, and

    Automatically applies optimal batch sizes based on hardware to maximize total inference throughput.

    C++aicomputer-visiondeep-learning
    Vezi pe GitHub↗10,414
  • halide/halideAvatar halide

    halide/Halide

    6,572Vezi pe GitHub↗

    Ships an automatic scheduler that uses a learned cost model and beam search to optimize pipeline performance.

    C++compilerdslgpu
    Vezi pe GitHub↗6,572
  • cocoindex-io/cocoindexAvatar cocoindex-io

    cocoindex-io/cocoindex

    6,117Vezi pe GitHub↗

    Cocoindex is an incremental data processing engine that builds and maintains live indexes for AI agents, with a core focus on codebase indexing and knowledge graph extraction. The engine uses a function-graph execution model where user-defined Python functions are composed into a directed acyclic graph, and it processes data incrementally so only changed source records or code paths are re-computed, avoiding full recomputation at any scale. It supports automatic schema inference from transformation pipeline type annotations and provides full data lineage tracing, tagging every output record wi

    Automatically batches GPU and ML workloads like text embeddings for higher throughput.

    Rustagentic-data-frameworkaiai-agents
    Vezi pe GitHub↗6,117
  • mosaicml/composerAvatar mosaicml

    mosaicml/composer

    5,485Vezi pe GitHub↗

    Composer este un framework de antrenare distribuită PyTorch conceput pentru scalarea modelelor de mari dimensiuni pe clustere GPU multi-nod. Acesta funcționează ca un antrenor de modele lingvistice mari (LLM), un optimizator de modele distribuite și un manager al ciclului de viață al antrenării. Proiectul se diferențiază ca o bibliotecă de regularizare pentru deep learning, oferind tehnici de optimizare specializate precum Sharpness Aware Minimization, MixUp și CutMix pentru a îmbunătăți generalizarea modelului. De asemenea, distinge fluxul de antrenare prin utilizarea warmup-ului pentru lungimea secvenței, înghețarea progresivă a straturilor și checkpointing-ul stării sharded pentru recuperarea modelelor la scară largă. Framework-ul acoperă o suprafață largă de capabilități, inclusiv orchestrarea antrenării distribuite, gestionarea hardware-ului cu precizie mixtă și streaming-ul de date cloud-native. Oferă, de asemenea, instrumente extinse de monitorizare și observabilitate pentru diagnosticarea memoriei GPU, detectarea divergenței antrenării și urmărirea throughput-ului. Proiectul include un launcher CLI pentru automatizarea execuției joburilor de antrenare multi-GPU pe mai multe noduri.

    Adjusts microbatch sizes and gradient accumulation rates dynamically to prevent out-of-memory errors.

    Python
    Vezi pe GitHub↗5,485
  • reactwg/react-18Avatar reactwg

    reactwg/react-18

    5,195Vezi pe GitHub↗

    Acest proiect este un grup de lucru software colaborativ și un track de lansare axat pe dezvoltarea tehnică și implementarea actualizării bibliotecii React 18. Acesta servește ca efort de coordonare comunitară și forum de discuții pentru gestionarea milestone-urilor și a seturilor de funcționalități ale acestei versiuni majore a framework-ului frontend. Grupul de lucru facilitează coordonarea lansărilor open source și planificarea versiunilor software printr-un grup distribuit de contribuitori. Se concentrează pe colectarea feedback-ului tehnic din comunitate și gestionarea discuțiilor publice pentru a rafina codul și documentația bibliotecii înainte de o lansare oficială. Domeniul de dezvoltare acoperă randarea concurrentă a interfeței utilizator, gestionarea stării frontend și rafinarea logicii interne de reconciliere și randare.

    Groups multiple state updates into a single render pass to improve performance and reduce repaints.

    Vezi pe GitHub↗5,195
  • uxlfoundation/onednnAvatar uxlfoundation

    uxlfoundation/oneDNN

    4,009Vezi pe GitHub↗

    oneDNN este o bibliotecă pentru accelerarea deep learning-ului care oferă blocuri de construcție optimizate pentru antrenarea și inferența rețelelor neuronale. Aceasta gestionează calculul tensorial pe hardware CPU și GPU, permițând execuția de primitive de înaltă performanță pentru antrenarea modelelor și optimizarea inferenței rețelelor neuronale. Proiectul se distinge prin optimizarea kernel-ului specifică hardware-ului și utilizarea compilării „just-in-time” pentru a viza seturi de instrucțiuni specifice ale procesorului. Suportă execuția rețelelor neuronale cuantizate folosind cuantizarea statică și dinamică pentru a reduce utilizarea memoriei și a crește throughput-ul. Biblioteca acoperă o gamă largă de capabilități, inclusiv primitive de deep learning precum convoluții, înmulțirea matricelor și execuția rețelelor neuronale recurente. Implementează optimizări avansate de performanță, inclusiv fuziunea operațiunilor, optimizarea grafului de calcul și gestionarea formatului memoriei. Integrarea este oferită printr-un ABI C stabil și un wrapper C++, cu suport pentru SYCL, OpenCL și biblioteci externe de algebră liniară. Sistemul include instrumente de observabilitate pentru profilarea performanței hardware, benchmarking-ul primitivelor și log-uri detaliate de execuție.

    Uses grouped memory formats to handle batches of varying sizes efficiently within a single operation.

    C++aarch64amxavx512
    Vezi pe GitHub↗4,009
  • iree-org/ireeAvatar iree-org

    iree-org/iree

    3,819Vezi pe GitHub↗

    IREE is an MLIR-based compiler toolchain and runtime designed to translate machine learning models from various frameworks into optimized binaries for execution across diverse hardware targets. It provides a unified pipeline to ingest models from PyTorch, TensorFlow, JAX, and ONNX, lowering them into a common intermediate representation for deployment on CPUs, GPUs, and bare-metal embedded systems. The project distinguishes itself through a bytecode virtual machine and a hardware abstraction layer that decouple high-level model logic from specific hardware instruction sets. It supports sophis

    Groups small, similar operations together to increase batch size and improve memory cache utilization.

    C++compilercudajax
    Vezi pe GitHub↗3,819
  • llm-d/llm-dAvatar llm-d

    llm-d/llm-d

    2,514Vezi pe GitHub↗

    llm-d is a distributed serving framework designed for large language model inference. It functions as an inference orchestrator and gateway, providing a control plane for deploying model replicas and managing hardware accelerators. The system includes a batch inference scheduler and a cache manager to coordinate request flow and memory utilization. The project is distinguished by a disaggregated serving architecture that separates prefill and decode execution phases across specialized workers to maximize throughput. It employs a hardware-agnostic control plane and tiered cache offloading, mov

    Provides a disaggregated prefill and decode topology specifically designed to maximize throughput for batch-intensive LLM workloads.

    Shell
    Vezi pe GitHub↗2,514
  1. Home
  2. Artificial Intelligence & ML
  3. Batch Size Tuning
  4. Automatic Batch Size Optimization

Explorează sub-etichetele

  • Automatic ML Workload Batching1 sub-tagAutomatic batching of GPU and ML workloads like text embeddings to achieve higher throughput without manual configuration. **Distinct from Automatic Batch Size Optimization:** Distinct from Automatic Batch Size Optimization: focuses on batching entire workloads rather than just optimizing batch size for a single model.
  • Dynamic Batch Size Adjustment1 sub-tagAutomatically adjusting batch sizes and gradient accumulation rates during runtime to optimize memory and throughput. **Distinct from Automatic Batch Size Optimization:** Focuses on runtime adjustment to prevent OOM errors, rather than static optimization of the most efficient batch size.
  • Schedule Optimization via Learned Cost ModelsUses a learned cost model and beam search to automatically find performant schedules for image processing pipelines. **Distinct from Automatic Batch Size Optimization:** Distinct from Automatic Batch Size Optimization: focuses on optimizing execution schedules for image pipelines, not batch sizes for ML training.