跳至內容

chelsea0x3b/dfdx

軟體分類 網路與資訊安全 (Network & Security) / VPN 與隱私 (VPN & Privacy)
軟體標籤
autodiff autodifferentiation autograd backpropagation cuda cuda-kernels cuda-support cuda-toolkit cudnn deep-learning deep-neural-networks gpu gpu-acceleration gpu-computing machine-learning neural-network rust rust-lang tensor
開發廠商 chelsea0x3b
Stars
1.9k
Forks
104
License
NOASSERTION
Version
vv0.13.0
Update
2024-07-23
OS
官方網站 官方網站 Git 儲存庫 Git 儲存庫

#### 產品特色

* 支援GPU加速與6維張量計算
* 編譯時檢查張量形狀與類型
* 提供大量張量操作與神經網絡元件
* 優化性能與減少unsafe碼使用
* 支援常數張量轉換與正常Rust陣列

#### 主要功能、特點

dfdx是一款專為Rust設計的深度學習庫，強調編譯時檢查與效能。它提供GPU加速的張量庫，支援6維張量計算，並在編譯時檢查張量形狀與類型。庫內包含大量張量操作，如matmul、conv2d等，並提供神經網絡元件如Linear、Conv2D等。支援常數張量轉換與正常Rust陣列。庫內使用Tuple來表示順序執行的模塊，並避免使用Rc>。所有操作結果都產生一個子節點，可直接將梯度記憶體移動到最後一個操作的子節點，實現更精確的梯度記憶體控制。

#### 指令

“`bash
# 無常用指令（自動生成）
“`