p-doom/crowd-code-dataset-0.2 huggingface dataset:
[terminal_command] TERMINAL ()
python3 -m MaxText.train MaxText/configs/base.yml \  run_name=h100_mfu_340m \  hardware=gpu \  dataset_type=synthetic \  steps=60 \  log_period=1 \  enable_checkpointing=False \  gcs_metrics=False \  metrics_file=/tmp/h100_mfu_metrics.jsonl \  base_output_directory=/tmp/maxtext \  attention=autoselected \  per_device_batch_size=4 \  base_emb_dim=1536 \  base_num_query_heads=12 \  base_num_kv_heads=12 \  base_mlp_dim=4096 \  base_num_decoder_layers=10 \  head_dim=128 \  logits_via_embedding=True