Rustで記述された最小限の自動微分エンジン
2025-06-20
Rustで記述された最小限の自動微分エンジンです。小さな多層パーセプトロンを訓練してXOR関数を学習し、単一パーセプトロンの計算グラフをgraph.htmlにレンダリングできます。コアはScalar構造体で、値、オプションの勾配、そしてそれを生成した操作を記述するEdgeを格納します。演算子のオーバーロードとヘルパー関数は、有向非巡回グラフを構築し、各エッジの局所的な導関数をキャッシュします。`backward()`は出力ノードから再帰的に勾配を伝播し、`Scalar::new_grad`で作成されたリーフノードに蓄積します。グラフは`plot::dump_graph`で視覚化できます。
開発