mlverse/torch

R Interface to Torch

解决的问题

提供对 PyTorch 库的原生 R 接口，使 R 用户无需安装独立的 Python 环境即可构建和训练机器学习模型。

工作原理

该包允许从 R 对象创建 torch 张量，并提供对 PyTorch 核心功能的访问，包括用于自动计算导数的 autograd 系统。

适用人群

希望实现深度学习和机器学习工作流的 R 编程语言使用者，包括数据科学家和研究人员。

主要亮点

  • PyTorch 功能的原生 R 实现
  • 支持张量操作以及 R 数组与 torch 张量之间的转换
  • 内置 autograd 功能，实现自动微分
  • 支持 CUDA，用于 GPU 加速

相关

  • 项目
    pytorch/pytorch

    一个基于 Python 的机器学习框架，提供 GPU 加速张量计算和动态 autograd 系统，用于构建深度神经网络。

  • 项目
    LaurentMazare/tch-rs

    PyTorch C++ API（libtorch）的 Rust 绑定，使 Rust 可以使用 PyTorch 的张量操作和深度学习工具。

  • 项目
    meta-pytorch/torchcodec

    一个用于将视频、音频和图像直接解码并编码为张量的 PyTorch 原生库，适用于机器学习训练和推理。

  • 项目
    facebookresearch/pytorch3d

    一个基于 PyTorch 的 3D 计算机视觉研究库，提供可微渲染以及用于操作 3D 网格和点云的高效工具。

  • 项目
    Ascend/pytorch

    A PyTorch adapter plugin that enables the PyTorch framework to run on Ascend AI processors, providing hardware acceleration for AI model training and inference.