256 Rank
Batch size 4 x 4 = 16
30 Epochs
LR: 3e-5