path	# lines of code
data/conceptual_captions/generate_conceptual_caption.py	65
data/envs/babyai/generate_random_score.py	82
data/envs/babyai/create_babyai_dataset.py	121
data/envs/babyai/bot_agent.py	501
data/envs/mujoco/create_mujoco_dataset.py	210
data/envs/mujoco/generate_random_score.py	54
data/envs/atari/generate_random_score.py	114
data/envs/atari/create_atari_dataset.py	134
data/envs/download_expert_scores.py	184
data/envs/metaworld/generate_dataset.py	101
data/envs/metaworld/push.py	22
data/envs/metaworld/train.py	65
data/envs/metaworld/generate_random_score.py	94
data/to_hub.py	146
data/ok-vqa/ok-vqa.py	86
scripts/download_all_datasets.py	20
scripts/generate_config_jat.py	60
scripts/train_jat_tokenized.py	105
scripts/eval_jat.py	143
scripts/tokenize_stream.py	108
scripts/train_jat.py	140
pyproject.toml	11
jat/processing_jat.py	207
jat/__init__.py	3
jat/configuration_jat.py	62
jat/modeling_jat.py	549
jat/eval/__init__.py	1
jat/eval/rl/__init__.py	2
jat/eval/rl/wrappers.py	92
jat/eval/rl/core.py	256
jat/utils.py	388
jat/utils_interleave_datasets.py	200
setup.py	66