Cong Lu

Results 2 issues of Cong Lu

The current PyTorch training routine in https://colab.research.google.com/github/google/brax/blob/main/notebooks/training_torch.ipynb lacks an option for `action_repeat` to be set. From some testing of the Torch version, it seems this is mandatory to learn things...

Please see FAQ for guide on adding new models, Grok-2 is now rank 3 on lmsys :)

enhancement
good first issue