1 个仓库
The process of designing model architectures based on sparse tensor network topologies.
Distinct from Sparse Model Architectures: Focuses on the network topology of sparse tensors rather than general conditional computation or routing.
Explore 1 awesome GitHub repository matching artificial intelligence & ml · Sparse Tensor Network Modeling. Refine with filters or upvote what's useful.
Amazon DSSTNE 是一个机器学习工具包和稀疏张量网络库,专为具有稀疏输入和输出的深度学习模型而设计。它提供了一个模型并行训练框架和一个 GPU 加速的稀疏引擎,以支持内存密集型网络。 该框架专门为推荐系统训练和大规模稀疏学习而设计。它实现了将大型权重矩阵和嵌入表分布在多个 GPU 设备上,以处理超过单个处理器内存容量的模型。 该项目涵盖了广泛的能力,包括分布式 GPU 计算、稀疏数据集处理以及可扩展稀疏张量网络的构建。这些实用程序允许在 GPU 集群上执行高性能机器学习操作和模型扩展。
Enables constructing machine learning models using scalable sparse tensor networks to handle large-scale data.