tree 0318066da340713479c7f476d9339930e43e9f19
parent 9f57e5ba424808bd6168ea5cd844f09d0195a2d7
author Austin Schuh <austin.linux@gmail.com> 1730006540 -0700
committer Austin Schuh <austin.linux@gmail.com> 1730006540 -0700
gpgsig -----BEGIN PGP SIGNATURE-----
 
 iQGzBAABCgAdFiEExcNA7Sy+j6Ly5hA4QVYw08sOnWYFAmcdzi4ACgkQQVYw08sO
 nWbFKwv/U2TyXT3e/cUWvdPsBo/bNAs1z9O0CDZHvveaNQ8veb2EhPcs0kJ88dbh
 CQWuFbOwdAoMzvNGN4EkbJ8E1Wo9l3lAl4BVWqJKaQoDqrJulyairLXHToiCdLvW
 IGOKQCtXJMUA/GnIoAt0ob9sRX8XyoQ0eNw0zmVPl511N4i0bdnAbHX9tnVxLHAr
 ZrLWAkaLYJzePcvr8DlUpta9gkxdh6OB5g7rFLy4k1+JV3s4n/2tD+5LLWBfuuns
 RLo83UU4zTgv93YnRCtz5pQSOA4ooyhARf3t96RzzE+hpgkMPUDOvElv/zIuUy6x
 yacCwzqQmtsek1hPxhvnqrncxsEClCMGhh31IghK3zWAEudtJM14rDP/oe+y5e27
 QOnypJq8E2cdYGeyY9J/zLlVvyPXj3bNvCSyAF+A4N+3g9tYkrSEVkIwQgKs3oqV
 EsGYNQ7X3ZlqI9ohWKFjPpvAtPMSHwquO37sPBKb2LU85m2LS5rmsK9h8bjSDlA6
 +Q7RAKdE
 =r8Eu
 -----END PGP SIGNATURE-----

Speed up writing experience significantly

The jax calculation to compute the reward needs to be JITed to be fast.
Add that, and add some logging to make it easy to measure.

Change-Id: I9e9c53cf6ac0180e05819ffaefc5834784fb6e93
Signed-off-by: Austin Schuh <austin.linux@gmail.com>