Add flag to toggle whether to create q values over trajectory figure

Bugfixes in openvla caching worker for new agent interface

Added arXiv link

Merge pull request #1 from MaxSobolMark/optimizer_resetting

When offline dataset size is known, adjust offline batch size accordingly

Restart optimizer state when switching to online training

Downgrade jax version to 0.4.24 to match paper

Initial commit