概要 https://github.com/pytorch/pytorch/tree/v0.4.0/torch/csrc/jit PyTorch は NN 定義をコンパイルし Python のユーザーコードを通らずモデルを実行できる機能を開発中。実は開発は普通にオープンに GitHub で行われているし master にじゃんじゃん入っているので試せる。以下の 2 種類の両方が実装されつつある。 実際に実行して作成された計算グラフを IR に変換する torch.jit.compile Python のコードを制御構文も含めて直接 IR に変換する torch.jit.script 以下は PyTorch 0.4.0 で実行。 In [1]: import torch In [2]: torch.__version__ Out[2]: '0.4.0' torch.jit.compile ま