Burn - Rustで書かれたディープラーニングフレームワーク
(github.com/burn-rs)- "Burn Unstoppable Rusty Neurons"
- PyTorchのように独自のニューラルネットワークモジュールを作成できるModule
- 学習時のmetric、logging、checkpointingをサポート
- バックエンドを選択可能なTensor crate
- CPU/GPUをサポートするTchバックエンド
- コンパイル時間が速いNdArrayバックエンド
- すべてのバックエンドを微分可能にするAutodiffバックエンド
- さまざまなユーティリティを提供するDataset crate
まだコメントはありません。