vorticity
divergence
specific_humidity
temperature
velocitiy_z

model_id: id23y8h1b4, based on training: 2zxrkmq0, epoch 40
token_size: [3, 9, 9]
num_tokens: [12, 6, 12]
ens dim: 16

vorticity id23y8h1b4 [3, 9, 9] [12, 6, 12] 16