51 个仓库
Tools that refine deep learning models by optimizing computational graphs and improving execution efficiency on hardware.
Explore 51 awesome GitHub repositories matching artificial intelligence & ml · Deep Learning Optimization. Refine with filters or upvote what's useful.
This project is an open-source, interactive educational platform designed to teach deep learning through a comprehensive, code-first curriculum. It provides a structured learning path that covers foundational mathematics, modern neural network architectures, and practical optimization techniques, enabling practitioners to master complex artificial intelligence concepts through hands-on experimentation. The platform distinguishes itself by integrating technical explanations with executable Jupyter notebooks. This design allows readers to modify code and hyperparameters in real-time, facilitati
Investigates how high-level model definitions are compiled into optimized execution graphs for hardware acceleration.
This project is a deep learning library designed for training neural networks on irregular data structures, including graphs, 3D meshes, and point clouds. It functions as an extension to the PyTorch framework, providing specialized layers and kernels that enable the processing of complex, non-Euclidean information. The library distinguishes itself through a geometric deep learning toolkit that manages the unique requirements of graph-based data. It utilizes sparse matrix-based message passing to aggregate information across nodes and employs dynamic computational graph construction to accommo
Compiles and accelerates neural network layers using just-in-time processing and specialized hardware instructions to improve training and inference speed.
Paddle is a deep learning framework designed for building, training, and deploying neural networks. It provides a platform for constructing models using tensor-based computations and supports both dynamic and static execution graphs to facilitate research and production workflows. The platform functions as a distributed machine learning system, enabling the scaling of training workloads across multiple nodes and hardware clusters. It includes a comprehensive toolkit for model deployment and optimization, allowing users to convert external model formats, compress trained models for resource-co
Provides a comprehensive platform for building and training complex neural networks using modular layers and tensor operations.
Triton is a parallel computing framework and high-level programming language designed for writing custom compute kernels. It functions as a deep learning compiler, translating complex mathematical operations into high-throughput instructions that maximize hardware utilization and memory efficiency on graphics processing units. The framework distinguishes itself through a hardware-agnostic compute abstraction that allows developers to define kernels without manual low-level tuning. It employs just-in-time compilation to generate optimized binary instructions at runtime, utilizing static data f
Translates complex mathematical operations into high-throughput compute instructions that maximize hardware utilization.
This project is a comprehensive computer vision library for the PyTorch ecosystem, providing a standardized collection of neural network architectures, datasets, and high-performance transformation utilities. It serves as a foundational framework for building, training, and deploying deep learning models, offering a centralized model registry that allows developers to instantiate architectures with pre-trained weights for tasks such as image classification, object detection, and semantic segmentation. The library distinguishes itself through its modular approach to data and compute management
Applies geometric and color transformations to visual datasets to improve model robustness and generalization during training.
Mamba is a deep learning framework designed for building and training sequence models that process long-range data dependencies with linear-time computational efficiency. By utilizing selective state space modeling, the library enables the construction of neural network architectures that replace traditional attention mechanisms with high-performance state space operations. The framework distinguishes itself through the use of data-dependent state gating, which allows the model to dynamically filter information flow based on the input sequence. To ensure high throughput, it incorporates hardw
Executes high-performance selective state space model operations using optimized hardware kernels.
This repository is a deep learning for natural language processing course and curriculum. It provides educational material and guides focused on neural network architectures used for processing natural language, speech signals, and text classification. The content includes instructional tutorials on sequence modeling and neural language modeling, covering the implementation of n-gram and recurrent neural networks. It also provides a framework for studying word embeddings to map linguistic meanings into numerical representations. The curriculum covers a broad range of capabilities, including
Provides technical guidance on maximizing memory bandwidth and throughput for deep learning model execution.
Albumentations is a computer vision image augmentation library designed to increase training data diversity for deep learning models. It provides a toolset for applying geometric and color transformations to images and annotations, including a specialized collection of 3D operations for volumetric data used in medical and scientific imaging. The library functions as an image mask and bounding box transformer, automatically updating masks, bounding boxes, and keypoints when images undergo geometric changes. This ensures that spatial alterations remain synchronized across images and their assoc
Provides tools for creating varied synthetic training samples through pixel and geometry modifications.
This project is a collection of optimized scripts, deployment patterns, and reference implementations designed for scaling and accelerating state-of-the-art AI models. It serves as a multi-domain model zoo and a distributed training framework, providing PyTorch reference implementations for training and deploying models on GPU-accelerated infrastructure. The repository distinguishes itself through an optimization suite focused on NVIDIA GPU hardware, utilizing automatic mixed precision and specialized math modes to increase training speed and throughput. It provides enterprise deployment patt
Implements processes for preparing and moving optimized models into production execution on target hardware.
TVM is a machine learning compiler framework designed to convert deep learning models from various frameworks into optimized machine code. It functions as a cross-platform deployment engine that transforms high-level model definitions into efficient, hardware-specific binaries for diverse computing architectures. The system utilizes a multi-level compilation pipeline that decouples algorithm logic from hardware implementation through tensor-operator abstractions. It employs a graph-level intermediate representation to perform cross-operator optimizations and memory planning before lowering co
Provides a toolkit for scheduling and tuning computational graphs to maximize execution speed on CPUs, GPUs, and accelerators.
MLAlgorithms 是一个教育性机器学习算法库,由从零开始用 Python 实现的核心预测模型组成。它作为开发者的参考资料,通过简洁、最小化的实现来学习这些模型的内部逻辑和数学原理。 代码库专注于算法实现的研究和机器学习教育,提供了一种在不依赖繁重外部库的情况下,通过构建组件来理解内部机制的方法。 该项目利用面向对象封装和基于 NumPy 的向量化来管理模型状态并执行数学运算。其架构通过使用纯 Python 逻辑来实现线性代数原语和模块化参数初始化,从而强调透明度。
Provides modular weight initialization strategies separated from the training loop to allow for various randomization techniques.
alpaca.cpp is a high-performance local inference engine implemented in C++ for executing instruction-tuned large language models. It serves as a quantized model runtime designed to load and run model tensors on local hardware with minimal dependencies, removing the requirement for a full Python environment. The project focuses on on-device text generation and the deployment of private AI chatbots. It utilizes model weight quantization to reduce memory requirements and increase inference speed on consumer-grade devices. The system covers hardware-optimized model execution through thread-pool
Optimizes memory bandwidth and throughput on local hardware to maximize model execution efficiency.
This project provides educational materials and courseware focused on the theoretical and practical foundations of distributed systems design. It serves as a comprehensive curriculum covering the disciplines of consensus, data consistency, reliability engineering, and scalability. The instructional content focuses on achieving cluster agreement through consensus algorithms and managing system-wide state via coordination frameworks. It includes a dedicated guide to data theory, exploring replication strategies, consistency models, and data convergence. The courseware covers a broad capability
Instructional material on improving performance by aligning memory barriers and pinning processors.
This project is a machine learning study guide and technical knowledge base. It serves as a version-controlled repository of mathematical formulas and algorithmic explanations, providing instructional material and reference notes for the study of artificial intelligence. The content is structured as a markdown-based knowledge base that pairs theoretical mathematical explanations directly with code implementations. This approach demonstrates model mechanics in practice across several specialized domains, including deep learning research, probabilistic graphical modeling, and reinforcement lear
Evaluates research on implicit bias and duality to improve convergence for stochastic gradient descent.
OneFlow is a deep learning framework and distributed execution engine designed for building, training, and deploying neural network architectures. It functions as a scalable neural network library that allows for the development of deep learning models and their execution across distributed hardware. The project includes a machine learning graph compiler used to optimize neural network execution graphs. This allows for the acceleration of model performance and the reduction of latency during both training and inference. The framework covers broad capability areas including large-scale model
Optimizes computational graphs to accelerate model execution and reduce latency during training and inference.
AutoKeras is an automated machine learning framework and Keras AutoML library designed to discover the most effective deep learning model structures for a given dataset. It functions as a tool for deep learning architecture search, eliminating manual hyperparameter tuning by automatically searching for and optimizing neural network architectures. The framework provides capabilities for benchmarking and refining neural network designs to maximize performance. It includes a system for containerized machine learning deployment, allowing environments to be packaged into containers to ensure consi
Refines neural network structures to maximize performance and accuracy for specific data.
Apex is a high-performance toolkit for PyTorch designed to coordinate distributed training, execute fused GPU kernels, manage mixed precision, and implement optimized distributed optimizers. It provides specialized tools for scaling model training across multiple GPUs and nodes to increase processing speed and throughput. The library features high-performance implementations of Adam and LAMB optimizers to reduce synchronization overhead and memory bottlenecks. It utilizes fused CUDA kernels to combine neural network operations, reducing memory overhead and increasing execution speed. The too
Optimizes deep learning training speed and memory efficiency via fused kernels and optimized normalization.
This project is a diffusion-based 3D generator and image-to-3D reconstruction system. It translates natural language descriptions or two-dimensional images into three-dimensional assets using neural radiance fields and diffusion models. The system utilizes score-distillation sampling and diffusion-based guidance to refine 3D shapes without requiring 3D training data. It includes specialized tools for transforming neural representations into exportable meshes with texture and material data, as well as a pipeline for iterative optimization of geometry and textures. The project covers a broad r
Accelerates deep learning convergence using adaptive momentum, gradient clipping, and weight decay.
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
Offers techniques for reducing model size and increasing inference speed through quantization and graph optimization.
This repository contains programming assignments and lecture notes from Andrew Ng's foundational deep learning course specialization on Coursera. The materials cover core neural network training techniques including optimization algorithms, normalization methods, regularization approaches, parameter initialization strategies, and learning rate scheduling to improve model convergence and generalization. The coursework explores design principles where successive neural network layers learn progressively more abstract feature representations from input data. It provides guidance on selecting ope
Randomly initialize weight matrices and bias vectors for each layer based on layer dimensions.