 [1m─────────────────────────────────────────────────────────────────────────[22m
 [1m                         [22m        Time                   Allocations      
                          ──────────────────────   ───────────────────────
     Tot / % measured:         4303s / 0.02%            223GiB / 0.10%    

 Section          ncalls     time   %tot     avg     alloc   %tot      avg
 ─────────────────────────────────────────────────────────────────────────
 backward pass        24    379ms  55.1%  15.8ms   46.3MiB  20.8%  1.93MiB
 jacobians            24    166ms  24.1%  6.90ms    121MiB  54.6%  5.06MiB
 forward pass         24    108ms  15.8%  4.51ms   45.4MiB  20.4%  1.89MiB
   rollout            61   93.3ms  13.6%  1.53ms   37.9MiB  17.0%   636KiB
   cost               61   13.7ms  2.00%   225μs   7.45MiB  3.35%   125KiB
 cost expansion       24   33.8ms  4.92%  1.41ms   8.46MiB  3.80%   361KiB
 copy                 24    431μs  0.06%  18.0μs    717KiB  0.31%  29.9KiB
 convergence          24    105μs  0.02%  4.38μs   1.47KiB  0.00%    62.7B
 [1m─────────────────────────────────────────────────────────────────────────[22m