日本語

ニュース

PyTorch互換APIで高階微分に対応、Python450行の自動微分ライブラリMyTorchが公開

MyTorchは、PyTorchのAPIを模した軽量な自動微分フレームワークです。

NumPyを使って計算処理を行い、グラフベースの逆モード自動微分を実装しています。

このライブラリは拡張性が容易で、torch.nnの追加やGPU実行への対応も視野に入れているとされています。

特に注目されるのは、スカラーおよび非スカラー変数に対する任意の高階微分の計算機能です。

PyTorchではcreate_graph=Trueが必要となる高階微分も、MyTorchでは追加設定なしで計算できます。

torch.autograd.backwardやtorch.autograd.gradといった標準的なAPIもサポートしています。


出典: MyTorch – Minimalist autograd in 450 lines of Python(HN 100pt・19コメント)(HN Search (backfill)、2026-01-04)