representations

cnn

mlp

rnn

vit

world_model

world_model_v2