Activity
configure logging level
configure logging level
add video logging to tensorboard
add video logging to tensorboard
change to linear for macos
change to linear for macos
works pretty smoothly now
works pretty smoothly now
much faster mnist training
much faster mnist training
better jaxcache location
better jaxcache location