#张量操作

tensordict - PyTorch张量集合操作的高效字典类工具
Github开源项目PyTorch机器学习数据处理TensorDict张量操作
TensorDict是一个继承张量属性的字典类,为PyTorch提供简洁的张量集合操作方法。它支持异步设备传输、节点间快速通信、张量形状操作和分布式计算,提高了代码的可读性、紧凑性和模块化。这个工具适用于模型训练、优化器实现等机器学习任务,能有效简化代码结构,提升开发效率。
einx - 跨框架张量运算统一接口库
Github开源项目深度学习跨框架兼容张量操作einx符号表示法
einx是一个Python库,为多个主流深度学习框架提供统一的张量操作接口。它采用类爱因斯坦符号系统表达基础张量操作的向量化,支持无缝集成现有代码。通过即时编译提升执行效率,einx不仅支持基础张量操作,还涵盖常见神经网络操作和可选的深度学习模块。
gotch - Go语言深度学习框架 基于PyTorch C++ API的高性能实现
Github开源项目深度学习神经网络PytorchGPU支持张量操作
Gotch是基于PyTorch C++ API (Libtorch)开发的Go语言深度学习框架。它实现了全面的张量操作、动态计算图、JIT接口和预训练模型加载等功能,同时支持CPU和GPU。Gotch采用纯Go API设计,便于构建和训练神经网络模型,适合深度学习研究和应用开发。
Mutable.ai - 智能代码文档生成平台
AI工具代码库机器学习框架AI专家零样本分类张量操作
Mutable.ai是一个基于人工智能的代码文档生成平台,自动分析代码结构并生成详细文档和Wiki。平台可回答复杂的代码相关问题,如多框架支持和文件功能。它旨在优化开发流程,提升团队效率,让开发者专注于核心编码工作,减少手动编写文档的时间。