wz-ml/termite

Optimize Termite

Opened this issue · 0 comments

wz-ml commented

The current implementation's rigorous, but slow! We can speed things up with the help of numpy/JAX.

  • Profile rollouts to see what the major time-sucks are.
  • Demolish them.