data: Store raw dataset models: Store trained models train_transformer_jax.py: Training script for Transformer-based dynamics model train_finetune_jax.py: Script for real-world finetuning