einx Logo

einx

跨框架张量运算统一接口库

einx张量操作深度学习符号表示法跨框架兼容Github开源项目

einx是一个Python库,为多个主流深度学习框架提供统一的张量操作接口。它采用类爱因斯坦符号系统表达基础张量操作的向量化,支持无缝集成现有代码。通过即时编译提升执行效率,einx不仅支持基础张量操作,还涵盖常见神经网络操作和可选的深度学习模块。

ortex - Elixir实现的ONNX模型并发推理框架
ElixirGithubONNX RuntimeOrtex开源项目机器学习模型模型推理
Ortex是基于ONNX Runtime的Elixir封装库,通过Nx.Serving实现ONNX模型的并发和分布式部署。该框架支持CUDA、TensorRT和Core ML等多种后端,能够高效加载和推理主流机器学习库导出的ONNX模型。Ortex还提供了便捷的存储型张量实现,为开发者提供了灵活且高效的机器学习模型部署方案。
onnx-tool - 专注于深度学习和自然语言处理的ONNX模型管理工具
Githubonnx-tool内存压缩开源项目形状推理模型压缩模型概要分析
ONNX-tool是一款强大的工具,支持ONNX模型的解析和编辑、推断和压缩。适用于自然语言处理和计算机视觉模型,提供模型构建、形状推断、激活压缩、权重压缩及计算图优化,以提升推理性能和存储效率。
TensorLayer - 高性能且灵活的深度学习和强化学习工具库
GithubTensorFlowTensorLayer开源软件开源项目强化学习深度学习
TensorLayer 是一个基于 TensorFlow 的深度学习和强化学习库,为研究人员和工程师提供多种可定制的神经网络层,简化复杂 AI 模型的构建。它设计独特,结合了高性能与灵活性,支持多种后端和硬件,并提供丰富的教程和应用实例。广泛应用于全球知名大学和企业,如谷歌、微软、阿里巴巴等。
safetensors - 安全高效的张量存储格式 支持零拷贝和延迟加载
GithubHugging Facesafetensors开源项目张量存储文件格式机器学习
safetensors是一种新的张量存储格式,旨在安全高效地保存机器学习模型。与pickle相比,它提供了更好的安全性,同时通过零拷贝技术保持高性能。该库支持Python和Rust,提供简单的API用于张量的存储和加载。safetensors支持延迟加载、布局控制和多种数据类型,适用于大规模模型的存储和分发。在加载速度和内存效率方面表现优异,尤其适合分布式环境。
OnnxStack - 高效机器学习集成框架 无缝对接.NET生态
.NETGithubONNX RuntimeStableDiffusion图像处理开源项目机器学习
OnnxStack为.NET开发者提供了一个便捷的机器学习集成方案。它与ONNX Runtime和Microsoft ML无缝对接,支持Stable Diffusion、图像放大和对象检测等AI模型。开发者可以在.NET环境中直接构建和运行机器学习应用,不再依赖Python。这个框架大大简化了AI开发流程,是.NET生态系统中的一个重要补充。
tiny-dnn - 轻量级C++14深度学习库,适用于嵌入式系统和物联网设备
C++14Githubtiny-dnn嵌入式系统开源项目深度学习物联网设备
tiny-dnn是一个为计算资源有限的嵌入式系统和物联网设备设计的C++14深度学习库。该库无需GPU,通过TBB线程和SSE/AVX向量化实现了高效性能,在13分钟内达到了98.8%的MNIST准确率。其便携的头文件形式使其易于集成,支持多种网络层类型、激活函数、损失函数和优化算法。tiny-dnn还能导入Caffe模型,适合学习和构建神经网络应用。
imodelsX - 多功能文本模型分析与优化库
GithubimodelsX可解释性开源项目文本模型机器学习自然语言处理
imodelsX是一个与Scikit-learn兼容的Python库,专注于文本模型和数据的解释、预测和优化。该库集成了多种可解释的建模技术,包括Tree-Prompt、iPrompt和Aug-Linear等。imodelsX还提供LLM封装器和数据集封装器等实用工具,简化文本数据处理流程。通过支持多种解释方法,imodelsX能够生成自然语言解释,并为用户提供易于实现的基线模型。
xuance - 多框架支持的深度强化学习算法库
GithubXuanCe多框架支持开源库开源项目深度强化学习算法实现
XuanCe是一个开源的深度强化学习算法库,支持PyTorch、TensorFlow和MindSpore等多种框架。它兼容单智能体和多智能体任务,提供丰富的算法实现。XuanCe设计模块化,易于学习和使用,运行速度快。支持经典控制、Box2D、MuJoCo、Atari等多种环境,为研究和开发提供全面的深度强化学习工具。
intel-extension-for-pytorch - 通过最新优化提升Intel硬件的深度学习性能
AIGPUsGithubIntel® Extension for PyTorchLLMs优化开源项目
Intel® Extension for PyTorch* 提供优化功能,利用Intel® AVX-512 VNNI、AMX以及XMX AI引擎,提升Intel CPU和GPU上的深度学习性能。该扩展优化了大规模语言模型(LLMs),如LLAMA、GPT-J、GPT-NEOX等,支持多种量化方法(如FP32、BF16、INT8、INT4)。此外,自2.3.0版本起,还引入了模块级优化API,为定制模型优化提供了更多选项。
pytorch - 能GPU加速的Python深度学习平台
GPU加速PyTorch深度学习神经网络
PyTorch是一个开源的提供强大GPU加速的张量计算和深度神经网络平台,基于动态autograd系统设计。它不仅支持广泛的科学计算需求,易于使用和扩展,还可以与Python的主流科学包如NumPy、SciPy无缝集成,是进行深度学习和AI研究的理想工具。