主 释放 火炬-Py + Nim pytorch的Nim前端,旨在主要自动生成并在内部使用ATen。 因为Nim可以编译为C ++,所以它不是包装器或绑定库。 它生成一对一的本地ATen代码。 pytorch的唯一要求是ATen的核心张量库。 因此,nimtorch具有极强的通用性,可以在任何类型的设备上进行编译。 当前状态 早期 从Declarations.yaml自动生成完整的ATen API Cuda支持(使用nim编译时添加-d:cuda) WASM支持(使用nim编译时添加-d:wasm) 自动从derivatives.yaml 。yaml,梯度过程 自动毕业 添