PyTorch
PyTorchモデルのモバイル向け展開にはPyTorch Mobileが存在する。PyTorch Mobileは最適化 (torch.utils.mobile_optimizer) や量子化 (torch.quantization)にも対応している。
PyTorchは自動微分と呼ばれるメソッドを利用する。recorderは実行された演算を記録し、その後、勾配の計算を行うときに記録した演算を逆方向にリプレイする。このメソッドは、ニューラルネットワークの構築時に特に強力であるため、順方向のパラメータの微分を計算することで、1エポックの計算にかかる時間を節約することができる。
TorchScriptとして後述。
TorchScriptコードは直接記述 (torch.jit.script) およびPyTorch実行トレース (torch.jit.trace) により生成される。
以下は前向き (forward) 関数のスクリプト変換の例である: