ステージ２ 自然なコードで表現する
| ステップ１１ 可変長の引数(順伝播編） ステップ１２ 可変長の引数（改善編） ステップ１３ 可変長の引数（逆伝播編） ステップ１４ 同じ変数を繰り返し使う ステップ１５ 複雑な計算グラフ（理論編） ステップ１６ 複雑な計算グラフ（実装編） ステップ１７ メモリ管理と循環参照 ステップ１８ メモリ使用量を減らすモード ステップ１９ 変数を使いやすく ステップ２０ 演算子のオーバーロード（１） ステップ２１ 演算子のオーバーロード（２） ステップ２２ 演算子のオーバーロード（３） ステップ２３ パッケージとしてまとめる ステップ２４ 複雑な関数の微分 |
このステージでは、より複雑な計算ができるように、「変数（Variable クラス）」と「関数（Function クラス）」を拡張する内容でした。具体的には、
・複数の入力を受け取る関数
・複数の出力を返す関数
・同じ変換を繰り返し使う
・関数の実行の優先順位を考える
・メモリ管理
・循環参照
・メモリ使用量を減らす
・変数に名前を付ける
・ndarray のインスタンス変数のように使う
・加算、減算、負数、乗算、割算、累乗の実装
・他の型と一緒に演算を行う
・＋や＊などの演算子が使える
・パッケージ化
・複雑な関数の微分（Sphere、matyas、Goldstein−Price）
などです。
コラムでは、Define−by−Run と、Define−and−Run について、それぞれのメリットやデメリットについて、分かりやすい解説があります。