Models/EVA02: add option to disable absolute positional embeddings

Models/EVA02: add support for variable number of register tokens

Models: add VGG

Simple Path to TFRecord Translator (#12)

ViT: make layernorm implementation selectable

EVA02: make layernorm implementation selectable

EVA02: add option to disable sub-LN

Training loops: commit current step

Revert "Training loops: on WandB, log the first epoch as 0 instead of 1"

Training loops: on WandB, log the first epoch as 0 instead of 1

Training loops: reset the metrics after reloading the full state from a

Training scripts: write nothing if --checkpoints-keep=0

pretraining_loop: better control over buffers

training_loop: better control over buffers

Models: better typing